All Downloads are FREE. Search and download functionalities are using the official Maven repository.

xin.alum.aim.socks.SocksServerHandler Maven / Gradle / Ivy

There is a newer version: 1.9.6
Show newest version
/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */
package xin.alum.aim.socks;

import com.google.protobuf.MessageLite;
import io.netty.buffer.AbstractReferenceCountedByteBuf;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.websocketx.BinaryWebSocketFrame;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import org.springframework.scheduling.annotation.Async;
import xin.alum.aim.AIM;
import xin.alum.aim.constant.AIMConstant;
import xin.alum.aim.handler.BaseServerHandler;
import xin.alum.aim.model.Reply;
import xin.alum.aim.model.Sent;

@ChannelHandler.Sharable
public class SocksServerHandler extends BaseServerHandler {

    @Override
    public void channelRead0(ChannelHandlerContext ctx, Object msg) {
        if (msg instanceof Sent) {
            switch (((Sent) msg).getKey()) {
                case AIMConstant.KEY_PING:
                    super.send(ctx.channel(), Reply.instance(AIMConstant.KEY_PONG));
                    break;
                default:
                    AIM.request.onRecive(ctx.channel(), (Sent) msg);
                    break;
            }
        } else if (msg instanceof AbstractReferenceCountedByteBuf) {
            AIM.request.onByte(ctx.channel(), (ByteBuf) msg);
        } else if (msg instanceof String) {
            AIM.request.onText(ctx.channel(), (String) msg);
        } else {
            logger.error("{} 未知消息类型", ctx.channel());
        }
    }

    @Async
    @Override
    public ChannelFuture send(Channel ch, Object msg) {
        //JSON-String
        if (msg == AIMConstant.KEY_PONG || msg == AIMConstant.KEY_PING) {
            msg = Reply.instance(msg.toString());
        } else if (msg instanceof String) {
            msg = new TextWebSocketFrame(((String) msg).concat(AIMConstant.TEXT_FRAME_Delimiters));
        } else //Proto
            if (msg instanceof MessageLite) {
                msg = new BinaryWebSocketFrame(Unpooled.wrappedBuffer(((MessageLite) msg).toByteArray()));
            } else if (msg instanceof MessageLite.Builder) {
                msg = new BinaryWebSocketFrame(Unpooled.wrappedBuffer(((MessageLite.Builder) msg).build().toByteArray()));
            }
        return super.send(ch, msg);
    }

}