com.github.thorbenkuck.network.connection.BlockingConnection Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of easy-net Show documentation

An EventStream driven, simple to use Client/Server framework

The newest version!

package com.github.thorbenkuck.network.connection;

import com.github.thorbenkuck.network.ThreadPools;
import com.github.thorbenkuck.network.utils.PropertyUtils;

import java.io.IOException;
import java.net.Socket;
import java.net.SocketAddress;

class BlockingConnection extends AbstractConnection {

	private final ReadingService readingService = new ReadingService();
	private final Thread thread = ThreadPools.newDaemonThread(readingService, "Blocking Connection Listener");
    private final boolean daemon = PropertyUtils.daemonWorkerThreads();
	private final DataConnection dataConnection;
	private final Socket socket;

	BlockingConnection(Socket socket) throws IOException {
		dataConnection = DataConnection.wrap(socket);
		setProtocol(new SizeFirstProtocol());
		this.socket = socket;
		socket.setKeepAlive(true);
		pipeInputStreams();
	}

	@Override
	public DataConnection getDataConnection() {
		return dataConnection;
	}

	@Override
	public void listen() {
		if (readingService.running) {
			return;
		}
        thread.setDaemon(daemon);
		thread.setName("TCP Listener");
		thread.start();
	}

	@Override
	public void closeSilently() {
		try {
			super.close();
		} catch (IOException ignored) {
		}
		readingService.running = false;
		dataConnection.closeSilent();
	}

	@Override
	public void close() throws IOException {
		super.close();
		readingService.running = false;
		dataConnection.close();
	}

	@Override
	public SocketAddress localAddress() {
		return socket.getLocalSocketAddress();
	}

	@Override
	public SocketAddress remoteAddress() {
		return socket.getRemoteSocketAddress();
	}

	@Override
	public boolean isOpen() {
		return socket.isConnected();
	}

	@Override
	public String toString() {
		return "TCPConnection{" +
				"readingService=" + readingService +
				'}';
	}

	private final class ReadingService implements Runnable {

		private boolean running = false;

		@Override
		public void run() {
			running = true;
			while (socket.isConnected() && running) {
				try {
					byte[] data = readFromProtocol();
					received(data);
				} catch (IOException e) {
					// EOF reached. No print needed
					running = false;
					acceptError(e);
					closeSilently();
				} catch (IllegalStateException ignored) {
				} catch (Throwable throwable) {
					acceptError(throwable);
				}
			}

			running = false;
			triggerDisconnectEvent();
		}

		@Override
		public String toString() {
			return "ReadingService {" +
					"running=" + running +
					'}';
		}
	}
}