com.ptsmods.mysqlw.table.TablePreset Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of MySQLw Show documentation
A wrapper for MySQL connections
There is a newer version: 1.5.1
package com.ptsmods.mysqlw.table;

import com.google.common.collect.ImmutableList;
import com.ptsmods.mysqlw.Database;
import com.google.common.collect.ImmutableMap;

import java.util.*;
import java.util.stream.Collectors;

public class TablePreset {

    /**
     * Create a new TablePreset with the given name.
     * @param name The name of the new TablePreset.
     * @return A new TablePreset with the given name.
     */
    public static TablePreset create(String name) {
        return new TablePreset(name);
    }

    private final Map> columns = new LinkedHashMap<>();
    private final List indices = new ArrayList<>();
    private String name;

    private TablePreset(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    /**
     * Changes the name of this TablePreset.
     * @param name The new name
     * @return This TablePreset
     */
    public TablePreset setName(String name) {
        this.name = name;
        return this;
    }

    /**
     * Puts all columns with their names into this TablePreset.
     * @param columns The columns to add.
     * @return This TablePreset
     */
    public TablePreset putAll(Map> columns) {
        this.columns.putAll(columns);
        return this;
    }

    /**
     * Puts a new column into this TablePreset.
     * @param name The name of the new column
     * @param structure The {@link ColumnStructure} of the column
     * @return This TablePreset
     */
    public TablePreset putColumn(String name, ColumnStructure structure) {
        columns.put(name, structure);
        return this;
    }

    /**
     * Removes a column by its name.
     * @param name The name of the column to remove.
     * @return This TablePreset
     */
    public TablePreset removeColumn(String name) {
        columns.remove(name);
        return this;
    }

    /**
     * Adds a {@link TableIndex} to this preset.
     * @param index The {@link TableIndex} to add
     * @return This TablePreset
     */
    public TablePreset addIndex(TableIndex index) {
        indices.add(index);
        return this;
    }

    /**
     * Removes an index by its type.
     * @param index The index to remove.
     * @return This TablePreset
     */
    public TablePreset removeIndex(TableIndex index) {
        indices.removeIf(index0 -> index0 == index);
        return this;
    }

    /**
     * @return An immutable copy of the map of columns in this
     */
    public Map> getColumns() {
        return ImmutableMap.copyOf(columns);
    }

    public Map build(Database.RDBMS type) {
        return columns.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> entry.getValue().toString(type), (e1, e2) -> e1, LinkedHashMap::new));
    }

    public String buildQuery(Database.RDBMS type) {
        StringBuilder query = new StringBuilder("CREATE TABLE IF NOT EXISTS " + getName() + " (");
        build(type).forEach((key, value) -> query.append(key).append(' ').append(value).append(", "));
        if (type != Database.RDBMS.SQLite) getIndices().forEach(index -> query.append(index).append(", ")); // Indices do not work on SQLite apparently when creating a table.
        if (getColumns().size() > 0) query.delete(query.length() - 2, query.length());
        query.append(");");
        return query.toString();
    }

    public List getIndices() {
        return ImmutableList.copyOf(indices);
    }

    public TablePreset create(Database db) {
        db.createTable(this);
        return this;
    }

    @Override
    public TablePreset clone() {
        TablePreset clone = new TablePreset(name);
        clone.putAll(columns.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> entry.getValue().clone())));
        clone.indices.addAll(indices.stream().map(TableIndex::clone).collect(Collectors.toList()));
        return clone;
    }

}