info.archinnov.achilles.internal.context.SchemaContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of achilles-core Show documentation
CQL implementation for Achilles using Datastax Java driver
There is a newer version: 6.1.0
/*
 * Copyright (C) 2012-2014 DuyHai DOAN
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package info.archinnov.achilles.internal.context;

import static info.archinnov.achilles.counter.AchillesCounter.ACHILLES_COUNTER_TABLE;

import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import com.datastax.driver.core.Cluster;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.TableMetadata;
import info.archinnov.achilles.internal.metadata.holder.EntityMeta;
import info.archinnov.achilles.internal.metadata.parsing.context.ParsingResult;
import info.archinnov.achilles.internal.table.SchemaReader;
import info.archinnov.achilles.internal.table.TableCreator;
import info.archinnov.achilles.internal.table.TableUpdater;
import info.archinnov.achilles.internal.table.TableValidator;

public class SchemaContext {

    private ConfigurationContext configContext;

    private Cluster cluster;

    private Session session;

    private String keyspaceName;

    private Map, EntityMeta> entityMetaMap;

    private boolean hasCounter;

    protected TableCreator tableCreator = TableCreator.Singleton.INSTANCE.get();
    protected TableValidator tableValidator = TableValidator.Singleton.INSTANCE.get();
    protected TableUpdater tableUpdater = TableUpdater.Singleton.INSTANCE.get();
    protected SchemaReader schemaReader = SchemaReader.Singleton.INSTANCE.get();

    public SchemaContext(ConfigurationContext configContext, Session session, String keyspaceName, Cluster cluster, ParsingResult parsingResult) {
        this.configContext = configContext;
        this.session = session;
        this.keyspaceName = keyspaceName;
        this.cluster = cluster;
        this.entityMetaMap = parsingResult.getMetaMap();
        this.hasCounter = parsingResult.hasSimpleCounter();
    }

    public Session getSession() {
        return session;
    }

    public boolean hasSimpleCounter() {
        return hasCounter;
    }

    public Set, EntityMeta>> entityMetaEntrySet() {
        return entityMetaMap.entrySet();
    }

    public void validateForEntity(EntityMeta entityMeta, TableMetadata tableMetaData) {
        tableValidator.validateForEntity(entityMeta, tableMetaData, configContext);
    }

    public void validateAchillesCounter() {
        tableValidator.validateAchillesCounter(cluster.getMetadata().getKeyspace(keyspaceName), keyspaceName);
    }

    public Map fetchTableMetaData() {
        return schemaReader.fetchTableMetaData(cluster, entityMetaMap.values());
    }

    public void createTableForEntity(EntityMeta entityMeta) {
        tableCreator.createTableForEntity(session, entityMeta, configContext);
    }

    public void createTableForCounter() {
        tableCreator.createTableForCounter(session, configContext);
    }

    public void updateForEntity(EntityMeta entityMeta, TableMetadata tableMetaData) {
        tableUpdater.updateTableForEntity(session, entityMeta, tableMetaData);
    }

    public boolean achillesCounterTableExists() {
        return cluster.getMetadata().getKeyspace(keyspaceName).getTable(ACHILLES_COUNTER_TABLE) != null;
    }
}