1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
# frozen_string_literal: true
begin
require "graphql-pro"
rescue LoadError => err
puts "Skipping GraphQL::Pro: #{err.message}"
end
class DummySchema < GraphQL::Schema
class Query < GraphQL::Schema::Object
field :str, String, fallback_value: "hello"
field :sleep, Float do
argument :seconds, Float
end
def sleep(seconds:)
Kernel.sleep(seconds)
seconds
end
end
query(Query)
class Subscription < GraphQL::Schema::Object
field :message, String do
argument :channel, String
end
end
subscription(Subscription)
DB_NUMBER = Rails.env.test? ? 1 : 2
use GraphQL::Tracing::DetailedTrace, redis: Redis.new(db: DB_NUMBER)
if defined?(GraphQL::Pro)
use GraphQL::Pro::OperationStore, redis: Redis.new(db: DB_NUMBER)
use GraphQL::Pro::PusherSubscriptions, redis: Redis.new(db: DummySchema::DB_NUMBER), pusher: MockPusher.new
class KeyNotRequiredLimiter < GraphQL::Enterprise::RuntimeLimiter
def limiter_key(query)
query.
context[:limiter_key] || "unlimited"
end
def limit_for(key, query)
key == "unlimited" ? nil : super
end
end
use KeyNotRequiredLimiter,
redis: Redis.new(db: DummySchema::DB_NUMBER),
limit_ms: 100
end
def self.detailed_trace?(query)
query.context[:profile]
end
end
# To preview rate limiter
# puts "Making Rate-limited requests..."
# 3.times.map do
# pp DummySchema.execute("{ sleep(seconds: 0.02) }", context: { limiter_key: "client-1" }).to_h
# end
# 3.times.map do
# pp DummySchema.execute("{ sleep(seconds: 0.110) }", context: { limiter_key: "client-2" }).to_h
# end
# puts " ... done"
# To preview subscription data in the dashboard:
# DummySchema.subscriptions.clear
# res1 = DummySchema.execute("subscription { message(channel: \"cats\") }")
# res2 = DummySchema.execute("subscription { message(channel: \"dogs\") }")
# DummySchema.subscriptions.trigger(:message, { channel: "cats" }, "meow")
|
