fix: add defer conn.Close() on planetmint grpc connection (#13)

Signed-off-by: Lorenz Herzberger <[email protected]>
rddl-network · Sep 3, 2024 · b51806d · b51806d
1 parent c0eaf9d
commit b51806d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/planetmint/planetmint_client.go b/planetmint/planetmint_client.go
@@ -34,6 +34,7 @@ func (pmc *Client) GetActiveDeviceCount() (count float64, err error) {
 	if err != nil {
 		return
 	}
+	defer conn.Close()
 
 	machineClient := types.NewQueryClient(conn)
 	response, err := machineClient.ActiveTrustAnchorCount(
@@ -52,6 +53,7 @@ func (pmc *Client) GetActivatedDeviceCount() (count float64, err error) {
 	if err != nil {
 		return
 	}
+	defer conn.Close()
 
 	machineClient := types.NewQueryClient(conn)
 	response, err := machineClient.ActivatedTrustAnchorCount(