Examples with MySQLRowDesc io.vertx.mysqlclient.impl.MySQLRowDesc used on opensource projects

Example 1 with MySQLRowDesc

use of io.vertx.mysqlclient.impl.MySQLRowDesc in project vertx-sql-client by eclipse-vertx.

the class QueryCommandBaseCodec method handleResultsetColumnDefinitionsDecodingCompleted.

protected void handleResultsetColumnDefinitionsDecodingCompleted() {
    commandHandlerState = CommandHandlerState.HANDLING_ROW_DATA_OR_END_PACKET;
    // use the column definitions if provided by execute or fetch response instead of prepare response
    MySQLRowDesc mySQLRowDesc = new MySQLRowDesc(columnDefinitions, format);
    decoder = new RowResultDecoder<>(cmd.collector(), mySQLRowDesc);
}

Example 2 with MySQLRowDesc

use of io.vertx.mysqlclient.impl.MySQLRowDesc in project Mycat2 by MyCATApache.

the class QueryCommandBaseCodec method handleResultsetColumnDefinitionsDecodingCompleted.

protected void handleResultsetColumnDefinitionsDecodingCompleted() {
    commandHandlerState = CommandHandlerState.HANDLING_ROW_DATA_OR_END_PACKET;
    Collector<Row, ?, T> collector = cmd.collector();
    MySQLRowDesc mySQLRowDesc = new MySQLRowDesc(columnDefinitions, format);
    if (collector instanceof MysqlCollector) {
        ((MysqlCollector) collector).onColumnDefinitions(mySQLRowDesc);
    }
    decoder = new RowResultDecoder<>(collector, /*cmd.isSingleton()*/
    mySQLRowDesc);
}

Example 3 with MySQLRowDesc

use of io.vertx.mysqlclient.impl.MySQLRowDesc in project Mycat2 by MyCATApache.

the class RowSetJdbcPreparedJdbcQuery method extracted.

public static <R> void extracted(Promise<SqlResult<R>> promise, Statement statement, ResultSet resultSet, Collector<Row, ?, R> collector) throws SQLException {
    try {
        if (resultSet == null) {
            Object o = collector.supplier().get();
            Function<Object, Object> finisher = (Function) collector.finisher();
            promise.complete(new MySqlResult<>(0, 0, 0, (R) finisher.apply(o), Collections.emptyList()));
            return;
        }
        JdbcRowMetaData metaData = new JdbcRowMetaData(resultSet.getMetaData());
        int columnCount = metaData.getColumnCount();
        List<ColumnDescriptor> columnDescriptors = new ArrayList<>();
        for (int i = 0; i < columnCount; i++) {
            int index = i;
            columnDescriptors.add(new ColumnDescriptor() {

                @Override
                public String name() {
                    return metaData.getColumnName(index);
                }

                @Override
                public boolean isArray() {
                    return false;
                }

                @Override
                public String typeName() {
                    return null;
                }

                @Override
                public JDBCType jdbcType() {
                    return JDBCType.valueOf(metaData.getColumnType(index));
                }
            });
        }
        RowDesc rowDesc = new RowDesc(metaData.getColumnList(), columnDescriptors);
        ColumnDefPacket[] columnDefPackets = new ColumnDefPacket[columnCount];
        for (int i = 0; i < columnCount; i++) {
            columnDefPackets[i] = new ColumnDefPacketImpl(metaData, i);
        }
        if (collector instanceof StreamMysqlCollector) {
            MySQLRowDesc mySQLRowDesc = new MySQLRowDesc(Arrays.asList(columnDefPackets).stream().map(packet -> {
                String catalog = new String(packet.getColumnCatalog());
                String schema = new String(packet.getColumnSchema());
                String table = new String(packet.getColumnTable());
                String orgTable = new String(packet.getColumnOrgTable());
                String name = new String(packet.getColumnName());
                String orgName = new String(packet.getColumnOrgName());
                int characterSet = packet.getColumnCharsetSet();
                long columnLength = packet.getColumnLength();
                DataType type = DataType.valueOf(packet.getColumnType() == 15 ? 253 : packet.getColumnType());
                int flags = packet.getColumnFlags();
                byte decimals = packet.getColumnDecimals();
                ColumnDefinition columnDefinition = new ColumnDefinition(catalog, schema, table, orgTable, name, orgName, characterSet, columnLength, type, flags, decimals);
                return columnDefinition;
            }).toArray(n -> new ColumnDefinition[n]), DataFormat.TEXT);
            ((StreamMysqlCollector) collector).onColumnDefinitions(mySQLRowDesc);
        }
        {
            Object supplier = collector.supplier().get();
            BiConsumer<Object, Row> accumulator = (BiConsumer) collector.accumulator();
            Function<Object, Object> finisher = (Function) collector.finisher();
            int count = 0;
            while (resultSet.next()) {
                JDBCRow jdbcRow = new MycatRow(rowDesc);
                for (int i = 0; i < columnCount; i++) {
                    jdbcRow.addValue(resultSet.getObject(i + 1));
                }
                count++;
                accumulator.accept(supplier, jdbcRow);
            }
            finisher.apply(supplier);
            resultSet.close();
            statement.close();
            promise.complete(new MySqlResult<>(count, 0, 0, (R) supplier, columnDescriptors));
        }
    } catch (Throwable throwable) {
        promise.tryFail(throwable);
    }
}

Example 4 with MySQLRowDesc

use of io.vertx.mysqlclient.impl.MySQLRowDesc in project Mycat2 by MyCATApache.

the class RowSetQuery method execute.

@Override
public Future<RowSet<Row>> execute() {
    VertxRowSetImpl vertxRowSet = new VertxRowSetImpl();
    StreamMysqlCollector streamMysqlCollector = new StreamMysqlCollector() {

        @Override
        public void onColumnDefinitions(MySQLRowDesc columnDefinitions) {
            vertxRowSet.setColumnDescriptor(columnDefinitions.columnDescriptor());
        }

        @Override
        public void onRow(Row row) {
            vertxRowSet.list.add(row);
        }
    };
    return runTextQuery(sql, mySQLClientSession.mySQLClientSession, streamMysqlCollector).map(voidSqlResult -> {
        vertxRowSet.setAffectRow(voidSqlResult.rowCount());
        vertxRowSet.setLastInsertId(voidSqlResult.property(MySQLClient.LAST_INSERTED_ID));
        return vertxRowSet;
    });
}

Example 5 with MySQLRowDesc

use of io.vertx.mysqlclient.impl.MySQLRowDesc in project Mycat2 by MyCATApache.

the class VertxMycatTextCollector method onColumnDefEof.

@Override
public void onColumnDefEof(MySQLPacket mySQLPacket, int startPos, int endPos) {
    rowResultDecoder = new MycatVertxRowResultDecoder(collector, new MySQLRowDesc(currentColumnDefList, DataFormat.TEXT));
    this.c = collector.supplier().get();
    this.accumulator = collector.accumulator();
    if (collector instanceof StreamMysqlCollector) {
        MySQLRowDesc mySQLRowDesc = new MySQLRowDesc(this.currentColumnDefList, DataFormat.TEXT);
        ((StreamMysqlCollector) collector).onColumnDefinitions(mySQLRowDesc);
    }
}