nl.renarj.jasdb.service.BagOperationUtil Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jasdb_localservice Show documentation
There is a newer version: 1.2.1
/*
 * The JASDB software and code is Copyright protected 2011 and owned by Renze de Vries
 * 
 * All the code and design principals in the codebase are also Copyright 2011 
 * protected and owned Renze de Vries. Any unauthorized usage of the code or the 
 * design and principals as in this code is prohibited.
 */
package nl.renarj.jasdb.service;

import nl.renarj.core.statistics.StatRecord;
import nl.renarj.core.statistics.StatisticsMonitor;
import nl.renarj.jasdb.api.SimpleEntity;
import nl.renarj.jasdb.core.exceptions.JasDBStorageException;
import nl.renarj.jasdb.core.exceptions.MetadataParseException;
import nl.renarj.jasdb.core.exceptions.RuntimeJasDBException;
import nl.renarj.jasdb.core.storage.RecordResult;
import nl.renarj.jasdb.core.streams.ClonableByteArrayInputStream;
import nl.renarj.jasdb.core.streams.ClonableDataStream;
import nl.renarj.jasdb.index.Index;
import nl.renarj.jasdb.index.keys.Key;
import nl.renarj.jasdb.index.keys.factory.KeyFactory;
import nl.renarj.jasdb.index.keys.impl.CompositeKey;
import nl.renarj.jasdb.index.keys.impl.UUIDKey;
import nl.renarj.jasdb.index.keys.keyinfo.KeyNameMapperImpl;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.util.HashSet;
import java.util.Set;

/**
 * Simple utility class that does Bag entity operations
 *
 * @author Renze de Vries
 */
public class BagOperationUtil {
    private static final Logger LOG = LoggerFactory.getLogger(BagOperationUtil.class);

    private static final String ENTITY_ENCODING = "UTF8";

    public static KeyNameMapperImpl DEFAULT_DOC_ID_MAPPER = new KeyNameMapperImpl();
    static {
        DEFAULT_DOC_ID_MAPPER.addMappedField(0, SimpleEntity.DOCUMENT_ID);
    }

    public static ClonableDataStream toStream(SimpleEntity entity) throws JasDBStorageException {
        try {
            String entityJson = SimpleEntity.toJson(entity);

            return new ClonableByteArrayInputStream(entityJson.getBytes(ENTITY_ENCODING));
        } catch(MetadataParseException e) {
            throw new JasDBStorageException("Unable to serialize to json", e);
        } catch(UnsupportedEncodingException e) {
            throw new JasDBStorageException("Unable to encode entity into Unicode", e);
        }
    }

    public static SimpleEntity toEntity(InputStream stream) throws JasDBStorageException {
        return SimpleEntity.fromStream(stream);
    }

    public static Set createEntityKeys(SimpleEntity entity, Index index) throws JasDBStorageException {
        StatRecord createKey = StatisticsMonitor.createRecord("bag:createKey");
        KeyFactory keyFactory = index.getKeyInfo().getKeyFactory();
        Set insertKeys;
        if(keyFactory.isMultiValueKey(entity)) {
            insertKeys = keyFactory.createMultivalueKeys(entity);
        } else {
            insertKeys = new HashSet<>();
            insertKeys.add(keyFactory.createKey(entity));
        }
        createKey.stop();
        return insertKeys;
    }
    
    public static void doIndexInsert(Set keys, Index index) throws JasDBStorageException {
        LOG.trace("Inserting {} keys into index: {}", keys.size(), index.getKeyInfo().getKeyName());
        StatRecord indexInsert = StatisticsMonitor.createRecord("bag:indexInsert");
        for(Key key : keys) {
            index.insertIntoIndex(key);
        }
        indexInsert.stop();
    }

    public static Key recordToKey(RecordResult recordResult) {
        try {
            SimpleEntity entity = toEntity(recordResult.getStream());
            return entityToKey(entity);
        } catch (JasDBStorageException e) {
            throw new RuntimeJasDBException("Unable to read record Document Id", e);
        }
    }

    public static Key entityToKey(SimpleEntity entity) throws JasDBStorageException {
        CompositeKey compositeKey = new CompositeKey();
        compositeKey.addKey(DEFAULT_DOC_ID_MAPPER, SimpleEntity.DOCUMENT_ID, new UUIDKey(entity.getInternalId()));

        return compositeKey;
    }
}