Haproxy太麻烦了，自己用netty写一个吧

简介

我们知道proxy 是提出的一个代理协议，通过这个协议,所有实现这个协议的proxy或者LBS,都可以附带真实客户端的IP地址和端口号，这使得proxy 在实际应用中非常有用。

这么优秀的协议，没有理由netty不支持。本文将会谈一下netty中对proxy 代理协议的支持。

netty对proxy 协议的支持

proxy 协议其实很简单，就是在请求前面带了proxy 信息。

在netty中这个信息叫做：

public final class HAProxyMessage extends AbstractReferenceCounted {
复制代码

是一个，这一点和很类似，说明保留着和很类似的特性。

根据proxy 协议，该协议可以分为两个版本，分别是v1和v2,其中v1版本是文本协议，而v2版本支持二进制的格式。

显然从代码编写和调试的角度来看v1更加友好，但是从程序的角度来看,v2可能性能更高。

中有个专门的on类，来表示proxy 的版本信息：

public enum HAProxyProtocolVersion {
    V1(VERSION_ONE_BYTE),
    V2(VERSION_TWO_BYTE);
复制代码

on是一个枚举类，在它里面定义了和proxy协议相对应的两个版本号。

在版本号之后是，在netty中用来表示：

public enum HAProxyCommand {
    LOCAL(HAProxyConstants.COMMAND_LOCAL_BYTE),
    PROXY(HAProxyConstants.COMMAND_PROXY_BYTE);
复制代码

也是一个枚举类，里面定义了两个的值，分别是local和proxy。

其中local表示该请求是代理服务器主动发起的，而不是客户端发起的，比如监控检测等请求。

proxy表示该请求是一个代理请求。

接下来是和,这两个字段用同一个byte来表示，所以这两个类都是ol的内部类。

先看下的定义：

    public enum AddressFamily {
        AF_UNSPEC(AF_UNSPEC_BYTE),
        AF_IPv4(AF_IPV4_BYTE),
        AF_IPv6(AF_IPV6_BYTE),
        AF_UNIX(AF_UNIX_BYTE);
复制代码

中定义了4个 类型，分别是,ipv4,ipv6和unix。分别对应未知,ipv4,ipv6和unix 。

再看下的定义：

    public enum TransportProtocol {
        UNSPEC(TRANSPORT_UNSPEC_BYTE),
        STREAM(TRANSPORT_STREAM_BYTE),
        DGRAM(TRANSPORT_DGRAM_BYTE);
复制代码

有3个值，分别是,和dgram。分别对应未知协议，http/https协议，udp/tcp协议。

因为和实际上是同一个byte，所以经过组合之后可以得到下面的几个枚举值：

    UNKNOWN(TPAF_UNKNOWN_BYTE, AddressFamily.AF_UNSPEC, TransportProtocol.UNSPEC),
    TCP4(TPAF_TCP4_BYTE, AddressFamily.AF_IPv4, TransportProtocol.STREAM),
    TCP6(TPAF_TCP6_BYTE, AddressFamily.AF_IPv6, TransportProtocol.STREAM),
    UDP4(TPAF_UDP4_BYTE, AddressFamily.AF_IPv4, TransportProtocol.DGRAM),
    UDP6(TPAF_UDP6_BYTE, AddressFamily.AF_IPv6, TransportProtocol.DGRAM),
    UNIX_STREAM(TPAF_UNIX_STREAM_BYTE, AddressFamily.AF_UNIX, TransportProtocol.STREAM),
    UNIX_DGRAM(TPAF_UNIX_DGRAM_BYTE, AddressFamily.AF_UNIX, TransportProtocol.DGRAM);
复制代码

以上的枚举值也是ol中定义的值。

接下来就是源ip地址，目标的ip地址，源端口和目标端口这几个值，定义为属性表示如下：

    private final String sourceAddress;
    private final String destinationAddress;
    private final int sourcePort;
    private final int destinationPort;
复制代码

最后，proxy 中还可以包含额外的字段tlv，tlv在netty中也是一种，使用表示：

public class HAProxyTLV extends DefaultByteBufHolder 
复制代码

因为tlv是key value结构，所以看下的构造函数：

    public HAProxyTLV(Type type, ByteBuf content) {
        this(type, Type.byteValueForType(type), content);
    }
复制代码

接受一个type和的value。

Type是一个枚举类，在netty中可以支持下面的值：

    public enum Type {
        PP2_TYPE_ALPN,
        PP2_TYPE_AUTHORITY,
        PP2_TYPE_SSL,
        PP2_TYPE_SSL_VERSION,
        PP2_TYPE_SSL_CN,
        PP2_TYPE_NETNS,
        OTHER;
复制代码

在中，tlv是一个list来保存的：

private final List tlvs;
复制代码

到此，所有所需要的参数都齐了，我们看下的构造函数：

    public HAProxyMessage(
            HAProxyProtocolVersion protocolVersion, HAProxyCommand command, HAProxyProxiedProtocol proxiedProtocol,
            String sourceAddress, String destinationAddress, int sourcePort, int destinationPort,
            List tlvs)
复制代码

会将所有的参数都存储到本地的变量中，供后续使用。

因为proxy 有两个版本，v1和v2，所以中提供了两个将编码为对象的方法，分别是：

static HAProxyMessage decodeHeader(ByteBuf header) 
复制代码

和：

static HAProxyMessage decodeHeader(String header)
复制代码

有了proxy 的java表示之后，我们再来看一下的编码解码器。

的编码解码器

netty对对象的支持表现在两个地方，netty提供了两个类分别对进行编码和解码，这两个类是r和r。

先看一下r:

public final class HAProxyMessageEncoder extends MessageToByteEncoder 
复制代码

r继承自，传入的泛型是，表示是将编码成为。

它的方法很简单，根据传入的版本信息，分别进行编码：

    protected void encode(ChannelHandlerContext ctx, HAProxyMessage msg, ByteBuf out) throws Exception {
        switch (msg.protocolVersion()) {
            case V1:
                encodeV1(msg, out);
                break;
            case V2:
                encodeV2(msg, out);
                break;
            default:
                throw new HAProxyProtocolException("Unsupported version: " + msg.protocolVersion());
        }
    }
复制代码

r是跟r相反的动作，是将接收到的解析成为：

public class HAProxyMessageDecoder extends ByteToMessageDecoder 
复制代码

因为有两个版本，那么怎么判断接收到的是哪个版本呢？

其实很简单，因为v1版本和v2版本的开始字符是不一样的，v1版本的开头是一个text:”PROXY”, v2版本的开头是一个固定的二进制串,如下所示：

    static final byte[] BINARY_PREFIX = {
            (byte) 0x0D,
            (byte) 0x0A,
            (byte) 0x0D,
            (byte) 0x0A,
            (byte) 0x00,
            (byte) 0x0D,
            (byte) 0x0A,
            (byte) 0x51,
            (byte) 0x55,
            (byte) 0x49,
            (byte) 0x54,
            (byte) 0x0A
    };
    static final byte[] TEXT_PREFIX = {
            (byte) 'P',
            (byte) 'R',
            (byte) 'O',
            (byte) 'X',
            (byte) 'Y',
    };
复制代码

看下它的方法实现：

    protected final void decode(ChannelHandlerContext ctx, ByteBuf in, List