Search in sources :

Example 1 with BatchTableSource

use of org.apache.flink.table.sources.BatchTableSource in project flink by apache.

the class TableSourceITCase method testBatchTableSourceTableAPI.

@Test
public void testBatchTableSourceTableAPI() throws Exception {
    ExecutionEnvironment env = ExecutionEnvironment.getExecutionEnvironment();
    BatchTableEnvironment tableEnv = TableEnvironment.getTableEnvironment(env, config());
    BatchTableSource csvTable = CommonTestData.getCsvTableSource();
    tableEnv.registerTableSource("persons", csvTable);
    Table result = tableEnv.scan("persons").select("id, first, last, score");
    DataSet<Row> resultSet = tableEnv.toDataSet(result, Row.class);
    List<Row> results = resultSet.collect();
    String expected = "1,Mike,Smith,12.3\n" + "2,Bob,Taylor,45.6\n" + "3,Sam,Miller,7.89\n" + "4,Peter,Smith,0.12\n" + "5,Liz,Williams,34.5\n" + "6,Sally,Miller,6.78\n" + "7,Alice,Smith,90.1\n" + "8,Kelly,Williams,2.34\n";
    compareResultAsText(results, expected);
}
Also used : BatchTableSource(org.apache.flink.table.sources.BatchTableSource) ExecutionEnvironment(org.apache.flink.api.java.ExecutionEnvironment) Table(org.apache.flink.table.api.Table) Row(org.apache.flink.types.Row) BatchTableEnvironment(org.apache.flink.table.api.java.BatchTableEnvironment) Test(org.junit.Test)

Example 2 with BatchTableSource

use of org.apache.flink.table.sources.BatchTableSource in project flink by apache.

the class TableSourceITCase method testBatchTableSourceSQL.

@Test
public void testBatchTableSourceSQL() throws Exception {
    ExecutionEnvironment env = ExecutionEnvironment.getExecutionEnvironment();
    BatchTableEnvironment tableEnv = TableEnvironment.getTableEnvironment(env, config());
    BatchTableSource csvTable = CommonTestData.getCsvTableSource();
    tableEnv.registerTableSource("persons", csvTable);
    Table result = tableEnv.sql("SELECT last, FLOOR(id), score * 2 FROM persons WHERE score < 20");
    DataSet<Row> resultSet = tableEnv.toDataSet(result, Row.class);
    List<Row> results = resultSet.collect();
    String expected = "Smith,1,24.6\n" + "Miller,3,15.78\n" + "Smith,4,0.24\n" + "Miller,6,13.56\n" + "Williams,8,4.68\n";
    compareResultAsText(results, expected);
}
Also used : BatchTableSource(org.apache.flink.table.sources.BatchTableSource) ExecutionEnvironment(org.apache.flink.api.java.ExecutionEnvironment) Table(org.apache.flink.table.api.Table) Row(org.apache.flink.types.Row) BatchTableEnvironment(org.apache.flink.table.api.java.BatchTableEnvironment) Test(org.junit.Test)

Aggregations

ExecutionEnvironment (org.apache.flink.api.java.ExecutionEnvironment)2 Table (org.apache.flink.table.api.Table)2 BatchTableEnvironment (org.apache.flink.table.api.java.BatchTableEnvironment)2 BatchTableSource (org.apache.flink.table.sources.BatchTableSource)2 Row (org.apache.flink.types.Row)2 Test (org.junit.Test)2