com.xxdb.route.LiteralListPartitionedTableRouter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api-java Show documentation
The messaging and data conversion protocol between Java and DolphinDB server
There is a newer version: 1.0.27
package com.xxdb.route;

import com.xxdb.data.*;

import java.util.*;

class LiteralListPartitionedTableRouter implements TableRouter {
    private Map> locationMap;
    LiteralListPartitionedTableRouter(AbstractVector values, BasicAnyVector locations) {
        initialize(values, locations);
    }

    @SuppressWarnings("unchecked")
	private void initialize(AbstractVector values, BasicAnyVector locations) {
        if (values.getDataType() != Entity.DATA_TYPE.DT_ANY || values.getDataForm() != Entity.DATA_FORM.DF_VECTOR) {
            throw new RuntimeException("expect a vector of partitioning lists");
        }
        if (values.rows() <= 0) {
            throw new RuntimeException("requires at least one partitioning list");
        }
        if (locations.rows() <= 0) {
            throw new RuntimeException("requires at least one location");
        }
        if (locations.getEntity(0).getDataType() != Entity.DATA_TYPE.DT_STRING) {
            throw new RuntimeException("location must be a string");
        }
        if (values.rows() != locations.rows()) {
            throw new RuntimeException("expect # locations == # partitioning lists");
        }
        this.locationMap = new HashMap<>();
        List[] locationListArray = new List[locations.rows()];
        boolean isScalar = locations.getEntity(0).getDataForm() == Entity.DATA_FORM.DF_SCALAR;
        if (isScalar) {
            for (int i = 0; i < locations.rows(); ++i) {
                BasicString location = (BasicString)locations.get(i);
                locationListArray[i] = Arrays.asList(location.getString());
            }
        } else {
            for (int i = 0; i < locations.rows(); ++i) {
                BasicStringVector locationVector = (BasicStringVector) locations.getEntity(i);
                locationListArray[i] = new ArrayList<>();
                for (int j = 0; j < locationVector.rows(); ++j) {
                    BasicString location = (BasicString)locationVector.get(j);
                    locationListArray[i].add(location.getString());
                }
            }
        }

        for (int i = 0; i < values.rows(); ++i) {
            AbstractVector partitioningList = (AbstractVector) ((BasicAnyVector) values).getEntity(i);
            if (partitioningList.rows() <= 0) {
                throw new RuntimeException("expect partitioning list to be nonempty");
            }
            if (partitioningList.getDataCategory() != Entity.DATA_CATEGORY.LITERAL) {
                throw new RuntimeException("expect partitioning column values to be LITERAL.");
            }
            for (int j = 0; j < partitioningList.rows(); ++j) {
                try {
                    String val = partitioningList.get(j).getString();
                    locationMap.put(val, locationListArray[i]);
                } catch (Exception e) {
                    throw new RuntimeException(e);
                }
            }
        }
    }

    @Override
    public String route(Scalar partitionColumn) {
        if (partitionColumn.getDataCategory() != Entity.DATA_CATEGORY.LITERAL)
            throw  new RuntimeException("invalid column category type" + partitionColumn.getDataCategory().name() + ", expect LITERAL category.");
        try {
            String stringVal = partitionColumn.getString();
            List locations = locationMap.get(stringVal);
            if (locations == null)
                throw new RuntimeException(partitionColumn.getNumber().longValue() + " not in any partitioning list!");
            return locations.get(0);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}