com.tinkerpop.rexster.protocol.msg.RexProBindings 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.msg;

import com.tinkerpop.rexster.protocol.serializer.msgpack.templates.ResultsConverter;
import org.msgpack.packer.Packer;
import org.msgpack.template.AbstractTemplate;
import org.msgpack.template.Templates;
import org.msgpack.unpacker.Unpacker;

import javax.script.SimpleBindings;
import java.io.IOException;
import java.util.Map;

/**
 * HashMap with a msgpack template
 *
 * @author Blake Eggleston (bdeggleston.github.com)
 */
public class RexProBindings extends SimpleBindings {
    public RexProBindings(Map stringObjectMap) {
        super(stringObjectMap);
    }

    public RexProBindings() {
    }

    //TODO: delete
    public static class SerializationTemplate 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