package com.zy.core.netty;
|
|
import com.core.common.SnowflakeIdWorker;
|
import com.zy.core.netty.cache.ChannelAttrKey;
|
import com.zy.core.netty.handle.PackageServerHandler;
|
import com.zy.core.netty.handle.ProtectorHandler;
|
import com.zy.core.netty.handle.ProtocolDecoder;
|
import com.zy.core.netty.handle.ProtocolEncoder;
|
import com.zy.core.netty.properties.TcpProperties;
|
import io.netty.channel.Channel;
|
import io.netty.channel.ChannelHandler;
|
import io.netty.channel.ChannelInitializer;
|
import io.netty.handler.timeout.IdleStateHandler;
|
import io.netty.util.Attribute;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.util.HashMap;
|
import java.util.Map;
|
|
|
/**
|
* handler管道
|
* 控制所有netty handler流向
|
* 待完成: 动态管理handler
|
* Created by vincent on 2019-04-02
|
*/
|
@Component
|
@ChannelHandler.Sharable
|
public class HandlerInitializer extends ChannelInitializer<Channel> {
|
|
@Autowired
|
private SnowflakeIdWorker snowflakeIdWorker;
|
@Autowired
|
private TcpProperties tcpProperties;
|
@Autowired
|
private ProtocolEncoder protocolEncoder;
|
@Autowired
|
private PackageServerHandler packageServerHandler;
|
@Autowired
|
private ProtectorHandler protectorHandler;
|
|
/**
|
* Set some channel handlers on channel pipeline
|
*/
|
@Override
|
protected void initChannel(Channel channel) {
|
channel.pipeline()
|
// 心跳
|
.addLast(new IdleStateHandler(tcpProperties.getHeartSeconds(), 0, 0))
|
// 编码器
|
.addLast(protocolEncoder)
|
// 解码器
|
.addLast(new ProtocolDecoder(snowflakeIdWorker))
|
// 校验码处理器
|
// .addLast(validateHandler)
|
// 认证处理器
|
// .addLast(vehAuthHandler)
|
// 业务处理器
|
.addLast(packageServerHandler)
|
// 通道保护器
|
.addLast(protectorHandler);
|
|
// Channel局部变量，相当于线程的ThreadLocal
|
// initAttrTrack(channel);
|
}
|
|
/**
|
* Init channel attr track
|
*/
|
private void initAttrTrack(Channel channel) {
|
Attribute<Map<String, Object>> coolTrackAttr = channel.attr(ChannelAttrKey.DATA_MAP_ATTR);
|
Map<String, Object> trackMap = new HashMap<>();
|
coolTrackAttr.setIfAbsent(trackMap);
|
}
|
|
}
|
