com.davfx.ninio.ping.util.PingClientCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ninio Show documentation
A Java NIO HTTP client/server as light as possible
The newest version!
package com.davfx.ninio.ping.util;

import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import com.davfx.ninio.common.Address;
import com.davfx.ninio.ping.PingClient;
import com.davfx.ninio.ping.PingClientConfigurator;
import com.davfx.ninio.ping.PingClientHandler;
import com.typesafe.config.Config;
import com.typesafe.config.ConfigFactory;

public final class PingClientCache implements AutoCloseable {
	private static final Config CONFIG = ConfigFactory.load();
	private static final int CACHE_EXPIRE_THRESHOLD = CONFIG.getInt("ping.cache.expire.threshold");

	private final PingClientConfigurator configurator;
	
	private static final class Hold {
		public final PingClient client;
		public final List handlers = new LinkedList<>();
		public PingClientHandler.Callback launchedCallback;
		public Hold(PingClient client) {
			this.client = client;
		}		
	}
	
	private final Map clients = new HashMap<>();

	public PingClientCache(PingClientConfigurator configurator) {
		this.configurator = configurator;
	}

	public static interface Connectable {
		void connect(PingClientHandler clientHandler);
	}
	
	public Connectable get(String host) {
		return get(new Address(host, PingClientConfigurator.DEFAULT_PORT));
		
	}
	public Connectable get(final Address address) {
		return new Connectable() {
			@Override
			public void connect(final PingClientHandler clientHandler) {
				if ((CACHE_EXPIRE_THRESHOLD > 0) && (clients.size() >= CACHE_EXPIRE_THRESHOLD)) {
					Iterator i = clients.values().iterator();
					while (i.hasNext()) {
						Hold c = i.next();
						if (c.handlers.isEmpty()) {
							if (c.launchedCallback != null) {
								c.launchedCallback.close();
							}
							i.remove();
							//%% c.client.close();
						}
					}
				}
				
				Hold c = clients.get(address);
				if (c == null) {
					c = new Hold(new PingClient(new PingClientConfigurator(configurator).withAddress(address)));
					
					final Hold cc = c;
					clients.put(address, cc);
					c.handlers.add(clientHandler);
					
					c.client.connect(new PingClientHandler() {
						@Override
						public void failed(IOException e) {
							clients.remove(address);
							for (PingClientHandler h : cc.handlers) {
								h.failed(e);
							}
							//%% cc.client.close();
						}
						@Override
						public void close() {
							clients.remove(address);
							for (PingClientHandler h : cc.handlers) {
								h.close();
							}
							//%% cc.client.close();
						}
						@Override
						public void launched(final Callback callback) {
							cc.launchedCallback = callback;
							for (PingClientHandler h : cc.handlers) {
								h.launched(new Callback() {
									@Override
									public void close() {
										// Connection never actually closed
										cc.handlers.remove(clientHandler);
									}
									@Override
									public void ping(String host, PingCallback pingCallback) {
										callback.ping(host, pingCallback);
									}
								});
							}
						}
					});
				} else {
					final Hold cc = c;
					
					cc.handlers.add(clientHandler);
					if (cc.launchedCallback != null) {
						clientHandler.launched(new PingClientHandler.Callback() {
							@Override
							public void close() {
								// Connection never actually closed
								cc.handlers.remove(clientHandler);
							}
							@Override
							public void ping(String host, PingCallback pingCallback) {
								cc.launchedCallback.ping(host, pingCallback);
							}
						});
					}
				}
			}
		};
	}
	
	@Override
	public void close() {
		for(Hold c : clients.values()) {
			if (c.launchedCallback != null) {
				c.launchedCallback.close();
			}
			//%% c.client.close();
		}
		clients.clear();
	}
}