Examples with MycatRelDataType io.mycat.beans.mycat.MycatRelDataType used on opensource projects

Example 1 with MycatRelDataType

use of io.mycat.beans.mycat.MycatRelDataType in project Mycat2 by MyCATApache.

the class VisualTablescanFactory method create.

@Override
public PhysicalPlan create(ComplierContext context) {
    RelOptTable table = tableScan.getRelNode().getTable();
    MycatLogicTable logicTable = table.unwrap(MycatLogicTable.class);
    TableHandler tableHandler = logicTable.getTable();
    Observable<Object[]> tableObservable = context.getTableObservable(tableHandler.getSchemaName(), tableHandler.getTableName());
    MycatRelDataType mycatRelDataTypeByCalcite = tableScan.getMycatRelDataTypeByCalcite();
    Schema schema = toArrowSchema(mycatRelDataTypeByCalcite);
    return ValuesPlan.create(schema, MycatRxJavaUtl.blockingIterable(tableObservable));
}

Example 2 with MycatRelDataType

use of io.mycat.beans.mycat.MycatRelDataType in project Mycat2 by MyCATApache.

the class LocalTableScan method getMycatRelDataType.

@Override
public MycatRelDataType getMycatRelDataType() {
    MycatLogicTable mycatTable = getTable().unwrap(MycatLogicTable.class);
    TableHandler tableTable = mycatTable.getTable();
    List<MycatField> mycatFields = tableTable.getColumns().stream().map(c -> c.toMycatField()).collect(Collectors.toList());
    return MycatRelDataType.of(mycatFields);
}

Example 3 with MycatRelDataType

use of io.mycat.beans.mycat.MycatRelDataType in project Mycat2 by MyCATApache.

the class DatasourcePoolTest method getSqlMycatRelDataType.

@NotNull
private static MycatRelDataType getSqlMycatRelDataType() {
    String sql = getSql("s", "t");
    MySqlCreateTableStatement sqlStatement = (MySqlCreateTableStatement) SQLUtils.parseSingleMysqlStatement(sql);
    MycatRelDataType mycatRelType = MycatRelDataType.getMycatRelType(sqlStatement);
    return mycatRelType;
}

Example 4 with MycatRelDataType

use of io.mycat.beans.mycat.MycatRelDataType in project Mycat2 by MyCATApache.

the class DatasourcePoolTest method main.

@SneakyThrows
public static void main(String[] args) {
    DatasourceConfig datasourceConfig = new DatasourceConfig();
    datasourceConfig.setDbType("mysql");
    datasourceConfig.setUser("root");
    datasourceConfig.setPassword("123456");
    datasourceConfig.setName("prototypeDs");
    datasourceConfig.setUrl("jdbc:mysql://localhost:3307/mysql");
    datasourceConfig.setMaxCon(10);
    Map<String, DatasourceConfig> datasources = Maps.of("prototypeDs", datasourceConfig);
    MetaClusterCurrent.register(JdbcConnectionManager.class, new JdbcConnectionManager(datasources, new DruidDatasourceProvider()));
    MetaClusterCurrent.register(IOExecutor.class, IOExecutor.DEFAULT);
    MycatDatasourcePool jdbcDs = new JdbcDatasourcePoolImpl("prototypeDs");
    NewMycatConnection newMycatConnection = jdbcDs.getConnection().toCompletionStage().toCompletableFuture().get();
    Observable<VectorSchemaRoot> observable = newMycatConnection.prepareQuery("select 1", Collections.emptyList(), new RootAllocator());
    List<VectorSchemaRoot> vectorSchemaRoots = observable.toList().blockingGet();
    System.out.println();
    // ///////////////////////////////////////////////////////////////////////////////////////////////////////////
    MycatDatasourcePool nativeDs = new VertxMySQLDatasourcePoolImpl(datasourceConfig, "prototypeDs");
    MycatRelDataType sqlMycatRelType = getSqlMycatRelDataType();
    TestResult testResult = getTestResult(jdbcDs);
    TestResult testResult2 = getTestResult(nativeDs);
    // List<String> collect = newMycatConnection.prepareQuery(querySql, Collections.emptyList()).toList().blockingGet().stream().map(i -> i.toString()).collect(Collectors.toList());
    MycatRowMetaData mycatRowMetaData1 = testResult.queryResult.getMycatRowMetaData();
    MycatRowMetaData mycatRowMetaData2 = testResult2.queryResult.getMycatRowMetaData();
    MycatRelDataType mycatRelDataType1 = mycatRowMetaData1.getMycatRelDataType();
    MycatRelDataType mycatRelDataType2 = mycatRowMetaData2.getMycatRelDataType();
    System.out.println(mycatRelDataType1);
    System.out.println(mycatRelDataType2);
    System.out.println("=====================================================================================================");
    boolean equals = mycatRelDataType1.equals(mycatRelDataType2);
    // Assert.assertTrue(equals);
    equals = mycatRelDataType1.equals(sqlMycatRelType);
    System.out.println(sqlMycatRelType);
    System.out.println(mycatRelDataType1);
    observable = newMycatConnection.prepareQuery("select * from testSchema.testColumnTable", Collections.emptyList(), new RootAllocator());
    vectorSchemaRoots = observable.toList().blockingGet();
    System.out.println();
}

Example 5 with MycatRelDataType

use of io.mycat.beans.mycat.MycatRelDataType in project Mycat2 by MyCATApache.

the class CalciteCompiler method convertValues.

public PhysicalPlan convertValues(Values values) {
    ImmutableList<ImmutableList<RexLiteral>> tuples = values.getTuples();
    ArrayList<Function[]> rowList = new ArrayList<>();
    for (ImmutableList<RexLiteral> tuple : tuples) {
        int size = tuple.size();
        Function[] functions = new Function[size];
        int index = 0;
        for (RexLiteral rexLiteral : tuple) {
            Function function = rexConverter.convertToFunction(rexLiteral);
            functions[index] = function;
            index++;
        }
        rowList.add(functions);
    }
    RelDataType rowType = values.getRowType();
    MycatRelDataType mycatRelDataType = MycatRelDataTypeUtil.getMycatRelDataType(rowType);
    Schema schema = FactoryUtil.toArrowSchema(mycatRelDataType);
    return executeCompiler.values(rowList, schema);
}