Vertx 实现webapi实战项目(四)

本节主要介绍使用消息解析和handler分发

一：定义一个常量类，储存消息id

public class HandlerCode {
    /***** 测试接口 ******/
    public static final short DAILY_LEVEL = 101;
}

 定义一个返回信息枚举类

public enum HttpStatus {
    OK(200, "success"),
    PLAYER_REGISTER(201,"head register"),
    REDIRECT_LOGIN(202,"remote device login"),

    ERROR(501, "error"),
    JSON_ERROR(502,"json error"),
    PARAMETER_ERROR(503,"parameter error");
    
    private final int code;
    private final String message;

    HttpStatus(int code, String message) {
        this.code = code;
        this.message = message;
    }

    public int code() {
        return code;
    }

    public String message() {
        return message;
    }
}

二：新建一个返回错误编码类

public class NetDownError extends AbstractDownMessage {
    public NetDownError(short requestId, HttpStatus status){
        this.messageId = requestId;
        this.resultCode = status.code();
    }

    @Override
    protected void encodeBody() {

    }
}

三：修改httpServerVerticle解析消息。红色为修改的地方。

public class HttpServerVerticle extends AbstractVerticle {
    private final Logger logger = LoggerFactory.getLogger(HttpServerVerticle.class);
    private HttpServer httpServer;

    private IMessageRecognizer recognizer;

    @Override
    public void start(Promise<Void> startPromise) throws Exception {
        recognizer = new MessageRecognizer();

        HttpServerOptions options = new HttpServerOptions()
                .setIdleTimeout(10000)
                .setIdleTimeoutUnit(TimeUnit.MILLISECONDS)
                .setTcpKeepAlive(true);

        httpServer = vertx.createHttpServer(options);
        httpServer.exceptionHandler( it -> {
            if(it instanceof IOException){
                logger.error("
 ---network io error:"+it.getMessage());
                return;
            }
            logger.error("
 ---net error:"+it.getMessage());
        });

//        httpServer.requestHandler(req -> {
//            HttpServerResponse resp = req.response().setStatusCode(200);
//            resp.putHeader("content-type", "application/json");
//            resp.putHeader("connection", "Keep-Alive");
//
//            JsonObject js = new JsonObject().put("ke","demos");
//            resp.end(js.encode());
//        });

        httpServer.requestHandler(this::handleRequest);
        httpServer.listen(8888,"127.0.0.1",res -> {
            if (res.succeeded()) {
                startPromise.complete();
                logger.info("HTTP server started on port 8888");
            } else {
                startPromise.fail(res.cause());
                logger.error(res.cause().getMessage());
            }
        });
    }

    private void handleRequest(HttpServerRequest request){
        request.bodyHandler(body -> {
            HttpServerResponse resp = request.response().setStatusCode(200);
            resp.putHeader("content-type", "application/json");
            resp.putHeader("connection", "Keep-Alive");

            //解析上传的json数据
            JsonObject upData = AbstractMessage.decodeUpMessage(body);
            if(upData == null){
                //上传json参数有误
                AbstractDownMessage output = new NetDownError((short)-1, HttpStatus.JSON_ERROR);
                output.encode();
                resp.end(output.SendMessage());

                logger.error("json error: 
" + body.toString());
                return;
            }

            AbstractDownMessage output;
            short messageId = upData.getInteger("mId",-1).shortValue();
            MessageFactory input = recognizer.recognize(messageId);
            if(input == null){
                output = new NetDownError((short)-1,HttpStatus.ERROR);
                output.encode();
                resp.end(output.SendMessage());
                return;
            }

            InterHandler handlerManager = HandlerManager.getInstance().getHandler(messageId);
            if(handlerManager == null){
                logger.error("not handler :"+messageId);
                output = new NetDownError(messageId,HttpStatus.ERROR);
                output.encode();
                resp.end(output.SendMessage());
                return;
            }

            input.decode(messageId,upData,request);
            handlerManager.handler((AbstractUpMessage)input,resp);
        });
    }

    @Override
    public void stop(){
        httpServer.close();
        logger.error(" AppLogin Server stop ------");
    }

四：因为还没有测试handler，所以启动服务器返回会返回code:501

项目结构：