001/*
002 * (C) Copyright 2017 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.ecm.platform.importer.mqueues.pattern.producer;
020
021import org.nuxeo.ecm.platform.importer.mqueues.pattern.message.DocumentMessage;
022
023import java.nio.file.Path;
024
025/**
026 * @since 9.1
027 */
028public class RandomDocumentMessageProducerFactory implements ProducerFactory<DocumentMessage> {
029    private final long nbDocuments;
030    private final String lang;
031    private final Path blobInfoDirectory;
032    private final int blobSizeKb;
033
034    /**
035     * Generates random document messages that contains random blob.
036     */
037    public RandomDocumentMessageProducerFactory(long nbDocuments, String lang, int blobSizeKb) {
038        this.nbDocuments = nbDocuments;
039        this.lang = lang;
040        this.blobInfoDirectory = null;
041        this.blobSizeKb = blobSizeKb;
042    }
043
044    /**
045     * Generates random documents messages that point to existing blobs.
046     */
047    public RandomDocumentMessageProducerFactory(long nbDocuments, String lang, Path blobInfoDirectory) {
048        this.nbDocuments = nbDocuments;
049        this.lang = lang;
050        this.blobInfoDirectory = blobInfoDirectory;
051        this.blobSizeKb = 0;
052    }
053
054    @Override
055    public ProducerIterator<DocumentMessage> createProducer(int producerId) {
056        return new RandomDocumentMessageProducer(producerId, nbDocuments, lang, blobInfoDirectory).withBlob(blobSizeKb, false);
057    }
058}