com.github.linyuzai.connection.loadbalance.netty.concept.NettyLoadBalanceHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of concept-connection-loadbalance-netty Show documentation

Load Balance 4 Netty

There is a newer version: 2.7.1

Show newest version

package com.github.linyuzai.connection.loadbalance.netty.concept;

import com.github.linyuzai.connection.loadbalance.core.concept.Connection;
import com.github.linyuzai.connection.loadbalance.core.extension.GroupSelector;
import io.netty.channel.*;
import io.netty.util.ReferenceCountUtil;
import lombok.Getter;
import lombok.RequiredArgsConstructor;

import java.util.HashMap;
import java.util.Map;

/**
 * Netty 负载均衡处理器。
 * 
 * Netty load balance handler.
 */
@Getter
@RequiredArgsConstructor
public class NettyLoadBalanceHandler extends ChannelInboundHandlerAdapter {

    private final NettyLoadBalanceConcept concept;

    private final Map metadata;

    public NettyLoadBalanceHandler(NettyLoadBalanceConcept concept) {
        this(concept, (Map) null);
    }

    public NettyLoadBalanceHandler(NettyLoadBalanceConcept concept, String group) {
        this(concept, getGroupMetadata(group));
    }

    private static Map getGroupMetadata(String group) {
        Map map = new HashMap<>();
        map.put(GroupSelector.KEY, group);
        return map;
    }

    protected Connection create(ChannelHandlerContext ctx, Map metadata) {
        return concept.createConnection(ctx, metadata);
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        concept.onEstablish(create(ctx, metadata));
        ctx.fireChannelActive();
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        concept.onClose(ctx.channel().id(), Connection.Type.CLIENT, "Inactive");
        ctx.fireChannelInactive();
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        concept.onError(ctx.channel().id(), Connection.Type.CLIENT, cause);
        ctx.fireExceptionCaught(cause);
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        concept.onMessage(ctx.channel().id(), Connection.Type.CLIENT, msg);
        ReferenceCountUtil.release(msg);
        //ctx.fireChannelRead(msg);
    }
}