/*******************************************************************************
* Copyright 2011 Netflix
*
* 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 com.netflix.astyanax.thrift;
import java.nio.ByteBuffer;
import java.util.Date;
import java.util.UUID;
import com.netflix.astyanax.AstyanaxConfiguration;
import com.netflix.astyanax.Clock;
import com.netflix.astyanax.ColumnMutation;
import com.netflix.astyanax.Execution;
import com.netflix.astyanax.Serializer;
import com.netflix.astyanax.clock.ConstantClock;
import com.netflix.astyanax.model.ConsistencyLevel;
import com.netflix.astyanax.retry.RetryPolicy;
import com.netflix.astyanax.serializers.BooleanSerializer;
import com.netflix.astyanax.serializers.ByteBufferSerializer;
import com.netflix.astyanax.serializers.ByteSerializer;
import com.netflix.astyanax.serializers.BytesArraySerializer;
import com.netflix.astyanax.serializers.DateSerializer;
import com.netflix.astyanax.serializers.DoubleSerializer;
import com.netflix.astyanax.serializers.FloatSerializer;
import com.netflix.astyanax.serializers.IntegerSerializer;
import com.netflix.astyanax.serializers.LongSerializer;
import com.netflix.astyanax.serializers.ShortSerializer;
import com.netflix.astyanax.serializers.StringSerializer;
import com.netflix.astyanax.serializers.UUIDSerializer;
public abstract class AbstractThriftColumnMutationImpl implements ColumnMutation {
protected final ByteBuffer key;
protected final ByteBuffer column;
protected Clock clock;
protected RetryPolicy retry;
protected ConsistencyLevel writeConsistencyLevel;
public AbstractThriftColumnMutationImpl(ByteBuffer key, ByteBuffer column, AstyanaxConfiguration config) {
this.key = key;
this.column = column;
this.clock = config.getClock();
this.retry = config.getRetryPolicy().duplicate();
this.writeConsistencyLevel = config.getDefaultWriteConsistencyLevel();
}
@Override
public ColumnMutation withRetryPolicy(RetryPolicy retry) {
this.retry = retry;
return this;
}
@Override
public ColumnMutation setConsistencyLevel(ConsistencyLevel consistencyLevel) {
writeConsistencyLevel = consistencyLevel;
return this;
}
@Override
public ColumnMutation withTimestamp(long timestamp) {
this.clock = new ConstantClock(timestamp);
return this;
}
@Override
public Execution<Void> putValue(String value, Integer ttl) {
return insertValue(StringSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(byte[] value, Integer ttl) {
return insertValue(BytesArraySerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(byte value, Integer ttl) {
return insertValue(ByteSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(short value, Integer ttl) {
return insertValue(ShortSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(int value, Integer ttl) {
return insertValue(IntegerSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(long value, Integer ttl) {
return insertValue(LongSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(boolean value, Integer ttl) {
return insertValue(BooleanSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(ByteBuffer value, Integer ttl) {
return insertValue(ByteBufferSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(Date value, Integer ttl) {
return insertValue(DateSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(float value, Integer ttl) {
return insertValue(FloatSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(double value, Integer ttl) {
return insertValue(DoubleSerializer.get().toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putValue(UUID value, Integer ttl) {
return insertValue(UUIDSerializer.get().toByteBuffer(value), ttl);
}
@Override
public <T> Execution<Void> putValue(T value, Serializer<T> serializer, Integer ttl) {
return insertValue(serializer.toByteBuffer(value), ttl);
}
@Override
public Execution<Void> putEmptyColumn(Integer ttl) {
return insertValue(ThriftUtils.EMPTY_BYTE_BUFFER, ttl);
}
protected abstract Execution<Void> insertValue(ByteBuffer value, Integer ttl);
}