com.infomaximum.network.session.TransportSession 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
The newest version!
package com.infomaximum.network.session;

import com.infomaximum.network.event.NetworkListener;
import com.infomaximum.network.executerequest.ExecuteRequest;
import com.infomaximum.network.packet.IPacket;
import com.infomaximum.network.protocol.Protocol;
import com.infomaximum.network.struct.HandshakeData;
import com.infomaximum.network.struct.RemoteAddress;
import com.infomaximum.network.struct.UpgradeRequest;
import com.infomaximum.network.transport.Transport;
import com.infomaximum.network.transport.TransportPacketHandler;
import org.eclipse.jetty.websocket.common.WebSocketSession;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

public abstract class TransportSession implements TransportPacketHandler {

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

    protected final Protocol protocol;
    protected final Transport transport;
    protected final Object channel;
    private final UpgradeRequest upgradeRequest;
    protected final Session session;
    private final ExecuteRequest requestQueue;

    private volatile List listeners;

    public TransportSession(Protocol protocol, Transport transport, Object channel, UpgradeRequest upgradeRequest) {
        this.protocol = protocol;
        this.transport = transport;
        this.channel = channel;
        this.upgradeRequest = upgradeRequest;
        this.requestQueue = new ExecuteRequest(this);

        this.session = new SessionImpl(this);
    }

    public Thread.UncaughtExceptionHandler getUncaughtExceptionHandler(){
        return protocol.uncaughtExceptionHandler;
    }

    public abstract void completedPhaseHandshake(HandshakeData handshakeData);

    public abstract void failPhaseHandshake(IPacket responsePacket);

    @Override
    public Session getSession() {
        return session;
    }

    public UpgradeRequest getUpgradeRequest() {
        return upgradeRequest;
    }

    public void incomingMessage(String message) {
        requestQueue.incomingPacket(message);
    }

    @Override
    public void send(IPacket packet) {
        try {
            transport.send(channel, packet);
        } catch (Throwable e) {
            if (!(e instanceof IOException)) {
                log.error("Exception", e);
            }
            try {
                transport.close(channel);
            } catch (Throwable ignore) {
            }
            destroyed();
        }
    }

    public void destroyed() {
        if (listeners!=null) {
            for (NetworkListener listener: listeners) {
                listener.onDisconnect(session);
            }
        }
        try {
            transport.close(channel);
        } catch (Throwable ignore) {
        }
    }

    public void addListener(NetworkListener listener) {
        if (listeners==null) {
            synchronized (this) {
                if (listeners==null) {
                    this.listeners = new CopyOnWriteArrayList<>();
                }
            }
        }
        listeners.add(listener);
        listener.onConnect(session);
    }

    public void removeListener(NetworkListener listener) {
        listeners.remove(listener);
    }

    public RemoteAddress buildRemoteAddress() {
        WebSocketSession webSocketSession = (WebSocketSession) channel;
        String endRemoteAddress = webSocketSession.getUpgradeRequest().getHeader("X-Real-IP");

        InetSocketAddress remoteAddress = (InetSocketAddress)webSocketSession.getRemoteSocketAddress();
        String rawRemoteAddress = "unknown";
        if (remoteAddress != null) {
            InetAddress address = remoteAddress.getAddress();
            if (address != null) {
                rawRemoteAddress = address.toString().split("/")[1];
            }
        }

        if (endRemoteAddress == null) {
            endRemoteAddress = rawRemoteAddress;
        }
        return new RemoteAddress(rawRemoteAddress, endRemoteAddress);
    }

    public String getXTraceId() {
        String xTraceId = ((WebSocketSession) channel).getUpgradeRequest().getHeader("X-Trace-Id");
        return xTraceId;
    }
}