src/sqlancer/h2/H2Constant.java - master - NikolayIvkin/sqlancer2

src/sqlancer/h2/H2Constant.java
135 строк3 KB
Manuel Rigger
[H2] Use the default SQLancer random data generation mechanism
20 авг 2020, 16:42
20 авг 2020, 16:425b16b77
package sqlancer.h2;

import sqlancer.common.ast.newast.Node;

public class H2Constant implements Node<H2Expression> {

    private H2Constant() {
    }

    public static class H2NullConstant extends H2Constant {

        @Override
        public String toString() {
            return "NULL";
        }

    }

    public static class H2IntConstant extends H2Constant {

        private final long value;

        public H2IntConstant(long value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.valueOf(value);
        }

    }

    public static class H2BoolConstant extends H2Constant {

        private final boolean value;

        public H2BoolConstant(boolean value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.valueOf(value);
        }

    }

    public static class H2StringConstant extends H2Constant {

        private final String value;

        public H2StringConstant(String value) {
            this.value = value;
        }

        @Override
        public String toString() {
            return String.format("'%s'", value.replace("'", "''"));
        }

    }

    public static class H2DoubleConstant extends H2Constant {

        private final double value;

        public H2DoubleConstant(double value) {
            this.value = value;
        }

        @Override
        public String toString() {
            if (value == Double.POSITIVE_INFINITY) {
                return "POWER(0, -1)";
            } else if (value == Double.NEGATIVE_INFINITY) {
                return "(-POWER(0, -1))";
            } else if (Double.compare(value, -0.0) == 0) {
                return "(-CAST(0 AS DOUBLE))";
            } else if (Double.isNaN(value)) {
                return "SQRT(-1)";
            } else {
                return String.valueOf(value);
            }
        }

    }

    public static class H2BinaryConstant extends H2Constant {

        private String value;

        public H2BinaryConstant(long value) {
            this.value = Long.toHexString(value);
            if (this.value.length() % 2 == 1) {
                this.value = '0' + this.value; // pad with leading zero if needed
            }
        }

        public String getValue() {
            return value;
        }

        @Override
        public String toString() {
            return "X'" + value + "'";
        }

    }

    public static Node<H2Expression> createIntConstant(long val) {
        return new H2IntConstant(val);
    }

    public static Node<H2Expression> createNullConstant() {
        return new H2NullConstant();
    }

    public static Node<H2Expression> createBoolConstant(boolean val) {
        return new H2BoolConstant(val);
    }

    public static Node<H2Expression> createStringConstant(String val) {
        return new H2StringConstant(val);
    }

    public static Node<H2Expression> createDoubleConstant(double val) {
        return new H2DoubleConstant(val);
    }

    public static Node<H2Expression> createBinaryConstant(long val) {
        return new H2BinaryConstant(val);
    }

}