1. 程式人生 > >Storm框架:如何根據業務條件選擇不同的bolt進行下發訊息

Storm框架:如何根據業務條件選擇不同的bolt進行下發訊息

Strom框架基本概念就不提了,這裡主要講的是Stream自定義ID的訊息流。預設spout、bolt都需實現介面方法declareOutputFields,程式碼如下:

@Override
public void declareOutputFields(OutputFieldsDeclarer outputFieldsDeclarer) {
    outputFieldsDeclarer.declare(new Fields("body"));
}

這種情況下發的訊息會被所有定義的bolts接收。我們如果需要根據得到的訊息型別來選擇不同的bolt,就需要用到Stream Grouping。

  • 首先通過訊息源的OutputFieldsDeclarer來定義發射多條訊息流stream

以下定義了兩種stream訊息流:email郵件、sms簡訊

@Override
public void declareOutputFields(OutputFieldsDeclarer outputFieldsDeclarer) {
    outputFieldsDeclarer.declareStream("email", new Fields("body"));
    outputFieldsDeclarer.declareStream("sms", new Fields("body"));
}
  • 然後我們通過對訊息內容進行分析判斷來決定發射指定的stream型別
@Override
public void execute(Tuple tuple) {
    String streamType;
    String value = tuple.getStringByField("body");
    # 邏輯判斷stub code
    if (value.startsWith("email:")) {
        streamType = "email";
    } else {
        streamType = "sms";
    }
    
    outputCollector.emit(streamType, new Values(value));
}
  • topology設定bolt的訊息源時通過localOrShuffleGrouping來設定只接收指定stream的訊息

FilterBolt通過對訊息進行加工處理,下發給bolts時會指定不同的stream,EmailNotifyBolt只接收email型別的stream訊息,SmsNotifyBolt只接收sms型別的stream訊息。

TopologyBuilder topologyBuilder = new TopologyBuilder();
topologyBuilder.setSpout("RabbitmqSpout", new RabbitmqSpout());
topologyBuilder.setBolt("FilterBolt", new FilterBolt()).shuffleGrouping("RabbitmqSpout");

topologyBuilder.setBolt("EmailNotifyBolt", new EmailNotifyBolt()).localOrShuffleGrouping("FilterBolt", "email");

topologyBuilder.setBolt("SmsNotifyBolt", new SmsNotifyBolt()).localOrShuffleGrouping("FilterBolt", "sms");