org.openlca.proto.io.server.DataSetReader Maven / Gradle / Ivy

Go to download
package org.openlca.proto.io.server;

import org.openlca.core.model.ModelType;
import org.openlca.proto.ProtoActor;
import org.openlca.proto.ProtoCurrency;
import org.openlca.proto.ProtoDQSystem;
import org.openlca.proto.ProtoEpd;
import org.openlca.proto.ProtoFlow;
import org.openlca.proto.ProtoFlowProperty;
import org.openlca.proto.ProtoImpactCategory;
import org.openlca.proto.ProtoImpactMethod;
import org.openlca.proto.ProtoLocation;
import org.openlca.proto.ProtoParameter;
import org.openlca.proto.ProtoProcess;
import org.openlca.proto.ProtoProductSystem;
import org.openlca.proto.ProtoProject;
import org.openlca.proto.ProtoResult;
import org.openlca.proto.ProtoSocialIndicator;
import org.openlca.proto.ProtoSource;
import org.openlca.proto.ProtoUnitGroup;
import org.openlca.proto.grpc.ProtoDataSet;
import org.openlca.proto.io.ProtoStoreReader;
import org.openlca.util.Strings;

import java.util.Collections;
import java.util.Objects;
import java.util.Set;

record DataSetReader(ProtoDataSet dataSet) implements ProtoStoreReader {

	static DataSetReader of(ProtoDataSet dataSet) {
		return new DataSetReader(dataSet);
	}

	ModelType getType() {
		if (dataSet.hasActor())
			return ModelType.ACTOR;
		if (dataSet.hasCurrency())
			return ModelType.CURRENCY;
		if (dataSet.hasDqSystem())
			return ModelType.DQ_SYSTEM;
		if (dataSet.hasEpd())
			return ModelType.EPD;
		if (dataSet.hasFlowProperty())
			return ModelType.FLOW_PROPERTY;
		if (dataSet.hasFlow())
			return ModelType.FLOW;
		if (dataSet.hasImpactCategory())
			return ModelType.IMPACT_CATEGORY;
		if (dataSet.hasImpactMethod())
			return ModelType.IMPACT_METHOD;
		if (dataSet.hasLocation())
			return ModelType.LOCATION;
		if (dataSet.hasParameter())
			return ModelType.PARAMETER;
		if (dataSet.hasProcess())
			return ModelType.PROCESS;
		if (dataSet.hasProductSystem())
			return ModelType.PRODUCT_SYSTEM;
		if (dataSet.hasProject())
			return ModelType.PROJECT;
		if (dataSet.hasResult())
			return ModelType.RESULT;
		if (dataSet.hasSocialIndicator())
			return ModelType.SOCIAL_INDICATOR;
		if (dataSet.hasSource())
			return ModelType.SOURCE;
		if (dataSet.hasUnitGroup())
			return ModelType.UNIT_GROUP;
		return null;
	}

	String getId() {
		if (dataSet.hasActor())
			return dataSet.getActor().getId();
		if (dataSet.hasCurrency())
			return dataSet.getCurrency().getId();
		if (dataSet.hasDqSystem())
			return dataSet.getDqSystem().getId();
		if (dataSet.hasEpd())
			return dataSet.getEpd().getId();
		if (dataSet.hasFlowProperty())
			return dataSet.getFlowProperty().getId();
		if (dataSet.hasFlow())
			return dataSet.getFlow().getId();
		if (dataSet.hasImpactCategory())
			return dataSet.getImpactCategory().getId();
		if (dataSet.hasImpactMethod())
			return dataSet.getImpactMethod().getId();
		if (dataSet.hasLocation())
			return dataSet.getLocation().getId();
		if (dataSet.hasParameter())
			return dataSet.getParameter().getId();
		if (dataSet.hasProcess())
			return dataSet.getProcess().getId();
		if (dataSet.hasProductSystem())
			return dataSet.getProductSystem().getId();
		if (dataSet.hasProject())
			return dataSet.getProject().getId();
		if (dataSet.hasResult())
			return dataSet.getResult().getId();
		if (dataSet.hasSocialIndicator())
			return dataSet.getSocialIndicator().getId();
		if (dataSet.hasSource())
			return dataSet.getSource().getId();
		if (dataSet.hasUnitGroup())
			return dataSet.getUnitGroup().getId();
		return null;
	}

	@Override
	public Set getIds(ModelType type) {
		var thisType = getType();
		if (type != thisType)
			return Collections.emptySet();
		var id = getId();
		return Strings.notEmpty(id)
			? Set.of(id)
			: Collections.emptySet();
	}

	@Override
	public ProtoActor getActor(String id) {
		if (!dataSet.hasActor())
			return null;
		return Objects.equals(id, dataSet.getActor().getId())
			? dataSet.getActor()
			: null;
	}

	@Override
	public ProtoCurrency getCurrency(String id) {
		if (!dataSet.hasCurrency())
			return null;
		return Objects.equals(id, dataSet.getCurrency().getId())
			? dataSet.getCurrency()
			: null;
	}

	@Override
	public ProtoDQSystem getDQSystem(String id) {
		if (!dataSet.hasDqSystem())
			return null;
		return Objects.equals(id, dataSet.getDqSystem().getId())
			? dataSet.getDqSystem()
			: null;
	}

	@Override
	public ProtoEpd getEpd(String id) {
		if (!dataSet.hasEpd())
			return null;
		return Objects.equals(id, dataSet.getEpd().getId())
			? dataSet.getEpd()
			: null;
	}

	@Override
	public ProtoFlowProperty getFlowProperty(String id) {
		if (!dataSet.hasFlowProperty())
			return null;
		return Objects.equals(id, dataSet.getFlowProperty().getId())
			? dataSet.getFlowProperty()
			: null;
	}

	@Override
	public ProtoFlow getFlow(String id) {
		if (!dataSet.hasFlow())
			return null;
		return Objects.equals(id, dataSet.getFlow().getId())
			? dataSet.getFlow()
			: null;
	}

	@Override
	public ProtoImpactCategory getImpactCategory(String id) {
		if (!dataSet.hasImpactCategory())
			return null;
		return Objects.equals(id, dataSet.getImpactCategory().getId())
			? dataSet.getImpactCategory()
			: null;
	}

	@Override
	public ProtoImpactMethod getImpactMethod(String id) {
		if (!dataSet.hasImpactMethod())
			return null;
		return Objects.equals(id, dataSet.getImpactMethod().getId())
			? dataSet.getImpactMethod()
			: null;
	}

	@Override
	public ProtoLocation getLocation(String id) {
		if (!dataSet.hasLocation())
			return null;
		return Objects.equals(id, dataSet.getLocation().getId())
			? dataSet.getLocation()
			: null;
	}

	@Override
	public ProtoParameter getParameter(String id) {
		if (!dataSet.hasParameter())
			return null;
		return Objects.equals(id, dataSet.getParameter().getId())
			? dataSet.getParameter()
			: null;
	}

	@Override
	public ProtoProcess getProcess(String id) {
		if (!dataSet.hasProcess())
			return null;
		return Objects.equals(id, dataSet.getProcess().getId())
			? dataSet.getProcess()
			: null;
	}

	@Override
	public ProtoProductSystem getProductSystem(String id) {
		if (!dataSet.hasProductSystem())
			return null;
		return Objects.equals(id, dataSet.getProductSystem().getId())
			? dataSet.getProductSystem()
			: null;
	}

	@Override
	public ProtoProject getProject(String id) {
		if (!dataSet.hasProject())
			return null;
		return Objects.equals(id, dataSet.getProject().getId())
			? dataSet.getProject()
			: null;
	}

	@Override
	public ProtoResult getResult(String id) {
		if (!dataSet.hasResult())
			return null;
		return Objects.equals(id, dataSet.getResult().getId())
			? dataSet.getResult()
			: null;
	}

	@Override
	public ProtoSocialIndicator getSocialIndicator(String id) {
		if (!dataSet.hasSocialIndicator())
			return null;
		return Objects.equals(id, dataSet.getSocialIndicator().getId())
			? dataSet.getSocialIndicator()
			: null;
	}

	@Override
	public ProtoSource getSource(String id) {
		if (!dataSet.hasSource())
			return null;
		return Objects.equals(id, dataSet.getSource().getId())
			? dataSet.getSource()
			: null;
	}

	@Override
	public ProtoUnitGroup getUnitGroup(String id) {
		if (!dataSet.hasUnitGroup())
			return null;
		return Objects.equals(id, dataSet.getUnitGroup().getId())
			? dataSet.getUnitGroup()
			: null;
	}
}