flink sql实时计算当天pv写入mysql

大数据技术与应用实战

发布于 2020-09-24 16:18:48

2.9K0

发布于 2020-09-24 16:18:48

文章被收录于专栏：大数据技术与应用实战大数据技术与应用实战

今天我们主要来讲一个很简单但是很常见的需求，实时计算出网站当天的pv值，然后将结果实时更新到mysql数据库，以供前端查询显示。

接下来我们看看如何用flink sql来实现这个简单的功能。

首先我们还是使用datagen生成测试数据，随机生成一些用户id

     String sourceSql = "CREATE TABLE datagen (\n" +
                " userid int,\n" +
                " proctime as PROCTIME()\n" +
                ") WITH (\n" +
                " 'connector' = 'datagen',\n" +
                " 'rows-per-second'='100',\n" +
                " 'fields.userid.kind'='random',\n" +
                " 'fields.userid.min'='1',\n" +
                " 'fields.userid.max'='100'\n" +
                ")";

定义mysql的sink，这里mysql是作为了一个upsert的sink，所以必须要一个主键，在mysql建表的时候我们指定了当天的日期作为主键，mysql ddl如下

CREATE TABLE `pv` (
  `day_str` varchar(100) NOT NULL,
  `pv` bigint(10) DEFAULT NULL,
  PRIMARY KEY (`day_str`)
)

Flink中的ddl要和mysql中对的上，也要指定主键。

    String mysqlsql = "CREATE TABLE pv (\n" +
                "  day_str STRING,\n" +
                "  pv bigINT,\n" +
                "  PRIMARY KEY (day_str) NOT ENFORCED\n" +
                ") WITH (\n" +
                "   'connector' = 'jdbc',\n" +
                "   'username' = 'root',\n" +
                "   'password' = 'root',\n" +
                "   'url' = 'jdbc:mysql://localhost:3306/test',\n" +
                "   'table-name' = 'pv'\n" +
                ")";

接下来我们写一个简单的查询：

     tEnv.executeSql("insert into pv SELECT DATE_FORMAT(proctime, 'yyyy-MM-dd') as day_str, count(*) \n" +
                "FROM datagen \n" +
                "GROUP BY DATE_FORMAT(proctime, 'yyyy-MM-dd')");

可能对于以前一直做批处理的同学来说会感到疑惑，对于流式处理来说，group by将会返回一个可撤回流(RetractStream)，转化成datastream，将会得到一个Tuple2<Boolean, T>对象，这个对象第一个字段如果是false表示数据要撤回，true表示数据是我们新添加的，第二个字段是实际的数据。在这里，我们将这个实时更新的结果写入到了mysql。这样mysql表，每天就会只有一个数据，系统会不断地更新pv字段。