com.github.jaemon.dinger.core.DingerMessageHandler Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of dinger-spring-boot-starter Show documentation
Show all versions of dinger-spring-boot-starter Show documentation
Dinger-SpringBoot集成钉钉/企业微信群机器人实现消息通知中间件
/*
* Copyright ©2015-2021 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.core;
import com.github.jaemon.dinger.core.annatations.Dinger;
import com.github.jaemon.dinger.core.entity.DingerProperties;
import com.github.jaemon.dinger.core.entity.MsgType;
import com.github.jaemon.dinger.core.entity.enums.DingerType;
import com.github.jaemon.dinger.core.entity.DingerResponse;
import com.github.jaemon.dinger.multi.MultiDingerConfigContainer;
import com.github.jaemon.dinger.multi.MultiDingerProperty;
import com.github.jaemon.dinger.multi.entity.MultiDingerConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import static com.github.jaemon.dinger.constant.DingerConstant.SPOT_SEPERATOR;
/**
* DingerMessageHandler
*
* @author Jaemon
* @since 1.0
*/
public class DingerMessageHandler
extends MultiDingerProperty
implements ParamHandler, MessageTransfer, ResultHandler {
private static final Logger log = LoggerFactory.getLogger(DingerMessageHandler.class);
protected DingerRobot dingerRobot;
protected DingerProperties dingerProperties;
@Override
public Map paramsHandler(Method method, DingerDefinition dingerDefinition, Object[] values) {
Map params = new HashMap<>();
int valueLength = values.length;
if (valueLength == 0) {
return params;
}
String[] keys = dingerDefinition.methodParams();
int[] genericIndex = dingerDefinition.genericIndex();
if (genericIndex.length > 0) {
for (int i : genericIndex) {
params.put(keys[i], values[i]);
}
return params;
}
int keyLength = keys.length;
if (keyLength == valueLength) {
for (int i = 0; i < valueLength; i++) {
params.put(keys[i], values[i]);
}
return params;
}
Parameter[] parameters = method.getParameters();
for (int i = 0; i < parameters.length; i++) {
Parameter parameter = parameters[i];
String paramName = parameter.getName();
com.github.jaemon.dinger.core.annatations.Parameter[] panno =
parameter.getDeclaredAnnotationsByType(com.github.jaemon.dinger.core.annatations.Parameter.class);
if (panno != null && panno.length > 0) {
paramName = panno[0].value();
}
params.put(paramName, values[i]);
}
return params;
}
@Override
public MsgType transfer(DingerDefinition dingerDefinition, Map params) {
MsgType message = copyProperties(dingerDefinition.message());
message.transfer(params);
return message;
}
@Override
public Object resultHandler(Class resultType, DingerResponse dingerResponse) {
String name = resultType.getName();
if (String.class.getName().equals(name)) {
return Optional.ofNullable(dingerResponse).map(e -> e.getData()).orElse(null);
} else if (DingerResponse.class.getName().equals(name)) {
return dingerResponse;
}
return null;
}
/**
* copyProperties
*
* @param src src
* @param T extends Message
* @return msg
*/
private T copyProperties(MsgType src) {
try {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(baos);
oos.writeObject(src);
ByteArrayInputStream byteIn = new ByteArrayInputStream(baos.toByteArray());
ObjectInputStream in = new ObjectInputStream(byteIn);
T dest = (T) in.readObject();
return dest;
} catch (Exception e) {
//
if (log.isDebugEnabled()) {
log.debug("copy properties error:", e);
}
return null;
}
}
/**
* 获取方法执行的Dinger
*
* @param method
* 代理方法
* @return
* 返回Dinger
*/
DingerType dingerType(Method method) {
Class dingerClass = method.getDeclaringClass();
if (dingerClass.isAnnotationPresent(Dinger.class)) {
return dingerClass.getAnnotation(Dinger.class).value();
}
return dingerProperties.getDefaultDinger();
}
/**
* 获取Dinger定义
*
*
* 优先级: local(dinger为空使用默认 {@link DingerMessageHandler#dingerType}) > multi > default({@link DingerMessageHandler#dingerType})
*
*
* @param useDinger
* 代理方法使用的Dinger
* @param dingerClassName
* 代理类全限定名
* @param keyName
* 代理方法全限定名
* @return
* dingerDefinition {@link DingerDefinition}
*/
DingerDefinition dingerDefinition(DingerType useDinger, String dingerClassName, String keyName) {
DingerDefinition dingerDefinition;
DingerConfig localDinger = DingerHelper.getLocalDinger();
// 优先使用用户设定 dingerConfig
if (localDinger == null) {
String dingerName = useDinger + SPOT_SEPERATOR + keyName;
dingerDefinition = AbstractDingerDefinitionResolver
.Container.INSTANCE.get(dingerName);
if (dingerDefinition == null) {
return null;
}
DingerConfig dingerMethodDefaultDingerConfig = dingerDefinition.dingerConfig();
// 判断是否是multiDinger
if (multiDinger()) {
MultiDingerConfig multiDingerConfig =
MultiDingerConfigContainer
.INSTANCE.get(useDinger, dingerClassName);
DingerConfig dingerConfig = null;
if (multiDingerConfig != null) {
// 拿到MultiDingerConfig中当前应该使用的DingerConfig
dingerConfig = multiDingerConfig.getAlgorithmHandler()
.dingerConfig(
multiDingerConfig.getDingerConfigs(),
dingerMethodDefaultDingerConfig
);
}
// use default dingerConfig
if (dingerConfig == null) {
dingerConfig = dingerMethodDefaultDingerConfig;
}
DingerHelper.assignDinger(dingerConfig);
} else {
DingerHelper.assignDinger(dingerMethodDefaultDingerConfig);
}
} else {
DingerType dingerType = localDinger.getDingerType();
if (dingerType == null) {
dingerType = useDinger;
}
keyName = dingerType + SPOT_SEPERATOR + keyName;
dingerDefinition = AbstractDingerDefinitionResolver
.Container.INSTANCE.get(keyName);
if (dingerDefinition == null) {
return null;
}
}
return dingerDefinition;
}
}