io.magician.tcp.TCPServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Magician Show documentation
There is a newer version: 2.0.7
package io.magician.tcp;

import io.magician.common.event.EventGroup;
import io.magician.tcp.handler.MagicianHandler;
import io.magician.tcp.codec.ProtocolCodec;
import io.magician.tcp.codec.impl.websocket.handler.WebSocketHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.InetSocketAddress;
import java.nio.channels.*;
import java.util.concurrent.Executors;

/**
 * tcp服务创建
 */
public class TCPServer {

    private Logger log = LoggerFactory.getLogger(TCPServer.class);

    /**
     * 连接超时时间
     */
    private int soTimeout = 10000;
    /**
     * io事件执行器组合
     * 每个端口对应里面的一个 事件执行器
     */
    private EventGroup ioEventGroup;
    /**
     * 业务事件执行器组合
     * 每个连接对应里面的一个事件执行器，一个事件执行器对应多个连接的多个任务
     */
    private EventGroup workerEventGroup;
    /**
     * 配置
     */
    private TCPServerConfig tcpServerConfig;

    /**
     * 创建一个默认事件组合的服务，用于监听端口
     */
    public TCPServer() {
        this(
                new EventGroup(1, Executors.newCachedThreadPool()),
                new EventGroup(3, Executors.newCachedThreadPool())
        );
    }

    /**
     * 自定义事件组合，创建服务用于监听端口
     *
     * @param ioEventGroup
     * @param workerEventGroup
     */
    public TCPServer(EventGroup ioEventGroup, EventGroup workerEventGroup) {
        this.ioEventGroup = ioEventGroup;
        this.workerEventGroup = workerEventGroup;
        this.tcpServerConfig = new TCPServerConfig();
    }

    /**
     * 连接超时时间
     *
     * @param soTimeout
     * @return
     */
    public TCPServer soTimeout(int soTimeout) {
        this.soTimeout = soTimeout;
        return this;
    }

    /**
     * 添加配置
     *
     * @param tcpServerConfig
     * @return
     */
    public TCPServer config(TCPServerConfig tcpServerConfig) {
        this.tcpServerConfig = tcpServerConfig;
        return this;
    }

    /**
     * 设置协议解析器
     *
     * @param protocolCodec
     * @return
     */
    public TCPServer protocolCodec(ProtocolCodec protocolCodec) {
        this.tcpServerConfig.setProtocolCodec(protocolCodec);
        return this;
    }

    /**
     * 设置处理器
     *
     * @param magicianHandler
     * @return
     */
    public TCPServer handler(String path, MagicianHandler magicianHandler) throws Exception {
        this.tcpServerConfig.addMagicianHandler(path, magicianHandler);
        return this;
    }

    /**
     * 设置处理器
     *
     * @param webSocketHandler
     * @return
     */
    public TCPServer webSocketHandler(String path, WebSocketHandler webSocketHandler) throws Exception {
        this.tcpServerConfig.addWebSocketHandler(path, webSocketHandler);
        return this;
    }

    /**
     * 开启服务
     *
     * @param port
     * @throws Exception
     */
    public void bind(int port) throws Exception {
        bind(port, 1000);
    }

    /**
     * 开启服务
     *
     * @param port
     * @param backLog
     * @throws Exception
     */
    public void bind(int port, int backLog) throws Exception {

        /* 开始监听端口 */
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.socket().setSoTimeout(soTimeout);
        serverSocketChannel.configureBlocking(false);
        serverSocketChannel.bind(new InetSocketAddress(port), backLog);

        /* 开启NIOSelector监听器 */
        ioEventGroup
                .getEventRunner()
                .addEvent(new TCPServerMonitorTask(serverSocketChannel, tcpServerConfig, ioEventGroup, workerEventGroup));

        /* 标识服务是否已经启动 */
        log.info("启动TCP服务成功, [port:" + port + "]");
    }
}