info.archinnov.achilles.internals.codec.ByteArrayCodec 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

/*
 * Copyright (C) 2012-2016 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.internals.codec;

import java.nio.ByteBuffer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.datastax.driver.core.utils.Bytes;

import info.archinnov.achilles.exception.AchillesTranscodingException;
import info.archinnov.achilles.type.codec.Codec;

public class ByteArrayCodec implements Codec {

    private static final Logger LOGGER = LoggerFactory.getLogger(ByteArrayCodec.class);

    @Override
    public Class sourceType() {
        return Byte[].class;
    }

    @Override
    public Class targetType() {
        return ByteBuffer.class;
    }

    @Override
    public ByteBuffer encode(Byte[] fromJava) throws AchillesTranscodingException {
        if (fromJava == null) return null;
        byte[] bytesPrimitive = new byte[fromJava.length];
        int i = 0;
        for (byte b : fromJava) bytesPrimitive[i++] = b;
        if (LOGGER.isTraceEnabled()) {
            LOGGER.trace(String.format("Encoding Byte[] '%s' to ByteBuffer", Bytes.toHexString(ByteBuffer.wrap(bytesPrimitive).duplicate())));
        }
        return ByteBuffer.wrap(bytesPrimitive);
    }

    @Override
    public Byte[] decode(ByteBuffer fromCassandra) {
        if (fromCassandra == null) return null;
        if (LOGGER.isTraceEnabled()) {
            LOGGER.trace(String.format("Decoding ByteBuffer '%s' to Byte[]", Bytes.toHexString(fromCassandra.duplicate())));
        }
        return readByteBuffer(fromCassandra);
    }

    private Byte[] readByteBuffer(Object fromCassandra) {
        ByteBuffer byteBuffer = (ByteBuffer) fromCassandra;
        byte[] bytes = new byte[byteBuffer.remaining()];
        byteBuffer.get(bytes);
        Byte[] byteObjects = new Byte[bytes.length];
        int i = 0;
        for (byte b : bytes) byteObjects[i++] = b;
        return byteObjects;
    }
}