com.jn.sqlhelper.common.transaction.TransactionDefinitionRegistry Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 the original author or authors.
 *
 * Licensed under the Apache, 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  http://www.gnu.org/licenses/lgpl-2.0.html
 *
 * 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 com.jn.sqlhelper.common.transaction;

import com.jn.langx.annotation.NonNull;
import com.jn.langx.registry.Registry;
import com.jn.langx.util.Preconditions;
import com.jn.langx.util.collection.Collects;
import com.jn.langx.util.function.Consumer2;
import com.jn.langx.util.function.Predicate2;
import com.jn.langx.util.struct.Holder;
import com.jn.sqlhelper.common.transaction.definition.parser.NamedTransactionDefinitionParser;
import com.jn.sqlhelper.common.transaction.definition.parser.TransactionDefinitionAnnotationParser;
import com.jn.sqlhelper.common.transaction.definition.parser.TransactionDefinitionParser;
import com.jn.sqlhelper.common.transaction.definition.parser.TransactionalAnnotationParser;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 事务定义Registry
 * @since 3.4.3
 */
public class TransactionDefinitionRegistry implements Registry {
    /**
     * 这里的 DataSourceKey，可以是一个确切的 key，也可以是个 keyPattern。
     * 不论是确切的key，还是个keyPattern，在DataSourceRegister里进行pattern匹配的。
     */
    private final ConcurrentHashMap> methodDataSourceKeyCache = new ConcurrentHashMap>();

    /**
     * parser 缓存
     */
    private Map, TransactionDefinitionAnnotationParser> annotationParserMap = new LinkedHashMap, TransactionDefinitionAnnotationParser>();

    private Map namedTransactionDefinitionParserMap = new LinkedHashMap();


    public TransactionDefinitionRegistry() {
        registerTransactionAnnotationParser(new TransactionalAnnotationParser());
    }

    @Override
    public void register(TransactionDefinition dataSourceKeyHolder) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void register(Method method, TransactionDefinition dataSourceKey) {
        Preconditions.checkNotNull(dataSourceKey);
        methodDataSourceKeyCache.putIfAbsent(method, new Holder(dataSourceKey));
    }

    public void register(TransactionDefinitionParser parser) {
        if (parser instanceof TransactionDefinitionAnnotationParser) {
            registerTransactionAnnotationParser((TransactionDefinitionAnnotationParser) parser);
        } else if (parser instanceof NamedTransactionDefinitionParser) {
            registerNamedTransactionParser((NamedTransactionDefinitionParser) parser);
        }
    }

    private void registerTransactionAnnotationParser(TransactionDefinitionAnnotationParser transactionDefinitionAnnotationParser) {
        if (transactionDefinitionAnnotationParser != null && transactionDefinitionAnnotationParser.getAnnotation() != null) {
            annotationParserMap.put(transactionDefinitionAnnotationParser.getAnnotation(), transactionDefinitionAnnotationParser);
        }
    }

    private void registerNamedTransactionParser(NamedTransactionDefinitionParser transactionDefinitionParser) {
        namedTransactionDefinitionParserMap.put(transactionDefinitionParser.getName(), transactionDefinitionParser);
    }

    @Override
    public void unregister(Method method) {
        methodDataSourceKeyCache.remove(method);
    }

    @Override
    public boolean contains(Method method) {
        return methodDataSourceKeyCache.containsKey(method);
    }

    @Override
    @NonNull
    public TransactionDefinition get(final Method method) {
        Holder holder = methodDataSourceKeyCache.get(method);
        if (holder == null) {
            // 第一次用到该方法
            synchronized (this) {
                final Holder holder0 = new Holder();

                // 优先使用注解
                Collects.forEach(annotationParserMap, new Consumer2, TransactionDefinitionAnnotationParser>() {
                    @Override
                    public void accept(Class annotationClass, TransactionDefinitionAnnotationParser parser) {
                        TransactionDefinition key = parser.parse(method);
                        if (key != null) {
                            holder0.set(key);
                        }
                    }
                }, new Predicate2, TransactionDefinitionAnnotationParser>() {
                    @Override
                    public boolean test(Class key, TransactionDefinitionAnnotationParser value) {
                        return !holder0.isNull();
                    }
                });

                // 再用 named transaction parser
                Collects.forEach(namedTransactionDefinitionParserMap, new Consumer2() {
                    @Override
                    public void accept(String parserName, NamedTransactionDefinitionParser parser) {
                        TransactionDefinition key = parser.parse(method);
                        if (key != null) {
                            holder0.set(key);
                        }
                    }
                }, new Predicate2() {
                    @Override
                    public boolean test(String parserName, NamedTransactionDefinitionParser value) {
                        return !holder0.isNull();
                    }
                });

                methodDataSourceKeyCache.putIfAbsent(method, holder0);
                holder = methodDataSourceKeyCache.get(method);
            }
        }
        return holder.get();
    }
}