org.teamapps.universaldb.index.reference.value.MultiReferenceEditValue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of universal-db Show documentation
Ultra fast TeamApps database
The newest version!
/*-
 * ========================LICENSE_START=================================
 * UniversalDB
 * ---
 * Copyright (C) 2014 - 2024 TeamApps.org
 * ---
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * =========================LICENSE_END==================================
 */
package org.teamapps.universaldb.index.reference.value;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;

public class MultiReferenceEditValue implements MultiReferenceValue {

	private boolean removeAll;
	private final List addReferences = new ArrayList<>();
	private final List removeReferences = new ArrayList<>();
	private final List setReferences = new ArrayList<>();

	public MultiReferenceEditValue() {
	}

	public MultiReferenceEditValue(DataInputStream dis) throws IOException {
		removeAll = dis.readBoolean();
		addReferences.addAll(readReferences(dis));
		removeReferences.addAll(readReferences(dis));
		setReferences.addAll(readReferences(dis));
	}

	public void write(DataOutputStream dos) throws IOException {
		dos.writeBoolean(removeAll);
		writeReferences(this.addReferences, dos);
		writeReferences(this.removeReferences, dos);
		writeReferences(this.setReferences, dos);
	}

	public void updateReferences(Map recordIdByCorrelationId) {
		addReferences.stream().filter(ref -> ref.getRecordId() == 0).forEach(ref -> ref.setRecordId(recordIdByCorrelationId.get(ref.getCorrelationId())));
		removeReferences.stream().filter(ref -> ref.getRecordId() == 0).forEach(ref -> ref.setRecordId(recordIdByCorrelationId.get(ref.getCorrelationId())));
		setReferences.stream().filter(ref -> ref.getRecordId() == 0).forEach(ref -> ref.setRecordId(recordIdByCorrelationId.get(ref.getCorrelationId())));
	}

	public ResolvedMultiReferenceUpdate getResolvedUpdateValue() {
		if (removeAll) {
			return ResolvedMultiReferenceUpdate.createRemoveAllReferences();
		} else if (!setReferences.isEmpty()) {
			return ResolvedMultiReferenceUpdate.createSetReferences(setReferences.stream().map(RecordReference::getRecordId).filter(id -> id != 0).collect(Collectors.toList()));
		} else {
			List addRecords = addReferences.stream().map(RecordReference::getRecordId).filter(id -> id != 0).collect(Collectors.toList());
			List removeRecords = removeReferences.stream().map(RecordReference::getRecordId).filter(id -> id != 0).collect(Collectors.toList());
			return ResolvedMultiReferenceUpdate.createAddRemoveReferences(addRecords, removeRecords);
		}
	}

	public void setRemoveAll() {
		this.removeAll = true;
		addReferences.clear();
		removeReferences.clear();
		setReferences.clear();
	}

	public void setReferences(List references) {
		setReferences.clear();
		removeAll = false;
		addReferences.clear();
		removeReferences.clear();
		Set referenceSet = new HashSet<>();
		references.forEach(reference -> {
			if (!referenceSet.contains(reference)) {
				referenceSet.add(reference);
				setReferences.add(reference);
			}
		});

	}

	public void addReferences(List references) {
		if (!setReferences.isEmpty()) {
			Set referenceSet = new HashSet<>(setReferences);
			references.forEach(reference -> {
				if (!referenceSet.contains(reference)) {
					setReferences.add(reference);
					referenceSet.add(reference);
				}
			});
		} else {
			Set addSet = new HashSet<>(addReferences);
			references.forEach(reference -> {
				removeReferences.remove(reference);
				if (!addSet.contains(reference)) {
					addReferences.add(reference);
				}
			});
		}
	}

	public void removeReferences(List references) {
		if (!setReferences.isEmpty()) {
			references.forEach(setReferences::remove);
		} else if (removeAll){
			references.forEach(addReferences::remove);
		} else {
			references.forEach(reference -> {
				if (addReferences.contains(reference)) {
					addReferences.remove(reference);
				} else {
					removeReferences.add(reference);
				}
			});
		}
	}

	public boolean isRemoveAll() {
		return removeAll;
	}

	public List getAddReferences() {
		return addReferences;
	}

	public List getRemoveReferences() {
		return removeReferences;
	}

	public List getSetReferences() {
		return setReferences;
	}

	@Override
	public MultiReferenceValueType getType() {
		return MultiReferenceValueType.EDIT_VALUE;
	}

	@Override
	public void writeValues(DataOutputStream dataOutputStream) throws IOException {
		dataOutputStream.writeBoolean(removeAll);
		writeReferences(this.addReferences, dataOutputStream);
		writeReferences(this.removeReferences, dataOutputStream);
		writeReferences(this.setReferences, dataOutputStream);
	}

	public void writeReferences(List references, DataOutputStream dataOutputStream) throws IOException {
		dataOutputStream.writeInt(references.size());
		for (RecordReference reference : references) {
			dataOutputStream.writeInt(reference.getRecordId());
			dataOutputStream.writeInt(reference.getCorrelationId());
		}
	}

	@Override
	public void readValues(DataInputStream dataInputStream) throws IOException {
		removeAll = dataInputStream.readBoolean();
		addReferences.addAll(readReferences(dataInputStream));
		removeReferences.addAll(readReferences(dataInputStream));
		setReferences.addAll(readReferences(dataInputStream));
	}

	private List readReferences(DataInputStream dataInputStream) throws IOException {
		List references = new ArrayList<>();
		int len = dataInputStream.readInt();
		for (int i = 0; i < len; i++) {
			int recordId = dataInputStream.readInt();
			int correlationId = dataInputStream.readInt();
			references.add(new RecordReference(recordId, correlationId));
		}
		return references;
	}

	@Override
	public String toString() {
		StringBuilder sb = new StringBuilder();
		if (removeAll) {
			sb.append("REMOVE ALL ENTRIES");
		}
		if (!removeReferences.isEmpty()) {
			if (sb.length() > 0) {
				sb.append(", ");
			}
			sb.append("REMOVE (").append(removeReferences.stream().map(val -> "" + val).collect(Collectors.joining(", "))).append(")");
		}
		if (!addReferences.isEmpty()) {
			if (sb.length() > 0) {
				sb.append(", ");
			}
			sb.append("ADD (").append(addReferences.stream().map(val -> "" + val).collect(Collectors.joining(", "))).append(")");
		}
		if (!setReferences.isEmpty()) {
			if (sb.length() > 0) {
				sb.append(", ");
			}
			sb.append("SET (").append(setReferences.stream().map(val -> "" + val).collect(Collectors.joining(", "))).append(")");
		}
		return sb.toString();
	}

	@Override
	public boolean isEditValue() {
		return false;
	}
}