BindingsTemplate.java

package com.tinkerpop.rexster.protocol.serializer.msgpack.templates;

import com.tinkerpop.rexster.protocol.msg.RexProBindings;
import org.msgpack.packer.Packer;
import org.msgpack.template.AbstractTemplate;
import org.msgpack.template.Templates;
import org.msgpack.unpacker.Unpacker;

import java.io.IOException;
import java.util.Map;

/**
 * @author Blake Eggleston (bdeggleston.github.com)
 */
public class BindingsTemplate extends AbstractTemplate<RexProBindings> {
    @Override
    public void write(Packer pk, RexProBindings v) throws IOException {
        write(pk, v, false);
    }

    public void write(final Packer pk, final RexProBindings v, final boolean required) throws IOException {
        RexProBindings bindings = v;
        if (bindings == null) {
            bindings = new RexProBindings();
        }
        pk.writeMapBegin(bindings.size());
        for (Map.Entry pair : bindings.entrySet()) {
            pk.write(pair.getKey());
            try {
                ResultsConverter.serializeObject(pair.getValue(), pk);
            } catch (Exception ex) {
                throw new IOException(ex.toString());
            }
        }
        pk.writeMapEnd();
    }

    @Override
    public RexProBindings read(Unpacker u, RexProBindings to) throws IOException {
        return read(u, to, false);
    }

    public RexProBindings read(final Unpacker u, final RexProBindings to, final boolean required) throws IOException {
        if (!required && u.trySkipNil()) {
            return null;
        }

        RexProBindings bindings;
        if (to != null) {
            bindings = to;
            bindings.clear();
        } else {
            bindings = new RexProBindings();
        }

        int n = u.readMapBegin();
        for (int i=0; i<n; i++) {
            final String key = u.read(Templates.TString);
            final Object val = ResultsConverter.deserializeObject(u.read(Templates.TValue));
            bindings.put(key, val);
        }
        u.readMapEnd();

        return bindings;
    }

    public static BindingsTemplate instance = new BindingsTemplate();
    static public BindingsTemplate getInstance() {
        return instance;
    }
}