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

    • io.micronaut.data.runtime.intercept.criteria.reactive.AbstractReactiveSpecificationInterceptor 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.intercept.criteria.reactive;

import io.micronaut.aop.MethodInvocationContext;
import io.micronaut.core.annotation.NonNull;
import io.micronaut.data.exceptions.DataAccessException;
import io.micronaut.data.intercept.RepositoryMethodKey;
import io.micronaut.data.model.Pageable;
import io.micronaut.data.model.Pageable.Mode;
import io.micronaut.data.model.query.JoinPath;
import io.micronaut.data.operations.RepositoryOperations;
import io.micronaut.data.operations.reactive.ReactiveCapableRepository;
import io.micronaut.data.operations.reactive.ReactiveCriteriaCapableRepository;
import io.micronaut.data.operations.reactive.ReactiveCriteriaRepositoryOperations;
import io.micronaut.data.operations.reactive.ReactiveRepositoryOperations;
import io.micronaut.data.runtime.intercept.criteria.AbstractSpecificationInterceptor;
import jakarta.persistence.criteria.CriteriaQuery;
import org.reactivestreams.Publisher;
import reactor.core.publisher.Mono;

import java.util.Set;

/**
 * Abstract reactive specification interceptor.
 *
 * @param  The declaring type
 * @param  The return type
 * @author Denis Stepanov
 * @since 3.2
 */
public abstract class AbstractReactiveSpecificationInterceptor extends AbstractSpecificationInterceptor {

    protected final ReactiveRepositoryOperations reactiveOperations;
    protected final ReactiveCriteriaRepositoryOperations reactiveCriteriaOperations;

    /**
     * Default constructor.
     *
     * @param operations The operations
     */
    protected AbstractReactiveSpecificationInterceptor(RepositoryOperations operations) {
        super(operations);
        if (operations instanceof ReactiveCapableRepository reactiveCapableRepository) {
            this.reactiveOperations = reactiveCapableRepository.reactive();
        } else {
            throw new DataAccessException("Datastore of type [" + operations.getClass() + "] does not support reactive operations");
        }
        if (reactiveOperations instanceof ReactiveCriteriaRepositoryOperations reactiveCriteriaRepositoryOperations) {
            reactiveCriteriaOperations = reactiveCriteriaRepositoryOperations;
        } else if (operations instanceof ReactiveCriteriaRepositoryOperations reactiveCriteriaRepositoryOperations) {
            reactiveCriteriaOperations = reactiveCriteriaRepositoryOperations;
        } else if (operations instanceof ReactiveCriteriaCapableRepository repository) {
            reactiveCriteriaOperations = repository.reactive();
        } else {
            reactiveCriteriaOperations = null;
        }
    }

    @NonNull
    protected final Publisher findAllReactive(RepositoryMethodKey methodKey, MethodInvocationContext context, Type type) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            CriteriaQuery criteriaQuery = buildQuery(context, type, methodJoinPaths);
            Pageable pageable = getPageable(context);
            if (pageable != null) {
                if (pageable.getMode() != Mode.OFFSET) {
                    throw new UnsupportedOperationException("Pageable mode " + pageable.getMode() + " is not supported by hibernate operations");
                }
                return reactiveCriteriaOperations.findAll(criteriaQuery, (int) pageable.getOffset(), pageable.getSize());
            }
            return reactiveCriteriaOperations.findAll(criteriaQuery);
        }
        return reactiveOperations.findAll(preparedQueryForCriteria(methodKey, context, type, methodJoinPaths));
    }

    @NonNull
    protected final Publisher findOneReactive(RepositoryMethodKey methodKey, MethodInvocationContext context, Type type) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            return reactiveCriteriaOperations.findOne(buildQuery(context, type, methodJoinPaths));
        }
        return reactiveOperations.findOne(preparedQueryForCriteria(methodKey, context, type, methodJoinPaths));
    }

    @NonNull
    protected final Publisher countReactive(RepositoryMethodKey methodKey, MethodInvocationContext context) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            return reactiveCriteriaOperations.findOne(buildCountQuery(context));
        }
        return reactiveOperations.findOne(preparedQueryForCriteria(methodKey, context, Type.COUNT, methodJoinPaths));
    }

    protected final Publisher existsReactive(RepositoryMethodKey methodKey, MethodInvocationContext context) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            return reactiveCriteriaOperations.findOne(buildExistsQuery(context, methodJoinPaths));
        }
        return Mono.from(reactiveOperations.findOne(preparedQueryForCriteria(methodKey, context, Type.EXISTS, methodJoinPaths)))
            .map(one -> one instanceof Boolean aBoolean ? aBoolean : one != null);
    }

    protected final Publisher deleteAllReactive(RepositoryMethodKey methodKey, MethodInvocationContext context) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            return reactiveCriteriaOperations.deleteAll(buildDeleteQuery(context));
        }
        return reactiveOperations.executeDelete(preparedQueryForCriteria(methodKey, context, Type.DELETE_ALL, methodJoinPaths));
    }

    protected final Publisher updateAllReactive(RepositoryMethodKey methodKey, MethodInvocationContext context) {
        Set methodJoinPaths = getMethodJoinPaths(methodKey, context);
        if (reactiveCriteriaOperations != null) {
            return reactiveCriteriaOperations.updateAll(buildUpdateQuery(context));
        }
        return reactiveOperations.executeUpdate(preparedQueryForCriteria(methodKey, context, Type.UPDATE_ALL, methodJoinPaths));
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy