src/sqlancer/mysql/gen/admin/MySQLFlush.java - master - NikolayIvkin/sqlancer2

src/sqlancer/mysql/gen/admin/MySQLFlush.java

50 строк2 KB

Patrick

Continuation of Reworking Connection to DB to support non-SQL connections

26 ноя 2020, 15:40

26 ноя 2020, 15:4062e302e

package sqlancer.mysql.gen.admin;

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

import sqlancer.Randomly;
import sqlancer.common.query.SQLQueryAdapter;
import sqlancer.mysql.MySQLGlobalState;
import sqlancer.mysql.MySQLSchema.MySQLTable;

/*
 * https://dev.mysql.com/doc/refman/8.0/en/flush.html#flush-tables-variants
 */
public class MySQLFlush {

    private final List<MySQLTable> tables;
    private final StringBuilder sb = new StringBuilder();

    public MySQLFlush(List<MySQLTable> tables) {
        this.tables = tables;
    }

    public static SQLQueryAdapter create(MySQLGlobalState globalState) {
        return new MySQLFlush(globalState.getSchema().getDatabaseTablesRandomSubsetNotEmpty()).generate();
    }

    private SQLQueryAdapter generate() {
        sb.append("FLUSH");
        if (Randomly.getBoolean()) {
            sb.append(" ");
            sb.append(Randomly.fromOptions("NO_WRITE_TO_BINLOG", "LOCAL"));
            sb.append(" ");
            // TODO: | RELAY LOGS [FOR CHANNEL channel] not fully implemented
            List<String> options = Randomly.nonEmptySubset("BINARY LOGS", "ENGINE LOGS", "ERROR LOGS", "GENERAL LOGS",
                    "HOSTS", "LOGS", "PRIVILEGES", "OPTIMIZER_COSTS", "RELAY LOGS", "SLOW LOGS", "STATUS",
                    "USER_RESOURCES");
            sb.append(options.stream().collect(Collectors.joining(", ")));
        } else {
            sb.append(" ");
            sb.append("TABLES");
            if (Randomly.getBoolean()) {
                sb.append(" ");
                sb.append(tables.stream().map(t -> t.getName()).collect(Collectors.joining(", ")));
                // TODO implement READ LOCK and other variants
            }
        }
        return new SQLQueryAdapter(sb.toString());
    }

}