All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • io.micronaut.data.runtime.support.DefaultRuntimeEntityRegistry Maven / Gradle / Ivy

    There is a newer version: 4.10.5
    Show newest version
    /*
 * Copyright 2017-2021 original authors
 *
 * 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
 *
 * https://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.
 */
package io.micronaut.data.runtime.support;

import io.micronaut.context.ApplicationContext;
import io.micronaut.context.ApplicationContextProvider;
import io.micronaut.core.annotation.NonNull;
import io.micronaut.context.BeanRegistration;
import io.micronaut.core.annotation.Internal;
import io.micronaut.core.type.Argument;
import io.micronaut.core.util.ArgumentUtils;
import io.micronaut.data.annotation.event.*;
import io.micronaut.data.model.runtime.PropertyAutoPopulator;
import io.micronaut.data.event.EntityEventListener;
import io.micronaut.data.model.runtime.RuntimeEntityRegistry;
import io.micronaut.data.model.runtime.RuntimePersistentEntity;
import io.micronaut.data.model.runtime.RuntimePersistentProperty;
import io.micronaut.data.model.runtime.AttributeConverterRegistry;
import io.micronaut.data.model.runtime.convert.AttributeConverter;
import io.micronaut.data.runtime.event.EntityEventRegistry;

import jakarta.inject.Singleton;
import java.lang.annotation.Annotation;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

/**
 * A registry for entities looking up instances of {@link RuntimeEntityRegistry}.
 *
 * @author graemerocher
 * @since 2.3.0
 */
@Singleton
@Internal
final class DefaultRuntimeEntityRegistry implements RuntimeEntityRegistry, ApplicationContextProvider {
    private final Map entities = new ConcurrentHashMap<>(10);
    private final Map, PropertyAutoPopulator> propertyPopulators;
    private final EntityEventRegistry eventRegistry;
    private final ApplicationContext applicationContext;
    private final AttributeConverterRegistry attributeConverterRegistry;

    /**
     * Default constructor.
     *
     * @param eventRegistry         The event registry
     * @param propertyPopulators    The property populators
     * @param applicationContext    The application context
     * @param attributeConverterRegistry The type converter registry
     */
    public DefaultRuntimeEntityRegistry(
            EntityEventRegistry eventRegistry,
            Collection>> propertyPopulators,
            ApplicationContext applicationContext,
            AttributeConverterRegistry attributeConverterRegistry) {
        this.eventRegistry = eventRegistry;
        this.propertyPopulators = new HashMap<>(propertyPopulators.size());
        this.attributeConverterRegistry = attributeConverterRegistry;
        for (BeanRegistration> propertyPopulator : propertyPopulators) {
            final PropertyAutoPopulator populator = propertyPopulator.getBean();
            final List> typeArguments = propertyPopulator.getBeanDefinition().getTypeArguments(PropertyAutoPopulator.class);
            if (!typeArguments.isEmpty()) {
                @SuppressWarnings("unchecked") final Class annotationType =
                        (Class) typeArguments.iterator().next().getType();
                if (this.propertyPopulators.containsKey(annotationType)) {
                    throw new IllegalStateException("Multiple property populators for annotation of type are not allowed: " + annotationType);
                } else {
                    this.propertyPopulators.put(annotationType, populator);
                }
            }
        }
        this.applicationContext = applicationContext;
    }

    @Override
    @NonNull
    public Object autoPopulateRuntimeProperty(@NonNull RuntimePersistentProperty persistentProperty, Object previousValue) {
        for (Map.Entry, PropertyAutoPopulator> entry : propertyPopulators.entrySet()) {
            if (persistentProperty.getAnnotationMetadata().hasAnnotation(entry.getKey())) {
                final PropertyAutoPopulator populator = entry.getValue();
                return Objects.requireNonNull(
                        populator.populate(persistentProperty, previousValue),
                        () -> "PropertyAutoPopulator illegally returned null: " + populator.getClass()
                );
            }
        }
        throw new IllegalStateException("Cannot auto populate property: " + persistentProperty.getName()  + " for entity: " + persistentProperty.getOwner().getName());
    }

    @NonNull
    @Override
    public EntityEventListener getEntityEventListener() {
        return eventRegistry;
    }

    @NonNull
    @Override
    public  RuntimePersistentEntity getEntity(@NonNull Class type) {
        ArgumentUtils.requireNonNull("type", type);
        // we need atomicity here, since entites are compared by identity (==)
        return entities.computeIfAbsent(type, this::newEntity);
    }

    @NonNull
    @Override
    public  RuntimePersistentEntity newEntity(@NonNull Class type) {
        return new RuntimePersistentEntity<>(type) {
            final boolean hasPrePersistEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PrePersist.class);
            final boolean hasPreRemoveEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PreRemove.class);
            final boolean hasPreUpdateEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PreUpdate.class);
            final boolean hasPostPersistEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PostPersist.class);
            final boolean hasPostRemoveEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PostRemove.class);
            final boolean hasPostUpdateEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PostUpdate.class);
            final boolean hasPostLoadEventListeners = eventRegistry.supports((RuntimePersistentEntity) this, PostLoad.class);

            @Override
            protected AttributeConverter resolveConverter(Class converterClass) {
                return attributeConverterRegistry.getConverter(converterClass);
            }

            @Override
            protected RuntimePersistentEntity getEntity(Class type) {
                return DefaultRuntimeEntityRegistry.this.getEntity(type);
            }

            @Override
            public boolean hasPostUpdateEventListeners() {
                return hasPostUpdateEventListeners;
            }

            @Override
            public boolean hasPostRemoveEventListeners() {
                return hasPostRemoveEventListeners;
            }

            @Override
            public boolean hasPostLoadEventListeners() {
                return hasPostLoadEventListeners;
            }

            @Override
            public boolean hasPrePersistEventListeners() {
                return hasPrePersistEventListeners;
            }

            @Override
            public boolean hasPreUpdateEventListeners() {
                return hasPreUpdateEventListeners;
            }

            @Override
            public boolean hasPreRemoveEventListeners() {
                return hasPreRemoveEventListeners;
            }

            @Override
            public boolean hasPostPersistEventListeners() {
                return hasPostPersistEventListeners;
            }
        };
    }

    @Override
    public ApplicationContext getApplicationContext() {
        return applicationContext;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy