欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

flink sink to redis

程序员文章站 2022-07-14 14:04:59
...

不多bb!
先看RedisCommand设置数据结构类型时和redis结构对应关系。

Data Type Redis Command [Sink]
HASH HSET
LIST RPUSH, LPUSH
SET SADD
PUBSUB PUBLISH
STRING SET
HYPER_LOG_LOG PFADD
SORTED_SET ZADD
SORTED_SET ZREM

具体的操作类:

    package sink;
    import org.apache.flink.api.common.functions.MapFunction;
    import org.apache.flink.api.java.tuple.Tuple2;
    import org.apache.flink.streaming.api.datastream.DataStream;
    import org.apache.flink.streaming.api.datastream.DataStreamSource;
    import org.apache.flink.streaming.api.datastream.SingleOutputStreamOperator;
    import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
    import org.apache.flink.streaming.connectors.redis.RedisSink;
    import org.apache.flink.streaming.connectors.redis.common.config.FlinkJedisPoolConfig;
    import org.apache.flink.streaming.connectors.redis.common.mapper.RedisCommand;
    import org.apache.flink.streaming.connectors.redis.common.mapper.RedisCommandDescription;
    import org.apache.flink.streaming.connectors.redis.common.mapper.RedisMapper;

    public class StreamingDemoToRedis {
        public static void main(String[] args) throws Exception{
            String hostname="hadoop01";
            String delimiter="\n";
            StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
            DataStreamSource<String> text = env.socketTextStream(hostname, 9000, delimiter);
            DataStream<Tuple2<String, String>> list_wordData = text.map(new MapFunction<String, Tuple2<String, String>>() {
                public Tuple2<String, String> map(String value) throws Exception {
                    return new Tuple2<>("list_word", value);
                }
            });
    
            //创建redis的配置
            FlinkJedisPoolConfig conf = new FlinkJedisPoolConfig.Builder().setHost(hostname).setPort(6379).build();
            //
            RedisSink<Tuple2<String, String>> redisSink = new RedisSink<Tuple2<String, String>>(conf, new MyRedisMapper());
            list_wordData.addSink(redisSink);
            env.execute("sink redis");
        }
        public static class MyRedisMapper implements RedisMapper<Tuple2<String, String>>{
    		//操作redis时选择的数据类型
            public RedisCommandDescription getCommandDescription() {
                return new RedisCommandDescription(RedisCommand.LPUSH);
            }
            //表示接受的数据中获取需要操作的redis的key值
            public String getKeyFromData(Tuple2<String, String> data) {
                return data.f0;
            }
            //表示接受的数据中获取需要操作的redis的value值
            public String getValueFromData(Tuple2<String, String> data) {
                return data.f1;
            }
        }
    
        /**
         * 结果为:
         * 192.168.189.128:6379> flushall
         * OK
         * 192.168.189.128:6379> lrange list_word 0 -1
         * 1) "ss"
         * 2) "rr"
         * 3) "ww"
         * 192.168.189.128:6379>
         */
    }