diff --git a/graph2/types/comment.go b/graph2/types/comment.go index 77b8641..1a7a41a 100644 --- a/graph2/types/comment.go +++ b/graph2/types/comment.go @@ -2,9 +2,9 @@ package types import ( "context" + "errors" - "git.aiterp.net/rpdata/api/models/characters" - + "git.aiterp.net/rpdata/api/internal/loader" "git.aiterp.net/rpdata/api/models" ) @@ -15,7 +15,12 @@ func (r *commentResolver) Character(ctx context.Context, obj *models.Comment) (* return nil, nil } - character, err := characters.FindID(obj.CharacterID) + loader := loader.FromContext(ctx) + if loader == nil { + return nil, errors.New("no loader") + } + + character, err := loader.Character("id", obj.CharacterID) if err != nil { return nil, err }