com.firefly.net.support.wrap.client.SimpleTcpClient Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of firefly-nettool Show documentation

There is a newer version: 5.0.0-dev6

package com.firefly.net.support.wrap.client;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicInteger;

import com.firefly.net.Client;
import com.firefly.net.Decoder;
import com.firefly.net.Encoder;
import com.firefly.net.tcp.aio.AsynchronousTcpClient;
import com.firefly.utils.concurrent.FuturePromise;

public class SimpleTcpClient {
	private String host;
	private int port;
	private SimpleTcpClientHandler handler;

	private Map connectionInfo = new ConcurrentHashMap();
	private Client client;
	private AtomicInteger sessionId = new AtomicInteger(0);

	public SimpleTcpClient(String host, int port, Decoder decoder, Encoder encoder, SimpleTcpClientHandler handler) {
		this.host = host;
		this.port = port;
		this.handler = handler;
		this.handler.setConnectionInfo(connectionInfo);
		this.client = new AsynchronousTcpClient(decoder, encoder, handler);
	}

	public Future connect() {
		return connect(0);
	}
	
	public Future connect(final long timeout) {
		final long t = timeout <= 0 ? 5000L : timeout;
		final FuturePromise future = new FuturePromise<>();
		
		connect(t, new SessionOpenedCallback(){

			@Override
			public void sessionOpened(TcpConnection connection) {
				future.succeeded(connection);
			}
		});
		return future;
	}
	
	public void connect(long timeout, SessionOpenedCallback callback) {
		int id = sessionId.getAndIncrement();
		ConnectionInfo info = new ConnectionInfo();
		info.callback = callback;
		info.timeout = timeout;
		connectionInfo.put(id, info);
		client.connect(host, port, id);
	}

	public void shutdown() {
		client.stop();
	}
}