com.infomaximum.network.transport.http.PacketWebSocketHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of network Show documentation

Library for creating a light http server

There is a newer version: 1.1.3p12

package com.infomaximum.network.transport.http;

import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.annotations.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.InetSocketAddress;

/**
 * Created by kris on 08.04.17.
 */
@WebSocket
public class PacketWebSocketHandler {

    private final static Logger log = LoggerFactory.getLogger(PacketWebSocketHandler.class);

    @OnWebSocketConnect
    public void afterConnectionEstablished(Session session) throws Exception {

        //Определяеи ip, проверяя загаловки возможно балансировщик добавит данные с реальным ip
        String remoteIpAddress = null;
        if (session.getUpgradeRequest().getHeaders() != null) {
            remoteIpAddress = session.getUpgradeRequest().getHeader("X-Real-IP");
        }

        if (remoteIpAddress == null) {
            remoteIpAddress = ((InetSocketAddress) session.getRemoteAddress()).getAddress().getHostAddress();
        }

        HttpTransport.instance.fireConnect(session, remoteIpAddress);//Оповещаем о новом подключении
    }

    @OnWebSocketMessage
    public void onMessage(Session session, String message) {
        HttpTransport.instance.fireIncomingMessage(session, message);
    }

    @OnWebSocketError
    public void handleTransportError(Session session, Throwable throwable) throws Exception {
        session.close();
        HttpTransport.instance.fireDisconnect(session, -1, throwable);//Соединение разрывается-оповещаем
    }

    @OnWebSocketClose
    public void afterConnectionClosed(Session session, int statusCode, String reason) throws Exception {
        HttpTransport.instance.fireDisconnect(session, statusCode, null);//Соединение разрывается-оповещаем
    }
}