Merge pull request #165 from BuxOrg/feature/utxo_functions

feat(BUX-184): add GetUtxosCount function
bitcoin-sv · Oct 5, 2023 · c7df2a4 · c7df2a4
2 parents 26c0b68 + 74a7bb6
commit c7df2a4
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 0 deletions.
diff --git a/transports/graphql.go b/transports/graphql.go
@@ -963,6 +963,33 @@ func (g *TransportGraphQL) GetUtxos(
 	return respData.Utxos, nil
 }
 
+// GetUtxosCount will get the count of utxos filtered by conditions and metadata
+func (g *TransportGraphQL) GetUtxosCount(
+	ctx context.Context,
+	conditions map[string]interface{},
+	metadata *buxmodels.Metadata,
+) (int64, ResponseError) {
+	reqBody := `
+		query ($conditions: Map, $metadata: Metadata) {
+			utxos_count (
+				conditions: $conditions
+				metadata: $metadata
+			)
+		}`
+
+	variables := map[string]interface{}{
+		"conditions": conditions,
+		"metadata":   metadata,
+	}
+
+	var respData int64
+	if err := g.doGraphQLQuery(ctx, reqBody, variables, &respData); err != nil {
+		return 0, err
+	}
+
+	return respData, nil
+}
+
 // UnreserveUtxos will unreserve utxos from draft transaction
 func (g *TransportGraphQL) UnreserveUtxos(ctx context.Context, referenceID string) ResponseError {
 	reqBody := `

diff --git a/transports/http.go b/transports/http.go
@@ -637,6 +637,29 @@ func (h *TransportHTTP) GetUtxos(ctx context.Context, conditions map[string]inte
 	return utxos, nil
 }
 
+// GetUtxosCount will get the count of utxos filtered by conditions and metadata
+func (h *TransportHTTP) GetUtxosCount(ctx context.Context, conditions map[string]interface{}, metadata *buxmodels.Metadata) (int64, ResponseError) {
+	jsonStr, err := json.Marshal(map[string]interface{}{
+		FieldConditions: conditions,
+		FieldMetadata:   processMetadata(metadata),
+	})
+	if err != nil {
+		return 0, WrapError(err)
+	}
+
+	var count int64
+	if err := h.doHTTPRequest(
+		ctx, http.MethodPost, "/utxo/count", jsonStr, h.xPriv, h.signRequest, &count,
+	); err != nil {
+		return 0, err
+	}
+	if h.debug {
+		log.Printf("utxos count: %v\n", count)
+	}
+
+	return count, nil
+}
+
 // UnreserveUtxos will unreserve utxos from draft transaction
 func (h *TransportHTTP) UnreserveUtxos(ctx context.Context, referenceID string) ResponseError {
 	jsonStr, err := json.Marshal(map[string]interface{}{

diff --git a/transports/interface.go b/transports/interface.go
@@ -48,6 +48,7 @@ type TransactionService interface {
 	UpdateTransactionMetadata(ctx context.Context, txID string, metadata *buxmodels.Metadata) (*buxmodels.Transaction, ResponseError)
 	GetUtxo(ctx context.Context, txID string, outputIndex uint32) (*buxmodels.Utxo, ResponseError)
 	GetUtxos(ctx context.Context, conditions map[string]interface{}, metadata *buxmodels.Metadata, queryParams *QueryParams) ([]*buxmodels.Utxo, ResponseError)
+	GetUtxosCount(ctx context.Context, conditions map[string]interface{}, metadata *buxmodels.Metadata) (int64, ResponseError)
 	UnreserveUtxos(ctx context.Context, referenceID string) ResponseError
 }