Flink自定义订单数据源

订单实体类：

package com.cnic.flink.entiey;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class Order {
    private String id;
    private Integer userId;
    private Integer money;
    private Long createTime;
}

自定义数据源：

package com.cnic.flink.func;

import com.cnic.flink.entiey.Order;
import org.apache.flink.streaming.api.functions.source.RichParallelSourceFunction;

import java.util.Random;
import java.util.UUID;

public class MyOrderSource extends RichParallelSourceFunction<Order> {

    boolean flag = true;


    public void run(SourceContext<Order> sourceContext) throws Exception {
        Random random = new Random();
        while (flag){
            Thread.sleep(1000);
            String id = UUID.randomUUID().toString();
            int userId = random.nextInt(3);
            int money = random.nextInt(101);
            long createTime = System.currentTimeMillis();
            sourceContext.collect(new Order(id, userId, money, createTime));
        }
    }

    public void cancel() {
        flag = false;
    }
}

从自定义数据源加载数据：

package com.daidai.mocksource;

import com.daidai.mocksource.domain.Order;
import com.daidai.mocksource.source.MyOrderSource;
import org.apache.flink.streaming.api.datastream.DataStreamSource;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;

public class MockSource {
    public static void main(String[] args) throws Exception {

        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
        DataStreamSource<Order> orderDataStreamSource = env.addSource(new MyOrderSource());
        orderDataStreamSource.print();

        env.execute();
    }
}

注意：

RichParallelSourceFunction是支持设置多并行度的，关于 RichParallelSourceFunction 与 RichSourceFunction 的区别，前者支持用户设置多并行度，后者不支持通过 setParallelism () 方法设置并行度大于1，默认的并行度为1。