Examples with SqlWriter org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.sql.SqlWriter used on opensource projects

Example 51 with SqlWriter

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.sql.SqlWriter in project calcite by apache.

the class SqlLiteralChainOperator method unparse.

public void unparse(SqlWriter writer, SqlCall call, int leftPrec, int rightPrec) {
    final SqlWriter.Frame frame = writer.startList("", "");
    SqlCollation collation = null;
    for (Ord<SqlNode> operand : Ord.zip(call.getOperandList())) {
        SqlLiteral rand = (SqlLiteral) operand.e;
        if (operand.i > 0) {
            // SQL:2003 says there must be a newline between string
            // fragments.
            writer.newlineAndIndent();
        }
        if (rand instanceof SqlCharStringLiteral) {
            NlsString nls = ((SqlCharStringLiteral) rand).getNlsString();
            if (operand.i == 0) {
                collation = nls.getCollation();
                // print with prefix
                writer.literal(nls.asSql(true, false));
            } else {
                // print without prefix
                writer.literal(nls.asSql(false, false));
            }
        } else if (operand.i == 0) {
            // print with prefix
            rand.unparse(writer, leftPrec, rightPrec);
        } else {
            // print without prefix
            if (rand.getTypeName() == SqlTypeName.BINARY) {
                BitString bs = (BitString) rand.getValue();
                writer.literal("'" + bs.toHexString() + "'");
            } else {
                writer.literal("'" + rand.toValue() + "'");
            }
        }
    }
    if (collation != null) {
        collation.unparse(writer, 0, 0);
    }
    writer.endList(frame);
}

Example 52 with SqlWriter

use of org.apache.beam.vendor.calcite.v1_28_0.org.apache.calcite.sql.SqlWriter in project calcite by apache.

the class MysqlSqlDialect method unparseFloor.

/**
 * Unparses datetime floor for MySQL. There is no TRUNC function, so simulate
 * this using calls to DATE_FORMAT.
 *
 * @param writer Writer
 * @param call Call
 */
private void unparseFloor(SqlWriter writer, SqlCall call) {
    SqlLiteral node = call.operand(1);
    TimeUnitRange unit = (TimeUnitRange) node.getValue();
    if (unit == TimeUnitRange.WEEK) {
        writer.print("STR_TO_DATE");
        SqlWriter.Frame frame = writer.startList("(", ")");
        writer.print("DATE_FORMAT(");
        call.operand(0).unparse(writer, 0, 0);
        writer.print(", '%x%v-1'), '%x%v-%w'");
        writer.endList(frame);
        return;
    }
    String format;
    switch(unit) {
        case YEAR:
            format = "%Y-01-01";
            break;
        case MONTH:
            format = "%Y-%m-01";
            break;
        case DAY:
            format = "%Y-%m-%d";
            break;
        case HOUR:
            format = "%Y-%m-%d %k:00:00";
            break;
        case MINUTE:
            format = "%Y-%m-%d %k:%i:00";
            break;
        case SECOND:
            format = "%Y-%m-%d %k:%i:%s";
            break;
        default:
            throw new AssertionError("MYSQL does not support FLOOR for time unit: " + unit);
    }
    writer.print("DATE_FORMAT");
    SqlWriter.Frame frame = writer.startList("(", ")");
    call.operand(0).unparse(writer, 0, 0);
    writer.sep(",", true);
    writer.print("'" + format + "'");
    writer.endList(frame);
}