storm trident实战 trident state
一、认识storm trident
trident可以理解为storm批处理的高级抽象,提供了分组、分区、聚合、函数等操作,提供一致性和恰好一次处理的语义。
1)元祖被作为batch处理
2)每个batch的元祖都被指定唯一的一个事物id,如果因为处理失败导致batch重发,也和保证和重发前一样的事物id
3)数据更新操作严格有序,比如batch1必须在batch2之前被成功处理,且如果batch1失败了,后面的处理也会失败。
假如: batch1处理1--20
batch2处理21--40
batch1处理失败,那么batch2也会失败
虽然数据更新操作严格有序,但是数据处理阶段也可以并行的,只是最后的持久化操作必须有序。
1.1 trident state
trident的状态具有仅仅处理一次,持续聚合的语义,使用trident来实现恰好一次的语义不需要开发人员去处理事务相关的工作,因为trident state已经帮我们封装好了,只需要编写类似于如下的代码:
topology.newStream("sentencestream", spout)
.each(new Fields("sentence"), new Split(), new Fields("word"))
.groupBy(new Fields("word"))
.persistentAggregate(new MyHbaseState.HbaseFactory(options), new Count(), new Fields("count"))
.parallelismHint(3);
所有处理事务逻辑都在MyHbaseState.HbaseFactory中处理了(这个是我自己定义的,trident支持在内存里面处理,类似于MemachedState.opaque)。
trident提供了一个StateFactory用来创建State对象的实例,行如:
public final class XFactory implements StateFactory{
public State makeState(Map conf,int partitonIndex,int numPartitions){
return new State();
}
}
1.2 persistentAggregate
persistentAggregate是trident中用来更新来源的状态,如果前面是一个分好组的流,trident希望你提供的状态实现MapState接口,其中key是分组的字段,
而聚合结果是状态的值。
1.3 实现MapStates
trident中实现MapState非常简单,只需要为这个类提供一个IBackingMap的接口实现接口。
二、实战
首先搭建好zk,storm,hadoop,hbase的分布式环境
master:
slave1:
slave2:
main方法:
public static void main(String[] args) throws AlreadyAliveException, InvalidTopologyException, AuthorizationException {
TridentTopology topology = new TridentTopology();
FixedBatchSpout spout = new FixedBatchSpout(new Fields("sentence"), 3,
new Values("tanjie is a good man"), new Values(
"what is your name"), new Values("how old are you"),
new Values("my name is tanjie"), new Values("i am 18"));
spout.setCycle(false);
tridentStreamToHbase(topology,spout);
Config config = new Config();
config.setDebug(false);
StormSubmitter.submitTopologyWithProgressBar("word_count_trident_state_HbaseState", config, topology.build());
}
tridentStreamToHbase方法:
private static TridentState tridentStreamToHbase(TridentTopology topology,
FixedBatchSpout spout) {
MyHbaseState.Options options = new MyHbaseState.Options();
options.setTableName("storm_trident_state");
options.setColumFamily("colum1");
options.setQualifier("q1");
/**
* 根据数据源拆分单词后,然后分区操作,在每个分区上又进行分组(hash算法),然后在每个分组上进行聚合
* 所以这里可能有多个分区,每个分区有多个分组,然后在多个分组上进行聚合
* 用来进行group的字段会以key的形式存在于State当中,聚合后的结果会以value的形式存储在State当中
*/
return topology.newStream("sentencestream", spout)
.each(new Fields("sentence"), new Split(), new Fields("word"))
.groupBy(new Fields("word"))
.persistentAggregate(new MyHbaseState.HbaseFactory(options), new Count(), new Fields("count"))
.parallelismHint(3);
}
MyHbaseState实现:
package com.storm.trident.state.hbase;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.client.Connection;
import org.apache.hadoop.hbase.client.ConnectionFactory;
import org.apache.hadoop.hbase.client.Get;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.client.Table;
import org.apache.storm.task.IMetricsContext;
import org.apache.storm.trident.state.JSONNonTransactionalSerializer;
import org.apache.storm.trident.state.JSONOpaqueSerializer;
import org.apache.storm.trident.state.JSONTransactionalSerializer;
import org.apache.storm.trident.state.Serializer;
import org.apache.storm.trident.state.State;
import org.apache.storm.trident.state.StateFactory;
import org.apache.storm.trident.state.StateType;
import org.apache.storm.trident.state.map.IBackingMap;
import org.apache.storm.trident.state.map.MapState;
import org.apache.storm.trident.state.map.OpaqueMap;
import org.apache.storm.trident.state.map.SnapshottableMap;
import org.apache.storm.tuple.Values;
import com.google.common.collect.Maps;
@SuppressWarnings({ "unchecked", "rawtypes" })
public class MyHbaseState<T> implements IBackingMap<T> {
private static final Map<StateType, Serializer> DEFAULT_SERIALZERS = Maps
.newHashMap();
private int partitionNum;
private Options<T> options;
private Serializer<T> serializer;
private Connection connection;
private Table table;
static {
DEFAULT_SERIALZERS.put(StateType.NON_TRANSACTIONAL,
new JSONNonTransactionalSerializer());
DEFAULT_SERIALZERS.put(StateType.TRANSACTIONAL,
new JSONTransactionalSerializer());
DEFAULT_SERIALZERS.put(StateType.OPAQUE, new JSONOpaqueSerializer());
}
public MyHbaseState(final Options<T> options, Map conf, int partitionNum) {
this.options = options;
this.serializer = options.serializer;
this.partitionNum = partitionNum;
try {
connection = ConnectionFactory.createConnection(HBaseConfiguration
.create());
table = connection.getTable(TableName.valueOf(options.tableName));
} catch (IOException e) {
e.printStackTrace();
}
}
public static class Options<T> implements Serializable {
/**
*
*/
private static final long serialVersionUID = 1L;
public Serializer<T> serializer = null;
public String globalkey = "$HBASE_STATE_GLOBAL$";
/**
* 表名
*/
public String tableName;
/**
* 列族
*/
public String columFamily;
/**
*
*/
public String qualifier;
public String getTableName() {
return tableName;
}
public void setTableName(String tableName) {
this.tableName = tableName;
}
public String getColumFamily() {
return columFamily;
}
public void setColumFamily(String columFamily) {
this.columFamily = columFamily;
}
public String getQualifier() {
return qualifier;
}
public void setQualifier(String qualifier) {
this.qualifier = qualifier;
}
}
protected static class HbaseFactory<T> implements StateFactory {
private static final long serialVersionUID = 1L;
private Options<T> options;
public HbaseFactory(Options<T> options) {
this.options = options;
if (this.options.serializer == null) {
this.options.serializer = DEFAULT_SERIALZERS
.get(StateType.OPAQUE);
}
}
@Override
public State makeState(Map conf, IMetricsContext metrics,
int partitionIndex, int numPartitions) {
System.out.println("partitionIndex:" + partitionIndex
+ ",numPartitions:" + numPartitions);
IBackingMap state = new MyHbaseState(options, conf, partitionIndex);
MapState mapState = OpaqueMap.build(state);
return new SnapshottableMap(mapState, new Values(options.globalkey));
}
}
@Override
public void multiPut(List<List<Object>> keys, List<T> values) {
List<Put> puts = new ArrayList<Put>(keys.size());
for (int i = 0; i < keys.size(); i++) {
Put put = new Put(toRowKey(keys.get(i)));
T val = values.get(i);
System.out.println("partitionIndex: " + this.partitionNum
+ ",key.get(i):" + keys.get(i) + "value值:" + val);
put.addColumn(this.options.columFamily.getBytes(),
this.options.qualifier.getBytes(),
this.options.serializer.serialize(val));
puts.add(put);
}
try {
this.table.put(puts);
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public List<T> multiGet(List<List<Object>> keys) {
List<Get> gets = new ArrayList<Get>();
for (final List<Object> key : keys) {
// LOG.info("Partition: {}, GET: {}", this.partitionNum, key);
Get get = new Get(toRowKey(key));
get.addColumn(this.options.columFamily.getBytes(),
this.options.qualifier.getBytes());
gets.add(get);
}
List<T> retval = new ArrayList<T>();
try {
// 批量获取所有rowKey的数据
Result[] results = this.table.get(gets);
for (final Result result : results) {
byte[] value = result.getValue(
this.options.columFamily.getBytes(),
this.options.qualifier.getBytes());
if (value != null) {
retval.add(this.serializer.deserialize(value));
} else {
retval.add(null);
}
}
} catch (IOException e) {
e.printStackTrace();
}
return retval;
}
private byte[] toRowKey(List<Object> keys) {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
try {
for (Object key : keys) {
bos.write(String.valueOf(key).getBytes());
}
bos.close();
} catch (IOException e) {
throw new RuntimeException("IOException creating HBase row key.", e);
}
return bos.toByteArray();
}
}
运行结果:
查看supervisor日志:
2016-12-23 11:34:25.576 STDIO [INFO] partitionIndex: 0,key.get(i):[good]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.582 STDIO [INFO] partitionIndex: 1,key.get(i):[name]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.582 STDIO [INFO] partitionIndex: 1,key.get(i):[are]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.585 STDIO [INFO] partitionIndex: 2,key.get(i):[what]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.585 STDIO [INFO] partitionIndex: 2,key.get(i):[your]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.585 STDIO [INFO] partitionIndex: 2,key.get(i):[tanjie]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.585 STDIO [INFO] partitionIndex: 2,key.get(i):[old]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.586 STDIO [INFO] partitionIndex: 2,key.get(i):[is]value值:[email protected][currTxid=1,prev=<null>,curr=2]
2016-12-23 11:34:25.586 STDIO [INFO] partitionIndex: 2,key.get(i):[a]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.586 STDIO [INFO] partitionIndex: 2,key.get(i):[how]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.587 STDIO [INFO] partitionIndex: 2,key.get(i):[you]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.587 STDIO [INFO] partitionIndex: 2,key.get(i):[man]value值:[email protected][currTxid=1,prev=<null>,curr=1]
2016-12-23 11:34:25.812 STDIO [INFO] partitionIndex: 2,key.get(i):[tanjie]value值:[email protected][currTxid=2,prev=1,curr=2]
2016-12-23 11:34:25.812 STDIO [INFO] partitionIndex: 2,key.get(i):[is]value值:[email protected][currTxid=2,prev=2,curr=3]
2016-12-23 11:34:25.815 STDIO [INFO] partitionIndex: 0,key.get(i):[am]value值:[email protected][currTxid=2,prev=<null>,curr=1]
2016-12-23 11:34:25.815 STDIO [INFO] partitionIndex: 0,key.get(i):[my]value值:[email protected][currTxid=2,prev=<null>,curr=1]
2016-12-23 11:34:25.826 STDIO [INFO] partitionIndex: 1,key.get(i):[i]value值:[email protected][currTxid=2,prev=<null>,curr=1]
2016-12-23 11:34:25.827 STDIO [INFO] partitionIndex: 1,key.get(i):[name]value值:[email protected][currTxid=2,prev=1,curr=2]
2016-12-23 11:34:25.828 STDIO [INFO] partitionIndex: 1,key.get(i):[18]value值:[email protected][currTxid=2,prev=<null>,curr=1]
查看hbase表