net.dreamlu.iot.mqtt.spring.client.MqttClientSubscribeDetector Maven / Gradle / Ivy
/*
* Copyright (c) 2019-2029, Dreamlu 卢春梦 ([email protected] & dreamlu.net).
*
* 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 net.dreamlu.iot.mqtt.spring.client;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import net.dreamlu.iot.mqtt.codec.MqttPublishMessage;
import net.dreamlu.iot.mqtt.core.client.IMqttClientMessageListener;
import net.dreamlu.iot.mqtt.core.client.IMqttClientSession;
import net.dreamlu.iot.mqtt.core.util.TopicUtil;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.ApplicationContext;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.lang.NonNull;
import org.springframework.util.ClassUtils;
import org.springframework.util.ReflectionUtils;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.Map;
/**
* MqttClient 订阅监听器
*
* @author L.cm
*/
@Slf4j
@RequiredArgsConstructor
public class MqttClientSubscribeDetector implements BeanPostProcessor {
private final ApplicationContext applicationContext;
@Override
public Object postProcessAfterInitialization(@NonNull Object bean, String beanName) throws BeansException {
Class userClass = ClassUtils.getUserClass(bean);
// 1. 查找类上的 MqttClientSubscribe 注解
if (bean instanceof IMqttClientMessageListener) {
MqttClientSubscribe subscribe = AnnotationUtils.findAnnotation(userClass, MqttClientSubscribe.class);
if (subscribe != null) {
IMqttClientSession clientSession = getMqttClientSession(applicationContext, subscribe.clientTemplateBean());
String[] topicFilters = getTopicFilters(applicationContext, subscribe.value());
clientSession.addSubscriptionList(topicFilters, subscribe.qos(), (IMqttClientMessageListener) bean);
}
} else {
// 2. 查找方法上的 MqttClientSubscribe 注解
ReflectionUtils.doWithMethods(userClass, method -> {
MqttClientSubscribe subscribe = AnnotationUtils.findAnnotation(method, MqttClientSubscribe.class);
if (subscribe != null) {
// 1. 校验必须为 public 和非 static 的方法
int modifiers = method.getModifiers();
if (Modifier.isStatic(modifiers)) {
throw new IllegalArgumentException("@MqttClientSubscribe on method " + method + " must not static.");
}
if (!Modifier.isPublic(modifiers)) {
throw new IllegalArgumentException("@MqttClientSubscribe on method " + method + " must public.");
}
// 2. 校验 method 入参数必须等于2
int paramCount = method.getParameterCount();
if (paramCount < 2 || paramCount > 4) {
throw new IllegalArgumentException("@MqttClientSubscribe on method " + method + " parameter count must 2 ~ 4.");
}
// 3. 校验 method 入参类型必须为 String、Map、MqttPublishMessage、ByteBuffer
Class[] parameterTypes = method.getParameterTypes();
checkParameterTypes(method, parameterTypes);
// 4. mqtt client bean
IMqttClientSession clientSession = getMqttClientSession(applicationContext, subscribe.clientTemplateBean());
String[] topicTemplates = subscribe.value();
String[] topicFilters = getTopicFilters(applicationContext, topicTemplates);
// 5. 监听器
MqttClientSubscribeMessageListener messageListener = new MqttClientSubscribeMessageListener(
bean, method, topicTemplates, topicFilters);
// 6. 订阅
clientSession.addSubscriptionList(topicFilters, subscribe.qos(), messageListener);
}
}, ReflectionUtils.USER_DECLARED_METHODS);
}
return bean;
}
/**
* 读取 IMqttClientSession
*
* @param applicationContext ApplicationContext
* @param beanName beanName
* @return IMqttClientSession
*/
private static IMqttClientSession getMqttClientSession(ApplicationContext applicationContext, String beanName) {
return applicationContext.getBean(beanName, MqttClientTemplate.class).getClientCreator().getClientSession();
}
/**
* 解析 Spring boot env 变量
*
* @param applicationContext ApplicationContext
* @param values values
* @return topic array
*/
private static String[] getTopicFilters(ApplicationContext applicationContext, String[] values) {
// 1. 替换 Spring boot env 变量
// 2. 替换订阅中的其他变量
return Arrays.stream(values)
.map(applicationContext.getEnvironment()::resolvePlaceholders)
.map(TopicUtil::getTopicFilter)
.toArray(String[]::new);
}
/**
* 校验方法参数
*
* @param method Method
* @param parameterTypes parameterTypes
*/
private static void checkParameterTypes(Method method, Class[] parameterTypes) {
for (Class parameterType : parameterTypes) {
if (String.class != parameterType && Map.class != parameterType && MqttPublishMessage.class != parameterType
&& byte[].class != parameterType && ByteBuffer.class != parameterType) {
throw new IllegalArgumentException("@MqttClientSubscribe on method " + method + " parameter type must String topic, Map topicVars, MqttPublishMessage message, byte[] payload or ByteBuffer payload.");
}
}
}
}