From fb7fb5938d43a3b5c4b960dba23c171551fb6467 Mon Sep 17 00:00:00 2001
From: Noah Metz <noah@metznet.ca>
Date: Mon, 10 Jul 2023 21:24:47 -0600
Subject: [PATCH] Changed NodeDef to use example instead of manually passing
 string and type

---
 context.go | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/context.go b/context.go
index 5ea84d9..da53138 100644
--- a/context.go
+++ b/context.go
@@ -23,12 +23,12 @@ type NodeDef struct {
   Reflect reflect.Type
 }
 
-func NewNodeDef(type_name string, reflect reflect.Type, load_func NodeLoadFunc, gql_type *graphql.Object) NodeDef {
+func NewNodeDef(example Node, load_func NodeLoadFunc, gql_type *graphql.Object) NodeDef {
   return NodeDef{
-    Type: NodeType(type_name),
+    Type: example.Type(),
     Load: load_func,
     GQLType: gql_type,
-    Reflect: reflect,
+    Reflect: reflect.TypeOf(example),
   }
 }
 
@@ -158,19 +158,19 @@ func NewContext(db * badger.DB, log Logger) * Context {
     Types: map[uint64]NodeDef{},
   }
 
-  err := ctx.RegisterNodeType(NewNodeDef("graph_node", reflect.TypeOf((*GraphNode)(nil)), LoadGraphNode, GQLTypeGraphNode()))
+  err := ctx.RegisterNodeType(NewNodeDef((*GraphNode)(nil), LoadGraphNode, GQLTypeGraphNode()))
   if err != nil {
     panic(err)
   }
-  err = ctx.RegisterNodeType(NewNodeDef("simple_lockable", reflect.TypeOf((*SimpleLockable)(nil)), LoadSimpleLockable, GQLTypeSimpleLockable()))
+  err = ctx.RegisterNodeType(NewNodeDef((*SimpleLockable)(nil), LoadSimpleLockable, GQLTypeSimpleLockable()))
   if err != nil {
     panic(err)
   }
-  err = ctx.RegisterNodeType(NewNodeDef("simple_thread", reflect.TypeOf((*SimpleThread)(nil)), LoadSimpleThread, GQLTypeSimpleThread()))
+  err = ctx.RegisterNodeType(NewNodeDef((*SimpleThread)(nil), LoadSimpleThread, GQLTypeSimpleThread()))
   if err != nil {
     panic(err)
   }
-  err = ctx.RegisterNodeType(NewNodeDef("gql_thread", reflect.TypeOf((*GQLThread)(nil)), LoadGQLThread, GQLTypeGQLThread()))
+  err = ctx.RegisterNodeType(NewNodeDef((*GQLThread)(nil), LoadGQLThread, GQLTypeGQLThread()))
   if err != nil {
     panic(err)
   }