com.netflix.astyanax.thrift.AbstractThriftColumnMutationImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of astyanax-thrift Show documentation
The newest version!
/*******************************************************************************
 * 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 putValue(String value, Integer ttl) {
        return insertValue(StringSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(byte[] value, Integer ttl) {
        return insertValue(BytesArraySerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(byte value, Integer ttl) {
        return insertValue(ByteSerializer.get().toByteBuffer(value), ttl);
    }
    
    @Override
    public Execution putValue(short value, Integer ttl) {
        return insertValue(ShortSerializer.get().toByteBuffer(value), ttl);
    }
    
    @Override
    public Execution putValue(int value, Integer ttl) {
        return insertValue(IntegerSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(long value, Integer ttl) {
        return insertValue(LongSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(boolean value, Integer ttl) {
        return insertValue(BooleanSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(ByteBuffer value, Integer ttl) {
        return insertValue(ByteBufferSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(Date value, Integer ttl) {
        return insertValue(DateSerializer.get().toByteBuffer(value), ttl);
    }
    
    @Override
    public Execution putValue(float value, Integer ttl) {
        return insertValue(FloatSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(double value, Integer ttl) {
        return insertValue(DoubleSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public Execution putValue(UUID value, Integer ttl) {
        return insertValue(UUIDSerializer.get().toByteBuffer(value), ttl);
    }

    @Override
    public  Execution putValue(T value, Serializer serializer, Integer ttl) {
        return insertValue(serializer.toByteBuffer(value), ttl);
    }

    @Override
    public Execution putEmptyColumn(Integer ttl) {
        return insertValue(ThriftUtils.EMPTY_BYTE_BUFFER, ttl);
    }

    protected abstract Execution insertValue(ByteBuffer value, Integer ttl);

}