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}