src/sqlancer/common/schema/AbstractTableColumn.java - master - NikolayIvkin/sqlancer2

src/sqlancer/common/schema/AbstractTableColumn.java

76 строк2 KB

Patrick

Rework Connection to DB to support non-SQL connections

23 ноя 2020, 03:28

23 ноя 2020, 03:2897b8850

package sqlancer.common.schema;

public class AbstractTableColumn<T extends AbstractTable<?, ?, ?>, U> implements Comparable<AbstractTableColumn<T, U>> {

    private final String name;
    private final U type;
    private T table;

    public AbstractTableColumn(String name, T table, U type) {
        this.name = name;
        this.table = table;
        this.type = type;
    }

    public String getName() {
        return name;
    }

    public String getFullQualifiedName() {
        if (table == null) {
            return getName();
        } else {
            return table.getName() + "." + getName();
        }
    }

    public void setTable(T table) {
        this.table = table;
    }

    public T getTable() {
        return table;
    }

    public U getType() {
        return type;
    }

    @Override
    public String toString() {
        if (table == null) {
            return String.format("%s: %s", getName(), getType());
        } else {
            return String.format("%s.%s: %s", table.getName(), getName(), getType());
        }
    }

    @Override
    public boolean equals(Object obj) {
        if (!(obj instanceof AbstractTableColumn)) {
            return false;
        } else {
            @SuppressWarnings("unchecked")
            AbstractTableColumn<T, U> c = (AbstractTableColumn<T, U>) obj;
            if (c.getTable() == null) {
                return getName().equals(c.getName());
            }
            return table.getName().contentEquals(c.getTable().getName()) && getName().equals(c.getName());
        }
    }

    @Override
    public int hashCode() {
        return getName().hashCode() + 11 * getType().hashCode();
    }

    @Override
    public int compareTo(AbstractTableColumn<T, U> o) {
        if (o.getTable().equals(this.getTable())) {
            return getName().compareTo(o.getName());
        } else {
            return o.getTable().compareTo(getTable());
        }
    }

}