info.archinnov.achilles.internal.table.TableUpdater 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
package info.archinnov.achilles.internal.table;

import static com.datastax.driver.core.DataType.counter;
import static com.datastax.driver.core.DataType.list;
import static com.datastax.driver.core.DataType.map;
import static com.datastax.driver.core.DataType.set;
import static com.google.common.collect.FluentIterable.from;
import static info.archinnov.achilles.internal.cql.TypeMapper.toCQLDataType;
import static info.archinnov.achilles.schemabuilder.SchemaBuilder.alterTable;
import java.util.List;
import java.util.Set;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.datastax.driver.core.ColumnMetadata;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.TableMetadata;
import com.google.common.base.Function;
import info.archinnov.achilles.internal.metadata.holder.EntityMeta;
import info.archinnov.achilles.internal.metadata.holder.PropertyMeta;

public class TableUpdater {
    private static final Logger log = LoggerFactory.getLogger(TableUpdater.class);
    private static final Logger DML_LOG = LoggerFactory.getLogger(TableCreator.ACHILLES_DDL_SCRIPT);

    private static final Function COLUMN_NAME_EXTRACTOR = new Function() {
        @Override
        public String apply(ColumnMetadata columnMetadata) {
            return columnMetadata.getName();
        }
    };


    public void updateTableForEntity(Session session, EntityMeta entityMeta, TableMetadata tableMetadata) {
        log.debug("Updating table for entityMeta {}", entityMeta.getClassName());

        if (!entityMeta.config().isSchemaUpdateEnabled()) {
            return;
        }

        List existingColumns = tableMetadata.getColumns();
        List propertyMetas = entityMeta.getAllMetasExceptId();
        Set columnNames = from(existingColumns).transform(COLUMN_NAME_EXTRACTOR).toSet();
        addNewPropertiesToBuilder(session, entityMeta,propertyMetas, columnNames);
    }

    private void addNewPropertiesToBuilder(Session session,EntityMeta entityMeta,List propertyMetas, Set columnNames) {
        final String tableName = entityMeta.config().getQualifiedTableName();

        for (PropertyMeta propertyMeta : propertyMetas) {
            String cqlColumnName = propertyMeta.getCQLColumnName();
            if (!columnNames.contains(cqlColumnName)) {
                Class valueClass = propertyMeta.structure().getCQLValueType();
                final boolean staticColumn = propertyMeta.structure().isStaticColumn();
                String alterTableScript = "";
                switch (propertyMeta.type()) {
                    case SIMPLE:
                        alterTableScript = alterTable(tableName).addColumn(cqlColumnName, staticColumn).type(toCQLDataType(valueClass));
                        session.execute(alterTableScript);
                        break;
                    case LIST:
                        alterTableScript = alterTable(tableName).addColumn(cqlColumnName, staticColumn).type(list(toCQLDataType(valueClass)));
                        session.execute(alterTableScript);
                        break;
                    case SET:
                        alterTableScript = alterTable(tableName).addColumn(cqlColumnName, staticColumn).type(set(toCQLDataType(valueClass)));
                        session.execute(alterTableScript);
                        break;
                    case MAP:
                        final Class