/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package backtype.storm.drpc; import backtype.storm.coordination.CoordinatedBolt.FinishedCallback; import backtype.storm.task.OutputCollector; import backtype.storm.task.TopologyContext; import backtype.storm.topology.BasicBoltExecutor; import backtype.storm.topology.IBasicBolt; import backtype.storm.topology.IRichBolt; import backtype.storm.topology.OutputFieldsDeclarer; import backtype.storm.tuple.Tuple; import backtype.storm.utils.KeyedRoundRobinQueue; import java.util.HashMap; import java.util.Map; public class KeyedFairBolt implements IRichBolt, FinishedCallback { IRichBolt _delegate; KeyedRoundRobinQueue<Tuple> _rrQueue; Thread _executor; FinishedCallback _callback; public KeyedFairBolt(IRichBolt delegate) { _delegate = delegate; } public KeyedFairBolt(IBasicBolt delegate) { this(new BasicBoltExecutor(delegate)); } public void prepare(Map stormConf, TopologyContext context, OutputCollector collector) { if (_delegate instanceof FinishedCallback) { _callback = (FinishedCallback) _delegate; } _delegate.prepare(stormConf, context, collector); _rrQueue = new KeyedRoundRobinQueue<Tuple>(); _executor = new Thread(new Runnable() { public void run() { try { while (true) { _delegate.execute(_rrQueue.take()); } } catch (InterruptedException e) { } } }); _executor.setDaemon(true); _executor.start(); } public void execute(Tuple input) { Object key = input.getValue(0); _rrQueue.add(key, input); } public void cleanup() { _executor.interrupt(); _delegate.cleanup(); } public void declareOutputFields(OutputFieldsDeclarer declarer) { _delegate.declareOutputFields(declarer); } public void finishedId(Object id) { if (_callback != null) { _callback.finishedId(id); } } @Override public Map<String, Object> getComponentConfiguration() { return new HashMap<String, Object>(); } }