Prepare gremlin for introduction of a worker type

This commit is contained in:
kortschak 2014-08-25 15:57:40 +09:30
parent a96f2ef6f3
commit 95170eb8ed
5 changed files with 141 additions and 160 deletions

View file

@ -25,21 +25,15 @@ import (
"github.com/google/cayley/quad" "github.com/google/cayley/quad"
) )
func getStrings(obj *otto.Object, field string) []string { func propertiesOf(obj *otto.Object, name string) []string {
strings := make([]string, 0) val, _ := obj.Get(name)
val, _ := obj.Get(field) if val.IsUndefined() {
if !val.IsUndefined() { return nil
export, _ := val.Export()
array := export.([]interface{})
for _, arg := range array {
strings = append(strings, arg.(string))
}
} }
return strings export, _ := val.Export()
return export.([]string)
} }
func getStringArgs(obj *otto.Object) []string { return getStrings(obj, "string_args") }
func buildIteratorTree(obj *otto.Object, ts graph.TripleStore) graph.Iterator { func buildIteratorTree(obj *otto.Object, ts graph.TripleStore) graph.Iterator {
if !isVertexChain(obj) { if !isVertexChain(obj) {
return iterator.NewNull() return iterator.NewNull()
@ -47,7 +41,7 @@ func buildIteratorTree(obj *otto.Object, ts graph.TripleStore) graph.Iterator {
return buildIteratorTreeHelper(obj, ts, iterator.NewNull()) return buildIteratorTreeHelper(obj, ts, iterator.NewNull())
} }
func makeListOfStringsFromArrayValue(obj *otto.Object) []string { func stringsFrom(obj *otto.Object) []string {
var output []string var output []string
lengthValue, _ := obj.Get("length") lengthValue, _ := obj.Get("length")
length, _ := lengthValue.ToInteger() length, _ := lengthValue.ToInteger()
@ -55,14 +49,10 @@ func makeListOfStringsFromArrayValue(obj *otto.Object) []string {
for index := uint32(0); index < ulength; index += 1 { for index := uint32(0); index < ulength; index += 1 {
name := strconv.FormatInt(int64(index), 10) name := strconv.FormatInt(int64(index), 10)
value, err := obj.Get(name) value, err := obj.Get(name)
if err != nil { if err != nil || !value.IsString() {
continue continue
} }
if !value.IsString() { output = append(output, value.String())
continue
}
s, _ := value.ToString()
output = append(output, s)
} }
return output return output
} }
@ -87,7 +77,7 @@ func buildIteratorFromValue(val otto.Value, ts graph.TripleStore) graph.Iterator
return buildIteratorTree(val.Object(), ts) return buildIteratorTree(val.Object(), ts)
case "Array": case "Array":
// Had better be an array of strings // Had better be an array of strings
strings := makeListOfStringsFromArrayValue(val.Object()) strings := stringsFrom(val.Object())
it := ts.FixedIterator() it := ts.FixedIterator()
for _, x := range strings { for _, x := range strings {
it.Add(ts.ValueOf(x)) it.Add(ts.ValueOf(x))
@ -101,8 +91,7 @@ func buildIteratorFromValue(val otto.Value, ts graph.TripleStore) graph.Iterator
fallthrough fallthrough
case "String": case "String":
it := ts.FixedIterator() it := ts.FixedIterator()
str, _ := val.ToString() it.Add(ts.ValueOf(val.String()))
it.Add(ts.ValueOf(str))
return it return it
default: default:
glog.Errorln("Trying to handle unsupported Javascript value.") glog.Errorln("Trying to handle unsupported Javascript value.")
@ -130,10 +119,9 @@ func buildInOutIterator(obj *otto.Object, ts graph.TripleStore, base graph.Itera
var tags []string var tags []string
one, _ := argArray.Get("1") one, _ := argArray.Get("1")
if one.IsString() { if one.IsString() {
s, _ := one.ToString() tags = append(tags, one.String())
tags = append(tags, s)
} else if one.Class() == "Array" { } else if one.Class() == "Array" {
tags = makeListOfStringsFromArrayValue(one.Object()) tags = stringsFrom(one.Object())
} }
for _, tag := range tags { for _, tag := range tags {
predicateNodeIterator.Tagger().Add(tag) predicateNodeIterator.Tagger().Add(tag)
@ -152,21 +140,19 @@ func buildInOutIterator(obj *otto.Object, ts graph.TripleStore, base graph.Itera
} }
func buildIteratorTreeHelper(obj *otto.Object, ts graph.TripleStore, base graph.Iterator) graph.Iterator { func buildIteratorTreeHelper(obj *otto.Object, ts graph.TripleStore, base graph.Iterator) graph.Iterator {
var it graph.Iterator var it graph.Iterator = base
it = base
// TODO: Better error handling // TODO: Better error handling
kindVal, _ := obj.Get("_gremlin_type")
stringArgs := getStringArgs(obj)
var subIt graph.Iterator var subIt graph.Iterator
prevVal, _ := obj.Get("_gremlin_prev") if prev, _ := obj.Get("_gremlin_prev"); !prev.IsObject() {
if !prevVal.IsObject() {
subIt = base subIt = base
} else { } else {
subIt = buildIteratorTreeHelper(prevVal.Object(), ts, base) subIt = buildIteratorTreeHelper(prev.Object(), ts, base)
} }
kind, _ := kindVal.ToString() stringArgs := propertiesOf(obj, "string_args")
switch kind { val, _ := obj.Get("_gremlin_type")
switch val.String() {
case "vertex": case "vertex":
if len(stringArgs) == 0 { if len(stringArgs) == 0 {
it = ts.NodesAllIterator() it = ts.NodesAllIterator()

View file

@ -21,31 +21,25 @@ import (
"github.com/robertkrimen/otto" "github.com/robertkrimen/otto"
) )
func BuildEnviron(ses *Session) *otto.Otto { func argsOf(call otto.FunctionCall) []string {
env := otto.New() var out []string
setupGremlin(env, ses)
return env
}
func concatStringArgs(call otto.FunctionCall) *[]interface{} {
outStrings := make([]interface{}, 0)
for _, arg := range call.ArgumentList { for _, arg := range call.ArgumentList {
if arg.IsString() { if arg.IsString() {
outStrings = append(outStrings, arg.String()) out = append(out, arg.String())
} }
if arg.IsObject() && arg.Class() == "Array" { if arg.IsObject() && arg.Class() == "Array" {
obj, _ := arg.Export() obj, _ := arg.Export()
for _, x := range obj.([]interface{}) { for _, x := range obj.([]interface{}) {
outStrings = append(outStrings, x.(string)) out = append(out, x.(string))
} }
} }
} }
return &outStrings return out
} }
func isVertexChain(obj *otto.Object) bool { func isVertexChain(obj *otto.Object) bool {
val, _ := obj.Get("_gremlin_type") val, _ := obj.Get("_gremlin_type")
if x, _ := val.ToString(); x == "vertex" { if val.String() == "vertex" {
return true return true
} }
val, _ = obj.Get("_gremlin_prev") val, _ = obj.Get("_gremlin_prev")
@ -55,8 +49,10 @@ func isVertexChain(obj *otto.Object) bool {
return false return false
} }
func setupGremlin(env *otto.Otto, ses *Session) { func (s *Session) setup(env *otto.Otto) *otto.Otto {
graph, _ := env.Object("graph = {}") graph, _ := env.Object("graph = {}")
env.Run("g = graph")
graph.Set("Vertex", func(call otto.FunctionCall) otto.Value { graph.Set("Vertex", func(call otto.FunctionCall) otto.Value {
call.Otto.Run("var out = {}") call.Otto.Run("var out = {}")
out, err := call.Otto.Object("out") out, err := call.Otto.Object("out")
@ -65,31 +61,32 @@ func setupGremlin(env *otto.Otto, ses *Session) {
return otto.TrueValue() return otto.TrueValue()
} }
out.Set("_gremlin_type", "vertex") out.Set("_gremlin_type", "vertex")
outStrings := concatStringArgs(call) args := argsOf(call)
if len(*outStrings) > 0 { if len(args) > 0 {
out.Set("string_args", *outStrings) out.Set("string_args", args)
} }
embedTraversals(env, ses, out) s.embedTraversals(env, out)
embedFinals(env, ses, out) s.embedFinals(env, out)
return out.Value() return out.Value()
}) })
env.Run("graph.V = graph.Vertex")
graph.Set("Morphism", func(call otto.FunctionCall) otto.Value { graph.Set("Morphism", func(call otto.FunctionCall) otto.Value {
call.Otto.Run("var out = {}") call.Otto.Run("var out = {}")
out, _ := call.Otto.Object("out") out, _ := call.Otto.Object("out")
out.Set("_gremlin_type", "morphism") out.Set("_gremlin_type", "morphism")
embedTraversals(env, ses, out) s.embedTraversals(env, out)
return out.Value() return out.Value()
}) })
env.Run("graph.M = graph.Morphism")
graph.Set("Emit", func(call otto.FunctionCall) otto.Value { graph.Set("Emit", func(call otto.FunctionCall) otto.Value {
value := call.Argument(0) value := call.Argument(0)
if value.IsDefined() { if value.IsDefined() {
ses.SendResult(&Result{val: &value}) s.SendResult(&Result{val: &value})
} }
return otto.NullValue() return otto.NullValue()
}) })
env.Run("graph.V = graph.Vertex")
env.Run("graph.M = graph.Morphism")
env.Run("g = graph")
return env
} }

View file

@ -26,45 +26,45 @@ import (
const TopResultTag = "id" const TopResultTag = "id"
func embedFinals(env *otto.Otto, ses *Session, obj *otto.Object) { func (s *Session) embedFinals(env *otto.Otto, obj *otto.Object) {
obj.Set("All", allFunc(env, ses, obj)) obj.Set("All", s.allFunc(env, obj))
obj.Set("GetLimit", limitFunc(env, ses, obj)) obj.Set("GetLimit", s.limitFunc(env, obj))
obj.Set("ToArray", toArrayFunc(env, ses, obj, false)) obj.Set("ToArray", s.toArrayFunc(env, obj, false))
obj.Set("ToValue", toValueFunc(env, ses, obj, false)) obj.Set("ToValue", s.toValueFunc(env, obj, false))
obj.Set("TagArray", toArrayFunc(env, ses, obj, true)) obj.Set("TagArray", s.toArrayFunc(env, obj, true))
obj.Set("TagValue", toValueFunc(env, ses, obj, true)) obj.Set("TagValue", s.toValueFunc(env, obj, true))
obj.Set("Map", mapFunc(env, ses, obj)) obj.Set("Map", s.mapFunc(env, obj))
obj.Set("ForEach", mapFunc(env, ses, obj)) obj.Set("ForEach", s.mapFunc(env, obj))
} }
func allFunc(env *otto.Otto, ses *Session, obj *otto.Object) func(otto.FunctionCall) otto.Value { func (s *Session) allFunc(env *otto.Otto, obj *otto.Object) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
it := buildIteratorTree(obj, ses.ts) it := buildIteratorTree(obj, s.ts)
it.Tagger().Add(TopResultTag) it.Tagger().Add(TopResultTag)
ses.limit = -1 s.limit = -1
ses.count = 0 s.count = 0
runIteratorOnSession(it, ses) s.runIterator(it)
return otto.NullValue() return otto.NullValue()
} }
} }
func limitFunc(env *otto.Otto, ses *Session, obj *otto.Object) func(otto.FunctionCall) otto.Value { func (s *Session) limitFunc(env *otto.Otto, obj *otto.Object) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
if len(call.ArgumentList) > 0 { if len(call.ArgumentList) > 0 {
limitVal, _ := call.Argument(0).ToInteger() limitVal, _ := call.Argument(0).ToInteger()
it := buildIteratorTree(obj, ses.ts) it := buildIteratorTree(obj, s.ts)
it.Tagger().Add(TopResultTag) it.Tagger().Add(TopResultTag)
ses.limit = int(limitVal) s.limit = int(limitVal)
ses.count = 0 s.count = 0
runIteratorOnSession(it, ses) s.runIterator(it)
} }
return otto.NullValue() return otto.NullValue()
} }
} }
func toArrayFunc(env *otto.Otto, ses *Session, obj *otto.Object, withTags bool) func(otto.FunctionCall) otto.Value { func (s *Session) toArrayFunc(env *otto.Otto, obj *otto.Object, withTags bool) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
it := buildIteratorTree(obj, ses.ts) it := buildIteratorTree(obj, s.ts)
it.Tagger().Add(TopResultTag) it.Tagger().Add(TopResultTag)
limit := -1 limit := -1
if len(call.ArgumentList) > 0 { if len(call.ArgumentList) > 0 {
@ -74,10 +74,10 @@ func toArrayFunc(env *otto.Otto, ses *Session, obj *otto.Object, withTags bool)
var val otto.Value var val otto.Value
var err error var err error
if !withTags { if !withTags {
array := runIteratorToArrayNoTags(it, ses, limit) array := s.runIteratorToArrayNoTags(it, limit)
val, err = call.Otto.ToValue(array) val, err = call.Otto.ToValue(array)
} else { } else {
array := runIteratorToArray(it, ses, limit) array := s.runIteratorToArray(it, limit)
val, err = call.Otto.ToValue(array) val, err = call.Otto.ToValue(array)
} }
@ -89,21 +89,21 @@ func toArrayFunc(env *otto.Otto, ses *Session, obj *otto.Object, withTags bool)
} }
} }
func toValueFunc(env *otto.Otto, ses *Session, obj *otto.Object, withTags bool) func(otto.FunctionCall) otto.Value { func (s *Session) toValueFunc(env *otto.Otto, obj *otto.Object, withTags bool) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
it := buildIteratorTree(obj, ses.ts) it := buildIteratorTree(obj, s.ts)
it.Tagger().Add(TopResultTag) it.Tagger().Add(TopResultTag)
limit := 1 limit := 1
var val otto.Value var val otto.Value
var err error var err error
if !withTags { if !withTags {
array := runIteratorToArrayNoTags(it, ses, limit) array := s.runIteratorToArrayNoTags(it, limit)
if len(array) < 1 { if len(array) < 1 {
return otto.NullValue() return otto.NullValue()
} }
val, err = call.Otto.ToValue(array[0]) val, err = call.Otto.ToValue(array[0])
} else { } else {
array := runIteratorToArray(it, ses, limit) array := s.runIteratorToArray(it, limit)
if len(array) < 1 { if len(array) < 1 {
return otto.NullValue() return otto.NullValue()
} }
@ -119,9 +119,9 @@ func toValueFunc(env *otto.Otto, ses *Session, obj *otto.Object, withTags bool)
} }
} }
func mapFunc(env *otto.Otto, ses *Session, obj *otto.Object) func(otto.FunctionCall) otto.Value { func (s *Session) mapFunc(env *otto.Otto, obj *otto.Object) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
it := buildIteratorTree(obj, ses.ts) it := buildIteratorTree(obj, s.ts)
it.Tagger().Add(TopResultTag) it.Tagger().Add(TopResultTag)
limit := -1 limit := -1
if len(call.ArgumentList) == 0 { if len(call.ArgumentList) == 0 {
@ -132,26 +132,26 @@ func mapFunc(env *otto.Otto, ses *Session, obj *otto.Object) func(otto.FunctionC
limitParsed, _ := call.Argument(0).ToInteger() limitParsed, _ := call.Argument(0).ToInteger()
limit = int(limitParsed) limit = int(limitParsed)
} }
runIteratorWithCallback(it, ses, callback, call, limit) s.runIteratorWithCallback(it, callback, call, limit)
return otto.NullValue() return otto.NullValue()
} }
} }
func tagsToValueMap(m map[string]graph.Value, ses *Session) map[string]string { func (s *Session) tagsToValueMap(m map[string]graph.Value) map[string]string {
outputMap := make(map[string]string) outputMap := make(map[string]string)
for k, v := range m { for k, v := range m {
outputMap[k] = ses.ts.NameOf(v) outputMap[k] = s.ts.NameOf(v)
} }
return outputMap return outputMap
} }
func runIteratorToArray(it graph.Iterator, ses *Session, limit int) []map[string]string { func (s *Session) runIteratorToArray(it graph.Iterator, limit int) []map[string]string {
output := make([]map[string]string, 0) output := make([]map[string]string, 0)
count := 0 count := 0
it, _ = it.Optimize() it, _ = it.Optimize()
for { for {
select { select {
case <-ses.kill: case <-s.kill:
return nil return nil
default: default:
} }
@ -160,20 +160,20 @@ func runIteratorToArray(it graph.Iterator, ses *Session, limit int) []map[string
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
output = append(output, tagsToValueMap(tags, ses)) output = append(output, s.tagsToValueMap(tags))
count++ count++
if limit >= 0 && count >= limit { if limit >= 0 && count >= limit {
break break
} }
for it.NextPath() { for it.NextPath() {
select { select {
case <-ses.kill: case <-s.kill:
return nil return nil
default: default:
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
output = append(output, tagsToValueMap(tags, ses)) output = append(output, s.tagsToValueMap(tags))
count++ count++
if limit >= 0 && count >= limit { if limit >= 0 && count >= limit {
break break
@ -184,20 +184,20 @@ func runIteratorToArray(it graph.Iterator, ses *Session, limit int) []map[string
return output return output
} }
func runIteratorToArrayNoTags(it graph.Iterator, ses *Session, limit int) []string { func (s *Session) runIteratorToArrayNoTags(it graph.Iterator, limit int) []string {
output := make([]string, 0) output := make([]string, 0)
count := 0 count := 0
it, _ = it.Optimize() it, _ = it.Optimize()
for { for {
select { select {
case <-ses.kill: case <-s.kill:
return nil return nil
default: default:
} }
if !graph.Next(it) { if !graph.Next(it) {
break break
} }
output = append(output, ses.ts.NameOf(it.Result())) output = append(output, s.ts.NameOf(it.Result()))
count++ count++
if limit >= 0 && count >= limit { if limit >= 0 && count >= limit {
break break
@ -207,13 +207,13 @@ func runIteratorToArrayNoTags(it graph.Iterator, ses *Session, limit int) []stri
return output return output
} }
func runIteratorWithCallback(it graph.Iterator, ses *Session, callback otto.Value, this otto.FunctionCall, limit int) { func (s *Session) runIteratorWithCallback(it graph.Iterator, callback otto.Value, this otto.FunctionCall, limit int) {
count := 0 count := 0
it, _ = it.Optimize() it, _ = it.Optimize()
glog.V(2).Infoln(it.DebugString(0)) glog.V(2).Infoln(it.DebugString(0))
for { for {
select { select {
case <-ses.kill: case <-s.kill:
return return
default: default:
} }
@ -222,7 +222,7 @@ func runIteratorWithCallback(it graph.Iterator, ses *Session, callback otto.Valu
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
val, _ := this.Otto.ToValue(tagsToValueMap(tags, ses)) val, _ := this.Otto.ToValue(s.tagsToValueMap(tags))
val, _ = callback.Call(this.This, val) val, _ = callback.Call(this.This, val)
count++ count++
if limit >= 0 && count >= limit { if limit >= 0 && count >= limit {
@ -230,13 +230,13 @@ func runIteratorWithCallback(it graph.Iterator, ses *Session, callback otto.Valu
} }
for it.NextPath() { for it.NextPath() {
select { select {
case <-ses.kill: case <-s.kill:
return return
default: default:
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
val, _ := this.Otto.ToValue(tagsToValueMap(tags, ses)) val, _ := this.Otto.ToValue(s.tagsToValueMap(tags))
val, _ = callback.Call(this.This, val) val, _ = callback.Call(this.This, val)
count++ count++
if limit >= 0 && count >= limit { if limit >= 0 && count >= limit {
@ -247,16 +247,16 @@ func runIteratorWithCallback(it graph.Iterator, ses *Session, callback otto.Valu
it.Close() it.Close()
} }
func runIteratorOnSession(it graph.Iterator, ses *Session) { func (s *Session) runIterator(it graph.Iterator) {
if ses.wantShape { if s.wantShape {
iterator.OutputQueryShapeForIterator(it, ses.ts, ses.shape) iterator.OutputQueryShapeForIterator(it, s.ts, s.shape)
return return
} }
it, _ = it.Optimize() it, _ = it.Optimize()
glog.V(2).Infoln(it.DebugString(0)) glog.V(2).Infoln(it.DebugString(0))
for { for {
select { select {
case <-ses.kill: case <-s.kill:
return return
default: default:
} }
@ -265,18 +265,18 @@ func runIteratorOnSession(it graph.Iterator, ses *Session) {
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
if !ses.SendResult(&Result{actualResults: &tags}) { if !s.SendResult(&Result{actualResults: &tags}) {
break break
} }
for it.NextPath() { for it.NextPath() {
select { select {
case <-ses.kill: case <-s.kill:
return return
default: default:
} }
tags := make(map[string]graph.Value) tags := make(map[string]graph.Value)
it.TagResults(tags) it.TagResults(tags)
if !ses.SendResult(&Result{actualResults: &tags}) { if !s.SendResult(&Result{actualResults: &tags}) {
break break
} }
} }

View file

@ -54,7 +54,7 @@ func NewSession(ts graph.TripleStore, timeout time.Duration, persist bool) *Sess
limit: -1, limit: -1,
timeout: timeout, timeout: timeout,
} }
g.env = BuildEnviron(&g) g.env = g.setup(otto.New())
if persist { if persist {
g.emptyEnv = g.env g.emptyEnv = g.env
} }

View file

@ -21,123 +21,122 @@ import (
"github.com/robertkrimen/otto" "github.com/robertkrimen/otto"
) )
func embedTraversals(env *otto.Otto, ses *Session, obj *otto.Object) { func (s *Session) embedTraversals(env *otto.Otto, obj *otto.Object) {
obj.Set("In", gremlinFunc("in", obj, env, ses)) obj.Set("In", gremlinFunc("in", obj, env, s))
obj.Set("Out", gremlinFunc("out", obj, env, ses)) obj.Set("Out", gremlinFunc("out", obj, env, s))
obj.Set("Is", gremlinFunc("is", obj, env, ses)) obj.Set("Is", gremlinFunc("is", obj, env, s))
obj.Set("Both", gremlinFunc("both", obj, env, ses)) obj.Set("Both", gremlinFunc("both", obj, env, s))
obj.Set("Follow", gremlinFunc("follow", obj, env, ses)) obj.Set("Follow", gremlinFunc("follow", obj, env, s))
obj.Set("FollowR", gremlinFollowR("followr", obj, env, ses)) obj.Set("FollowR", gremlinFollowR("followr", obj, env, s))
obj.Set("And", gremlinFunc("and", obj, env, ses)) obj.Set("And", gremlinFunc("and", obj, env, s))
obj.Set("Intersect", gremlinFunc("and", obj, env, ses)) obj.Set("Intersect", gremlinFunc("and", obj, env, s))
obj.Set("Union", gremlinFunc("or", obj, env, ses)) obj.Set("Union", gremlinFunc("or", obj, env, s))
obj.Set("Or", gremlinFunc("or", obj, env, ses)) obj.Set("Or", gremlinFunc("or", obj, env, s))
obj.Set("Back", gremlinBack("back", obj, env, ses)) obj.Set("Back", gremlinBack("back", obj, env, s))
obj.Set("Tag", gremlinFunc("tag", obj, env, ses)) obj.Set("Tag", gremlinFunc("tag", obj, env, s))
obj.Set("As", gremlinFunc("tag", obj, env, ses)) obj.Set("As", gremlinFunc("tag", obj, env, s))
obj.Set("Has", gremlinFunc("has", obj, env, ses)) obj.Set("Has", gremlinFunc("has", obj, env, s))
obj.Set("Save", gremlinFunc("save", obj, env, ses)) obj.Set("Save", gremlinFunc("save", obj, env, s))
obj.Set("SaveR", gremlinFunc("saver", obj, env, ses)) obj.Set("SaveR", gremlinFunc("saver", obj, env, s))
} }
func gremlinFunc(kind string, prevObj *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value { func gremlinFunc(kind string, prev *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
call.Otto.Run("var out = {}") call.Otto.Run("var out = {}")
out, _ := call.Otto.Object("out") out, _ := call.Otto.Object("out")
out.Set("_gremlin_type", kind) out.Set("_gremlin_type", kind)
out.Set("_gremlin_values", call.ArgumentList) out.Set("_gremlin_values", call.ArgumentList)
out.Set("_gremlin_prev", prevObj) out.Set("_gremlin_prev", prev)
outStrings := concatStringArgs(call) args := argsOf(call)
if len(*outStrings) > 0 { if len(args) > 0 {
out.Set("string_args", *outStrings) out.Set("string_args", args)
} }
embedTraversals(env, ses, out) ses.embedTraversals(env, out)
if isVertexChain(call.This.Object()) { if isVertexChain(call.This.Object()) {
embedFinals(env, ses, out) ses.embedFinals(env, out)
} }
return out.Value() return out.Value()
} }
} }
func gremlinBack(kind string, prevObj *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value { func gremlinBack(kind string, prev *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
call.Otto.Run("var out = {}") call.Otto.Run("var out = {}")
out, _ := call.Otto.Object("out") out, _ := call.Otto.Object("out")
out.Set("_gremlin_type", kind) out.Set("_gremlin_type", kind)
out.Set("_gremlin_values", call.ArgumentList) out.Set("_gremlin_values", call.ArgumentList)
outStrings := concatStringArgs(call) args := argsOf(call)
if len(*outStrings) > 0 { if len(args) > 0 {
out.Set("string_args", *outStrings) out.Set("string_args", args)
} }
var otherChain *otto.Object var otherChain *otto.Object
var thisObj *otto.Object var thisObj *otto.Object
if len(*outStrings) != 0 { if len(args) != 0 {
otherChain, thisObj = reverseGremlinChainTo(call.Otto, prevObj, (*outStrings)[0].(string)) otherChain, thisObj = reverseGremlinChainTo(call.Otto, prev, args[0])
} else { } else {
otherChain, thisObj = reverseGremlinChainTo(call.Otto, prevObj, "") otherChain, thisObj = reverseGremlinChainTo(call.Otto, prev, "")
} }
out.Set("_gremlin_prev", thisObj) out.Set("_gremlin_prev", thisObj)
out.Set("_gremlin_back_chain", otherChain) out.Set("_gremlin_back_chain", otherChain)
embedTraversals(env, ses, out) ses.embedTraversals(env, out)
if isVertexChain(call.This.Object()) { if isVertexChain(call.This.Object()) {
embedFinals(env, ses, out) ses.embedFinals(env, out)
} }
return out.Value() return out.Value()
} }
} }
func gremlinFollowR(kind string, prevObj *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value { func gremlinFollowR(kind string, prev *otto.Object, env *otto.Otto, ses *Session) func(otto.FunctionCall) otto.Value {
return func(call otto.FunctionCall) otto.Value { return func(call otto.FunctionCall) otto.Value {
call.Otto.Run("var out = {}") call.Otto.Run("var out = {}")
out, _ := call.Otto.Object("out") out, _ := call.Otto.Object("out")
out.Set("_gremlin_type", kind) out.Set("_gremlin_type", kind)
out.Set("_gremlin_values", call.ArgumentList) out.Set("_gremlin_values", call.ArgumentList)
outStrings := concatStringArgs(call) args := argsOf(call)
if len(*outStrings) > 0 { if len(args) > 0 {
out.Set("string_args", *outStrings) out.Set("string_args", args)
} }
if len(call.ArgumentList) == 0 { if len(call.ArgumentList) == 0 {
return prevObj.Value() return prev.Value()
} }
arg := call.Argument(0) arg := call.Argument(0)
if isVertexChain(arg.Object()) { if isVertexChain(arg.Object()) {
return prevObj.Value() return prev.Value()
} }
newChain, _ := reverseGremlinChainTo(call.Otto, arg.Object(), "") newChain, _ := reverseGremlinChainTo(call.Otto, arg.Object(), "")
out.Set("_gremlin_prev", prevObj) out.Set("_gremlin_prev", prev)
out.Set("_gremlin_followr", newChain) out.Set("_gremlin_followr", newChain)
embedTraversals(env, ses, out) ses.embedTraversals(env, out)
if isVertexChain(call.This.Object()) { if isVertexChain(call.This.Object()) {
embedFinals(env, ses, out) ses.embedFinals(env, out)
} }
return out.Value() return out.Value()
} }
} }
func reverseGremlinChainTo(env *otto.Otto, prevObj *otto.Object, tag string) (*otto.Object, *otto.Object) { func reverseGremlinChainTo(env *otto.Otto, prev *otto.Object, tag string) (*otto.Object, *otto.Object) {
env.Run("var _base_object = {}") env.Run("var _base_object = {}")
base, err := env.Object("_base_object") base, err := env.Object("_base_object")
if err != nil { if err != nil {
glog.Error(err) glog.Error(err)
return otto.NullValue().Object(), otto.NullValue().Object() return otto.NullValue().Object(), otto.NullValue().Object()
} }
if isVertexChain(prevObj) { if isVertexChain(prev) {
base.Set("_gremlin_type", "vertex") base.Set("_gremlin_type", "vertex")
} else { } else {
base.Set("_gremlin_type", "morphism") base.Set("_gremlin_type", "morphism")
} }
return reverseGremlinChainHelper(env, prevObj, base, tag) return reverseGremlinChainHelper(env, prev, base, tag)
} }
func reverseGremlinChainHelper(env *otto.Otto, chain *otto.Object, newBase *otto.Object, tag string) (*otto.Object, *otto.Object) { func reverseGremlinChainHelper(env *otto.Otto, chain *otto.Object, newBase *otto.Object, tag string) (*otto.Object, *otto.Object) {
kindVal, _ := chain.Get("_gremlin_type") kindVal, _ := chain.Get("_gremlin_type")
kind, _ := kindVal.ToString() kind := kindVal.String()
if tag != "" { if tag != "" {
if kind == "tag" { if kind == "tag" {
tags := getStringArgs(chain) tags := propertiesOf(chain, "string_args")
for _, t := range tags { for _, t := range tags {
if t == tag { if t == tag {
return newBase, chain return newBase, chain
@ -174,8 +173,7 @@ func reverseGremlinChainHelper(env *otto.Otto, chain *otto.Object, newBase *otto
func debugChain(obj *otto.Object) bool { func debugChain(obj *otto.Object) bool {
val, _ := obj.Get("_gremlin_type") val, _ := obj.Get("_gremlin_type")
x, _ := val.ToString() glog.V(2).Infoln(val)
glog.V(2).Infoln(x)
val, _ = obj.Get("_gremlin_prev") val, _ = obj.Get("_gremlin_prev")
if val.IsObject() { if val.IsObject() {
return debugChain(val.Object()) return debugChain(val.Object())