private CachedStatement buildCachedStatement(final String sql, final Supplier<CalciteConnection> connectionSupplier, final ProcessSession session, final FlowFile flowFile, final RecordReaderFactory recordReaderFactory) throws SQLException { final CalciteConnection connection = connectionSupplier.get(); final SchemaPlus rootSchema = connection.getRootSchema(); final FlowFileTable<?, ?> flowFileTable = new FlowFileTable<>(session, flowFile, recordReaderFactory, getLogger()); rootSchema.add("FLOWFILE", flowFileTable); rootSchema.setCacheEnabled(false); final PreparedStatement stmt = connection.prepareStatement(sql); return new CachedStatement(stmt, flowFileTable, connection); }
connection = DriverManager.getConnection("jdbc:calcite:", properties); final CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); final SchemaPlus rootSchema = calciteConnection.getRootSchema();
/** A trojan-horse method, subject to change without notice. */ @VisibleForTesting public static DataContext createDataContext(CalciteConnection connection) { return ((CalciteConnectionImpl) connection) .createDataContext(ImmutableMap.of(), CalciteSchema.from(connection.getRootSchema())); }
/** A trojan-horse method, subject to change without notice. */ @VisibleForTesting public static DataContext createDataContext(CalciteConnection connection) { return ((CalciteConnectionImpl) connection) .createDataContext(ImmutableMap.of(), CalciteSchema.from(connection.getRootSchema())); }
public Connection apply(Connection connection) throws SQLException { if (schema != null) { CalciteConnection con = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = con.getRootSchema(); rootSchema.add(name, schema); } connection.setSchema(name); return connection; } }
public Connection apply(Connection connection) throws SQLException { if (schema != null) { CalciteConnection con = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = con.getRootSchema(); rootSchema.add(name, schema); } connection.setSchema(name); return connection; } }
private CachedStatement buildCachedStatement(final String sql, final Supplier<CalciteConnection> connectionSupplier, final ProcessSession session, final FlowFile flowFile, final RecordReaderFactory recordReaderFactory) throws SQLException { final CalciteConnection connection = connectionSupplier.get(); final SchemaPlus rootSchema = connection.getRootSchema(); final FlowFileTable<?, ?> flowFileTable = new FlowFileTable<>(session, flowFile, recordReaderFactory, getLogger()); rootSchema.add("FLOWFILE", flowFileTable); rootSchema.setCacheEnabled(false); final PreparedStatement stmt = connection.prepareStatement(sql); return new CachedStatement(stmt, flowFileTable, connection); }
protected ConnectionPostProcessor newSchema(final String schemaName, final String tableName, final Table table) { return connection -> { CalciteConnection con = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = con.getRootSchema(); SchemaPlus schema = rootSchema.add(schemaName, new AbstractSchema()); schema.add(tableName, table); connection.setSchema(schemaName); return connection; }; }
protected ConnectionPostProcessor newSchema(final String schemaName, final String tableName, final Table table) { return connection -> { CalciteConnection con = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = con.getRootSchema(); SchemaPlus schema = rootSchema.add(schemaName, new AbstractSchema()); schema.add(tableName, table); connection.setSchema(schemaName); return connection; }; }
private Connection setupConnectionWithNestedAnyTypeTable() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); schema.add("nested", new NestedCollectionWithAnyTypeTable()); return connection; }
private Connection setupConnectionWithNestedTable() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); schema.add("nested", new NestedCollectionTable()); return connection; }
private Connection setupConnectionWithNestedTable() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); schema.add("nested", new NestedCollectionTable()); return connection; }
private Connection setupConnectionWithNestedAnyTypeTable() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); schema.add("nested", new NestedCollectionWithAnyTypeTable()); return connection; }
private Connection getConnectionWithMultiplyFunction() throws ClassNotFoundException, SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); final TableFunction table = TableFunctionImpl.create(Smalls.MULTIPLICATION_TABLE_METHOD); schema.add("multiplication", table); return connection; }
private Connection getConnectionWithMultiplyFunction() throws ClassNotFoundException, SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); SchemaPlus schema = rootSchema.add("s", new AbstractSchema()); final TableFunction table = TableFunctionImpl.create(Smalls.MULTIPLICATION_TABLE_METHOD); schema.add("multiplication", table); return connection; }
public Connection apply(Connection connection) throws SQLException { CalciteConnection con = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = con.getRootSchema(); switch (schemaSpec) { case CLONE_FOODMART: case JDBC_FOODMART_WITH_LATTICE: addSchema(rootSchema, SchemaSpec.JDBC_FOODMART); /* fall through */ default: addSchema(rootSchema, schemaSpec); } con.setSchema(schemaSpec.schemaName); return connection; } }
@Before public void setUp() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); rootSchema.add("test", new ReflectiveSchema(new TestSchema())); calciteConnection.setSchema("test"); this.conn = calciteConnection; }
@Before public void setUp() throws SQLException { Connection connection = DriverManager.getConnection("jdbc:calcite:"); CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class); SchemaPlus rootSchema = calciteConnection.getRootSchema(); rootSchema.add("test", new ReflectiveSchema(new TestSchema())); calciteConnection.setSchema("test"); this.conn = calciteConnection; }
public CalciteConnection createConnection() throws SQLException { CalciteConnection connection = (CalciteConnection) new AutoTempDriver(objects) .connect("jdbc:calcite:", new Properties()); final SchemaPlus rootSchema = connection.getRootSchema(); rootSchema.add("hr", new ReflectiveSchema(new HrSchema())); connection.setSchema("hr"); return connection; } })
public CalciteConnection createConnection() throws SQLException { CalciteConnection connection = (CalciteConnection) new AutoTempDriver(objects) .connect("jdbc:calcite:", new Properties()); final SchemaPlus rootSchema = connection.getRootSchema(); rootSchema.add("hr", new ReflectiveSchema(new HrSchema())); connection.setSchema("hr"); return connection; } })