com.tinkerpop.rexster.protocol.serializer.msgpack.templates.BindingsTemplate Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rexster-protocol Show documentation

RexPro is a binary protocol for Rexster graph server.

There is a newer version: 2.6.0

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 {
    @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