package alice.daemon;

import java.io.EOFException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.ClosedChannelException;

import org.msgpack.unpacker.Unpacker;

import alice.codesegment.SingletonMessage;
import alice.datasegment.Command;
import alice.datasegment.CommandType;
import alice.datasegment.DataSegment;
import alice.datasegment.DataSegmentManager;
import alice.topology.manager.keeparive.RespondData;

public class IncomingUdpConnection extends IncomingTcpConnection {
	// receive Data set into LocalDataSegment now but need to set into MulticastDataSegment.
	// and this implement has problem. If over 65507 data receive, can not read.
    // but  Max data length is 65507 because of the max length of UDP payload
	
	public MulticastConnection receiver;
	public MulticastConnection sender;

	public IncomingUdpConnection(MulticastConnection s, MulticastConnection r, DataSegmentManager manager) {
		super(manager);
		receiver = r;
		sender = s;
	}
	
	@Override
	public void run() {
		while (true){
			try {
			    // Max data length is 65507 because of the max length of UDP payload
				ByteBuffer receive = ByteBuffer.allocate(65507); 
				receiver.receive(receive);
				Unpacker unpacker = SingletonMessage.getInstance().createBufferUnpacker(receive);
				receive.flip();
				CommandMessage msg = unpacker.read(CommandMessage.class);
				byte[] val = unpacker.readByteArray();
				CommandType type = CommandType.getCommandTypeFromId(msg.type);
				switch (type) {
				case UPDATE:
					getLocalDataSegmentManager().getDataSegmentKey(msg.key)
						.runCommand(new Command(type, null, null, val, 0, 0, null, null, reverseKey));
					break;
				case PUT:
					getLocalDataSegmentManager().getDataSegmentKey(msg.key)
						.runCommand(new Command(type, null, null, val, 0, 0, null, null, reverseKey));
					break;
				case PEEK:
					getLocalDataSegmentManager().getDataSegmentKey(msg.key)
						.runCommand(new Command(type, null, null, null, msg.index, msg.seq, sender, null, null, msg.flag));
					break;
				case TAKE:
					getLocalDataSegmentManager().getDataSegmentKey(msg.key)
						.runCommand(new Command(type, null, null, null, msg.index, msg.seq, sender, null, null, msg.flag));
					break;	
				case REMOVE:
					getLocalDataSegmentManager().getDataSegmentKey(msg.key)
						.runCommand(new Command(type, null, null, null, 0, 0, null, null, null));
					break;
				case REPLY:
					Command cmd = manager.getAndRemoveCmd(msg.seq);
					cmd.cs.ids.reply(cmd.receiver, new Command(type, null, null, val, msg.index, msg.seq, null, null, null));
					cmd=null;
					break;
				case PING:
					DataSegment.get(reverseKey).response(msg.key);
					break;
				case RESPONSE:
					DataSegment.getLocal().put(msg.key, new RespondData(reverseKey, System.currentTimeMillis()));
					break;
				default:
					break;
				}
				
			} catch (ClosedChannelException e) {
				return;
			} catch (EOFException e) {
				return;
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}

}