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 */
017package org.nuxeo.ecm.platform.importer.queue.producer;
018
019import com.codahale.metrics.Counter;
020import com.codahale.metrics.MetricRegistry;
021import com.codahale.metrics.SharedMetricRegistries;
022import org.nuxeo.ecm.platform.importer.log.ImporterLogger;
023import org.nuxeo.ecm.platform.importer.queue.AbstractTaskRunner;
024import org.nuxeo.ecm.platform.importer.queue.manager.QueuesManager;
025import org.nuxeo.ecm.platform.importer.source.SourceNode;
026import org.nuxeo.runtime.metrics.MetricsService;
027
028import java.util.Random;
029
030/**
031 * @since 8.3
032 */
033public abstract class AbstractProducer extends AbstractTaskRunner implements Producer {
034
035    final protected ImporterLogger log;
036
037    protected QueuesManager qm;
038
039    protected final MetricRegistry registry = SharedMetricRegistries.getOrCreate(MetricsService.class.getName());
040
041    protected final Counter producerCounter;
042
043    protected final Random rand;
044
045    public AbstractProducer(ImporterLogger log) {
046        assert(log != null);
047        this.log = log;
048        producerCounter = registry.counter(MetricRegistry.name("nuxeo", "importer", "queue", "producer"));
049        rand = new Random(System.currentTimeMillis());
050    }
051
052    @Override
053    public void init(QueuesManager qm) {
054        this.qm = qm;
055    }
056
057    protected void dispatch(SourceNode node) throws InterruptedException {
058        int idx = getTargetQueue(node, qm.count());
059        qm.put(idx, node);
060        producerCounter.inc();
061        incrementProcessed();
062    }
063
064    @Override
065    public int getTargetQueue(SourceNode bh, int nbQueues) {
066        return rand.nextInt(nbQueues);
067    }
068}