io.evitadb.api.requestResponse.data.mutation.reference.RemoveReferenceGroupMutation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evita_api Show documentation
Module contains external API of the evitaDB.
The newest version!
/*
 *
 *                         _ _        ____  ____
 *               _____   _(_) |_ __ _|  _ \| __ )
 *              / _ \ \ / / | __/ _` | | | |  _ \
 *             |  __/\ V /| | || (_| | |_| | |_) |
 *              \___| \_/ |_|\__\__,_|____/|____/
 *
 *   Copyright (c) 2023-2024
 *
 *   Licensed under the Business Source License, Version 1.1 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *   https://github.com/FgForrest/evitaDB/blob/master/LICENSE
 *
 *   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.
 */

package io.evitadb.api.requestResponse.data.mutation.reference;

import io.evitadb.api.exception.InvalidMutationException;
import io.evitadb.api.requestResponse.cdc.Operation;
import io.evitadb.api.requestResponse.data.Droppable;
import io.evitadb.api.requestResponse.data.ReferenceContract;
import io.evitadb.api.requestResponse.data.ReferenceContract.GroupEntityReference;
import io.evitadb.api.requestResponse.data.structure.Reference;
import io.evitadb.api.requestResponse.schema.EntitySchemaContract;
import io.evitadb.utils.Assert;
import lombok.EqualsAndHashCode;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.io.Serial;
import java.util.Optional;

/**
 * This mutation allows to remove {@link GroupEntityReference} in the {@link Reference}.
 *
 * @author Jan Novotný ([email protected]), FG Forrest a.s. (c) 2021
 */
@EqualsAndHashCode(callSuper = true)
public class RemoveReferenceGroupMutation extends ReferenceMutation {
	@Serial private static final long serialVersionUID = -564814790916765844L;

	public RemoveReferenceGroupMutation(@Nonnull ReferenceKey referenceKey) {
		super(referenceKey);
	}

	@Nonnull
	@Override
	public ReferenceContract mutateLocal(@Nonnull EntitySchemaContract entitySchema, @Nullable ReferenceContract existingValue) {
		Assert.isTrue(
			existingValue != null && existingValue.exists(),
			() -> new InvalidMutationException("Cannot remove reference " + referenceKey + " - reference doesn't exist!")
		);
		Assert.isTrue(
			existingValue.getGroup().filter(Droppable::exists).isPresent(),
			() -> new InvalidMutationException("Cannot remove reference " + referenceKey + " group - no group is currently set!")
		);

		final Optional existingReferenceGroup = existingValue.getGroup();
		Assert.isTrue(
			existingReferenceGroup.filter(Droppable::exists).isPresent(),
			() -> new InvalidMutationException("Cannot remove reference group - no reference group is set on reference " + referenceKey + "!")
		);
		return new Reference(
			entitySchema,
			existingValue.version() + 1,
			existingValue.getReferenceName(), existingValue.getReferencedPrimaryKey(),
			existingValue.getReferencedEntityType(), existingValue.getReferenceCardinality(),
			existingReferenceGroup
				.filter(Droppable::exists)
				.map(it -> new GroupEntityReference(
						it.getType(),
						it.getPrimaryKey(),
						it.version() + 1,
						true
					)
				).orElseThrow(() -> new InvalidMutationException("Cannot remove reference group - no reference group is set on reference " + referenceKey + "!")),
			existingValue.getAttributeValues(),
			existingValue.dropped()
		);
	}

	@Override
	public long getPriority() {
		// we need that this mutation is before insert/remove reference itself
		return PRIORITY_REMOVAL * 2;
	}

	@Nonnull
	@Override
	public Operation operation() {
		return Operation.REMOVE;
	}

	@Override
	public ReferenceKey getComparableKey() {
		return referenceKey;
	}

	@Override
	public String toString() {
		return "Remove reference group `" + referenceKey + "`";
	}
}