update example todo add directive with location QUERY and MUTATION

example/todo/todo.go

@@ -15,13 +15,20 @@ import (
 var you = &User{ID: 1, Name: "You"}
 var them = &User{ID: 2, Name: "Them"}
 
+func getUserId(ctx context.Context) int {
+	if id, ok := ctx.Value("userId").(int); ok {
+		return id
+	}
+	return you.ID
+}
+
 func New() Config {
 	c := Config{
 		Resolvers: &resolvers{
 			todos: []*Todo{
 				{ID: 1, Text: "A todo not to forget", Done: false, owner: you},
 				{ID: 2, Text: "This is the most important", Done: false, owner: you},
-				{ID: 3, Text: "Somebody else's todo", Done: true, owner: them},
+				{ID: 3, Text: "Somebody else's todo", Done: false, owner: them},
 				{ID: 4, Text: "Please do this or else", Done: false, owner: you},
 			},
 			lastID: 4,
@@ -38,13 +45,16 @@ func New() Config {
 				return nil, fmt.Errorf("obj cant be owned")
 			}
 
-			if ownable.Owner().ID != you.ID {
+			if ownable.Owner().ID != getUserId(ctx) {
 				return nil, fmt.Errorf("you dont own that")
 			}
 		}
 
 		return next(ctx)
 	}
+	c.Directives.User = func(ctx context.Context, obj interface{}, next graphql.Resolver, id int) (interface{}, error) {
+		return next(context.WithValue(ctx, "userId", id))
+	}
 	return c
 }
 

example/todo/todo_test.go

@@ -55,6 +55,18 @@ func TestTodo(t *testing.T) {
 		require.Equal(t, "Very important", resp.UpdateTodo.Text)
 	})
 
+	t.Run("update the todo status by user id", func(t *testing.T) {
+		var resp struct {
+			UpdateTodo struct {
+				Text string
+				Done bool
+			}
+		}
+		c.MustPost(`mutation @user(id:2){ updateTodo(id: 3, changes:{done:true}) { text, done } }`, &resp)
+
+		require.Equal(t, "Somebody else's todo", resp.UpdateTodo.Text)
+	})
+
 	t.Run("select with alias", func(t *testing.T) {
 		var resp struct {
 			A struct{ Text string }