CqlStyleMutationQuery.java

package com.netflix.astyanax.cql.writes;

import java.util.List;
import java.util.concurrent.atomic.AtomicReference;

import com.netflix.astyanax.cql.CqlKeyspaceImpl.KeyspaceContext;
import com.netflix.astyanax.cql.schema.CqlColumnFamilyDefinitionImpl;
import com.netflix.astyanax.cql.util.CFQueryContext;
import com.netflix.astyanax.model.ColumnFamily;
import com.netflix.astyanax.model.ConsistencyLevel;

public class CqlStyleMutationQuery {

	protected final KeyspaceContext ksContext;
	protected final CFQueryContext<?,?> cfContext;
	protected final List<CqlColumnMutationImpl<?,?>> mutationList;
	
	protected AtomicReference<Boolean> deleteRow;
	
	protected final AtomicReference<Long> defaultTimestamp;
	protected final AtomicReference<Integer> defaultTTL; 
	protected final ConsistencyLevel consistencyLevel;

	private static final String USING = " USING ";
	private static final String TTL = " TTL ";
	private static final String AND = " AND";
	private static final String TIMESTAMP = " TIMESTAMP ";
	
	public CqlStyleMutationQuery(KeyspaceContext ksCtx, CFQueryContext<?,?> cfCtx, 
								 List<CqlColumnMutationImpl<?,?>> mutationList, AtomicReference<Boolean> deleteRow, 
								 AtomicReference<Integer> ttl, AtomicReference<Long> timestamp, ConsistencyLevel consistencyLevel) {
		
		this.ksContext = ksCtx;
		this.cfContext = cfCtx;
		
		this.mutationList = mutationList;
		this.deleteRow = deleteRow;
		this.defaultTTL = ttl;
		this.defaultTimestamp = timestamp;
		this.consistencyLevel = consistencyLevel;
		
		if (this.consistencyLevel != null) {
			cfContext.setConsistencyLevel(consistencyLevel);
		}
	}
	
	public String getDeleteEntireRowQuery() {
		ColumnFamily<?,?> cf = cfContext.getColumnFamily();
		CqlColumnFamilyDefinitionImpl cfDef = (CqlColumnFamilyDefinitionImpl)cf.getColumnFamilyDefinition();
		return "DELETE FROM " + ksContext.getKeyspace() + "." + cf.getName() + 
				" WHERE " + cfDef.getPartitionKeyColumnDefinition().getName() + " = ?;";
	}

	public void appendWriteOptions(StringBuilder sb, Integer overrideTTL, Long overrideTimestamp) {
		
		Integer ttl = overrideTTL != null ? overrideTTL : defaultTTL.get();
		Long timestamp = overrideTimestamp != null ? overrideTimestamp : defaultTimestamp.get();
				
		if (ttl != null || timestamp != null) {
			sb.append(USING);
		}
		
		if (ttl != null) {
			sb.append(TTL + ttl);
		}
		
		if (timestamp != null) {
			if (ttl != null) {
				sb.append(AND);
			}
			sb.append(TIMESTAMP + timestamp);
		}
	}
	
}