package com.ld.igds.protocol.quantity.shuhan.server;

import com.ld.igds.protocol.quantity.shuhan.ShuHanUtils;
import com.ld.io.api.*;
import com.ld.io.netty.NettyServer;
import lombok.extern.slf4j.Slf4j;

import java.util.List;

/**
 * 配置主服务信息
 *
 * @author vince
 */
@Slf4j
public class ShuhanServerEngine {

	public static int PORT = 9303;

	public static NettyServer ioServer = null;
	// 配置消息接收类
	private static MsgConsumer messageConsume = new MsgConsumer();
	// 监听会话的创建与销毁
	private static SessionListener ioSessionListener = new SessionListener();

	// 心跳提供
	private static HeartbeatProvider heartbeatProvider = new HeartbeatProviderImpl();

	public static void start() throws Exception {
		start(PORT);
	}

	public static void start(int port) throws Exception {
		// 配置Server的配置
		IoServerOption ioServerOption = new IoServerOption(port);

		ioServerOption
				.setSplitDecoderType(SplitByteDecoderType.DELIMITER_SYMBOL);
		
		ioServerOption.setDelimiter(ShuHanUtils.MSG_END
				.getBytes(ShuHanUtils.CHARSET));

		// 配置心跳执行时间
		ioServerOption.setReaderIdleTime(ShuHanUtils.HEART_BEAT_TIME);

		ioServer = new NettyServer(ioServerOption, messageConsume,
				ioSessionListener, heartbeatProvider);

		ioServer.startup();

		log.info("* ========================");
		log.info("* ");
		log.info("* 蜀汉数量监测服务启动，端口号={}", port);
		log.info("* ");
		log.info("* ========================");

		System.out.println("* ========================");
		System.out.println("* ");
		System.out.println("* 数量监测服务启动，端口号=" + port);
		System.out.println("* ");
		System.out.println("* ========================");
	}

	/**
	 * 发送信息
	 *
	 * @param ip
	 * @param port
	 * @param msg
	 * @return
	 */
	public static InvokeResult push(String ip, int port, byte[] msg) {
		IoSession session = getSession(ip, port);
		if (null == session) {
			return InvokeResult.CHANNEL_CLOSED;
		}
		return session.invoke(msg);
	}

	/**
	 * 主动销毁当前连接
	 *
	 * @param ip
	 * @param port
	 * @return
	 */
	public static InvokeResult destroy(String ip, int port) {
		IoSession session = getSession(ip, port);

		if (null == session) {
			return InvokeResult.SUCCESS;
		}
		// 执行连接销毁
		session.destroy();
		return InvokeResult.SUCCESS;
	}

	/**
	 * 根据SessionKey获取Session信息
	 *
	 * @param ip
	 * @return
	 */
	public static IoSession getSession(String ip, int port) {
		IoSessionQuery sessionFactory = ioServer.getSessionQuery();
		List<IoSession> sessions = sessionFactory.getAllSession();
		for (IoSession ioSession : sessions) {
			if (ip.equals(ioSession.getAddress())
					&& port == ioSession.getPort()) {
				return ioSession;
			}
		}
		return null;
	}
}