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}