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

io.evitadb.api.requestResponse.data.mutation.attribute.AttributeMutation Maven / Gradle / Ivy

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.attribute;

import io.evitadb.api.exception.InvalidMutationException;
import io.evitadb.api.requestResponse.data.AttributesContract.AttributeKey;
import io.evitadb.api.requestResponse.data.AttributesContract.AttributeValue;
import io.evitadb.api.requestResponse.data.EntityContract;
import io.evitadb.api.requestResponse.data.mutation.NamedLocalMutation;
import io.evitadb.api.requestResponse.data.structure.Attributes;
import io.evitadb.api.requestResponse.mutation.Mutation;
import io.evitadb.api.requestResponse.schema.AttributeSchemaContract;
import io.evitadb.api.requestResponse.schema.CatalogSchemaContract;
import io.evitadb.api.requestResponse.schema.EntitySchemaEditor.EntitySchemaBuilder;
import io.evitadb.api.requestResponse.schema.EvolutionMode;
import io.evitadb.dataType.ContainerType;
import io.evitadb.dataType.EvitaDataTypes;
import io.evitadb.dataType.Predecessor;
import io.evitadb.dataType.ReferencedEntityPredecessor;
import io.evitadb.utils.Assert;
import lombok.EqualsAndHashCode;
import lombok.Getter;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.io.Serial;
import java.io.Serializable;
import java.util.Locale;
import java.util.function.BiConsumer;
import java.util.stream.Collectors;

/**
 * Attribute {@link Mutation} allows to execute mutation operations on {@link Attributes} of the {@link EntityContract}
 * object. Each attribute change increments {@link AttributeValue#version()} by one, attribute removal only sets
 * tombstone flag on an attribute value and doesn't really remove it. Possible removal will be taken care of during
 * compaction process, leaving attributes in place allows to see last assigned value to the attribute and also consult
 * last version of the attribute.
 *
 * These traits should help to manage concurrent transactional process as updates to the same entity could be executed
 * safely and concurrently as long as attribute modification doesn't overlap. Some mutations may also overcome same
 * attribute concurrent modification if it's safely additive (i.e. incrementation / decrementation and so on).
 *
 * Exact mutations also allows engine implementation to safely update only those indexes that the change really affects
 * and doesn't require additional analysis.
 *
 * @author Jan Novotný ([email protected]), FG Forrest a.s. (c) 2021
 */
@EqualsAndHashCode
public abstract class AttributeMutation implements NamedLocalMutation {
	@Serial private static final long serialVersionUID = 8615227519108169551L;
	/**
	 * Identification of the attribute that the mutation affects.
	 */
	@Nonnull
	@Getter protected final AttributeKey attributeKey;

	protected AttributeMutation(@Nonnull AttributeKey attributeKey) {
		this.attributeKey = attributeKey;
	}

	@Nonnull
	@Override
	public String containerName() {
		return attributeKey.attributeName();
	}

	@Nonnull
	@Override
	public ContainerType containerType() {
		return ContainerType.ATTRIBUTE;
	}

	@Override
	public AttributeKey getComparableKey() {
		return attributeKey;
	}

	public void verifyOrEvolveSchema(
		@Nonnull CatalogSchemaContract catalogSchema,
		@Nonnull EntitySchemaBuilder entitySchemaBuilder,
		@Nullable AttributeSchemaContract attributeSchema,
		@Nonnull Serializable attributeValue,
		@Nonnull BiConsumer schemaEvolutionApplicator
	) throws InvalidMutationException {
		// when attribute definition is known execute first encounter formal verification
		if (attributeSchema != null) {
			// we need to unwrap array classes - we need to check base class for compatibility with Comparable
			final Class attributeClass;
			if (attributeValue instanceof Object[]) {
				attributeClass = attributeValue.getClass().getComponentType();
			} else {
				attributeClass = attributeValue.getClass();
			}
			Assert.isTrue(
				(attributeSchema.getType().isPrimitive() ?
					EvitaDataTypes.getWrappingPrimitiveClass(attributeSchema.getType()) : attributeSchema.getType())
					.isInstance(attributeValue),
				() -> new InvalidMutationException(
					"Invalid type: `" + attributeValue.getClass() + "`! " +
						"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` was already stored as type " + attributeSchema.getType() + ". " +
						"All values of attribute `" + attributeKey.attributeName() + "` must respect this data type!"
				)
			);
			if (attributeSchema.isSortable()) {
				Assert.isTrue(
					Comparable.class.isAssignableFrom(attributeClass) || attributeValue instanceof Predecessor || attributeValue instanceof ReferencedEntityPredecessor,
					() -> new InvalidMutationException(
						"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` is sortable and needs to implement " +
							"Comparable interface (or be Predecessor/ReferencedEntityPredecessor), but it doesn't: `" + attributeValue.getClass() + "`!"
					)
				);
			}
			if (attributeSchema.isLocalized()) {
				Assert.isTrue(
					attributeKey.localized(),
					() -> new InvalidMutationException(
						"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` was already stored as localized value. " +
							"All values of attribute `" + attributeKey.attributeName() + "` must be localized now " +
							"- use different attribute name for locale independent variant of attribute!"
					)
				);
				final Locale locale = attributeKey.locale();
				if (!entitySchemaBuilder.getLocales().contains(locale)) {
					if (entitySchemaBuilder.allows(EvolutionMode.ADDING_LOCALES)) {
						// evolve schema automatically
						schemaEvolutionApplicator.accept(catalogSchema, entitySchemaBuilder);
					} else {
						throw new InvalidMutationException(
							"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` is localized to `" + locale + "` which is not allowed by the schema" +
								" (allowed are only: " + entitySchemaBuilder.getLocales().stream().map(Locale::toString).collect(Collectors.joining(", ")) + "). " +
								"You must first alter entity schema to be able to add this localized attribute to the entity!"
						);
					}
				}
			} else {
				Assert.isTrue(
					!attributeKey.localized(),
					() -> new InvalidMutationException(
						"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` was not stored as localized value. " +
							"No values of attribute `" + attributeKey.attributeName() + "` can be localized now " +
							"- use different attribute name for localized variant of attribute!"
					)
				);
			}
			// else check whether adding attributes on the fly is allowed
		} else if (entitySchemaBuilder.allows(EvolutionMode.ADDING_ATTRIBUTES)) {
			// evolve schema automatically
			schemaEvolutionApplicator.accept(catalogSchema, entitySchemaBuilder);
		} else {
			throw new InvalidMutationException(
				"Unknown attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "``! " +
					"You must first alter entity schema to be able to add this attribute to the entity!"
			);
		}
	}

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy