summaryrefslogtreecommitdiff
path: root/java-plan2/src/org/msgpack/schema/MapSchema.java
blob: e72cf635834d3ad71258fc9140f2332c019eddd7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package org.msgpack.schema;

import java.util.List;
import java.util.Map;
import java.util.HashMap;
import java.io.IOException;
import org.msgpack.*;

public class MapSchema extends Schema {
	private Schema keyType;
	private Schema valueType;

	public MapSchema(Schema keyType, Schema valueType)
	{
		super("map");
		this.keyType = keyType;
		this.valueType = valueType;
	}

	public Schema getKeyType()
	{
		return keyType;
	}

	public Schema getValueType()
	{
		return valueType;
	}

	@Override
	public String getFullName()
	{
		return "HashList<"+keyType.getFullName()+", "+valueType.getFullName()+">";
	}

	@Override
	public String getExpression()
	{
		return "(map "+keyType.getExpression()+" "+valueType.getExpression()+")";
	}

	@Override
	@SuppressWarnings("unchecked")
	public void pack(Packer pk, Object obj) throws IOException
	{
		if(obj == null) {
			pk.packNil();
			return;
		}
		Map<Object,Object> d = (Map<Object,Object>)obj;
		pk.packMap(d.size());
		for(Map.Entry<Object,Object> e : d.entrySet()) {
			keyType.pack(pk, e.getKey());
			valueType.pack(pk, e.getValue());
		}
	}

	@Override
	@SuppressWarnings("unchecked")
	public Object convert(GenericObject obj)
	{
		Map<GenericObject,GenericObject> d = obj.asMap();
		Map<Object,Object> g = new HashMap<Object,Object>();
		for(Map.Entry<GenericObject,GenericObject> e : d.entrySet()) {
			g.put(keyType.convert(e.getKey()), valueType.convert(e.getValue()));
		}
		return g;
	}
}