com.infomaximum.database.utils.SchemaTableCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rdao Show documentation
Library for creating a light cluster
The newest version!
package com.infomaximum.database.utils;

import com.infomaximum.database.schema.dbstruct.DBField;
import com.infomaximum.database.schema.dbstruct.DBHashIndex;
import com.infomaximum.database.schema.dbstruct.DBSchema;
import com.infomaximum.database.schema.dbstruct.DBTable;
import com.infomaximum.database.schema.table.FieldReference;

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

public class SchemaTableCache {

    private final DBSchema schema;
    private final Map uuidTableMap;
    private final Map> uuidForeignTableFieldReferences;


    public SchemaTableCache(List tables, DBSchema schema) {
        this.schema = schema;
        this.uuidTableMap = tables.stream()
                .collect(Collectors.toMap(table -> buildUuidTable(table.getName(), table.getNamespace()),
                        table -> table));
        this.uuidForeignTableFieldReferences = new HashMap<>();
        putFieldReferences(tables);
    }

    public DBTable getTable(String tableName, String namespace) {
        return uuidTableMap.get(buildUuidTable(tableName, namespace));
    }

    public void newTable(DBTable table) {
        uuidTableMap.put(buildUuidTable(table.getName(), table.getNamespace()), table);
        putFieldReferences(table);
    }

    public void createField(DBField field, DBTable table) {
        if (field.isForeignKey()) {
            putFieldReference(field, table);
        }
    }

    public void removeField(DBField field, DBTable table) {
        if (field.isForeignKey()) {
            DBTable referencedTable = schema.getTableById(field.getForeignTableId());
            uuidForeignTableFieldReferences.get(buildUuidTable(referencedTable.getName(), referencedTable.getNamespace()))
                    .remove(new FieldReference(table.getName(), table.getNamespace(), null));
        }
    }

    public void removeTable(String name, String namespace) {
        uuidTableMap.remove(buildUuidTable(name, namespace));
        uuidForeignTableFieldReferences.remove(buildUuidTable(name, namespace));
    }

    public Set getTableReference(String name, String namespace) {
        return uuidForeignTableFieldReferences.getOrDefault(buildUuidTable(name, namespace), Collections.emptySet());
    }

    private String buildUuidTable(String tableName, String namespace) {
        return tableName + "." + namespace;
    }

    private void putFieldReferences(DBTable table) {
        for (DBField field : table.getSortedFields()) {
            if (field.isForeignKey()) {
                putFieldReference(field, table);
            }
        }
    }

    private void putFieldReferences(List tables) {
        tables.forEach(this::putFieldReferences);
    }

    private void putFieldReference(DBField field, DBTable table) {
        DBTable referencedTable = schema.getTableById(field.getForeignTableId());
        FieldReference fieldReference = new FieldReference(table.getName(), table.getNamespace(), new DBHashIndex(field));
        String tableUuid = buildUuidTable(referencedTable.getName(), referencedTable.getNamespace());
        Set references = uuidForeignTableFieldReferences.get(tableUuid);
        if (references == null) {
            references = new HashSet<>();
            references.add(fieldReference);
            uuidForeignTableFieldReferences.put(tableUuid, references);
        } else {
            references.add(fieldReference);
        }
    }
}