io.magician.tcp.codec.impl.http.routing.RoutingParsing Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Magician Show documentation
There is a newer version: 2.0.7
package io.magician.tcp.codec.impl.http.routing;

import io.magician.common.constant.CommonConstant;
import io.magician.common.util.ChannelUtil;
import io.magician.tcp.TCPServerConfig;
import io.magician.tcp.codec.impl.http.model.HttpHeaders;
import io.magician.tcp.codec.impl.http.request.MagicianHttpExchange;
import io.magician.tcp.codec.impl.websocket.handler.WebSocketHandler;
import io.magician.tcp.codec.impl.websocket.constant.WebSocketConstant;
import io.magician.tcp.codec.impl.http.constant.ReqMethod;
import io.magician.tcp.handler.MagicianHandler;

import java.util.Map;

/**
 * 路由解析，判断是http还是websocket
 */
public class RoutingParsing {

    /**
     * 配置类
     */
    private TCPServerConfig tcpServerConfig;

    /**
     * 路由跳转
     * 判断此次http请求 是否是为了建立websocket连接，之后调用对应的方法做接下去的业务逻辑
     */
    private RoutingJump routingJump;

    public RoutingParsing(TCPServerConfig tcpServerConfig){
        this.tcpServerConfig = tcpServerConfig;
        routingJump = new RoutingJump(tcpServerConfig);
    }

    /**
     * 根据判断结果进行不同的处理
     * @param httpExchange
     * @throws Exception
     */
    public void parsing(MagicianHttpExchange httpExchange) throws Exception {

        Map magicianHandlerMap = tcpServerConfig.getMagicianHandlerMap();
        Map webSocketHandlerMap = tcpServerConfig.getWebSocketHandlerMap();

        String uri = httpExchange.getRequestURI().toString();
        uri = getUri(uri);
        if(uri.equals(CommonConstant.FAVICON_ICO)){
            ChannelUtil.destroy(httpExchange);
            return;
        }

        /* 判断是否为webSocket */
        if(isWebSocket(httpExchange)){
            WebSocketHandler webSocketHandler = webSocketHandlerMap.get(uri);
            if(webSocketHandler != null){
                /* 如果是socket就建立连接 */
                routingJump.websocket(httpExchange, webSocketHandler);
                return;
            }
            throw new Exception("没有找到对应的websocketHandler，handler:[" + uri + "]");
        }

        /* 不是webSocket的话，就当http处理 */
        MagicianHandler rootServerHandler = magicianHandlerMap.get("/");
        MagicianHandler rouServerHandler = magicianHandlerMap.get(uri);
        if(rootServerHandler == null && rouServerHandler == null){
            throw new Exception("没有找到对应的httpHandler，handler:[" + uri + "]");
        }

        if(rootServerHandler != null){
            routingJump.http(httpExchange, rootServerHandler);
        }
        if(rouServerHandler != null){
            routingJump.http(httpExchange, rouServerHandler);
        }
    }

    /**
     * 判断是不是socket连接
     * @param httpExchange
     * @return
     */
    private boolean isWebSocket(MagicianHttpExchange httpExchange){
        if(!ReqMethod.GET.getCode().equals(httpExchange.getRequestMethod())){
            return false;
        }

        HttpHeaders httpHeaders = httpExchange.getRequestHeaders();
        String upgrade = httpHeaders.get(WebSocketConstant.UPGRADE);
        String connection = httpHeaders.get(WebSocketConstant.CONNECTION);
        String swKey = httpHeaders.get(WebSocketConstant.SEC_WEBSOCKET_KEY);
        if(upgrade == null || connection == null || swKey == null){
            return false;
        }

        if(!connection.toUpperCase().equals(WebSocketConstant.UPGRADE)){
            return false;
        }

        return true;
    }

    /**
     * 获取纯净的uri
     * @param uri
     * @return
     */
    private String getUri(String uri){
        if(!uri.startsWith("/")){
            uri = "/"+uri;
        }
        int endIndex = uri.length();
        if(uri.lastIndexOf("?") > -1){
            endIndex = uri.lastIndexOf("?");
        }
        return uri.substring(0, endIndex).toUpperCase();
    }
}