一个以SpringBoot框架为基础开发的消息发送中间件

/* * Copyright ©2015-2023 Jaemon. All Rights Reserved. * * 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 * * http://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 com.github.jaemon.dinger.multi; import com.github.jaemon.dinger.constant.DingerConstant; import com.github.jaemon.dinger.core.entity.enums.DingerType; import com.github.jaemon.dinger.core.entity.enums.ExceptionEnum; import com.github.jaemon.dinger.exception.MultiDingerRegisterException; import com.github.jaemon.dinger.listeners.DingerListenersProperty; import com.github.jaemon.dinger.multi.annotations.MultiDinger; import com.github.jaemon.dinger.multi.entity.MultiDingerAlgorithmDefinition; import com.github.jaemon.dinger.core.DingerConfig; import com.github.jaemon.dinger.exception.DingerException; import com.github.jaemon.dinger.multi.algorithm.AlgorithmHandler; import com.github.jaemon.dinger.multi.annotations.EnableMultiDinger; import com.github.jaemon.dinger.multi.annotations.MultiHandler; import com.github.jaemon.dinger.multi.entity.MultiDingerConfig; import com.github.jaemon.dinger.utils.DingerUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.support.AbstractBeanDefinition; import org.springframework.beans.factory.support.BeanDefinitionBuilder; import org.springframework.beans.factory.support.BeanDefinitionRegistry; import org.springframework.beans.factory.support.GenericBeanDefinition; import org.springframework.context.annotation.ImportBeanDefinitionRegistrar; import org.springframework.core.Ordered; import org.springframework.core.annotation.AnnotationAttributes; import org.springframework.core.type.AnnotationMetadata; import java.util.*; import static com.github.jaemon.dinger.constant.DingerConstant.SPOT_SEPERATOR; import static com.github.jaemon.dinger.core.entity.enums.ExceptionEnum.MULTIDINGER_ALGORITHM_EXCEPTION; import static com.github.jaemon.dinger.core.entity.enums.ExceptionEnum.MULTIDINGER_ANNOTATTION_EXCEPTION; import static com.github.jaemon.dinger.multi.MultiDingerConfigContainer.GLOABL_KEY; /** * MultiDingerScannerRegistrar * * @author Jaemon * @since 1.0 */ public class MultiDingerScannerRegistrar extends DingerListenersProperty implements ImportBeanDefinitionRegistrar, Ordered { private static final Logger log = LoggerFactory.getLogger(MultiDingerScannerRegistrar.class); /** * 算法{@link AlgorithmHandler}容器 * * <blockquote> * {<br> * key: dingerClassName | {@link MultiDingerConfigContainer#GLOABL_KEY}(key) + {@link DingerConstant#SPOT_SEPERATOR} + {@link AlgorithmHandler}.simpleName<br> * value: {@link MultiDingerAlgorithmDefinition}<br> * }<br> * </blockquote> */ protected final static Map<String, MultiDingerAlgorithmDefinition> MULTIDINGER_ALGORITHM_DEFINITION_MAP = new HashMap<>(); @Override public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) { try { doScanAndRegister(importingClassMetadata, registry); } catch (DingerException ex) { throw new MultiDingerRegisterException(ex.getPairs(), ex.getMessage()); } catch (Exception ex) { throw new DingerException(ex, ExceptionEnum.UNKNOWN); } finally { emptyDingerClasses(); } } private void doScanAndRegister(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) { boolean debugEnabled = log.isDebugEnabled(); if (!importingClassMetadata.hasAnnotation(EnableMultiDinger.class.getName())) { log.warn("import class can't find EnableMultiDinger annotation."); return; } AnnotationAttributes annotationAttributes = AnnotationAttributes.fromMap( importingClassMetadata.getAnnotationAttributes(EnableMultiDinger.class.getName()) ); AnnotationAttributes[] value = annotationAttributes.getAnnotationArray("value"); boolean aloneMulti = value.length == 0; log.info("multi dinger register and is it global register? {}.", !aloneMulti); // 指定多机器人配置处理逻辑 if (aloneMulti) { // 处理需要执行MultiDinger逻辑的dingerClass List<Class<?>> dingerClasses = dingerClasses(); if (dingerClasses.isEmpty()) { log.warn("dinger class is empty, so no need to deal with multiDinger."); return; } multiDingerHandler(registry, dingerClasses); if (!MultiDingerConfigContainer.INSTANCE.isEmpty()) { MultiDingerProperty.multiDinger = true; } // 全局多机器人配置处理逻辑 } else { for (AnnotationAttributes attributes : value) { DingerType dinger = attributes.getEnum("dinger"); Class<? extends DingerConfigHandler> handler = attributes.getClass("handler"); if (debugEnabled) { log.debug("enable {} global multi dinger, and multiDinger handler class={}.", dinger, handler.getName()); } DingerConfigHandler dingerConfigHandler = BeanUtils.instantiateClass(handler); registerHandler(registry, dinger, dinger + DingerConstant.SPOT_SEPERATOR + GLOABL_KEY, dingerConfigHandler); MultiDingerProperty.multiDinger = true; } } } /** * 处理DingerClass绑定的MultiHandler并注册{@link AlgorithmHandler} * * @param registry * 注册器 * @param dingerClasses * dingerClass集 */ private void multiDingerHandler(BeanDefinitionRegistry registry, List<Class<?>> dingerClasses) { boolean debugEnabled = log.isDebugEnabled(); int valid = 0; for (Class<?> dingerClass : dingerClasses) { if (dingerClass.isAnnotationPresent(MultiHandler.class)) { MultiHandler multiDinger = dingerClass.getAnnotation(MultiHandler.class); MultiDinger value = multiDinger.value(); Class<? extends DingerConfigHandler> dingerConfigHandler = value.handler(); String beanName = dingerConfigHandler.getSimpleName(); // 如果DingerClass指定的MultiHandler对应的处理器为接口，则直接跳过 if (dingerConfigHandler.isInterface()) { log.warn("dingerClass={} handler className={} is interface and skip.", dingerClass.getSimpleName(), beanName); continue; } String key = dingerClass.getName(); DingerConfigHandler handler = DingerUtils.newInstance(dingerConfigHandler); DingerType dinger = value.dinger(); registerHandler(registry, dinger, dinger + DingerConstant.SPOT_SEPERATOR + key, handler); if (debugEnabled) { log.debug("regiseter multi dinger for dingerClass={} and dingerConfigHandler={}.", dingerClass.getSimpleName(), beanName); } valid++; } } if (valid == 0) { log.warn("enable global multi dinger but none dinger interface be decorated wi