MqttClientConfiguration.java 3.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
/*
 * Copyright (c) 2019-2029, Dreamlu 卢春梦 (596392912@qq.com & 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;

19
import net.dreamlu.iot.mqtt.core.client.IMqttClientConnectListener;
20
import net.dreamlu.iot.mqtt.core.client.IMqttClientSession;
21 22
import net.dreamlu.iot.mqtt.core.client.MqttClient;
import net.dreamlu.iot.mqtt.core.client.MqttClientCreator;
浅梦2013's avatar
浅梦2013 已提交
23 24
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
25 26 27
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
28 29 30
import org.springframework.util.StringUtils;

import java.nio.charset.StandardCharsets;
31 32 33 34 35 36 37

/**
 * mqtt client 配置
 *
 * @author L.cm
 */
@Configuration(proxyBeanMethods = false)
浅梦2013's avatar
浅梦2013 已提交
38 39 40 41 42
@ConditionalOnProperty(
	prefix = MqttClientProperties.PREFIX,
	name = "enabled",
	havingValue = "true"
)
43 44 45 46
@EnableConfigurationProperties(MqttClientProperties.class)
public class MqttClientConfiguration {

	@Bean
浅梦2013's avatar
浅梦2013 已提交
47
	public MqttClientCreator mqttClientCreator(MqttClientProperties properties,
48
											   ObjectProvider<IMqttClientSession> clientSessionObjectProvider,
49
											   ObjectProvider<IMqttClientConnectListener> clientConnectListenerObjectProvider,
浅梦2013's avatar
浅梦2013 已提交
50 51
											   ObjectProvider<MqttClientCustomizer> customizers) {
		MqttClientCreator clientCreator = MqttClient.create()
52 53 54 55
			.name(properties.getName())
			.ip(properties.getIp())
			.port(properties.getPort())
			.username(properties.getUserName())
浅梦2013's avatar
浅梦2013 已提交
56 57 58
			.password(properties.getPassword())
			.clientId(properties.getClientId())
			.readBufferSize(properties.getReadBufferSize())
59
			.maxBytesInMessage(properties.getMaxBytesInMessage())
60
			.maxClientIdLength(properties.getMaxClientIdLength())
浅梦2013's avatar
浅梦2013 已提交
61 62
			.keepAliveSecs(properties.getKeepAliveSecs())
			.reconnect(properties.isReconnect())
63 64
			.reInterval(properties.getReInterval())
			.setRetryCount(properties.getRetryCount())
浅梦2013's avatar
浅梦2013 已提交
65 66 67 68 69
			.version(properties.getVersion())
			.cleanSession(properties.isCleanSession())
			.bufferAllocator(properties.getBufferAllocator());
		Integer timeout = properties.getTimeout();
		if (timeout != null && timeout > 0) {
浅梦2013's avatar
浅梦2013 已提交
70
			clientCreator.timeout(timeout);
浅梦2013's avatar
浅梦2013 已提交
71
		}
72 73 74
		// 构造遗嘱消息
		MqttClientProperties.WillMessage willMessage = properties.getWillMessage();
		if (willMessage != null && StringUtils.hasText(willMessage.getTopic())) {
浅梦2013's avatar
浅梦2013 已提交
75 76 77 78 79 80 81 82
			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));
				}
			});
83
		}
84 85
		// 客户端 session
		clientSessionObjectProvider.ifAvailable(clientCreator::clientSession);
86 87
		// 配置客户端链接监听器
		clientConnectListenerObjectProvider.ifAvailable(clientCreator::connectListener);
浅梦2013's avatar
浅梦2013 已提交
88 89 90
		// 自定义处理
		customizers.ifAvailable((customizer) -> customizer.customize(clientCreator));
		return clientCreator;
91 92 93 94 95 96 97
	}

	@Bean
	public MqttClientTemplate mqttClientTemplate(MqttClientCreator mqttClientCreator) {
		return new MqttClientTemplate(mqttClientCreator);
	}

98 99 100 101 102
	@Bean
	public MqttClientSubscribeDetector mqttClientSubscribeDetector(MqttClientTemplate clientTemplate) {
		return new MqttClientSubscribeDetector(clientTemplate);
	}

103
}