src/sqlancer/mongodb/ast/MongoDBSelect.java - master - NikolayIvkin/sqlancer2

src/sqlancer/mongodb/ast/MongoDBSelect.java
104 строки3 KB
Patrick
Add a variation where the count is compared to the number of returned documents in MongoDB
23 мар 2021, 01:03
23 мар 2021, 01:035897c86
package sqlancer.mongodb.ast;

import java.util.List;

import sqlancer.common.ast.newast.Node;
import sqlancer.mongodb.test.MongoDBColumnTestReference;

public class MongoDBSelect<E> implements Node<E> {

    private final String mainTableName;
    private final MongoDBColumnTestReference joinColumn;
    List<MongoDBColumnTestReference> projectionColumns;
    List<MongoDBColumnTestReference> lookupList;
    boolean hasFilter;
    Node<E> filterClause;
    boolean hasComputed;
    List<Node<E>> computedClauses;
    private boolean withCountClause;

    public MongoDBSelect(String mainTableName, MongoDBColumnTestReference joinColumn) {
        this.mainTableName = mainTableName;
        this.joinColumn = joinColumn;
    }

    public String getMainTableName() {
        return mainTableName;
    }

    public MongoDBColumnTestReference getJoinColumn() {
        return joinColumn;
    }

    public void setProjectionList(List<MongoDBColumnTestReference> fetchColumns) {
        if (fetchColumns == null || fetchColumns.isEmpty()) {
            throw new IllegalArgumentException();
        }
        this.projectionColumns = fetchColumns;
    }

    public List<MongoDBColumnTestReference> getProjectionList() {
        if (projectionColumns == null) {
            throw new IllegalStateException();
        }
        return projectionColumns;
    }

    public void setLookupList(List<MongoDBColumnTestReference> lookupList) {
        if (lookupList == null || lookupList.isEmpty()) {
            throw new IllegalArgumentException();
        }
        this.lookupList = lookupList;
    }

    public List<MongoDBColumnTestReference> getLookupList() {
        if (lookupList == null) {
            throw new IllegalStateException();
        }
        return lookupList;
    }

    public void setFilterClause(Node<E> filterClause) {
        if (filterClause == null) {
            hasFilter = false;
            this.filterClause = null;
            return;
        }
        hasFilter = true;
        this.filterClause = filterClause;
    }

    public Node<E> getFilterClause() {
        return filterClause;
    }

    public boolean hasFilter() {
        return hasFilter;
    }

    public void setComputedClause(List<Node<E>> computedClause) {
        if (computedClause == null) {
            hasComputed = false;
            this.computedClauses = null;
            return;
        }
        hasComputed = true;
        this.computedClauses = computedClause;
    }

    public List<Node<E>> getComputedClause() {
        return computedClauses;
    }

    public boolean hasComputed() {
        return hasComputed;
    }

    public boolean getWithCountClause() {
        return withCountClause;
    }

    public void setWithCountClause(boolean withCountClause) {
        this.withCountClause = withCountClause;
    }
}