package cn.flightfeather.supervision.infrastructure.service.Impl;
|
|
import cn.flightfeather.supervision.infrastructure.service.AbstractNettyServer;
|
import cn.flightfeather.supervision.infrastructure.service.NettyConfig;
|
import io.netty.bootstrap.ServerBootstrap;
|
import io.netty.channel.Channel;
|
import io.netty.channel.ChannelInitializer;
|
import io.netty.channel.ChannelOption;
|
import io.netty.channel.socket.nio.NioServerSocketChannel;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.PreDestroy;
|
import java.util.Map;
|
import java.util.Set;
|
|
@Service("tcpServer")
|
public class NettyTCPServer extends AbstractNettyServer {
|
|
private static final Logger LOG = LoggerFactory.getLogger(NettyTCPServer.class);
|
|
private ServerBootstrap serverBootstrap;
|
|
@Autowired
|
public NettyTCPServer(NettyConfig nettyConfig,
|
ChannelInitializer<? extends Channel> channelInitializer) {
|
super(nettyConfig, channelInitializer);
|
}
|
|
@Override
|
public void startServer() throws Exception {
|
try {
|
serverBootstrap = new ServerBootstrap();
|
Map<ChannelOption<?>, Object> channelOptions = nettyConfig.getChannelOptions();
|
if (null != channelOptions) {
|
Set<ChannelOption<?>> keySet = channelOptions.keySet();
|
// 获取configuration配置到channelOption
|
for (ChannelOption option : keySet) {
|
serverBootstrap.option(option, channelOptions.get(option));
|
}
|
}
|
// reactor多线程模型，配置bossGroup和workGroup
|
// bossGroup和workGroup使用spring容器管理
|
serverBootstrap.group(getBossGroup(), getWorkerGroup())
|
.channel(NioServerSocketChannel.class)
|
.childHandler(getChannelInitializer());
|
// 绑定端口，启动并监听
|
Channel serverChannel = serverBootstrap.bind(nettyConfig.getSocketAddress()).sync()
|
.channel();
|
ALL_CHANNELS.add(serverChannel);
|
} catch (Exception e) {
|
LOG.error("TCP Server start error {}, going to shut down", e);
|
super.stopServer();
|
throw e;
|
}
|
}
|
|
@Override
|
@PreDestroy
|
public void stopServer() throws Exception {
|
super.stopServer();
|
}
|
|
@Override
|
public TransmissionProtocol getTransmissionProtocol() {
|
return TRANSMISSION_PROTOCOL.TCP;
|
}
|
|
// 配置自己的initializer
|
@Override
|
public void setChannelInitializer(ChannelInitializer<? extends Channel> initializer) {
|
this.channelInitializer = initializer;
|
serverBootstrap.childHandler(initializer);
|
}
|
|
@Override
|
public String toString() {
|
return "NettyTCPServer [socketAddress=" + nettyConfig.getSocketAddress()
|
+ ", portNumber=" + nettyConfig.getPortNumber() + "]";
|
}
|
|
}
|
