gremlin-orm
gremlin-orm is an ORM for graph databases in Node.js. Currently working on Neo4j and Microsoft Azure Cosmos DB with more to come in the future.
Installation
$ npm install --save @rrnara/gremlin-orm
Example
const genericPool = require('generic-pool');
const GremlinOrm = require('gremlin-orm');
const Gremlin = require('gremlin');
const Action = require('gremlin-orm/Action');
// const g = new gremlinOrm('neo4j'); // connects to localhost:8182 by default
const dbLogger = (msg) => console.log(msg);
const dbErrorLogger = (msg) => console.error(msg);
const factory = {
create: function() {
const user = '/dbs/--TESTDB--/colls/--TESTGRAPH--';
const primaryKey = '--TEST-PRIMARY-KEY--';
const authenticator = new Gremlin.driver.auth.PlainTextSaslAuthenticator(user, primaryKey);
const options = {
authenticator,
traversalsource: 'g',
rejectUnauthorized: true,
mimeType: 'application/vnd.gremlin-v2.0+json'
};
const client = new Gremlin.driver.Client('wss://--TEST-COSMOSDB--.gremlin.cosmos.azure.com:443/', options);
client.addListener('log', dbLogger);
client.addListener('socketError', dbErrorLogger);
return client;
},
destroy: function(client) {
client.close();
}
};
const connectionPool = genericPool.createPool(factory, { min: 0, max: 10, evictionRunIntervalMillis: 120000, idleTimeoutMillis: 230000 });
connectionPool.on('factoryCreateError', dbErrorLogger);
connectionPool.on('factoryDestroyError', dbErrorLogger);
const g = new GremlinOrm(['azure', 'pk', 'test-partition'], connectionPool, true); // Can use dbLogger as parameter instead of true
const Person = g.define(
'person',
{
name: {
type: g.STRING,
required: true
},
age: {
type: g.NUMBER
}
},
{
getDetails: function(suffix) { return `${this.name} ${suffix}, Age ${this.age}`; }
}
);
Person.create(req.body, (error, result) => {
if (error) {
res.send(error);
}
else {
console.log(result.invoke('getDetails', 'Jr'));
res.send(result);
// result formatted as nice JSON Object
/*
{
"id": "1",
"label": "person",
"name": "Bob",
"age": 20
}
*/
}
});
Documentation
Initialization
Initialize the gremlin-orm instance with parameters matching the gremlin-javascript createClient()
initialization - with the addition of the dialect argument.
Arguments
-
dialect
(string or Array): Required argument that takes string ('neo4j'
) or array (['azure', '<partitionName>', '<defaultPartitionValue>']
). -
client
: Gremlin Client
Methods
Defining Models
- define - alias for defineVertex
- defineVertex - define a new Vertex model
- defineEdge - define a new Edge model
Generic Methods
- query - run a Gremlin query string on a Model
- queryRaw - perform a raw query on the gremlin-orm root and return raw data
- update - update specific props on an existing vertex or edge
- delete - delete an existing vertex or edge
- order - order the results by property and asc/desc
- limit - limit the number of results returned
- range - Array range of results returned
- invoke - Invoke methods defined along with schema
Vertex Methods
- create - create a new vertex
- find - find first vertex with matching properties
- findAll - find all vertices with matching properties
- findActionAll - find all vertices when particular action is applied on the properties
- createEdge - define a new edge relationship to another vertex(es)
- findEdge - find edges directly connected to the relevant vertex(es)
- findRelated - find all vertices connected to initial vertex(es) through a type of edge with optional properties
-
findImplicit - find all vertices which have the same edge relations
in
that the current vertex(es) hasout
to another vertex
Edge Methods
- create - create new edge relationship(s) by passing in two vertices or sets of vertices
- find - find first edge with matching properties
- findAll - find all edges with matching properties
- findVertex - find all vertices that are connected by the relevant edges
Action Methods
Refer to Tinkerpop documentation Only custom query option is Or - to perform Or condition queries
Method Chaining
In order to avoid sacrificing the power of Gremlin traversals, method calls in this ORM can take advantage of method chaining. Any read-only method will avoid running its database query and instead pass its Gremlin query string to the next method in the chain if it is not given a callback. Note: All create, update, and delete methods require a callback and can not have more methods chained after.
Example
// Only makes one call to the database
Person.find({'name': 'John'}).findRelated('knows', {'since': '2015'}, (error, result) => {
// Send people John knows to client
})
Additionally, results returned in the form of JSON objects will retain their relevant model methods for additional queries.
// Makes two calls to the database
Person.find({'name': 'John'}), (error, result) => {
let john = result;
john.findRelated('knows', {'since': '2015'}, (error, result) => {
// Send people John knows to client
})
})
Defining Models
This ORM utilizes Model definitions similar to Sequelize to add structure to developing servers around graph databases. Queries outside of the constraints of pre-defined models can be run using the generic .query
or .queryRaw
.
define(label, schema)
.define
is an alias for defineVertex
defineVertex(label, schema)
.defineVertex
defines a new instance of the VertexModel
class - see generic and vertex model methods
Arguments
-
label
: Label to be used on all vertices of this model -
schema
: A schema object which defines allowed property keys and allowed values/types for each key -
methods
: An object which defines methods that can be called on instance of model
Example
const Person = g.define(
'person',
{
name: {
type: g.STRING,
required: true
},
age: {
type: g.NUMBER
}
},
{
getDetails: function (suffix) { return `${this.name} ${suffix}, Age ${this.age}`; }
}
);
defineEdge(label, schema)
.defineEdge
defines a new instance of the EdgeModel
class - see generic and edge model methods
Arguments
-
label
: Label to be used on all edges of this model -
schema
: A schema object which defines allowed property keys and allowed values/types for each key -
methods
: An object which defines methods that can be called on instance of model
Example
const Knows = g.defineEdge(
'knows',
{
from: {
type: g.STRING
},
since: {
type: g.DATE
}
},
{
relationshipDetails: function () { return `Since ${this.since}`; }
}
);
Model Data types
The following options are available when defining model schemas:
-
type
: Use Sequelize-like constants to define data types. Date properties will be returned as javascript Date objects unless returning raw data. The following data type constants are currently available with possibly more in the future.g.STRING
g.NUMBER
g.DATE
g.BOOLEAN
-
required
(default = false): If true, will not allow saving to database if not present or empty
Generic Methods
query(queryString, [raw, callback])
.query
takes a raw Gremlin query string and runs it on the object it is called on.
Arguments
-
queryString
: Gremlin query as a string -
raw
(optional, default = false): If true, will return the raw data from the graph database instead of normally formatted JSON -
callback
(optional, required if raw is true): Some callback function with (error, result) arguments.
Returns
- If callback is given, returns array where 0th index is array of Vertex results and 1th index is array of Edge results (even if either is empty) -- this helps expose the correct model methods if the query returns edges from a query on a vertex or vis versa.
Example
let query = ".as('a').out('created').as('b').in('created').as('c').dedup('a','b').select('a','b','c')"
Person.find({'name': 'John'}).query(query, true, (error, result) => {
// send raw data to client
});
queryRaw(queryString, callback)
.queryRaw
performs a raw query on the gremlin-orm root and returns raw data
Arguments
-
queryString
: Gremlin query as a string -
callback
: Some callback function with (error, result) arguments
Example
// query must be a full Gremlin query string
let query = "g.V(1).as('a').out('created').as('b').in('created').as('c').dedup('a','b').select('a','b','c')"
g.queryRaw(query, (error, result) => {
// send raw data to client
});
update({props}, callback)
.update
takes a properties object and updates the relevant properties on the model instance it is called on.
Arguments
-
props
: Object containing key value pairs of properties to update -
callback
: Some callback function with (error, result) arguments
Example
Person.find({'name': 'John'}).update({'age', 30}, (error, result) => {
// send data to client
});
delete(callback)
.delete
removes the object(s) it is called on from the database.
Arguments
-
callback
: Some callback function with (error, result) arguments
Example
Person.find({'name', 'John'}, (error, result) => {
if (result) result.delete((error, result) => {
// check if successful delete
});
});
order(property, order, [callback])
.order
sorts the results by a property in ascending or descending order
Arguments
-
property
: Name of property to order by -
order
: Order to sort - 'ASC' or 'DESC' -
callback
(optional): Some callback function with (error, result) arguments
Example
Person.findAll({'occupation': 'developer'}).order('age', 'DESC', (error, result) => {
// Return oldest developers first
});
limit(num, [callback])
.limit
limits the query to only the first num
objects
Arguments
-
num
: Max number of results to return -
callback
(optional): Some callback function with (error, result) arguments
Example
Person.find({'name': 'John'}).findEdge('knows').limit(100, (error, result) => {
// Return first 100 people that John knows
});
range(start, end, [callback])
.range
Array range of results returned
Arguments
-
start
: Starting index of results to return (inclusive) -
end
: Ending index of results to return (exclusive) -
callback
(optional): Some callback function with (error, result) arguments
Example
Person.find({'name': 'John'}).findEdge('knows').range(20, 40, (error, result) => {
// Return results 20-39 people that John knows
});
invoke(methodName, ...args)
.invoke
takes a method name as key and invokes it with arguments provided.
Arguments
-
methodName
: Method name as string key as defined along with schema -
args
(optional): Arguments to pass to the method
Returns
- Same return value as the method invoked
Example
Person.create(req.body, (error, result) => {
if (error) {
res.send(error);
}
else {
console.log(result.invoke('getDetails', 'Jr')); // displays "${result.name} Jr"
}
});
Vertex Methods
create({props}, callback)
.create
creates a new vertex with properties matching props object
Arguments
-
props
: Object containing key value pairs of properties matching defined Model schema -
callback
: Some callback function with (error, result) arguments
Returns
- Returns the newly created vertex object (with a unique ID) or an error object of failed schema checks
Example
Person.create({'name': 'John', 'age': 30}, (error, result) => {
// Returns the newly created vertex
/*
{
"id": "1",
"label": "person",
"name": "John",
"age": 30
}
*/
});
find({props}, [callback])
.find
finds the first vertex with properties matching props object
Arguments
-
props
: Object containing key value pairs of properties -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns the first matching vertex as an object
Example
Person.find({'name': 'John'}, (error, result) => {
// Returns first vertex found matching props
/*
{
"id": "1",
"label": "person",
"name": "John",
"age": 30
}
*/
});
findAll({props}, [callback])
.findAll
finds the all vertices with properties matching props object
Arguments
-
props
: Object containing key value pairs of properties -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all vertices matching props as objects
Example
Person.findAll({'age': 30}, (error, result) => {
// Returns array of matching vertices
/*
[
{
"id": "1",
"label": "person",
"name": "John",
"age": 30
},
{
"id": "2",
"label": "person",
"name": "Jane",
"age": 30
}
]
*/
});
findActionAll({action}, {props}, [callback])
.findActionAll
find all vertices when particular action is applied on the properties
Arguments
-
action
: Action providing context to operation to perform over the props -
props
: Object containing key value pairs of properties -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all vertices matching props as objects
Example
Person.findActionAll(Action.Or, { name: ['Abc', 'Def'], age: 75 }, (error, results) => {
if (error) {
console.error(error);
} else {
console.log(`Results: ${results.length}`);
console.log(results);
}
})
createEdge(edge, {props}, vertex, [both,] callback)
.createEdge
creates new edge relationships from starting vertex(es) to vertex(es) passed in.
Arguments
-
edge
: Edge model. If a string label is passed, no schema check will be done - edge model is recommended -
props
: Object containing key value pairs of properties to place on new edges -
vertex
: Vertex model instances or vertex model query -
both
(optional, default = false): If true, will create edges in both directions -
callback
: Some callback function with (error, result) arguments
Returns
- Returns an array containing all new created edges
Examples
// Chaining vertex methods
Person.findAll({'age': 20}).createEdge(Uses, {'frequency': 'daily'}, Website.find({'name': 'Facebook'}), (error, result) => {
// Result is array of newly created edges from everyone with age 20 to the website 'Facebook'
});
// Calling .createEdge on model instances
Person.findAll({'occupation': 'web developer'}, (error, results) => {
let developers = results;
Language.findAll({'name': ['Javascript', 'HTML', 'CSS']}, (error, results) => {
let languages = results;
developers.createEdge(Uses, {}, languages, (error, result) => {
// Result is array of newly created edge objects from each web developers
// to each of the 3 important components of web development
});
});
});
// Creating edges both ways
Person.find({'name': 'Jane'}, (error, result) => {
let jane = result;
Person.find({'name' : 'John'}).createEdge(Knows, {since: '1999'}, jane, true, (error, result) => {
// Creates two edges so that John knows Jane and Jane also knows John
})
});
findEdge(edge, {props}, [callback])
.findEdge
finds edges directly connected to the relevant vertex(es)
Arguments
-
edge
: Edge model. If a string label is passed, no schema check will be done - edge model is recommended -
props
: Object containing key value pairs of properties to match on edge relationships -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all connected edges
Examples
Person.find({'name': 'John'}).findEdge(Knows, {'from': 'school'}, (error, result) => {
// Result is array of edge objects representing all the 'knows' relationships of John
// where John knows the person from school (edge model property)
});
findRelated(edge, {props}, depth, [inModel, callback])
.findRelated
finds vertices related through the desired edge relationship.
Arguments
-
edge
: Edge model. If a string label is passed, no schema check will be done - edge model is recommended. -
props
: Object containing key value pairs of properties to match on edge relationships -
depth
: Depth of edge traversals to make -
inModel
(optional, default =this
): Vertex model of results to find. Can pass a vertex model (Person
) or label string ('person'
) -- vertex model is recommended. -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all related vertices
Examples
Person.find({'name': 'John'}).findRelated(Knows, {}, 2, (error, result) => {
// Result is array of Person records representing John's friends of friends
});
Person.find({'name': 'John'}).findRelated(Likes, {}, 1, Movie, (error, result) => {
// Result is array of Movie records that John likes.
});
findImplicit(edge, {props}, [callback])
.findImplicit
finds vertices that are related to another vertex the same way the original vertex is.
Arguments
-
edge
: Edge model. If a string label is passed, no schema check will be done - edge model is recommended. -
props
: Object containing key value pairs of properties to match on edge relationships -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all related vertices
Examples
Person.find({'name': 'John'}).findImplicit('created', {}, (error, result) => {
// Result is array of vertex objects representing people who have co-created things that John created
});
Edge Methods
create(out, in, {props}, callback)
.create
creates an index from out
vertex(es) to the in
vertex(es)
Arguments
-
out
: Vertex instance(s) or find/findAll method call -
in
: Vertex instance(s) or find/findAll method call -
props
: Object containing key value pairs of properties to add on the new edge -
both
(optional, default = false): If true, will create edges in both directions -
callback
: Some callback function with (error, result) arguments
Returns
- Returns newly created edge object
Examples
Person.find({'name': 'Joe'}, (error, result) => {
let joe = result;
Knows.create(Person.find({'name': 'John'}), joe, {'since': 2015}, (error, result) => {
// Returns the newly created edge
/*
{
"id": "1",
"label": "knows",
"since": "2015",
"outV": "1", // John's id
"inV": "2", // Joe's id
}
*/
});
});
find({props}, [callback])
.find
finds the first edge with properties matching props object
Arguments
-
props
: Object containing key value pairs of properties -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns the first matching edge as an object
Example
Knows.find({'since': 2015}, (error, result) => {
// Returns first edge found matching props
/*
{
"id": "1",
"label": "knows",
"since": 2015,
"outV": 1,
"inV": 123
}
*/
});
findAll({props}, [callback])
.findAll
finds the all edges with properties matching props object
Arguments
-
props
: Object containing key value pairs of properties -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all edges matching props as objects
Example
Knows.findAll({'since': 2015}, (error, result) => {
// Returns array of matching edges
/*
[
{
"id": "1",
"label": "knows",
"since": 2015,
"outV": 1,
"inV": 123
},
{
"id": "2",
"label": "knows",
"since": 2015,
"outV": 1,
"inV": 200
}
]
*/
});
findVertex(vertexModel, {props}, [callback])
.findVertex
finds the all vertices with properties matching props object connected by the relevant edge(s)
Arguments
-
vertexModel
: Vertex model. If a string label is passed, no schema check will be done - vertex model is recommended. -
props
: Object containing key value pairs of properties to find on vertices -
callback
(optional): Some callback function with (error, result) arguments
Returns
- Returns an array containing all vertices connected by current edge(s)
Example
Knows.find({'through': 'school'}).findVertex(Person, {'occupation': 'developer'}, (error, result) => {
// Result is array of people who are developers who know other people through school
});
Action Methods
P_eq(object)
-
object
: Object to check equality with
P_neq(object)
-
object
: Object to check inequality with
P_lt(object)
-
object
: Object to check less than with
P_lte(object)
-
object
: Object to check less than equal to with
P_gt(object)
-
object
: Object to check greater than with
P_gte(object)
-
object
: Object to check greater than equal to with
P_inside(object1, object2)
-
object1
: Object to check greater than with -
object2
: Object to check less than with
P_outside(object1, object2)
-
object1
: Object to check less than with -
object2
: Object to check greater than with
P_between(object1, object2)
-
object1
: Object to check greater than equal to with -
object2
: Object to check less than equal to with
P_within(objects)
-
objects
: Array to have one of the values
P_without(objects)
-
objects
: Array to not have the value
TextP_startingWith(str)
-
str
: String to start with
TextP_endingWith(str)
-
str
: String to end with
TextP_containing(str)
-
str
: String contained
TextP_notStartingWith(str)
-
str
: String to not start with
TextP_notEndingWith(str)
-
str
: String to not end with
TextP_notContaining(str)
-
str
: String to not be contained
Has_And(key, value)
-
key
: key to query over (an array field ideally) -
value
: An array ideally, where the key needs to contain all the provided values
Or(props)
-
props
: key/value to perform OR query over
Example
Person with age greater than 35
Person.findAll({ age: Action.P_gt(35) }, (error, results) => {
if (error) {
console.error(error);
} else {
console.log(`Results: ${results.length}`);
console.log(results);
}
})
Person with both the tags 'Abc' and 'Def'
Person.findActionAll(Action.Has_And, { tags: ['Abc', 'Def'] }, (error, results) => {
if (error) {
console.error(error);
} else {
console.log(`Results: ${results.length}`);
console.log(results);
}
})
Person with name either 'Abc' or 'Def' or age is 75
Person.findActionAll(Action.Or, { name: ['Abc', 'Def'], age: 75 }, (error, results) => {
if (error) {
console.error(error);
} else {
console.log(`Results: ${results.length}`);
console.log(results);
}
})
License
This project is licensed under the MIT License