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

Example 1 with MycatField

use of io.mycat.beans.mycat.MycatField 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 2 with MycatField

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

the class FactoryUtil method toArrowSchema.

public static Schema toArrowSchema(MycatRelDataType mycatRelDataTypeByCalcite) {
    List<MycatField> fieldList = mycatRelDataTypeByCalcite.getFieldList();
    int columnCount = fieldList.size();
    ImmutableList.Builder<org.apache.arrow.vector.types.pojo.Field> builder = ImmutableList.builder();
    for (int i = 0; i < columnCount; i++) {
        MycatField mycatField = fieldList.get(i);
        String columnName = mycatField.getName();
        MycatDataType columnType = mycatField.getMycatDataType();
        boolean nullable = mycatField.isNullable();
        InnerType innerType;
        switch(columnType) {
            case BOOLEAN:
                innerType = InnerType.BOOLEAN_TYPE;
                break;
            case BIT:
            case UNSIGNED_LONG:
                innerType = InnerType.UINT64_TYPE;
                break;
            case TINYINT:
                innerType = InnerType.INT8_TYPE;
                break;
            case UNSIGNED_TINYINT:
                innerType = InnerType.UINT8_TYPE;
                break;
            case SHORT:
                innerType = InnerType.INT16_TYPE;
                break;
            case UNSIGNED_SHORT:
            case YEAR:
                innerType = InnerType.UINT16_TYPE;
                break;
            case INT:
                innerType = InnerType.INT32_TYPE;
                break;
            case UNSIGNED_INT:
                innerType = InnerType.UINT32_TYPE;
                break;
            case LONG:
                innerType = InnerType.INT64_TYPE;
                break;
            case DOUBLE:
                innerType = InnerType.DOUBLE_TYPE;
                break;
            case DECIMAL:
                innerType = InnerType.DECIMAL_TYPE;
                break;
            case DATE:
                innerType = InnerType.DATE_TYPE;
                break;
            case DATETIME:
                innerType = InnerType.DATETIME_MILLI_TYPE;
                break;
            case TIME:
                innerType = InnerType.TIME_MILLI_TYPE;
                break;
            case CHAR:
            case VARCHAR:
                innerType = InnerType.STRING_TYPE;
                break;
            case BINARY:
                innerType = InnerType.BINARY_TYPE;
                break;
            case FLOAT:
                innerType = InnerType.FLOAT_TYPE;
                break;
            default:
                throw new IllegalStateException("Unexpected value: " + columnType);
        }
        org.apache.arrow.vector.types.pojo.Field field = FieldBuilder.of(columnName, innerType.getArrowType(), nullable).toArrow();
        builder.add(field);
    }
    return new Schema(builder.build());
}

Example 3 with MycatField

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

the class MycatRelDataTypeUtil method getMycatRelDataType.

public static MycatRelDataType getMycatRelDataType(RelDataType relDataType) {
    List<RelDataTypeField> fieldList = relDataType.getFieldList();
    ImmutableList.Builder<MycatField> builder = ImmutableList.builder();
    for (RelDataTypeField relDataTypeField : fieldList) {
        SqlTypeName sqlTypeName = relDataTypeField.getType().getSqlTypeName();
        int jdbcOrdinal = sqlTypeName.getJdbcOrdinal();
        MycatDataType mycatDataType = MycatDataType.fromJdbc(JDBCType.valueOf(jdbcOrdinal), true);
        MycatField mycatField = MycatField.of(relDataTypeField.getName(), mycatDataType, relDataTypeField.getType().isNullable());
        builder.add(mycatField);
    }
    return MycatRelDataType.of(builder.build());
}