yangui28 / nettyserver Goto Github PK

public void messageReceived(ChannelHandlerContext ctx, Object msg) {
        if (msg instanceof FullHttpRequest) {//如果是HTTP请求，进行HTTP操作
            handleHttpRequest(ctx, (FullHttpRequest) msg);
        } else if (msg instanceof WebSocketFrame) {//如果是Websocket请求，则进行websocket操作
            handleWebSocketFrame(ctx, (WebSocketFrame) msg);
        }
    }

private void handleHttpRequest(ChannelHandlerContext ctx, FullHttpRequest req) {
        logger.warn("uri:" + req.uri());
        if (req.uri().startsWith("/ws/join")) {//如果urL开头为/ws/join则升级为websocket
            mac = wsBeforeHandler(ctx, req);
            if (mac == null || mac.length() < 1) {
                RespTools.paraErrorBack(ctx,req,null);
                return;
            }
            WebSocketServerHandshakerFactory wsFactory = new WebSocketServerHandshakerFactory(
                    getWebSocketLocation(req), null, true);
            handshaker = wsFactory.newHandshaker(req);
            if (handshaker == null) {
                WebSocketServerHandshakerFactory.sendUnsupportedVersionResponse(ctx.channel());
            } else {
                handshaker.handshake(ctx.channel(), req);
            }
        } else {//是HTTP请求则路由到Action
            RouteResult<Action> routeResult = rs.getRouter().route(req.method(), req.uri());
            Action action = routeResult.target();
            action.act(ctx, req);
        }
    }

private void handleWebSocketFrame(ChannelHandlerContext ctx, WebSocketFrame frame) {

        // Check for closing frame
        if (frame instanceof CloseWebSocketFrame) {
            handshaker.close(ctx.channel(), (CloseWebSocketFrame) frame.retain());
            return;
        }
        if (frame instanceof PingWebSocketFrame) {
            ctx.write(new PongWebSocketFrame(frame.content().retain()));
            return;
        }
        if (frame instanceof TextWebSocketFrame) {
            devicePool.join(ctx.channel(), mac);
            String json = ((TextWebSocketFrame) frame).text();
            Logic.ReqRespType data= JsonTools.read(json,Logic.ReqRespType.class);
           //...
            return;
        }
    }

public class RouterSetting {
    @Autowired
    private Router<Action> router;
    @Autowired
    private GetVerifyCodeAct getVerifyCodeAct;//w
    @Autowired
    private LoginAct loginAct;
    @Autowired
    private RegisterAct registerAcc;

    public Router<Action> getRouter() {
        routerConfig(this.router);
        return this.router;
    }

    private void routerConfig(Router<Action> r) {
        r.POST("api/get_verify_code", getVerifyCodeAct);
        r.ANY("api/login", loginAct);
        r.GET("api/register", registerAcc);
    }
}

@Controller
public class LoginAct implements Action {
    private static final Logger logger = LoggerFactory.getLogger(LoginAct.class);
    @Override
    public void act(ChannelHandlerContext ctx, FullHttpRequest req) {
        String ip = HttpTools.getIp(req);
        String body = Convert.buf2Str(req.content());
        Get.Login get = JsonTools.read(body, Get.Login.class);//1.得到HTTP传来的json数据解析为javabean
        Sub.Register back;//构建返回给客户端的javabean的实例
        //...
        HttpTools.sendCorrectResp(ctx, req, back);//返回给客户端HTTP Response
    }
}

private void addSession(long userId, String ip) {
        Logic.DeviceSession session = new Logic.DeviceSession(ip, "");
        Cache.add(userId + "", session, "6mn");//设置session的缓存时间为6分钟
        //debugSession(userId);
    }