flink自定义数据源——获取当前时间戳

1.新建maven工程——myflink

2.导入pom.xml依赖

<properties>
    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    <maven.compiler.source>1.8</maven.compiler.source>
    <maven.compiler.target>1.8</maven.compiler.target>
    <flink.version>1.7.2</flink.version>
  </properties>

  <dependencies>
    <dependency>
      <groupId>junit</groupId>
      <artifactId>junit</artifactId>
      <version>4.11</version>
      <scope>test</scope>
    </dependency>
    <!-- https://mvnrepository.com/artifact/org.apache.flink/flink-scala -->
    <dependency>
      <groupId>org.apache.flink</groupId>
      <artifactId>flink-scala_2.11</artifactId>
      <version>${flink.version}</version>
    </dependency>
    <!-- https://mvnrepository.com/artifact/org.apache.flink/flink-streaming-scala -->
    <dependency>
      <groupId>org.apache.flink</groupId>
      <artifactId>flink-streaming-scala_2.11</artifactId>
      <version>${flink.version}</version>
    </dependency>
    <!-- https://mvnrepository.com/artifact/org.apache.flink/flink-clients -->
    <dependency>
      <groupId>org.apache.flink</groupId>
      <artifactId>flink-clients_2.11</artifactId>
      <version>${flink.version}</version>
    </dependency>
  </dependencies>

3.创建MySource.scala

package cn.alisa.myflink.exp

import java.util.Date

import org.apache.flink.streaming.api.functions.source.SourceFunction
import org.apache.flink.streaming.api.scala._
//自定义数据源
//一秒内多个线程并发读这一秒
class MySource extends SourceFunction[String]{
  var flag=true;
  override def run(sourceContext: SourceFunction.SourceContext[String]): Unit = {
    while(flag){
      //采集数据
      sourceContext.collect(new Date().getTime()+"")
    }
  }

  override def cancel(): Unit ={
    flag=false
  }
}
//测试
object Test{
  def main(args: Array[String]): Unit = {
    val environment = StreamExecutionEnvironment.getExecutionEnvironment
    val ds: DataStream[String] = environment.addSource(new MySource)
    ds.print()
    environment.execute("mysource")
  }
}

运行结果如下：