All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.aerospike.mapper.tools.mappers.ArrayMapper Maven / Gradle / Ivy

package com.aerospike.mapper.tools.mappers;

import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.List;

import com.aerospike.mapper.tools.DeferredObjectLoader;
import com.aerospike.mapper.tools.TypeMapper;
import com.aerospike.mapper.tools.utils.TypeUtils;
import com.aerospike.mapper.tools.DeferredObjectLoader.DeferredObject;
import com.aerospike.mapper.tools.DeferredObjectLoader.DeferredObjectSetter;
import com.aerospike.mapper.tools.DeferredObjectLoader.DeferredSetter;

public class ArrayMapper extends TypeMapper {

    private final Class instanceClass;
    private final boolean supportedWithoutTranslation;
    private final TypeMapper instanceClassMapper;
    private final Boolean allowBatch;

    public ArrayMapper(final Class instanceClass, final TypeMapper instanceClassMapper, final boolean allowBatch) {
        this.instanceClass = instanceClass;
        this.supportedWithoutTranslation = TypeUtils.isByteType(instanceClass);
        this.instanceClassMapper = instanceClassMapper;
        this.allowBatch = allowBatch;
    }

    @Override
    public Object toAerospikeFormat(Object value) {
        if (value == null) {
            return null;
        }
        int length = Array.getLength(value);
        if (this.supportedWithoutTranslation) {
            return value;
        }

        List results = new ArrayList<>();
        for (int i = 0; i < length; i++) {
            results.add(this.instanceClassMapper.toAerospikeFormat(Array.get(value, i)));
        }
        return results;
    }

    @Override
    public Object fromAerospikeFormat(Object value) {
        if (value == null) {
            return null;
        }
        List list = (List) value;
        if (this.supportedWithoutTranslation) {
            return value;
        }

        Object result = Array.newInstance(instanceClass, list.size());
        for (int i = 0; i < list.size(); i++) {

            Object obj = list.get(i);
            Object item = this.instanceClassMapper.fromAerospikeFormat(obj);
            if (!allowBatch || (!(item instanceof DeferredObject))) {
                Array.set(result, i, item);
            } else {
                final int thisIndex = i;
                DeferredSetter setter = object -> Array.set(result, thisIndex, object);
                DeferredObjectLoader.add(new DeferredObjectSetter(setter, (DeferredObject) item));
            }
        }
        return result;
    }
}