servereventhandler.java

来自「基于java nio实现的http服务器」· Java 代码 · 共 52 行

package org.sse.server;

import java.io.*;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import org.sse.http.ServletContext;

/**
 * @author Taylor Cowan
 */
public class ServerEventHandler {

	private ByteBuffer byteBuffer = ByteBuffer.allocateDirect(1024);
	private ServletContext context;
	private Queue q;
	private int socketConnects = 0;
	
	public ServerEventHandler(String root, int workerThreads) throws IOException {
		q = new Queue();
		context = new ServletContext(new File(root));
		for (int i = 0; i < workerThreads; i++)
			new RequestHandlerThread(context,q).start();			
	}

	public void disconnectClient(SelectionKey key) throws IOException {
		key.attach(null);
		key.channel().close();
	}

	public void acceptNewClient(Selector selector, SelectionKey key)
		throws IOException, ClosedChannelException {
		ServerSocketChannel server = (ServerSocketChannel) key.channel();
		SocketChannel channel = server.accept();
		channel.configureBlocking(false);
		SelectionKey readKey = channel.register(selector, SelectionKey.OP_READ);
		readKey.attach(new Client(readKey, q));
	}

	public void readDataFromSocket(SelectionKey key) throws IOException {
		int count = ((SocketChannel)key.channel()).read(byteBuffer);
		if ( count > 0) {
			byteBuffer.flip();
			byte[] data = new byte[count];
			byteBuffer.get(data, 0, count);
			((Client)key.attachment()).write(data);
		} else if ( count < 0) {
			key.channel().close();
		}
		byteBuffer.clear();
	}
}