SimpleMessageHandle.java

package org.codesharp.traffic;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Stack;

public class SimpleMessageHandle implements MessageHandle {
	public byte getCommand(Object msg) {
		return ((Message) msg).Command;
	}
	
	public byte getStatus(Object msg) {
		return ((Message) msg).Status;
	}
	
	public Object getDestination(Object msg) {
		return ((Message) msg).Destination;
	}
	
	public Object getNext(Object msg) {
		Stack<Object> path = ((Message) msg).Path;
		return path.isEmpty() ? null : path.pop();
	}
	
	public Object append(Object msg, Object from) {
		((Message) msg).Path.add(from);
		return msg;
	}
	
	public Object unknownDestination(Object msg) {
		// FIXME add status to message
		((Message) msg).Command = Commands.ACK;
		return msg;
	}
	
	public static class Message {
		public byte Command;
		public byte Status;
		public Object Destination;
		public Map<Object, Object> Headers = new HashMap<Object, Object>();
		public Stack<Object> Path = new Stack<Object>();
		public Object Body;
		
		@Override
		public String toString() {
			return String.format("%s|%s|%s|%s|%s|%s",
					this.Command,
					this.Status,
					this.Destination,
					this.Headers,
					Arrays.toString(this.Path.toArray()),
					this.Body);
		}
	}
}