wu-lazy-cloud-network/netty-proxy-server/src/main/java/com/luck/server/Constant.java

package com.luck.server;
 
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
 
import io.netty.channel.Channel;
import io.netty.util.AttributeKey;
import io.netty.util.internal.StringUtil;
 
public class Constant {
	/** 客户端服务channel */
	public static Channel clientChannel = null;
 
	/** 绑定channel_id */
	public static final AttributeKey<String> VID = AttributeKey.newInstance("vid");
 
	/** 访客，客户服务channel */
	public static Map<String, Channel> vcc = new ConcurrentHashMap<>();
 
	/** 访客，访客服务channel */
	public static Map<String, Channel> vvc = new ConcurrentHashMap<>();
 
	/** 服务代理端口 */
	public static int visitorPort = 16002;
 
	/** 服务端口 */
	public static int serverPort = 16001;
 
	/**
	 * 清除连接
	 * 
	 * @param vid 访客ID
	 */
	public static void clearVccVvc(String vid) {
		if (StringUtil.isNullOrEmpty(vid)) {
			return;
		}
		Channel clientChannel = vcc.get(vid);
		if (null != clientChannel) {
			clientChannel.attr(VID).set(null);
			vcc.remove(vid);
		}
		Channel visitorChannel = vvc.get(vid);
		if (null != visitorChannel) {
			visitorChannel.attr(VID).set(null);
			vvc.remove(vid);
		}
	}
 
	/**
	 * 清除关闭连接
	 * 
	 * @param vid 访客ID
	 */
	public static void clearVccVvcAndClose(String vid) {
		if (StringUtil.isNullOrEmpty(vid)) {
			return;
		}
		Channel clientChannel = vcc.get(vid);
		if (null != clientChannel) {
			clientChannel.attr(VID).set(null);
			vcc.remove(vid);
			clientChannel.close();
		}
		Channel visitorChannel = vvc.get(vid);
		if (null != visitorChannel) {
			visitorChannel.attr(VID).set(null);
			vvc.remove(vid);
			visitorChannel.close();
		}
	}
}