30 lines
610 B
Ruby
30 lines
610 B
Ruby
|
module GraphqlHelpers
|
||
|
|
# Run a loader's named resolver
|
||
|
|
def resolve(kls, name, obj: nil, args: {}, ctx: {})
|
||
|
|
kls[name].call(obj, args, ctx)
|
||
|
|
end
|
||
|
|
|
||
|
|
# Runs a block inside a GraphQL::Batch wrapper
|
||
|
|
def batch(max_queries: nil, &blk)
|
||
|
|
wrapper = proc do
|
||
|
|
GraphQL::Batch.batch do
|
||
|
|
result = yield
|
||
|
|
|
||
|
|
if result.is_a?(Array)
|
||
|
|
Promise.all(result)
|
||
|
|
else
|
||
|
|
result
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
if max_queries
|
||
|
|
result = nil
|
||
|
|
expect { result = wrapper.call }.not_to exceed_query_limit(max_queries)
|
||
|
|
result
|
||
|
|
else
|
||
|
|
wrapper.call
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|