com.firefly.net.tcp.aio.AsynchronousTcpClient 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.tcp.aio;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.StandardSocketOptions;
import java.nio.channels.AsynchronousChannelGroup;
import java.nio.channels.AsynchronousSocketChannel;
import java.nio.channels.CompletionHandler;
import java.util.concurrent.LinkedTransferQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

import com.firefly.net.Client;
import com.firefly.net.Config;
import com.firefly.net.Decoder;
import com.firefly.net.Encoder;
import com.firefly.net.EventManager;
import com.firefly.net.Handler;
import com.firefly.net.event.DefaultEventManager;
import com.firefly.utils.lang.AbstractLifeCycle;
import com.firefly.utils.log.Log;
import com.firefly.utils.log.LogFactory;
import com.firefly.utils.time.Millisecond100Clock;

public class AsynchronousTcpClient extends AbstractLifeCycle implements Client {
	
	private static Log log = LogFactory.getInstance().getLog("firefly-system");
    private Config config;
    private AtomicInteger sessionId = new AtomicInteger(0);
    private AsynchronousChannelGroup group;
    private AsynchronousTcpWorker worker;
    
    public AsynchronousTcpClient() {}
    
    public AsynchronousTcpClient(Config config) {
    	this.config = config;
    }
    
    public AsynchronousTcpClient(Decoder decoder, Encoder encoder, Handler handler) {
        config = new Config();
        config.setDecoder(decoder);
        config.setEncoder(encoder);
        config.setHandler(handler);
    }
    
    public AsynchronousTcpClient(Decoder decoder, Encoder encoder, Handler handler, int timeout) {
        config = new Config();
        config.setDecoder(decoder);
        config.setEncoder(encoder);
        config.setHandler(handler);
        config.setTimeout(timeout);
    }

	@Override
	public void setConfig(Config config) {
		this.config = config;
	}

	@Override
	public int connect(String host, int port) {
		int id = sessionId.getAndIncrement();
		connect(host, port, id);
		return id;
	}

	@Override
	public void connect(String host, int port, int id) {
		start();
		try {
			final AsynchronousSocketChannel socketChannel = AsynchronousSocketChannel.open(group);
			socketChannel.setOption(StandardSocketOptions.SO_REUSEADDR, true);
			socketChannel.setOption(StandardSocketOptions.SO_KEEPALIVE, true);
			socketChannel.setOption(StandardSocketOptions.TCP_NODELAY, false);
			socketChannel.connect(new InetSocketAddress(host, port), id, new CompletionHandler(){

				@Override
				public void completed(Void result, Integer sessionId) {
					worker.registerChannel(socketChannel, sessionId);
				}

				@Override
				public void failed(Throwable t, Integer sessionId) {
					log.error("session {} connect error", t, sessionId);
				}});
		} catch (IOException e) {
			log.error("client connect error", e);
		}
	}

	@Override
	protected void init() {
		try {
			group = AsynchronousChannelGroup.withThreadPool(new ThreadPoolExecutor(
					config.getAsynchronousCorePoolSize(),
					config.getAsynchronousMaximumPoolSize(), 
					config.getAsynchronousPoolKeepAliveTime(), 
					TimeUnit.MILLISECONDS, 
					new LinkedTransferQueue(),
					new ThreadFactory(){

						@Override
						public Thread newThread(Runnable r) {
							return new Thread(r, "firefly asynchronous client thread");
						}
					}));
			log.info(config.toString());
			EventManager eventManager = new DefaultEventManager(config);
			worker = new AsynchronousTcpWorker(config, eventManager);
		} catch (IOException e) {
			log.error("initialization client channel group error", e);
		}
	}

	@Override
	protected void destroy() {
		if(group != null) {
			group.shutdown();
		}
		LogFactory.getInstance().stop();
		Millisecond100Clock.stop();
	}

}