org.snapscript.agent.event.MapMarshaller Maven / Gradle / Ivy

Go to download

package org.snapscript.agent.event;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

public class MapMarshaller {
   
   public Map> readMap(DataInput input) throws IOException {
      Map> map = new TreeMap>();
      int count = input.readInt();
      
      for(int i = 0; i < count; i++) {
         Map criteria = new HashMap();
         String name = input.readUTF();
         int size = input.readInt();
         
         for(int j = 0; j < size; j++) {
            String key = input.readUTF();
            String value = input.readUTF();
            
            criteria.put(key, value);
         }
         map.put(name, criteria);
      }
      return map;
   }
   
   public void writeMap(DataOutput output, Map> map) throws IOException {
      Set names = map.keySet();
      int count = map.size();
      
      output.writeInt(count);
      
      for(String name : names) {
         Map criteria = map.get(name);
         Set keys = criteria.keySet();
         int size = criteria.size();
         
         output.writeUTF(name);
         output.writeInt(size);
         
         for(String key : keys) {
            String value = criteria.get(key);
            
            output.writeUTF(key);
            output.writeUTF(value);
         }
      }
   }
}