br.com.anteros.mqtt.broker.service.impl.MqttBrokerServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of anteros-mqtt-broker Show documentation
Anteros mqtt tools.
The newest version!
/*
 * 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 br.com.anteros.mqtt.broker.service.impl;

import br.com.anteros.mqtt.broker.service.IMqttBrokerService;
import br.com.anteros.mqtt.broker.enums.RedisKeys;
import br.com.anteros.mqtt.broker.model.ServerNode;
import net.dreamlu.mica.core.utils.StringPool;
import net.dreamlu.mica.redis.cache.MicaRedisCache;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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 {
	@Autowired
	private MicaRedisCache redisCache;

	@Override
	public List getNodes() {
		Set nodeKeySet = redisCache.scan(RedisKeys.SERVER_NODES.getKey(StringPool.STAR));
		if (nodeKeySet.isEmpty()) {
			return Collections.emptyList();
		}
		int beginIndex = RedisKeys.SERVER_NODES.getKey().length();
		List list = new ArrayList<>();
		for (String nodeKey : nodeKeySet) {
			String nodeName = nodeKey.substring(beginIndex);
			list.add(new ServerNode(nodeName, redisCache.get(nodeKey)));
		}
		return list;
	}

	@Override
	public long getOnlineClientSize() {
		Set 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 getOnlineClients() {
		Set keySet = redisCache.scan(RedisKeys.CONNECT_STATUS.getKey(StringPool.STAR));
		if (keySet.isEmpty()) {
			return Collections.emptyList();
		}
		List clientList = new ArrayList<>();
		for (String redisKey : keySet) {
			Set members = redisCache.sMembers(redisKey);
			if (members != null && !members.isEmpty()) {
				clientList.addAll(members);
			}
		}
		return clientList;
	}

	@Override
	public long getOnlineClientSize(String nodeName) {
		Long count = redisCache.getSetOps().size(RedisKeys.CONNECT_STATUS.getKey(nodeName));
		return count == null ? 0L : count;
	}

	@Override
	public List getOnlineClients(String nodeName) {
		Set members = redisCache.sMembers(RedisKeys.CONNECT_STATUS.getKey(nodeName));
		if (members == null || members.isEmpty()) {
			return Collections.emptyList();
		}
		return new ArrayList<>(members);
	}

}