view src/alice/daemon/OutboundTcpConnection.java @ 41:f9334781344a

add close api
author kazz <kazz@cr.ie.u-ryukyu.ac.jp>
date Thu, 02 Feb 2012 10:48:39 +0900
parents b5a21baf0b07
children 73158dc54c59
line wrap: on
line source

package alice.daemon;

import java.io.IOException;
import java.nio.ByteBuffer;

import org.msgpack.MessagePack;

import alice.datasegment.Command;
import alice.datasegment.CommandType;

public class OutboundTcpConnection extends Thread {
	
	public Connection connection;
	
	public OutboundTcpConnection(Connection connection) {
		this.connection = connection;
	}
	
	public CommandMessage convert(Command cmd) {
		return new CommandMessage(cmd.type.id, cmd.index, cmd.seq, cmd.key, cmd.val);
	}
	
	public void run() {
		MessagePack msgpack = new MessagePack();
		while (true) {
			try {
				Command cmd = connection.sendQueue.take();
				switch (cmd.type) {
				case CLOSE:
					connection.socket.close();
					return;
				case FINISH:
					System.exit(0);
					return;
				}
				CommandMessage cmdMsg = convert(cmd);
				byte[] buf = msgpack.write(cmdMsg);
				ByteBuffer buffer = ByteBuffer.allocateDirect(4 + buf.length);
				buffer.putInt(buf.length);
				buffer.put(buf);
				buffer.flip();
				connection.socket.getChannel().write(buffer);
			} catch (InterruptedException e) {
				e.printStackTrace();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
}