info.archinnov.achilles.statement.CQLStatementGenerator Maven / Gradle / Ivy

Go to download
/**
 *
 * Copyright (C) 2012-2013 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.statement;

import static com.datastax.driver.core.querybuilder.QueryBuilder.*;
import info.archinnov.achilles.context.CQLDaoContext;
import info.archinnov.achilles.entity.metadata.EntityMeta;
import info.archinnov.achilles.entity.metadata.PropertyMeta;
import info.archinnov.achilles.entity.metadata.PropertyType;
import info.archinnov.achilles.exception.AchillesException;
import info.archinnov.achilles.query.slice.CQLSliceQuery;
import info.archinnov.achilles.statement.prepared.CQLSliceQueryPreparedStatementGenerator;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import com.datastax.driver.core.PreparedStatement;
import com.datastax.driver.core.Query;
import com.datastax.driver.core.Statement;
import com.datastax.driver.core.querybuilder.Delete;
import com.datastax.driver.core.querybuilder.Insert;
import com.datastax.driver.core.querybuilder.QueryBuilder;
import com.datastax.driver.core.querybuilder.Select;
import com.datastax.driver.core.querybuilder.Select.Selection;
import com.datastax.driver.core.querybuilder.Update;
import com.datastax.driver.core.querybuilder.Update.Assignments;
import com.google.common.collect.FluentIterable;

public class CQLStatementGenerator {

	private CQLSliceQueryStatementGenerator sliceQueryGenerator = new CQLSliceQueryStatementGenerator();
	private CQLSliceQueryPreparedStatementGenerator sliceQueryPreparedGenerator = new CQLSliceQueryPreparedStatementGenerator();

	public  Query generateSelectSliceQuery(CQLSliceQuery sliceQuery, int limit) {
		EntityMeta meta = sliceQuery.getMeta();

		Select select = generateSelectEntity(meta);
		select = select.limit(limit);
		if (sliceQuery.getCQLOrdering() != null) {
			select.orderBy(sliceQuery.getCQLOrdering());
		}

		Statement where = sliceQueryGenerator.generateWhereClauseForSelectSliceQuery(sliceQuery, select);

		return where.setConsistencyLevel(sliceQuery.getConsistencyLevel());
	}

	public  PreparedStatement generateIteratorSliceQuery(CQLSliceQuery sliceQuery, CQLDaoContext daoContext) {
		EntityMeta meta = sliceQuery.getMeta();

		Select select = generateSelectEntity(meta);
		select = select.limit(sliceQuery.getLimit());
		if (sliceQuery.getCQLOrdering() != null) {
			select.orderBy(sliceQuery.getCQLOrdering());
		}

		Statement where = sliceQueryPreparedGenerator.generateWhereClauseForIteratorSliceQuery(sliceQuery, select);

		PreparedStatement preparedStatement = daoContext.prepare(where);
		preparedStatement.setConsistencyLevel(sliceQuery.getConsistencyLevel());
		return preparedStatement;
	}

	public  Query generateRemoveSliceQuery(CQLSliceQuery sliceQuery) {
		EntityMeta meta = sliceQuery.getMeta();

		Delete delete = QueryBuilder.delete().from(meta.getTableName());
		Statement where = sliceQueryGenerator.generateWhereClauseForDeleteSliceQuery(sliceQuery, delete);

		return where.setConsistencyLevel(sliceQuery.getConsistencyLevel());
	}

	public Select generateSelectEntity(EntityMeta entityMeta) {
		PropertyMeta idMeta = entityMeta.getIdMeta();

		Selection select = select();

		generateSelectForPrimaryKey(idMeta, select);

		List eagerMetas = FluentIterable.from(entityMeta.getEagerMetas())
				.filter(PropertyType.excludeIdType).toImmutableList();

		for (PropertyMeta pm : eagerMetas) {
			select.column(pm.getPropertyName());
		}
		return select.from(entityMeta.getTableName());
	}

	public Insert generateInsert(Object entity, EntityMeta entityMeta) {
		PropertyMeta idMeta = entityMeta.getIdMeta();
		Insert insert = insertInto(entityMeta.getTableName());
		generateInsertPrimaryKey(entity, idMeta, insert);

		List nonProxyMetas = FluentIterable.from(entityMeta.getAllMetasExceptIdMeta())
				.filter(PropertyType.excludeCounterType).toImmutableList();

		List fieldMetas = new ArrayList(nonProxyMetas);
		fieldMetas.remove(idMeta);

		for (PropertyMeta pm : fieldMetas) {
			Object value = pm.getValueFromField(entity);
			value = encodeValueForCassandra(pm, value);
			insert.value(pm.getPropertyName(), value);
		}
		return insert;
	}

	public Update.Assignments generateUpdateFields(Object entity, EntityMeta entityMeta, List pms) {
		PropertyMeta idMeta = entityMeta.getIdMeta();
		Update update = update(entityMeta.getTableName());

		int i = 0;
		Assignments assignments = null;
		for (PropertyMeta pm : pms) {
			Object value = pm.getValueFromField(entity);
			value = encodeValueForCassandra(pm, value);
			if (i == 0) {
				assignments = update.with(set(pm.getPropertyName(), value));
			} else {
				assignments.and(set(pm.getPropertyName(), value));
			}
			i++;
		}
		return generateWhereClauseForUpdate(entity, idMeta, assignments);
	}

	private Update.Assignments generateWhereClauseForUpdate(Object entity, PropertyMeta idMeta, Assignments update) {
		Object primaryKey = idMeta.getPrimaryKey(entity);
		if (idMeta.isEmbeddedId()) {
			Update.Where where = null;
			int index = 0;

			List componentNames = idMeta.getComponentNames();
			List