package alice.datasegment;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SocketChannel;

import org.apache.log4j.Logger;
import org.msgpack.type.Value;

import alice.codesegment.CodeSegment;
import alice.daemon.Connection;
import alice.daemon.IncomingTcpConnection;
import alice.daemon.OutboundTcpConnection;

public class RemoteDataSegmentManager extends DataSegmentManager {
	
	Connection connection;
	Logger logger = Logger.getLogger(RemoteDataSegmentManager.class);
	
	// TODO: delete this constructor later
	@Deprecated
	public RemoteDataSegmentManager(Connection connection) {
		this.connection = connection;
		new IncomingTcpConnection(connection, this, "").start();
		new OutboundTcpConnection(connection).start();
		new Thread(replyThread, "RemoteDataSegmentManager-"
		+ connection.socket.getInetAddress().getHostName()
		+ ":" + connection.socket.getPort()).start();
	}
	
	public RemoteDataSegmentManager(String connectionKey, final String reverseKey, final String hostName, final int port) {
		connection = new Connection();
		final RemoteDataSegmentManager manager = this;
		new Thread(replyThread, "RemoteDataSegmentManager-" + connectionKey).start();
		new Thread("Connect-" + connectionKey) {
			public void run() {
				boolean connect = true;
				do {
					try {
						SocketChannel sc = SocketChannel.open(new InetSocketAddress(hostName, port));
						connection.socket = sc.socket();
						connect = false;
						logger.info("Connect to "
								+ connection.socket.getInetAddress().getHostName()
								+ ":" + connection.socket.getPort());
					} catch (IOException e) {
						try {
							Thread.sleep(500);
						} catch (InterruptedException e1) {
							e1.printStackTrace();
						}
					}
				} while (connect);
				new IncomingTcpConnection(connection, manager, reverseKey).start();
				new OutboundTcpConnection(connection).start();
			}
		}.start();
	}
	
	@Override
	public void put(String key, Value val) {
		connection.sendCommand(new Command(CommandType.PUT, null, key, val, 0, 0, null, null, null));
	}

	@Override
	public void update(String key, Value val) {
		connection.sendCommand(new Command(CommandType.UPDATE, null, key, val, 0, 0, null, null, null));
	}

	@Override
	public void take(DataSegmentReceiver receiver, String key, int index, CodeSegment cs) {
		int seq = this.seq.getAndIncrement();
		Command cmd = new Command(CommandType.TAKE, receiver, key, null, index, seq, replyQueue, cs, null);
		seqHash.put(seq, cmd);
		connection.sendCommand(cmd);		
	}

	@Override
	public void peek(DataSegmentReceiver receiver, String key, int index, CodeSegment cs) {
		int seq = this.seq.getAndIncrement();
		Command cmd = new Command(CommandType.PEEK, receiver, key, null, index, seq, replyQueue, cs, null);
		seqHash.put(seq, cmd);
		connection.sendCommand(cmd);
	}

	@Override
	public void remove(String key) {
		connection.sendCommand(new Command(CommandType.REMOVE, null, key, null, 0, 0, null, null, null));
	}

}