Merge remote-tracking branch 'origin/master'
夜笙歌
1 year ago
commit
44bc09806c
@ -1,25 +0,0 @@
|
|||||||
package com.hw.mqtt.domain;
|
|
||||||
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import lombok.Data;
|
|
||||||
import lombok.NoArgsConstructor;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @author WenJY
|
|
||||||
* @date 2023年03月14日 13:46
|
|
||||||
*/
|
|
||||||
@Data
|
|
||||||
@NoArgsConstructor
|
|
||||||
@AllArgsConstructor
|
|
||||||
public class ServerNode {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 节点名称
|
|
||||||
*/
|
|
||||||
private String name;
|
|
||||||
/**
|
|
||||||
* ip:port
|
|
||||||
*/
|
|
||||||
private String peerHost;
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,62 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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 com.hw.mqtt.service;
|
|
||||||
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* mqtt broker 服务
|
|
||||||
*
|
|
||||||
* @author L.cm
|
|
||||||
*/
|
|
||||||
public interface IMqttBrokerService {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 获取在线客户端数量
|
|
||||||
* @author WenJY
|
|
||||||
* @date 2023-03-14 13:51
|
|
||||||
* @return long
|
|
||||||
*/
|
|
||||||
long getOnlineClientSize();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 获取所有在线的客户端
|
|
||||||
* @author WenJY
|
|
||||||
* @date 2023-03-14 13:51
|
|
||||||
* @return java.util.List<java.lang.String>
|
|
||||||
*/
|
|
||||||
List<String> getOnlineClients();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 向指定主题发送消息
|
|
||||||
* @author WenJY
|
|
||||||
* @date 2023-03-14 13:50
|
|
||||||
* @param topic
|
|
||||||
* @param payload
|
|
||||||
* @return boolean
|
|
||||||
*/
|
|
||||||
boolean publish(String topic,String payload);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 主动关闭指定客户端连接
|
|
||||||
* @author WenJY
|
|
||||||
* @date 2023-03-18 9:23
|
|
||||||
* @param clientId
|
|
||||||
*/
|
|
||||||
boolean closeClientById(String clientId);
|
|
||||||
}
|
|
||||||
@ -1,96 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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 com.hw.mqtt.service.impl;
|
|
||||||
|
|
||||||
import com.hw.mqtt.enums.RedisKeys;
|
|
||||||
import com.hw.mqtt.service.IMqttBrokerService;
|
|
||||||
import net.dreamlu.iot.mqtt.spring.server.MqttServerTemplate;
|
|
||||||
import net.dreamlu.mica.core.utils.StringPool;
|
|
||||||
import net.dreamlu.mica.redis.cache.MicaRedisCache;
|
|
||||||
import org.slf4j.Logger;
|
|
||||||
import org.slf4j.LoggerFactory;
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.stereotype.Service;
|
|
||||||
|
|
||||||
import java.nio.charset.StandardCharsets;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Collections;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Set;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* mqtt broker 服务
|
|
||||||
*
|
|
||||||
* @author L.cm
|
|
||||||
*/
|
|
||||||
@Service
|
|
||||||
public class MqttBrokerServiceImpl implements IMqttBrokerService {
|
|
||||||
private static final Logger logger = LoggerFactory.getLogger(MqttBrokerServiceImpl.class);
|
|
||||||
@Autowired private MicaRedisCache redisCache;
|
|
||||||
@Autowired private MqttServerTemplate server;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public long getOnlineClientSize() {
|
|
||||||
Set<String> keySet = redisCache.scan(RedisKeys.CONNECT_STATUS.getKey(StringPool.STAR));
|
|
||||||
if (keySet.isEmpty()) {
|
|
||||||
return 0L;
|
|
||||||
}
|
|
||||||
long result = 0;
|
|
||||||
for (String redisKey : keySet) {
|
|
||||||
Long count = redisCache.getSetOps().size(redisKey);
|
|
||||||
if (count != null) {
|
|
||||||
result += count;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<String> getOnlineClients() {
|
|
||||||
Set<String> keySet = redisCache.scan(RedisKeys.CONNECT_STATUS.getKey(StringPool.STAR));
|
|
||||||
if (keySet.isEmpty()) {
|
|
||||||
return Collections.emptyList();
|
|
||||||
}
|
|
||||||
List<String> clientList = new ArrayList<>();
|
|
||||||
for (String redisKey : keySet) {
|
|
||||||
Set<String> members = redisCache.sMembers(redisKey);
|
|
||||||
if (members != null && !members.isEmpty()) {
|
|
||||||
clientList.addAll(members);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return clientList;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean publish(String topic, String payload) {
|
|
||||||
boolean result = server.publishAll(topic, payload.getBytes(StandardCharsets.UTF_8));
|
|
||||||
logger.info("Mqtt publishAll result:{};topic:{};payload:{}", result, topic, payload);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public boolean closeClientById(String clientId) {
|
|
||||||
try{
|
|
||||||
server.close(clientId);
|
|
||||||
}catch (Exception ex){
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,40 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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 com.hw.mqtt.util;
|
|
||||||
|
|
||||||
import net.dreamlu.mica.core.utils.CharPool;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* redis 工具
|
|
||||||
*
|
|
||||||
* @author L.cm
|
|
||||||
*/
|
|
||||||
public class RedisUtil {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* 转换成 redis 的 pattern 规则
|
|
||||||
*
|
|
||||||
* @return pattern
|
|
||||||
*/
|
|
||||||
public static String getTopicPattern(String topicFilter) {
|
|
||||||
// mqtt 分享主题 $share/{ShareName}/{filter}
|
|
||||||
return topicFilter
|
|
||||||
.replace(CharPool.PLUS, CharPool.STAR)
|
|
||||||
.replace(CharPool.HASH, CharPool.STAR);
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue