feat: add raw subscriber

LilithGames · Nov 11, 2022 · f26bbb7 · f26bbb7
1 parent 1e07081
commit f26bbb7
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/agent/subscriber.go b/pkg/agent/subscriber.go
@@ -211,3 +211,15 @@ func NewGqlSubscriber(name string, query interface{}) *GqlSubscriber {
 	}
 	return subscriber
 }
+
+func NewGqlRawSubscriber(name, query string) *GqlSubscriber {
+	subscriber := &GqlSubscriber{}
+	subscriber.SubTopic = func(ticker core.Ticker, client interface{}) error {
+		tick := ticker.(*Tick)
+		subClient := client.(*graphql.SubscriptionClient)
+		wrapHandler := subscriber.GqlSubscriberWrapHandler(name, tick)
+		_, err := subClient.SubscribeRaw(query, subscriber.variables, wrapHandler)
+		return err
+	}
+	return subscriber
+}