Add Sub to GeyserClient

Prophet-Solutions · Aug 24, 2024 · fa1a4c1 · fa1a4c1
1 parent 9e797a6
commit fa1a4c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/yellowstone-geyser/client.go b/yellowstone-geyser/client.go
@@ -291,6 +291,7 @@ func (gc *YellowstoneGeyserClient) Subscribe(
 
 	// Store the subscription request in the client.
 	gc.Stream = &stream
+	gc.StreamSubscription = &subscription
 
 	// Return the stream for receiving updates.
 	return nil

diff --git a/yellowstone-geyser/types.go b/yellowstone-geyser/types.go
@@ -8,7 +8,8 @@ import (
 // YellowstoneGeyserClient is the main client struct that holds the gRPC connection
 // and the GeyserClient for communicating with the Yellowstone Geyser service.
 type YellowstoneGeyserClient struct {
-	GRPCConn *grpc.ClientConn
-	Stream   *pb.Geyser_SubscribeClient
-	Client   pb.GeyserClient
+	GRPCConn           *grpc.ClientConn
+	Stream             *pb.Geyser_SubscribeClient
+	StreamSubscription *pb.SubscribeRequest
+	Client             pb.GeyserClient
 }