io.github.icodegarden.commons.nio.java.ClientNioSelector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of commons-nio-java Show documentation
The newest version!
package io.github.icodegarden.commons.nio.java;

import java.io.Closeable;
import java.io.IOException;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.util.Iterator;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.github.icodegarden.commons.lang.exception.remote.ExceedExpectedRemoteException;
import io.github.icodegarden.commons.lang.util.SystemUtils;

/**
 * 
 * @author Fangfang.Xu
 *
 */
public class ClientNioSelector implements Closeable {
	private static final Logger log = LoggerFactory.getLogger(ClientNioSelector.class);

	/**
	 * 
	 * @param name
	 * @return
	 * @throws IOException
	 */
	public static ClientNioSelector openNew(String name) throws ExceedExpectedRemoteException {
		ClientNioSelector clientNioSelector = new ClientNioSelector(name);
		try {
			clientNioSelector.start();
		} catch (IOException e) {
			throw new ExceedExpectedRemoteException(e);
		}
		return clientNioSelector;
	}

	private volatile boolean closed = true;
	private final String name;
	private Selector selector;

	private ClientNioSelector(String name) {
		this.name = name;
	}

	public void registerRead(ClientNioEventListener nioEventListener) throws ClosedChannelException {
		nioEventListener.socketChannel().register(selector, SelectionKey.OP_READ, nioEventListener);
	}

	private void start() throws IOException {
		selector = Selector.open();
		closed = false;

		new Thread("NioClient-Selector-Thread-" + name) {
			public void run() {
				while (!closed) {
					try {
						int count;
						if (SystemUtils.isWindowsPlatform()) {
							count = selector.select(100);
							/**
							 * 在windows下不加这个会有问题
							 */
							System.out.print("");// slow
						} else {
							count = selector.select(100);
						}

						if (count > 0) {
							Set selectedKeys = selector.selectedKeys();
							Iterator iterator = selectedKeys.iterator();
							while (iterator.hasNext()) {
								SelectionKey key = iterator.next();
								try {
									if (key.isReadable()) {
										ClientNioEventListener nioEventListener = (ClientNioEventListener) key
												.attachment();
										nioEventListener.onRead(key);
									} else if (key.isWritable()) {
										//
									}
								} catch (Throwable e) {
									log.error("ex on SelectionKey handle", e);
								} finally {
									iterator.remove();
								}
							}
						}
					} catch (Throwable e) {
						log.error("ex on Client Nio Select loop handle", e);
					}
				}
			};
		}.start();
	}

	@Override
	public void close() throws IOException {
		closed = true;
		selector.wakeup();
	}
}