Examples with Expr com.pingcap.tidb.tipb.Expr used on opensource projects

Example 6 with Expr

use of com.pingcap.tidb.tipb.Expr in project client-java by tikv.

the class ProtoConverter method visit.

@Override
protected Expr visit(Constant node, Object context) {
    Expr.Builder builder = Expr.newBuilder();
    DataType type = node.getDataType();
    if (node.getValue() == null) {
        builder.setTp(ExprType.Null);
    } else {
        // can mark it cannot be pushed down to coprocessor.
        if (node.isOverflowed()) {
            throw new UnsupportedOperationException("overflowed value cannot be pushed down to coprocessor");
        }
        builder.setTp(type.getProtoExprType());
        CodecDataOutput cdo = new CodecDataOutput();
        type.encode(cdo, EncodeType.PROTO, node.getValue());
        builder.setVal(cdo.toByteString());
        builder.setFieldType(toPBFieldType(getType(node)));
    }
    return builder.build();
}

Example 7 with Expr

use of com.pingcap.tidb.tipb.Expr in project tispark by pingcap.

the class ProtoConverter method visit.

@Override
protected Expr visit(Not node, Object context) {
    ScalarFuncSig protoSig = null;
    DataType dataType = getType(node);
    switch(dataType.getType()) {
        case TypeDecimal:
            protoSig = ScalarFuncSig.UnaryNotDecimal;
            break;
        case TypeDouble:
        case TypeFloat:
            protoSig = ScalarFuncSig.UnaryNotReal;
            break;
        case TypeInt24:
        case TypeLong:
        case TypeShort:
        case TypeLonglong:
        case TypeTiny:
            protoSig = ScalarFuncSig.UnaryNotInt;
            break;
        default:
    }
    Objects.requireNonNull(protoSig, "unary not can not find proper proto signature.");
    Expr.Builder builder = scalarToPartialProto(node, context);
    builder.setSig(protoSig);
    builder.setFieldType(toPBFieldType(getType(node)));
    return builder.build();
}

Example 8 with Expr

use of com.pingcap.tidb.tipb.Expr in project tispark by pingcap.

the class ProtoConverter method scalarToPartialProto.

// Generate protobuf builder with partial data encoded.
// Scalar Signature is left alone
private Expr.Builder scalarToPartialProto(Expression node, Object context) {
    Expr.Builder builder = Expr.newBuilder();
    // Scalar function type
    builder.setTp(ExprType.ScalarFunc);
    // Return type
    builder.setFieldType(toPBFieldType(getType(node)));
    for (Expression child : node.getChildren()) {
        Expr exprProto = child.accept(this, context);
        builder.addChildren(exprProto);
    }
    return builder;
}

Example 9 with Expr

use of com.pingcap.tidb.tipb.Expr in project tispark by pingcap.

the class ProtoConverter method visit.

@Override
protected Expr visit(LogicalBinaryExpression node, Object context) {
    ScalarFuncSig protoSig;
    switch(node.getCompType()) {
        case AND:
            protoSig = ScalarFuncSig.LogicalAnd;
            break;
        case OR:
            protoSig = ScalarFuncSig.LogicalOr;
            break;
        case XOR:
            protoSig = ScalarFuncSig.LogicalXor;
            break;
        default:
            throw new TiExpressionException(String.format("Unknown comparison type %s", node.getCompType()));
    }
    Expr.Builder builder = scalarToPartialProto(node, context);
    builder.setSig(protoSig);
    builder.setFieldType(toPBFieldType(getType(node)));
    return builder.build();
}

Example 10 with Expr

use of com.pingcap.tidb.tipb.Expr in project tispark by pingcap.

the class ProtoConverter method visit.

@Override
protected Expr visit(StringRegExpression node, Object context) {
    // assume after type coerce, children should be compatible
    ScalarFuncSig protoSig;
    switch(node.getRegType()) {
        case STARTS_WITH:
        case CONTAINS:
        case ENDS_WITH:
        case LIKE:
            protoSig = ScalarFuncSig.LikeSig;
            break;
        default:
            throw new TiExpressionException(String.format("Unknown reg type %s", node.getRegType()));
    }
    Expr.Builder builder = scalarToPartialProto(node, context);
    builder.setSig(protoSig);
    return builder.build();
}