Flink Java 之 MySQL Sink

1. 实体类

package com.daidai.sink.domain;

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

@Data
@AllArgsConstructor
@NoArgsConstructor
public class User {
    private Integer id;
    private String username;
    private String password;
}

2. 自定义MySQL sink 继承 RichSinkFunction

package com.daidai.sink.mysqlsink;

import com.daidai.sink.domain.User;
import org.apache.flink.configuration.Configuration;
import org.apache.flink.streaming.api.functions.sink.RichSinkFunction;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;

public class MyMySqlSink extends RichSinkFunction<User> {

    private Connection connection = null;
    private PreparedStatement prepareStatement = null;

    @Override
    public void open(Configuration parameters) throws Exception {
        connection = DriverManager.getConnection(
                "jdbc:mysql://localhost:3306/flink",
                "root",
                "000000"
        );
        String sql = "insert into user values(?, ?, ?)";
        prepareStatement = connection.prepareStatement(sql);
    }

    @Override
    public void invoke(User value, Context context) throws Exception {
        prepareStatement.setInt(1, value.getId());
        prepareStatement.setString(2, value.getUsername());
        prepareStatement.setString(3, value.getPassword());
        prepareStatement.execute();
    }

    @Override
    public void close() throws Exception {
        if (prepareStatement != null) {
            prepareStatement.close();
        }
        if (connection != null) {
            connection.close();
        }
    }


}

3. 添加 MySQL sink

package com.daidai.sink;

import com.daidai.sink.domain.User;
import com.daidai.sink.mysqlsink.MyMySqlSink;
import org.apache.flink.streaming.api.datastream.DataStreamSource;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;

public class MySqlSink {
    public static void main(String[] args) throws Exception {
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();

        DataStreamSource<User> userSource = env.fromElements(new User(1, "admin", "admin"));

        userSource.addSink(new MyMySqlSink());

        env.execute();

    }

}