mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package graphql
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/vektah/gqlparser/v2/ast"
|
|
)
|
|
|
|
type key string
|
|
|
|
const resolverCtx key = "resolver_context"
|
|
|
|
// Deprecated: Use FieldContext instead
|
|
type ResolverContext = FieldContext
|
|
|
|
type FieldContext struct {
|
|
Parent *FieldContext
|
|
// The name of the type this field belongs to
|
|
Object string
|
|
// These are the args after processing, they can be mutated in middleware to change what the resolver will get.
|
|
Args map[string]interface{}
|
|
// The raw field
|
|
Field CollectedField
|
|
// The index of array in path.
|
|
Index *int
|
|
// The result object of resolver
|
|
Result interface{}
|
|
// IsMethod indicates if the resolver is a method
|
|
IsMethod bool
|
|
// IsResolver indicates if the field has a user-specified resolver
|
|
IsResolver bool
|
|
}
|
|
|
|
type FieldStats struct {
|
|
// When field execution started
|
|
Started time.Time
|
|
|
|
// When argument marshaling finished
|
|
ArgumentsCompleted time.Time
|
|
|
|
// When the field completed running all middleware. Not available inside field middleware!
|
|
Completed time.Time
|
|
}
|
|
|
|
func (r *FieldContext) Path() ast.Path {
|
|
var path ast.Path
|
|
for it := r; it != nil; it = it.Parent {
|
|
if it.Index != nil {
|
|
path = append(path, ast.PathIndex(*it.Index))
|
|
} else if it.Field.Field != nil {
|
|
path = append(path, ast.PathName(it.Field.Alias))
|
|
}
|
|
}
|
|
|
|
// because we are walking up the chain, all the elements are backwards, do an inplace flip.
|
|
for i := len(path)/2 - 1; i >= 0; i-- {
|
|
opp := len(path) - 1 - i
|
|
path[i], path[opp] = path[opp], path[i]
|
|
}
|
|
|
|
return path
|
|
}
|
|
|
|
// Deprecated: Use GetFieldContext instead
|
|
func GetResolverContext(ctx context.Context) *ResolverContext {
|
|
return GetFieldContext(ctx)
|
|
}
|
|
|
|
func GetFieldContext(ctx context.Context) *FieldContext {
|
|
if val, ok := ctx.Value(resolverCtx).(*FieldContext); ok {
|
|
return val
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func WithFieldContext(ctx context.Context, rc *FieldContext) context.Context {
|
|
rc.Parent = GetFieldContext(ctx)
|
|
return context.WithValue(ctx, resolverCtx, rc)
|
|
}
|
|
|
|
func equalPath(a ast.Path, b ast.Path) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
|
|
for i := 0; i < len(a); i++ {
|
|
if a[i] != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|