net.dreamlu.iot.mqtt.spring.client.config.MqttClientConfiguration 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.config;
import net.dreamlu.iot.mqtt.codec.MqttTopicSubscription;
import net.dreamlu.iot.mqtt.core.client.IMqttClientConnectListener;
import net.dreamlu.iot.mqtt.core.client.MqttClient;
import net.dreamlu.iot.mqtt.core.client.MqttClientCreator;
import net.dreamlu.iot.mqtt.spring.client.MqttClientSubscribeDetector;
import net.dreamlu.iot.mqtt.spring.client.MqttClientTemplate;
import net.dreamlu.iot.mqtt.spring.client.event.SpringEventMqttClientConnectListener;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.StringUtils;
import org.tio.core.ssl.SSLEngineCustomizer;
import org.tio.core.ssl.SslConfig;
import java.nio.charset.StandardCharsets;
import java.util.List;
/**
* mqtt client 配置
*
* @author L.cm
*/
@Configuration(proxyBeanMethods = false)
@ConditionalOnProperty(
prefix = MqttClientProperties.PREFIX,
name = "enabled",
havingValue = "true",
matchIfMissing = true
)
@EnableConfigurationProperties(MqttClientProperties.class)
public class MqttClientConfiguration {
@Bean
@ConditionalOnMissingBean
public IMqttClientConnectListener springEventMqttClientConnectListener(ApplicationEventPublisher eventPublisher) {
return new SpringEventMqttClientConnectListener(eventPublisher);
}
@Bean
public MqttClientCreator mqttClientCreator(MqttClientProperties properties, ObjectProvider sslCustomizers) {
MqttClientCreator clientCreator = MqttClient.create()
.name(properties.getName())
.ip(properties.getIp())
.port(properties.getPort())
.username(properties.getUserName())
.password(properties.getPassword())
.clientId(properties.getClientId())
.readBufferSize((int) properties.getReadBufferSize().toBytes())
.maxBytesInMessage((int) properties.getMaxBytesInMessage().toBytes())
.maxClientIdLength(properties.getMaxClientIdLength())
.keepAliveSecs(properties.getKeepAliveSecs())
.reconnect(properties.isReconnect())
.reInterval(properties.getReInterval())
.retryCount(properties.getRetryCount())
.reSubscribeBatchSize(properties.getReSubscribeBatchSize())
.version(properties.getVersion())
.cleanSession(properties.isCleanSession())
.bufferAllocator(properties.getBufferAllocator())
.statEnable(properties.isStatEnable());
Integer timeout = properties.getTimeout();
if (timeout != null && timeout > 0) {
clientCreator.timeout(timeout);
}
if (properties.isDebug()) {
clientCreator.debug();
}
// 开启 ssl
MqttClientProperties.Ssl ssl = properties.getSsl();
if (ssl.isEnabled()) {
SslConfig sslConfig = SslConfig.forClient(ssl.getKeystorePath(), ssl.getKeystorePass(), ssl.getTruststorePath(), ssl.getTruststorePass());
clientCreator.sslConfig(sslConfig);
sslCustomizers.ifAvailable(sslConfig::setSslEngineCustomizer);
}
// 构造遗嘱消息
MqttClientProperties.WillMessage willMessage = properties.getWillMessage();
if (willMessage != null && StringUtils.hasText(willMessage.getTopic())) {
clientCreator.willMessage(builder -> {
builder.topic(willMessage.getTopic())
.qos(willMessage.getQos())
.retain(willMessage.isRetain());
if (StringUtils.hasText(willMessage.getMessage())) {
builder.message(willMessage.getMessage().getBytes(StandardCharsets.UTF_8));
}
});
}
// 全局订阅
List globalSubscribe = properties.getGlobalSubscribe();
if (globalSubscribe != null && !globalSubscribe.isEmpty()) {
clientCreator.globalSubscribe(globalSubscribe);
}
return clientCreator;
}
@Bean(MqttClientTemplate.DEFAULT_CLIENT_TEMPLATE_BEAN)
@ConditionalOnMissingBean(name = MqttClientTemplate.DEFAULT_CLIENT_TEMPLATE_BEAN)
public MqttClientTemplate mqttClientTemplate(MqttClientCreator mqttClientCreator) {
return new MqttClientTemplate(mqttClientCreator);
}
@Bean
public MqttClientSubscribeDetector mqttClientSubscribeDetector(ApplicationContext applicationContext) {
return new MqttClientSubscribeDetector(applicationContext);
}
}