Extract mustCloseGRPCConnection

services/ledger_service.go

@@ -46,21 +46,13 @@ func (ls LedgerService) QueryDIDDoc(did string) (cheqd.Did, cheqd.Metadata, bool
 	}
 
 	conn, err := ls.openGRPCConnection(serverAddr)
-	defer func(conn *grpc.ClientConn) {
-		if conn == nil {
-			return
-		}
-		err := conn.Close()
-		if err != nil {
-			log.Panic().Err(err).Msg("QueryDIDDoc: failed to close connection")
-			panic(err)
-		}
-	}(conn)
 	if err != nil {
 		log.Error().Err(err).Msg("QueryDIDDoc: failed connection")
 		return cheqd.Did{}, cheqd.Metadata{}, false, err
 	}
 
+	defer mustCloseGRPCConnection(conn)
+
 	log.Info().Msgf("Querying did doc: %s", did)
 	client := cheqd.NewQueryClient(conn)
 	didDocResponse, err := client.Did(context.Background(), &cheqd.QueryGetDidRequest{Id: did})
@@ -79,21 +71,13 @@ func (ls LedgerService) QueryResource(did string, resourceId string) (resource.R
 	}
 
 	conn, err := ls.openGRPCConnection(serverAddr)
-	defer func(conn *grpc.ClientConn) {
-		if conn == nil {
-			return
-		}
-		err := conn.Close()
-		if err != nil {
-			log.Panic().Err(err).Msg("QueryResource: failed to close connection")
-			panic(err)
-		}
-	}(conn)
 	if err != nil {
 		log.Error().Err(err).Msg("QueryResource: failed connection")
 		return resource.Resource{}, false, err
 	}
 
+	defer mustCloseGRPCConnection(conn)
+
 	log.Info().Msgf("Querying did resource: %s, %s", did, resourceId)
 
 	client := resource.NewQueryClient(conn)
@@ -146,6 +130,17 @@ func (ls LedgerService) openGRPCConnection(addr string) (conn *grpc.ClientConn,
 	return conn, nil
 }
 
+func mustCloseGRPCConnection(conn *grpc.ClientConn) {
+	if conn == nil {
+		return
+	}
+	err := conn.Close()
+	if err != nil {
+		log.Panic().Err(err).Msg("QueryDIDDoc: failed to close connection")
+		panic(err)
+	}
+}
+
 func (ls LedgerService) GetNamespaces() []string {
 	keys := make([]string, 0, len(ls.ledgers))
 	for k := range ls.ledgers {