Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
net.dreamlu.iot.mqtt.spring.client.MqttClientTemplate 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 net.dreamlu.iot.mqtt.codec.MqttQoS;
import net.dreamlu.iot.mqtt.core.client.*;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.beans.factory.SmartInitializingSingleton;
import org.springframework.core.Ordered;
import org.tio.client.ClientChannelContext;
import org.tio.client.TioClient;
import org.tio.client.TioClientConfig;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;
/**
* mqtt client 模板
*
* @author wsq（冷月宫主）
*/
public class MqttClientTemplate implements SmartInitializingSingleton, DisposableBean, Ordered {
public static final String DEFAULT_CLIENT_TEMPLATE_BEAN = "mqttClientTemplate";
private final MqttClientCreator clientCreator;
private final ObjectProvider clientConnectListenerObjectProvider;
private final ObjectProvider customizersObjectProvider;
private final List tempSubscriptionList;
private MqttClient client;
public MqttClientTemplate(MqttClientCreator clientCreator,
ObjectProvider clientConnectListenerObjectProvider) {
this(clientCreator, clientConnectListenerObjectProvider, null);
}
public MqttClientTemplate(MqttClientCreator clientCreator,
ObjectProvider clientConnectListenerObjectProvider,
ObjectProvider customizersObjectProvider) {
this.clientCreator = clientCreator;
this.clientConnectListenerObjectProvider = clientConnectListenerObjectProvider;
this.customizersObjectProvider = customizersObjectProvider;
this.tempSubscriptionList = new ArrayList<>();
}
/**
* 订阅
*
* @param topicFilter topicFilter
* @param listener MqttMessageListener
* @return MqttClient
*/
public MqttClient subQos0(String topicFilter, IMqttClientMessageListener listener) {
return client.subscribe(topicFilter, MqttQoS.AT_MOST_ONCE, listener);
}
/**
* 订阅
*
* @param topicFilter topicFilter
* @param listener MqttMessageListener
* @return MqttClient
*/
public MqttClient subQos1(String topicFilter, IMqttClientMessageListener listener) {
return client.subscribe(topicFilter, MqttQoS.AT_LEAST_ONCE, listener);
}
/**
* 订阅
*
* @param topicFilter topicFilter
* @param listener MqttMessageListener
* @return MqttClient
*/
public MqttClient subQos2(String topicFilter, IMqttClientMessageListener listener) {
return client.subscribe(topicFilter, MqttQoS.EXACTLY_ONCE, listener);
}
/**
* 订阅
*
* @param mqttQoS MqttQoS
* @param topicFilter topicFilter
* @param listener MqttMessageListener
* @return MqttClient
*/
public MqttClient subscribe(MqttQoS mqttQoS, String topicFilter, IMqttClientMessageListener listener) {
return client.subscribe(topicFilter, mqttQoS, listener);
}
/**
* 订阅
*
* @param topicFilters topicFilter 数组
* @param mqttQoS MqttQoS
* @param listener MqttMessageListener
* @return MqttClient
*/
public MqttClient subscribe(String[] topicFilters, MqttQoS mqttQoS, IMqttClientMessageListener listener) {
return client.subscribe(topicFilters, mqttQoS, listener);
}
/**
* 批量订阅
*
* @param subscriptionList 订阅集合
* @return MqttClient
*/
public MqttClient subscribe(List subscriptionList) {
return client.subscribe(subscriptionList);
}
/**
* 取消订阅
*
* @param topicFilter topicFilter
* @return MqttClient
*/
public MqttClient unSubscribe(String... topicFilter) {
return client.unSubscribe(topicFilter);
}
/**
* 取消订阅
*
* @param topicFilters topicFilter 集合
* @return MqttClient
*/
public MqttClient unSubscribe(List topicFilters) {
return client.unSubscribe(topicFilters);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息内容
* @return 是否发送成功
*/
public boolean publish(String topic, byte[] payload) {
return client.publish(topic, payload, MqttQoS.AT_MOST_ONCE);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息内容
* @param qos MqttQoS
* @return 是否发送成功
*/
public boolean publish(String topic, byte[] payload, MqttQoS qos) {
return client.publish(topic, payload, qos, false);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息体
* @param qos MqttQoS
* @param retain 是否在服务器上保留消息
* @return 是否发送成功
*/
public boolean publish(String topic, byte[] payload, MqttQoS qos, boolean retain) {
return client.publish(topic, payload, qos, retain);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息体
* @return 是否发送成功
*/
public boolean publish(String topic, ByteBuffer payload) {
return client.publish(topic, payload, MqttQoS.AT_MOST_ONCE);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息体
* @param qos MqttQoS
* @return 是否发送成功
*/
public boolean publish(String topic, ByteBuffer payload, MqttQoS qos) {
return client.publish(topic, payload, qos, false);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息体
* @param retain 是否在服务器上保留消息
* @return 是否发送成功
*/
public boolean publish(String topic, ByteBuffer payload, boolean retain) {
return client.publish(topic, payload, MqttQoS.AT_MOST_ONCE, retain);
}
/**
* 发布消息
*
* @param topic topic
* @param payload 消息体
* @param qos MqttQoS
* @param retain 是否在服务器上保留消息
* @return 是否发送成功
*/
public boolean publish(String topic, ByteBuffer payload, MqttQoS qos, boolean retain) {
return client.publish(topic, payload, qos, retain);
}
/**
* 重连
*/
public void reconnect() {
client.reconnect();
}
/**
* 重连到新的服务端节点
*
* @param ip ip
* @param port port
* @return 是否成功
*/
public boolean reconnect(String ip, int port) {
return client.reconnect(ip, port);
}
/**
* 断开 mqtt 连接
*
* @return 是否成功
*/
public boolean disconnect() {
return client.disconnect();
}
/**
* 获取 TioClient
*
* @return TioClient
*/
public TioClient getTioClient() {
return client.getTioClient();
}
/**
* 获取配置
*
* @return MqttClientCreator
*/
public MqttClientCreator getClientCreator() {
return clientCreator;
}
/**
* 获取 ClientTioConfig
*
* @return ClientTioConfig
*/
public TioClientConfig getClientTioConfig() {
return client.getClientTioConfig();
}
/**
* 获取 ClientChannelContext
*
* @return ClientChannelContext
*/
public ClientChannelContext getContext() {
return client.getContext();
}
/**
* 判断客户端跟服务端是否连接
*
* @return 是否已经连接成功
*/
public boolean isConnected() {
return client.isConnected();
}
/**
* 判断客户端跟服务端是否断开连接
*
* @return 是否断连
*/
public boolean isDisconnected() {
return client.isDisconnected();
}
/**
* 获取 MqttClient
*
* @return MqttClient
*/
public MqttClient getMqttClient() {
return client;
}
/**
* 添加启动时的临时订阅
*
* @param topicFilters topicFilters
* @param qos MqttQoS
* @param messageListener IMqttClientMessageListener
*/
void addSubscriptionList(String[] topicFilters, MqttQoS qos, IMqttClientMessageListener messageListener) {
for (String topicFilter : topicFilters) {
tempSubscriptionList.add(new MqttClientSubscription(qos, topicFilter, messageListener));
}
}
@Override
public void afterSingletonsInstantiated() {
// 配置客户端连接监听器
clientConnectListenerObjectProvider.ifAvailable(clientCreator::connectListener);
// 自定义处理
if (customizersObjectProvider != null) {
customizersObjectProvider.ifAvailable(customizer -> customizer.customize(clientCreator));
}
// 连接超时时间，如果没设置，改成 3s，减少因连不上卡顿时间
Integer timeout = clientCreator.getTimeout();
if (timeout == null) {
clientCreator.timeout(3);
}
// 使用同步连接，不过如果连不上会卡一会
client = clientCreator.connectSync();
// 添加订阅并清理零时订阅存储
client.subscribe(tempSubscriptionList);
tempSubscriptionList.clear();
}
@Override
public void destroy() {
client.stop();
}
@Override
public int getOrder() {
return Ordered.LOWEST_PRECEDENCE;
}
}
