io.evitadb.api.requestResponse.data.mutation.attribute.AttributeMutation 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
Show all versions of evita_api Show documentation
Module contains external API of the evitaDB.
/*
*
* _ _ ____ ____
* _____ _(_) |_ __ _| _ \| __ )
* / _ \ \ / / | __/ _` | | | | _ \
* | __/\ V /| | || (_| | |_| | |_) |
* \___| \_/ |_|\__\__,_|____/|____/
*
* Copyright (c) 2023
*
* 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.LocalMutation;
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.EvitaDataTypes;
import io.evitadb.dataType.Predecessor;
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 LocalMutation {
@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;
}
@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,
() -> new InvalidMutationException(
"Attribute `" + attributeKey.attributeName() + "` in schema `" + entitySchemaBuilder.getName() + "` is sortable and needs to implement " +
"Comparable interface (or be Predecessor), 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!"
);
}
}
}