graphvent/gql_query.go

package graphvent
import (
  "time"
  "github.com/graphql-go/graphql"
  "github.com/graphql-go/graphql/language/ast"
)

func GetFieldNames(p graphql.ResolveParams) []string {
  names := []string{}

  for _, node := range(p.Info.FieldASTs) {
    for _, sel := range(node.SelectionSet.Selections) {
      names = append(names, sel.(*ast.Field).Name.Value)
    }
  }

  return names
}

var QueryNode = &graphql.Field{
  Type: InterfaceNode.Interface,
  Args: graphql.FieldConfigArgument{
    "id": &graphql.ArgumentConfig{
      Type: graphql.String,
    },
  },
  Resolve: func(p graphql.ResolveParams) (interface{}, error) {
    ctx, err := PrepResolve(p)
    if err != nil {
      return nil, err
    }


    id, err := ExtractID(p, "id")
    if err != nil {
      return nil, err
    }

    fields := GetFieldNames(p)
    ctx.Context.Log.Logf("gql", "RESOLVE_NODE(%s): %+v", id, fields)

    // Get a list of fields that will be written
    ext_fields, err := ctx.GQLContext.GetACLFields(p.Info.FieldName, fields)
    if err != nil {
      return nil, err
    }
    // Create a read signal, send it to the specified node, and add the wait to the response map if the send returns no error
    read_signal := NewReadSignal(ext_fields)

    ctx.Ext.resolver_reads_lock.Lock()
    ctx.Ext.resolver_reads[read_signal.UUID] = ctx.ID
    ctx.Ext.resolver_reads_lock.Unlock()

    err = ctx.Context.Send(ctx.Server.ID, id, read_signal)
    if err != nil {
      ctx.Ext.resolver_reads_lock.Lock()
      delete(ctx.Ext.resolver_reads, read_signal.UUID)
      ctx.Ext.resolver_reads_lock.Unlock()
      return nil, err
    }

    // Wait for the response, returning an error on timeout
    response, err := WaitForReadResult(ctx.Chan, time.Millisecond*100, read_signal.UUID)
    if err != nil {
      return nil, err
    }

    return NodeResult{id, response}, nil
  },
}

var QuerySelf = &graphql.Field{
  Type: InterfaceNode.Default,
  Resolve: func(p graphql.ResolveParams) (interface{}, error) {
    ctx, err := PrepResolve(p)
    if err != nil {
      return nil, err
    }

    ctx.Context.Log.Logf("gql", "FIELDS: %+v", GetFieldNames(p))

    return nil, nil
  },
}