001/*
002 * (C) Copyright 2016 Nuxeo SA (http://nuxeo.com/) and others.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 *
016 * Contributors:
017 *     bdelbosc
018 */
019package org.nuxeo.importer.stream.consumer;
020
021import org.nuxeo.importer.stream.message.BlobMessage;
022import org.nuxeo.lib.stream.pattern.consumer.Consumer;
023import org.nuxeo.lib.stream.pattern.consumer.ConsumerFactory;
024
025import static org.apache.commons.lang3.StringUtils.isEmpty;
026
027/**
028 * @since 9.1
029 */
030public class BlobMessageConsumerFactory implements ConsumerFactory<BlobMessage> {
031    protected final String blobProviderName;
032
033    protected final BlobInfoWriter blobInfoWriter;
034
035    protected final String watermarkPrefix;
036
037    protected final String persistBlobPath;
038
039    /**
040     * Blob Consumer factory requires a blob providerName that is present in Nuxeo instance running the consumer. The
041     * writer is used to store the blob information.
042     */
043    public BlobMessageConsumerFactory(String blobProviderName, BlobInfoWriter blobInfoWriter) {
044        this(blobProviderName, blobInfoWriter, null, null);
045    }
046
047    /**
048     * Blob consumer that change the input blob to add a random watermark.
049     */
050    public BlobMessageConsumerFactory(String blobProviderName, BlobInfoWriter blobInfoWriter, String watermarkPrefix) {
051        this(blobProviderName, blobInfoWriter, watermarkPrefix, null);
052    }
053
054    /**
055     * Blob consumer that change the input blob to add a random watermark and persist the generated blobs.
056     */
057    public BlobMessageConsumerFactory(String blobProviderName, BlobInfoWriter blobInfoWriter, String watermarkPrefix,
058            String persistBlobPath) {
059        this.blobProviderName = blobProviderName;
060        this.blobInfoWriter = blobInfoWriter;
061        this.watermarkPrefix = watermarkPrefix;
062        this.persistBlobPath = persistBlobPath;
063    }
064
065    @Override
066    public Consumer<BlobMessage> createConsumer(String consumerId) {
067        if (isEmpty(watermarkPrefix)) {
068            return new BlobMessageConsumer(consumerId, blobProviderName, blobInfoWriter);
069        }
070        return new BlobWatermarkMessageConsumer(consumerId, blobProviderName, blobInfoWriter, watermarkPrefix,
071                persistBlobPath);
072    }
073}