TryRexProSessioned.java

package com.tinkerpop.rexster.client;

import com.tinkerpop.rexster.Tokens;
import com.tinkerpop.rexster.protocol.msg.ScriptRequestMessage;
import com.tinkerpop.rexster.protocol.msg.ScriptResponseMessage;
import org.msgpack.MessagePack;

import java.io.IOException;
import java.util.Iterator;
import java.util.Map;
import java.util.UUID;
/**
 * A bit of an experiment.
 */
public class TryRexProSessioned {

    public static void main(String[] args) {
        //bigCalls();
        lotsOfCalls(false);
    }

    private static void bigCalls() {

        RemoteRexsterSession session = new RemoteRexsterSession("localhost", 8184, 100, "", "");
        session.open();

        long start = System.currentTimeMillis();

        long checkpoint = System.currentTimeMillis();

        try {
            ScriptResponseMessage resultMessage = (ScriptResponseMessage) session.sendRequest(
                    createScriptRequestMessage(session, "g=rexster.getGraph('gratefulgraph');g.V;"), 100);
            resultMessage = (ScriptResponseMessage) session.sendRequest(
                    createScriptRequestMessage(session, "g.E;"), 100);
            System.out.println((checkpoint - start) + ":" + (System.currentTimeMillis() - checkpoint));

        } catch (Exception ex) {
            ex.printStackTrace();
        }


    }

    private static void lotsOfCalls(boolean doJson){

        RemoteRexsterSession session = new RemoteRexsterSession("localhost", 8184, 100, "", "");
        session.open();

        MessagePack msgpack = new MessagePack();

        long start = System.currentTimeMillis();


        long checkpoint = System.currentTimeMillis();

        try {

            ScriptResponseMessage resultMessage = (ScriptResponseMessage) session.sendRequest(
                    createScriptRequestMessage(session, "g=rexster.getGraph('gratefulgraph');g.V;"), 100);

            int counter = 1;
            Iterator itty = ((Iterable) resultMessage.Results).iterator();
            while (itty.hasNext()){
                final Map<String,Object> map = (Map<String, Object>) itty.next();
                final String vId = (String) map.get(Tokens._ID);

                ScriptResponseMessage vertexResultMessage = (ScriptResponseMessage) session.sendRequest(
                        createScriptRequestMessage(session, "g.v(" + vId + ")"), 100);

                System.out.println(vertexResultMessage.Results);
                counter++;
            }

            /*
            resultMessage = (ScriptResponseMessage) session.sendRequest(
                    createScriptRequestMessage(session, "g.E;"), 100);

            unpacker = msgpack.createBufferUnpacker(resultMessage.Results);

            itty = unpacker.iterator();
            while (itty.hasNext()){
                final Map<String,Value> map = new Converter(msgpack, itty.next()).read(tMap(TString, TValue));
                final String eId = map.get(Tokens._ID).asRawValue().getString();

                ScriptResponseMessage edgeResultMessage = (ScriptResponseMessage) session.sendRequest(
                        createScriptRequestMessage(session, "g.e(" + eId + ")"), 100);

                unpacker = msgpack.createBufferUnpacker(edgeResultMessage.Results);
                System.out.println(unpacker.read(tMap(TString, TValue)));
            }
            */

            long end = System.currentTimeMillis() - checkpoint;
            System.out.println((checkpoint - start) + ":" + end);
            System.out.println(counter / (end / 1000));
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    private static ScriptRequestMessage createScriptRequestMessage(RemoteRexsterSession session, String script) throws IOException {
        ScriptRequestMessage scriptMessage = new ScriptRequestMessage();
        scriptMessage.setSessionAsUUID(session.getSessionKey());
        scriptMessage.Script = script;

        scriptMessage.LanguageName = "groovy";
        scriptMessage.metaSetInSession(true);
        scriptMessage.setRequestAsUUID(UUID.randomUUID());
        return scriptMessage;
    }
}