disable payload batch endpoint for p2p only

kadena-io · Aug 25, 2024 · 34df7ad · 34df7ad
1 parent ea6734d
commit 34df7ad
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/Chainweb/Payload/RestAPI.hs b/src/Chainweb/Payload/RestAPI.hs
@@ -87,7 +87,7 @@ newtype PayloadBatchLimit = PayloadBatchLimit Natural
     deriving newtype (Ord, Enum, Num, Real, Integral, ToJSON, FromJSON)
 
 p2pPayloadBatchLimit :: PayloadBatchLimit
-p2pPayloadBatchLimit = 20
+p2pPayloadBatchLimit = 0
 
 defaultServicePayloadBatchLimit :: PayloadBatchLimit
 defaultServicePayloadBatchLimit = 1000

diff --git a/src/Chainweb/Payload/RestAPI/Server.hs b/src/Chainweb/Payload/RestAPI/Server.hs
@@ -78,7 +78,7 @@ payloadHandler db k mh = liftIO (lookupPayloadDataWithHeight db mh k) >>= \case
         [ "reason" .= ("key not found" :: String)
         , "key" .= k
         ]
-    Just e -> return e
+    Just e -> return e 
 
 -- -------------------------------------------------------------------------- --
 -- POST Payload Batch Handler
@@ -89,8 +89,12 @@ payloadBatchHandler
     -> PayloadDb tbl
     -> BatchBody
     -> Handler PayloadDataList
-payloadBatchHandler _batchLimit _db _ks =
-    throwError (ServerError 404 "" "" [])
+payloadBatchHandler batchLimit db ks
+  = liftIO (PayloadDataList . catMaybes <$> lookupPayloadDataWithHeightBatch db ks')
+  where
+      limit = take (int batchLimit)
+      ks' | WithoutHeights xs <- ks = limit (fmap (Nothing,) xs)
+          | WithHeights    xs <- ks = limit (fmap (over _1 Just) xs)
 
 -- -------------------------------------------------------------------------- --
 -- GET Outputs Handler