org.tio.server.ServerTioConfig Java Examples

public WsServerStarter(WsServerConfig wsServerConfig, IWsMsgHandler wsMsgHandler, TioUuid tioUuid, SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor)
        throws IOException {
	if (tioExecutor == null) {
		tioExecutor = Threads.getTioExecutor();
	}

	if (groupExecutor == null) {
		groupExecutor = Threads.getGroupExecutor();
	}

	this.wsServerConfig = wsServerConfig;
	this.wsMsgHandler = wsMsgHandler;
	wsServerAioHandler = new WsServerAioHandler(wsServerConfig, wsMsgHandler);
	wsServerAioListener = new WsServerAioListener();
	serverTioConfig = new ServerTioConfig("Tio Websocket Server", wsServerAioHandler, wsServerAioListener, tioExecutor, groupExecutor);
	serverTioConfig.setHeartbeatTimeout(0);
	serverTioConfig.setTioUuid(tioUuid);
	serverTioConfig.setReadBufferSize(1024 * 30);
	tioServer = new TioServer(serverTioConfig);
}
 

/**
 * 
 * @param ip 可以为null
 * @param port 如果为null，则用默认的端口
 * @param tioExecutor
 * @param groupExecutor
 * @author tanyaowu
 */
public static void start(String ip, Integer port, SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor) {
	if (port == null) {
		port = Const.PORT;
	}
	aioHandler = new FlashPolicyServerAioHandler();
	serverTioConfig = new ServerTioConfig("tio flash policy server", aioHandler, aioListener, tioExecutor, groupExecutor);
	serverTioConfig.setHeartbeatTimeout(Const.HEARTBEAT_TIMEOUT);
	tioServer = new TioServer(serverTioConfig);

	try {
		tioServer.start(ip, port);
	} catch (Throwable e) {
		log.error(e.toString(), e);
		System.exit(1);
	}

	checkAllChannels();
}
 

/**
 *  删除clientip为指定值的所有连接
 * @param serverTioConfig
 * @param ip
 * @param remark
 * @param closeCode
 */
public static void remove(ServerTioConfig serverTioConfig, String ip, String remark, CloseCode closeCode) {
	SetWithLock<ChannelContext> setWithLock = serverTioConfig.ips.clients(serverTioConfig, ip);
	if (setWithLock == null) {
		return;
	}

	setWithLock.handle(new ReadLockHandler<Set<ChannelContext>>() {
		@Override
		public void handler(Set<ChannelContext> set) {
			for (ChannelContext channelContext : set) {
				Tio.remove(channelContext, remark, closeCode);
			}
		}
	});
}
 

public boolean add(String ip) {
	//先添加到黑名单列表
	cache.put(ip, SystemTimer.currTime);

	if (serverTioConfig != null) {
		//删除相关连接
		Tio.remove(serverTioConfig, ip, "ip[" + ip + "]被加入了黑名单, " + serverTioConfig.getName());
	} else {
		TioConfig.ALL_SERVER_GROUPCONTEXTS.stream().forEach(new Consumer<ServerTioConfig>() {
			@Override
			public void accept(ServerTioConfig tioConfig) {
				Tio.remove(tioConfig, ip, "ip[" + ip + "]被加入了黑名单, " + tioConfig.getName());

			}
		});
	}

	return true;
}
 

private void initTioServerTioConfig()
{
    serverTioConfig = new ServerTioConfig(GROUP_CONTEXT_NAME, serverAioHandler, serverAioListener);
    if (ipStatListener != null) {
        serverTioConfig.setIpStatListener(ipStatListener);
        // fixed bug for IpStatListener not work
        serverTioConfig.ipStats.addDurations(serverProperties.getIpStatDurations());
    }
    if(serverAioListener != null) {
        serverTioConfig.setServerAioListener(serverAioListener);
    }
    if (groupListener != null) {
        serverTioConfig.setGroupListener(groupListener);
    }
    if (serverProperties.getHeartbeatTimeout() > 0) {
        serverTioConfig.setHeartbeatTimeout(serverProperties.getHeartbeatTimeout());
    }
    //cluster config
    if (clusterConfig != null) {
        serverTioConfig.setTioClusterConfig(clusterConfig);
    }
    //ssl config
    if (serverSslProperties.isEnabled()){
        try {
            serverTioConfig.useSsl(serverSslProperties.getKeyStore(), serverSslProperties.getTrustStore(), serverSslProperties.getPassword());
        }catch (Exception e){
            //catch and log
            logger.error("init ssl config error",e);
        }
    }
}
 

private void init(HttpConfig httpConfig, HttpRequestHandler requestHandler, SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor) {
	String system_timer_period = System.getProperty("tio.system.timer.period");
	if (StrUtil.isBlank(system_timer_period)) {
		System.setProperty("tio.system.timer.period", "50");
	}

	this.httpConfig = httpConfig;
	this.httpRequestHandler = requestHandler;
	httpConfig.setHttpRequestHandler(this.httpRequestHandler);
	this.httpServerAioHandler = new HttpServerAioHandler(httpConfig, requestHandler);
	httpServerAioListener = new HttpServerAioListener();
	String name = httpConfig.getName();
	if (StrUtil.isBlank(name)) {
		name = "Tio Http Server";
	}
	serverTioConfig = new ServerTioConfig(name, httpServerAioHandler, httpServerAioListener, tioExecutor, groupExecutor);
	serverTioConfig.setHeartbeatTimeout(1000 * 20);
	serverTioConfig.setShortConnection(true);
	serverTioConfig.setReadBufferSize(TcpConst.MAX_DATA_LENGTH);
	//		serverTioConfig.setAttribute(TioConfigKey.HTTP_SERVER_CONFIG, httpConfig);
	serverTioConfig.setAttribute(TioConfigKey.HTTP_REQ_HANDLER, this.httpRequestHandler);

	tioServer = new TioServer(serverTioConfig);

	HttpUuid imTioUuid = new HttpUuid();
	serverTioConfig.setTioUuid(imTioUuid);
}
 

/**
 * 
 * @param tioExecutor
 * @param groupExecutor
 * @author: tanyaowu
 */
public TioConfig(SynThreadPoolExecutor tioExecutor, ThreadPoolExecutor groupExecutor) {
	super();
	ALL_GROUPCONTEXTS.add(this);
	if (this instanceof ServerTioConfig) {
		ALL_SERVER_GROUPCONTEXTS.add((ServerTioConfig) this);
	} else {
		ALL_CLIENT_GROUPCONTEXTS.add((ClientTioConfig) this);
	}

	if (ALL_GROUPCONTEXTS.size() > 20) {
		log.warn("已经产生{}个TioConfig对象，t-io作者怀疑你在误用t-io", ALL_GROUPCONTEXTS.size());
	}
	this.id = ID_ATOMIC.incrementAndGet() + "";

	this.ipStats = new IpStats(this, null);

	this.tioExecutor = tioExecutor;
	if (this.tioExecutor == null) {
		this.tioExecutor = Threads.getTioExecutor();
	}

	this.groupExecutor = groupExecutor;
	if (this.groupExecutor == null) {
		this.groupExecutor = Threads.getGroupExecutor();
	}

	closeRunnable = new CloseRunnable(this.tioExecutor);
}
 

/**
 * @return the serverTioConfig
 */
public ServerTioConfig getServerTioConfig() {
	return serverTioConfig;
}
 

@Bean
public ServerTioConfig serverTioConfig(TioServerBootstrap bootstrap){
    return bootstrap.getServerTioConfig();
}
 

public ServerTioConfig getServerTioConfig() {
    return serverTioConfig;
}
 

public ServerTioConfig getServerTioConfig() {
    return serverTioConfig;
}
 

@Bean
public ServerTioConfig wsServerTioConfig(TioWebSocketServerBootstrap bootstrap){
    return bootstrap.getServerTioConfig();
}
 

/**
 * @return the serverTioConfig
 */
public ServerTioConfig getServerTioConfig() {
	return serverTioConfig;
}
 

public IpBlacklist(String id, ServerTioConfig serverTioConfig) {
	this.id = id;
	this.serverTioConfig = serverTioConfig;
	this.cacheName = CACHE_NAME_PREFIX + this.id;
	this.cache = CaffeineCache.register(this.cacheName, TIME_TO_LIVE_SECONDS, TIME_TO_IDLE_SECONDS, null);
}
 

/**
 * 删除clientip为指定值的所有连接
 * @param serverTioConfig
 * @param ip
 * @param remark
 */
public static void remove(ServerTioConfig serverTioConfig, String ip, String remark) {
	remove(serverTioConfig, ip, remark, (CloseCode) null);
}