src/sqlancer/databend/gen/DatabendInsertGenerator.java - master - NikolayIvkin/sqlancer2

src/sqlancer/databend/gen/DatabendInsertGenerator.java
57 строк2 KB
Yisong Han
fix(databend): mvn verify succeed
09 сен 2022, 19:09
09 сен 2022, 19:095f07a2d
package sqlancer.databend.gen;

import java.util.List;
import java.util.stream.Collectors;

import sqlancer.common.gen.AbstractInsertGenerator;
import sqlancer.common.query.ExpectedErrors;
import sqlancer.common.query.SQLQueryAdapter;
import sqlancer.common.schema.AbstractTableColumn;
import sqlancer.databend.DatabendErrors;
import sqlancer.databend.DatabendProvider.DatabendGlobalState;
import sqlancer.databend.DatabendSchema.DatabendColumn;
import sqlancer.databend.DatabendSchema.DatabendTable;
import sqlancer.databend.DatabendToStringVisitor;

public class DatabendInsertGenerator extends AbstractInsertGenerator<DatabendColumn> {

    private final DatabendGlobalState globalState;
    private final ExpectedErrors errors = new ExpectedErrors();

    public DatabendInsertGenerator(DatabendGlobalState globalState) {
        this.globalState = globalState;
    }

    public static SQLQueryAdapter getQuery(DatabendGlobalState globalState) {
        return new DatabendInsertGenerator(globalState).generate();
    }

    private SQLQueryAdapter generate() {
        sb.append("INSERT INTO ");
        DatabendTable table = globalState.getSchema().getRandomTable(t -> !t.isView());
        List<DatabendColumn> columns = table.getRandomNonEmptyColumnSubset();
        sb.append(table.getName());
        sb.append("(");
        sb.append(columns.stream().map(AbstractTableColumn::getName).collect(Collectors.joining(", ")));
        sb.append(")");
        sb.append(" VALUES ");
        insertColumns(columns);
        DatabendErrors.addInsertErrors(errors);
        return new SQLQueryAdapter(sb.toString(), errors);
    }

    @Override
    protected void insertValue(DatabendColumn column) {
        // TODO: 等Databend实现DEFAULT关键字,暂时注入普通的value
        // if (Randomly.getBooleanWithRatherLowProbability()) {
        // sb.append("DEFAULT");
        // } else {
        // sb.append(DatabendToStringVisitor.asString(new DatabendExpressionGenerator(globalState).generateConstant()));
        // }
        String value = DatabendToStringVisitor.asString(new DatabendNewExpressionGenerator(globalState)
                .generateConstant(column.getType().getPrimitiveDataType(), column.isNullable())); // 生成一个与column相同的常量类型
        sb.append(value);

    }

}