diametric 0.1.1-java

data/ext/diametric/DiametricListenableFuture.java

@@ -0,0 +1,47 @@
+package diametric;
+
+import org.jruby.Ruby;
+import org.jruby.RubyClass;
+import org.jruby.RubyObject;
+import org.jruby.anno.JRubyClass;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.javasupport.JavaUtil;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+import datomic.ListenableFuture;
+
+@JRubyClass(name = "Diametric::Persistence::ListenableFuture")
+public class DiametricListenableFuture extends RubyObject {
+    private static final long serialVersionUID = 2083281771243513904L;
+    // future is supposed to datomic.ListenableFuture type;
+    private Object future = null;
+
+    public DiametricListenableFuture(Ruby runtime, RubyClass klazz) {
+        super(runtime, klazz);
+    }
+    
+    void init(Object future) {
+        this.future = future;
+    }
+    
+    @JRubyMethod
+    public IRubyObject to_java(ThreadContext context) {
+        return JavaUtil.convertJavaToUsableRubyObject(context.getRuntime(), future);
+    }
+    
+    @JRubyMethod
+    public IRubyObject get(ThreadContext context) {
+        Ruby runtime = context.getRuntime();
+        if (future == null) return runtime.getNil();
+        try {
+            Object result = ((ListenableFuture)future).get();
+            RubyClass clazz = (RubyClass)runtime.getClassFromPath("Diametric::Persistence::Object");
+            DiametricObject diametric_object = (DiametricObject)clazz.allocate();
+            diametric_object.update(result);
+            return diametric_object;
+        } catch (Throwable t) {
+            throw runtime.newRuntimeError(t.getMessage());
+        }
+    }
+}

data/ext/diametric/DiametricObject.java

@@ -0,0 +1,66 @@
+package diametric;
+
+import org.jruby.Ruby;
+import org.jruby.RubyClass;
+import org.jruby.RubyObject;
+import org.jruby.anno.JRubyClass;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.javasupport.JavaUtil;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+@JRubyClass(name = "Diametric::Persistence::Object")
+public class DiametricObject extends RubyObject {
+    private static final long serialVersionUID = -4198258841171995687L;
+    private Object java_object = null;
+
+    public DiametricObject(Ruby runtime, RubyClass klazz) {
+        super(runtime, klazz);
+    }
+    
+    void update(Object java_object) {
+        this.java_object = java_object;
+    }
+    
+    @Override
+    public Object clone() throws CloneNotSupportedException {
+        return super.clone();
+    }
+    
+    Object toJava() {
+        return java_object;
+    }
+    
+    @JRubyMethod(name="new", meta=true)
+    public static IRubyObject rbNew(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        RubyClass clazz = (RubyClass)context.getRuntime().getClassFromPath("Diametric::Persistence::Object");
+        DiametricObject diametric_object = (DiametricObject)clazz.allocate();
+        diametric_object.update(DiametricUtils.convertRubyToJava(context, arg));
+        return diametric_object;
+    }
+
+    @JRubyMethod
+    public IRubyObject to_java(ThreadContext context) {
+        return JavaUtil.convertJavaToUsableRubyObject(context.getRuntime(), java_object);
+    }
+    
+    @JRubyMethod(name="==", required=1)
+    public IRubyObject op_equal(ThreadContext context, IRubyObject arg) {
+        Ruby runtime = context.getRuntime();
+        if (arg instanceof DiametricObject) {
+            DiametricObject other = (DiametricObject)arg;
+            if (java_object.toString().equals(other.toJava().toString())) {
+                return runtime.getTrue();
+            } else {
+                return runtime.getFalse();
+            }
+        } else {
+            return runtime.getFalse();
+        }
+    }
+
+    @JRubyMethod
+    public IRubyObject to_s(ThreadContext context) {
+        return context.getRuntime().newString(java_object.toString());
+    }
+}

data/ext/diametric/DiametricPeer.java

@@ -0,0 +1,414 @@
+package diametric;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+import org.jruby.Ruby;
+import org.jruby.RubyArray;
+import org.jruby.RubyBoolean;
+import org.jruby.RubyClass;
+import org.jruby.RubyFixnum;
+import org.jruby.RubyHash;
+import org.jruby.RubyModule;
+import org.jruby.RubyString;
+import org.jruby.RubySymbol;
+import org.jruby.anno.JRubyMethod;
+import org.jruby.anno.JRubyModule;
+import org.jruby.runtime.Block;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+import clojure.lang.PersistentHashSet;
+import datomic.Connection;
+import datomic.Peer;
+
+@JRubyModule(name="Diametric::Persistence::Peer")
+public class DiametricPeer extends RubyModule {
+    private static final long serialVersionUID = 8659857729004427581L;
+    
+    protected DiametricPeer(Ruby runtime) {
+        super(runtime);
+    }
+    
+    private static DiametricConnection saved_connection = null;
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject connect(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        String uriOrMap = null;
+        if (arg instanceof RubyString) {
+            uriOrMap = DiametricUtils.rubyStringToJava(arg);
+        } else if (arg instanceof RubyHash) {
+            RubySymbol key = RubySymbol.newSymbol(context.getRuntime(), "uri");
+            RubyString value = (RubyString)((RubyHash)arg).op_aref(context, key);
+            uriOrMap = DiametricUtils.rubyStringToJava(value);
+        } else {
+            throw context.getRuntime().newArgumentError("Argument should be a String or Hash");
+        }
+        if (uriOrMap == null )
+            throw context.getRuntime().newArgumentError("Argument should be a String or Hash with :uri key");
+        
+        RubyClass clazz = (RubyClass) context.getRuntime().getClassFromPath("Diametric::Persistence::Connection");
+        DiametricConnection rubyConnection = (DiametricConnection)clazz.allocate();
+        try {
+            // what value will be returned when connect fails? API doc doesn't tell anything.
+            Connection connection = (Connection) clojure.lang.RT.var("datomic.api", "connect").invoke(uriOrMap);
+            rubyConnection.init(connection);
+            saved_connection = rubyConnection;
+            return rubyConnection;
+        } catch (Exception e) {
+            // Diametric doesn't require creating database before connect.
+            if (e.getMessage().contains(":peer/db-not-found") && (Boolean)clojure.lang.RT.var("datomic.api", "create-database").invoke(uriOrMap)) {
+                Connection connection = (Connection) clojure.lang.RT.var("datomic.api", "connect").invoke(uriOrMap);
+                rubyConnection.init(connection);
+                saved_connection = rubyConnection;
+                return rubyConnection;
+            }
+        }
+        throw context.getRuntime().newRuntimeError("Failed to create connection");
+    }
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject create_database(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        String uriOrMap = DiametricUtils.rubyStringToJava(arg);
+        if (uriOrMap == null)
+            throw context.getRuntime().newArgumentError("Argument should be a String");
+        try {
+            boolean status = (Boolean)clojure.lang.RT.var("datomic.api", "create-database").invoke(uriOrMap);
+            return RubyBoolean.newBoolean(context.getRuntime(), status);
+        } catch (Exception e) {
+            throw context.getRuntime().newRuntimeError("Datomic Error: " + e.getMessage());
+        }
+    }
+    
+    @JRubyMethod(meta=true, required=2, rest=true)
+    public static IRubyObject rename_database(ThreadContext context, IRubyObject klazz, IRubyObject[] args) {
+        if (args.length != 2) return context.getRuntime().getNil();
+        String uriOrMap = DiametricUtils.rubyStringToJava(args[0]);
+        if (uriOrMap == null) return context.getRuntime().getNil();
+        String newName = DiametricUtils.rubyStringToJava(args[1]);
+        if (newName == null) return context.getRuntime().getNil();
+        try {
+            boolean status = (Boolean)clojure.lang.RT.var("datomic.api", "rename-database").invoke(uriOrMap, newName);
+            return RubyBoolean.newBoolean(context.getRuntime(), status);
+        } catch (Exception e) {
+            throw context.getRuntime().newRuntimeError("Datomic Error: " + e.getMessage());
+        }
+    }
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject delete_database(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        String uriOrMap = DiametricUtils.rubyStringToJava(arg);
+        if (uriOrMap == null) return context.getRuntime().getNil();
+        try {
+            boolean status = (Boolean)clojure.lang.RT.var("datomic.api", "delete-database").invoke(uriOrMap);
+            return RubyBoolean.newBoolean(context.getRuntime(), status);
+        } catch (Exception e) {
+            throw context.getRuntime().newRuntimeError("Datomic Error: " + e.getMessage());
+        }
+    }
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject shutdown(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        if (!(arg instanceof RubyBoolean)) {
+            throw context.getRuntime().newArgumentError("Wrong argument type.");
+        }
+        Boolean shutdownClojure = (Boolean) arg.toJava(Boolean.class);
+        try {
+            clojure.lang.RT.var("datomic.api", "shutdown").invoke(shutdownClojure);
+        } catch (Exception e) {
+            throw context.getRuntime().newRuntimeError("Datomic Error: " + e.getMessage());
+        }
+        return context.getRuntime().getNil();
+    }
+
+    /**
+     * Constructs a semi-sequential UUID useful for creating UUIDs that don't fragment indexes
+     * 
+     * @param context
+     * @param klazz
+     * @return java.util.UUID. a UUID whose most significant 32 bits are currentTimeMillis rounded to seconds
+     */
+    @JRubyMethod(meta=true)
+    public static IRubyObject squuid(ThreadContext context, IRubyObject klazz) {
+        RubyClass clazz = (RubyClass) context.getRuntime().getClassFromPath("Diametric::Persistence::UUID");
+        diametric.DiametricUUID ruby_uuid = (diametric.DiametricUUID)clazz.allocate();
+        try {
+            java.util.UUID java_uuid = (UUID) clojure.lang.RT.var("datomic.api", "squuid").invoke();
+            ruby_uuid.init(java_uuid);
+            return ruby_uuid;
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError("Datomic Exception: " + t.getMessage());
+        }
+    }
+    
+    /**
+     * Gets the time part of a squuid
+     * 
+     * @param context
+     * @param klazz
+     * @param arg diametric.UUID. squuid -  a UUID created by squuid()
+     * @return the time in the format of System.currentTimeMillis
+     */
+    @JRubyMethod(meta=true)
+    public static IRubyObject squuid_time_millis(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        if (!(arg instanceof diametric.DiametricUUID)) {
+            throw context.getRuntime().newArgumentError("Wrong argument type.");
+        }
+        java.util.UUID squuid = ((diametric.DiametricUUID)arg).getUUID();
+        if (squuid == null) return context.getRuntime().getNil();
+        long value;
+        try {
+            value = (Long) clojure.lang.RT.var("datomic.api", "squuid-time-millis").invoke(squuid);
+            return RubyFixnum.newFixnum(context.getRuntime(), value);
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError("Datomic Exception: " + t.getMessage());
+        }
+    }
+    
+    /**
+     * Generates a temp id in the designated partition
+     * In case the second argument is given,
+     * it should be an idNumber from -1 (inclusive) to -1000000 (exclusive).
+     * 
+     * @param context
+     * @param klazz
+     * @param args the first argument: String. a partition, which is a keyword identifying the partition.
+     * @return
+     */
+    @JRubyMethod(meta=true, required=1, optional=1)
+    public static IRubyObject tempid(ThreadContext context, IRubyObject klazz, IRubyObject[] args) {
+        if (args.length < 1 || args.length > 2) {
+            throw context.getRuntime().newArgumentError("Wrong number of arguments");
+        }
+        String partition = DiametricUtils.rubyStringToJava(args[0]);
+        RubyClass clazz = (RubyClass)context.getRuntime().getClassFromPath("Diametric::Persistence::Object");
+        DiametricObject diametric_object = (DiametricObject)clazz.allocate();
+        try {
+            clojure.lang.Var clj_var = clojure.lang.RT.var("datomic.api", "tempid");
+            if (args.length > 1 && (args[1] instanceof RubyFixnum)) {
+                long idNumber = (Long) args[1].toJava(Long.class);
+                diametric_object.update(clj_var.invoke(partition, idNumber));
+            } else {
+                diametric_object.update(clj_var.invoke(partition));
+            }
+            return diametric_object;
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError(t.getMessage());
+        }
+    }
+    
+    /**
+     * 
+     * @param context
+     * @param klazz
+     * @param args Both 2 arguments should be DiametricObject.
+     *             The first argument should hold clojure.lang.PersistentArrayMap.
+     *             The second one should hold datomic.db.DbId.
+     * @return
+     */
+    @JRubyMethod(meta=true, required=2, rest=true)
+    public static IRubyObject resolve_tempid(ThreadContext context, IRubyObject klazz, IRubyObject[] args) {
+        if (args.length != 2) {
+            throw context.getRuntime().newArgumentError("Wrong number of arguments");
+        }
+        Map map;
+        DiametricObject ruby_object;
+        if ((args[0] instanceof DiametricObject) && (args[1] instanceof DiametricObject)) {
+            map = (Map) ((DiametricObject)args[0]).toJava();
+            ruby_object = ((DiametricObject)args[1]);
+        } else {
+            throw context.getRuntime().newArgumentError("Wrong argument type.");
+        }
+        try {
+            Object dbid = clojure.lang.RT.var("datomic.api", "resolve-tempid")
+                            .invoke(map.get(Connection.DB_AFTER), map.get(Connection.TEMPIDS), ruby_object.toJava());
+            ruby_object.update(dbid);
+            return ruby_object;
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError("Datomic Exception: " + t.getMessage());
+        }
+    }
+    
+    @JRubyMethod(meta=true, required=2, rest=true)
+    public static IRubyObject q(ThreadContext context, IRubyObject klazz, IRubyObject[] args) {
+        Ruby runtime = context.getRuntime();
+        if (args.length < 2) {
+            throw runtime.newArgumentError("Wrong number of arguments");
+        }
+        if (!(args[0] instanceof RubyString)) {
+            throw runtime.newArgumentError("The first arg should be a query string");
+        }
+        if (!(args[1] instanceof DiametricDatabase)) {
+            throw runtime.newArgumentError("The second arg should be a database.");
+        }
+        String query = (String)args[0].toJava(String.class);
+        Object database = ((DiametricDatabase)args[1]).toJava();
+
+        Collection<List<Object>> results = null;
+        try {
+            if (args.length == 2) {
+                results = (Collection<List<Object>>) clojure.lang.RT.var("datomic.api", "q").invoke(query, database);
+            } else if ((args.length == 3) && (args[2] instanceof RubyArray)) {
+                RubyArray ruby_inputs = (RubyArray)args[2];
+                if (ruby_inputs.getLength() == 0) {
+                    results = Peer.q(query, database);
+                } else {
+                    Object[] inputs = new Object[ruby_inputs.getLength()];
+                    for (int i=0; i<inputs.length; i++) {
+                        inputs[i] = DiametricUtils.convertRubyToJava(context, ruby_inputs.shift(context));
+                    }
+                    //System.out.println("OH INPUTS ARE: ");
+                    //for (int i=0; i<inputs.length; i++) {
+                    //    System.out.println("OH: " + inputs[i]);
+                    //}
+                    results = (Collection<List<Object>>) clojure.lang.RT.var("datomic.api", "q").invoke(query, database, inputs);
+                }
+            } else {
+                Object[] inputs = new Object[args.length-2];
+                for (int i=0; i<inputs.length; i++) {
+                    inputs[i] = DiametricUtils.convertRubyToJava(context, args[i+2]);
+                }
+                results = (Collection<List<Object>>) clojure.lang.RT.var("datomic.api", "q").invoke(query, database, inputs);
+            }
+        } catch (Throwable t) {
+            throw runtime.newRuntimeError("Datomic Exception: " + t.getMessage());
+        }
+
+        if (results == null) return context.getRuntime().getNil();
+        RubyArray ruby_results = RubyArray.newArray(context.getRuntime());
+        for (List list : results) {
+            RubyArray ruby_elements = RubyArray.newArray(context.getRuntime());
+            for (Object element : list) {
+                //System.out.println("OH ELEMENT IS: " + element + " [" + element.getClass().getCanonicalName() +"]");
+                ruby_elements.append(DiametricUtils.convertJavaToRuby(context, element));
+            }
+            ruby_results.append(ruby_elements);
+        }
+        return ruby_results;
+    }
+    
+    private static List<RubyModule> bases = new ArrayList<RubyModule>();
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject included(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        Ruby runtime = context.getRuntime();
+        if (arg instanceof RubyModule) {
+            RubyModule base = (RubyModule)arg;
+            bases.add(base);
+            base.instance_variable_set(RubyString.newString(context.getRuntime(), "@peer"), runtime.getTrue());
+            IRubyObject common = runtime.getClassFromPath("Diametric::Persistence::Common");
+            base.send(context, RubySymbol.newSymbol(runtime, "include"), common, Block.NULL_BLOCK);
+            IRubyObject classmethods = runtime.getClassFromPath("Diametric::Persistence::Peer::ClassMethods");
+            base.send(context, RubySymbol.newSymbol(runtime, "extend"), classmethods, Block.NULL_BLOCK);
+        }
+        return runtime.getNil();
+    }
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject connect(ThreadContext context, IRubyObject klazz) {
+        if (saved_connection == null) return context.getRuntime().getNil();
+        return saved_connection;
+    }
+    
+    @JRubyMethod(meta=true)
+    public static IRubyObject create_schemas(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        if (!(arg instanceof DiametricConnection))
+            throw context.getRuntime().newArgumentError("Argument should be Connection.");
+        IRubyObject result = context.getRuntime().getNil();
+        for (RubyModule base : bases) {
+            if (base.respondsTo("schema")) {
+                IRubyObject schema = base.send(context, RubySymbol.newSymbol(context.getRuntime(), "schema"), Block.NULL_BLOCK);
+                result = ((DiametricConnection)arg).transact(context, schema);
+            }
+        }
+        return result;
+    }
+
+    @JRubyMethod(meta=true)
+    public static IRubyObject transact(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        return saved_connection.transact(context, arg);
+    }
+
+    @JRubyMethod(meta=true)
+    public static IRubyObject get(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        Ruby runtime = context.getRuntime();
+        Object dbid = null;
+        if ((arg instanceof DiametricObject) && (((DiametricObject)arg).to_java(context) instanceof RubyFixnum)) {
+            dbid = ((DiametricObject)arg).toJava();
+        } else if (arg instanceof RubyFixnum) {
+            dbid = ((RubyFixnum)arg).toJava(Long.class);
+        } else {
+            throw runtime.newArgumentError("Argument should be dbid");
+        }
+        if (saved_connection == null) throw runtime.newRuntimeError("Connection is not established");
+        try {
+            Object database = clojure.lang.RT.var("datomic.api", "db").invoke(saved_connection.toJava());
+            Object entity = clojure.lang.RT.var("datomic.api", "entity").invoke(database, dbid);
+            RubyClass clazz = (RubyClass) context.getRuntime().getClassFromPath("Diametric::Persistence::Entity");
+            DiametricEntity ruby_entity = (DiametricEntity)clazz.allocate();
+            ruby_entity.init(entity);
+            return ruby_entity;
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError(t.getMessage());
+        }
+    }
+
+    @JRubyMethod(meta=true)
+    public static IRubyObject retract_entity(ThreadContext context, IRubyObject klazz, IRubyObject arg) {
+        Object dbid = DiametricUtils.convertRubyToJava(context, arg);
+        List query = datomic.Util.list((datomic.Util.list(":db.fn/retractEntity", dbid)));
+        try {
+            clojure.lang.RT.var("datomic.api", "transact-async").invoke(saved_connection.toJava(), query);
+        } catch (Throwable t) {
+            throw context.getRuntime().newRuntimeError("Datomic error: " + t.getMessage());
+        }
+        return context.getRuntime().getNil();
+    }
+
+    /**
+     * 
+     * @param context
+     * @param klazz
+     * @param args database, dbid, query
+     * @return
+     */
+    @JRubyMethod(meta=true, required=3, rest=true)
+    public static IRubyObject reverse_q(ThreadContext context, IRubyObject klazz, IRubyObject[] args) {
+        Ruby runtime = context.getRuntime();
+        if (args[0] instanceof DiametricDatabase &&
+                (args[1] instanceof DiametricObject || args[1] instanceof RubyFixnum) &&
+                args[2] instanceof RubyString) {
+            Object database = ((DiametricDatabase)args[0]).toJava();
+            Long dbid = (Long)DiametricUtils.convertRubyToJava(context, args[1]);
+            String query_string = (String)args[2].toJava(String.class);
+            try {
+                Object entity = clojure.lang.RT.var("datomic.api", "entity").invoke(database, dbid);
+                clojure.lang.PersistentHashSet set =
+                         (PersistentHashSet) clojure.lang.RT.var("clojure.core", "get").invoke(entity, query_string);
+
+                if (set == null) {
+                    return RubyArray.newEmptyArray(runtime);
+                }
+
+                RubyArray array = RubyArray.newArray(runtime, set.size());
+                Iterator iter = set.iterator();
+                while (iter.hasNext()) {
+                    Object e = iter.next();
+                    RubyClass clazz = (RubyClass) context.getRuntime().getClassFromPath("Diametric::Persistence::Entity");
+                    DiametricEntity ruby_entity = (DiametricEntity)clazz.allocate();
+                    ruby_entity.init(e);
+                    array.append(ruby_entity);
+                }
+                return array;
+            } catch (Throwable t) {
+                throw runtime.newRuntimeError("Datomic Error: " + t.getMessage());
+            }
+        } else {
+            throw runtime.newArgumentError("Arguments should be 'database, dbid, query_string'");
+        }
+    }
+}