com.jaemon.dingtalk.multi.MultiDingerScannerRegistrar Maven / Gradle / Ivy
The newest version!
/*
* Copyright ©2015-2020 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.jaemon.dingtalk.multi;
import com.jaemon.dingtalk.dinger.DingerConfig;
import com.jaemon.dingtalk.exception.DingTalkException;
import com.jaemon.dingtalk.listeners.ApplicationEventTimeTable;
import com.jaemon.dingtalk.multi.algorithm.AlgorithmHandler;
import com.jaemon.dingtalk.multi.annotations.EnableMultiDinger;
import com.jaemon.dingtalk.multi.annotations.MultiHandler;
import com.jaemon.dingtalk.multi.entity.MultiDingerAlgorithmDefinition;
import com.jaemon.dingtalk.multi.entity.MultiDingerConfig;
import com.jaemon.dingtalk.utils.DingTalkUtils;
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 org.springframework.objenesis.instantiator.util.ClassUtils;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import static com.jaemon.dingtalk.constant.DkConstant.SPOT_SEPERATOR;
import static com.jaemon.dingtalk.entity.enums.ExceptionEnum.GLOBAL_MULTIDINGER_CONFIG_EXCEPTION;
import static com.jaemon.dingtalk.entity.enums.ExceptionEnum.MULTIDINGER_ALGORITHM_EXCEPTION;
import static com.jaemon.dingtalk.multi.MultiDingerConfigContainer.GLOABL_KEY;
/**
* MultiDingerScannerRegistrar
*
* @author Jaemon
* @since 3.0
*/
public class MultiDingerScannerRegistrar implements ImportBeanDefinitionRegistrar, Ordered {
private static final Logger log = LoggerFactory.getLogger(MultiDingerScannerRegistrar.class);
/**
* 算法{@link AlgorithmHandler}容器
*
*
* {
* key: dingerClassName | {@link MultiDingerConfigContainer#GLOABL_KEY}(key) + {@link com.jaemon.dingtalk.constant.DkConstant#SPOT_SEPERATOR} + {@link AlgorithmHandler}.simpleName
* value: {@link MultiDingerAlgorithmDefinition}
* }
*
*/
protected final static Map MULTIDINGER_ALGORITHM_DEFINITION_MAP = new HashMap<>();
@Override
public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
boolean debugEnabled = log.isDebugEnabled();
log.info("ready to execute multiDingerScanner...");
try {
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())
);
Class value = annotationAttributes.getClass("value");
// 指定多机器人配置处理逻辑
if (value.isInterface()) {
if (DingerConfigHandler.class.equals(value)) {
// 处理需要执行MultiDinger逻辑的dingerClass
List> dingerClasses = ApplicationEventTimeTable.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 {
throw new DingTalkException(GLOBAL_MULTIDINGER_CONFIG_EXCEPTION);
}
// 全局多机器人配置处理逻辑
} else {
if (debugEnabled) {
log.debug("enable global multi dinger, and multiDinger handler class={}.", value.getName());
}
DingerConfigHandler dingerConfigHandler = BeanUtils.instantiateClass(value);
registerHandler(registry, GLOABL_KEY, dingerConfigHandler);
MultiDingerProperty.multiDinger = true;
}
} finally {
ApplicationEventTimeTable.emptyDingerClasses();
}
}
/**
* 处理DingerClass绑定的MultiHandler并注册{@link AlgorithmHandler}
*
* @param registry
* 注册器
* @param dingerClasses
* dingerClass集
*/
private void multiDingerHandler(BeanDefinitionRegistry registry, List> dingerClasses) {
boolean debugEnabled = log.isDebugEnabled();
for (Class dingerClass : dingerClasses) {
if (dingerClass.isAnnotationPresent(MultiHandler.class)) {
if (debugEnabled) {
log.debug("dingerClass={} configured with MultiDinger annotation.", dingerClass.getSimpleName());
}
MultiHandler multiDinger = dingerClass.getAnnotation(MultiHandler.class);
Class dingerConfigHandler = multiDinger.value();
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 = ClassUtils.newInstance(dingerConfigHandler);
registerHandler(registry, key, handler);
}
}
}
/**
* 注册 AlgorithmHandler
*
* @param registry
* 注册器
* @param key
* 当前dingerClass类名
* @param dingerConfigHandler
* dingerClass指定的multiHandler处理器
*/
private void registerHandler(BeanDefinitionRegistry registry, String key, DingerConfigHandler dingerConfigHandler) {
// 获取当前指定算法类名, 默认四种，或使用自定义
Class algorithm = dingerConfigHandler.algorithmHandler();
// if empty? use default dinger config
List dingerConfigs = dingerConfigHandler.dingerConfigs();
if (algorithm == null) {
throw new DingTalkException(MULTIDINGER_ALGORITHM_EXCEPTION);
}
// check dingerConfig
List dcs =
dingerConfigs
.stream()
.filter(
e -> DingTalkUtils.isNotEmpty(
e.getTokenId()
)
)
.collect(Collectors.toList());
long injectionCnt = Arrays.stream(algorithm.getDeclaredFields()).filter(e -> e.isAnnotationPresent(Autowired.class)).count();
// 如果无注入对象，直接反射算法处理器对象
AnalysisEnum mode = AnalysisEnum.REFLECT;
if (injectionCnt == 0) {
// create algorithm instance
AlgorithmHandler algorithmHandler = ClassUtils.newInstance(algorithm);
MultiDingerConfigContainer.INSTANCE.put(
key, new MultiDingerConfig(algorithmHandler, dcs)
);
} else {
BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(algorithm);
// beanDefinitionBuilder.addPropertyReference("dingerService", "dingerService");
// beanDefinitionBuilder.addPropertyValue("dingerName", "Jaemon");
AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition();
beanDefinition.setAutowireMode(GenericBeanDefinition.AUTOWIRE_BY_TYPE);
// 将当前算法注册到Spring容器中
String beanName = key + SPOT_SEPERATOR + algorithm.getSimpleName();
registry.registerBeanDefinition(beanName, beanDefinition);
MULTIDINGER_ALGORITHM_DEFINITION_MAP.put(
beanName, new MultiDingerAlgorithmDefinition(key, algorithm, dcs)
);
mode = AnalysisEnum.SPRING_CONTAINER;
}
if (log.isDebugEnabled()) {
log.debug("key={}, algorithm={} analysis through mode {}.",
key, algorithm.getSimpleName(), mode);
}
}
@Override
public int getOrder() {
return LOWEST_PRECEDENCE - 1;
}
public enum AnalysisEnum {
/**
* 反射方式
*/
REFLECT,
/**
* Spring容器
*/
SPRING_CONTAINER
;
}
}