com.ebuddy.cassandra.dao.AbstractThriftStructuredDataSupport Maven / Gradle / Ivy

Go to download
package com.ebuddy.cassandra.dao;

import java.util.HashMap;
import java.util.Map;

import org.apache.commons.lang3.Validate;

import com.ebuddy.cassandra.Path;
import com.ebuddy.cassandra.StructuredDataSupport;
import com.ebuddy.cassandra.databind.CustomTypeResolverBuilder;
import com.ebuddy.cassandra.structure.DefaultPath;
import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * Abstract implementation of StructuredDataSupport for Thrift.
 *
 * @param  the type of the row key
 *
 * @author Eric Zoerner [email protected]
 */
public abstract class AbstractThriftStructuredDataSupport implements StructuredDataSupport {
    private  static final int MAX_CODE_POINT = 0x10FFFF;
    protected final ObjectMapper writeMapper;
    protected final ObjectMapper readMapper;

    protected AbstractThriftStructuredDataSupport() {
        readMapper = new ObjectMapper();
        writeMapper = new ObjectMapper();
        writeMapper.setDefaultTyping(new CustomTypeResolverBuilder());
    }

    protected void validateArgs(K rowKey, Path path) {
        Validate.isTrue(!path.isEmpty(), "Path must not be empty");
        Validate.notNull(rowKey, "Row key must not be empty");
    }

    protected final String getFinishString(String start) {
        int startCodePointCount = start.codePointCount(0, start.length());
        int finishCodePointCount = startCodePointCount + 1;
        int[] finishCodePoints = new int[finishCodePointCount];
        for (int i = 0; i < startCodePointCount; i++) {
            finishCodePoints[i] = start.codePointAt(i);
        }
        finishCodePoints[finishCodePointCount - 1] = MAX_CODE_POINT;
        return new String(finishCodePoints, 0, finishCodePointCount);
    }

    /**
     * Convert strings to paths and remove the start of the paths that match the inputPath.
     */
    protected final Map getTerminalPathMap(Path inputPath, Map columnsMap) {
        Map pathMap = new HashMap(columnsMap.size());
        for (Map.Entry entry : columnsMap.entrySet()) {
            Path path = DefaultPath.fromEncodedPathString(entry.getKey());
            if (!path.startsWith(inputPath)) {
                throw new IllegalStateException("unexpected path found in database:" + path);
            }
            path = path.tail(inputPath.size());
            pathMap.put(path, entry.getValue());
        }
        return pathMap;
    }

    @Override
    public void writeToPath(K rowKey, Path path, Object value) {
        writeToPath(rowKey, path, value, null);
    }

    @Override
    public void deletePath(K rowKey, Path path) {
        deletePath(rowKey, path, null);
    }

    @Override
    public Path createPath(String... elements) {
        return DefaultPath.fromStrings(elements);
    }
}