feat: PRT - adding stateful api header for hanging relays (#1712)

lavanet · Sep 25, 2024 · ab51b73 · ab51b73
1 parent db8248c
commit ab51b73
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/protocol/common/endpoints.go b/protocol/common/endpoints.go
@@ -28,6 +28,7 @@ const (
 	NODE_ERRORS_PROVIDERS_HEADER_NAME               = "Lava-Node-Errors-providers"
 	REPORTED_PROVIDERS_HEADER_NAME                  = "Lava-Reported-Providers"
 	USER_REQUEST_TYPE                               = "lava-user-request-type"
+	STATEFUL_API_HEADER                             = "lava-stateful-api"
 	LAVA_IDENTIFIED_NODE_ERROR_HEADER               = "lava-identified-node-error"
 	LAVAP_VERSION_HEADER_NAME                       = "Lavap-Version"
 	LAVA_CONSUMER_PROCESS_GUID                      = "lava-consumer-process-guid"

diff --git a/protocol/rpcconsumer/rpcconsumer_server.go b/protocol/rpcconsumer/rpcconsumer_server.go
@@ -1306,6 +1306,15 @@ func (rpccs *RPCConsumerServer) appendHeadersToRelayResult(ctx context.Context,
 			})
 	}
 
+	// add stateful API (hanging, transactions)
+	if protocolMessage.GetApi().Category.Stateful == common.CONSISTENCY_SELECT_ALL_PROVIDERS {
+		metadataReply = append(metadataReply,
+			pairingtypes.Metadata{
+				Name:  common.STATEFUL_API_HEADER,
+				Value: "true",
+			})
+	}
+
 	// add user requested API
 	metadataReply = append(metadataReply,
 		pairingtypes.Metadata{