GraphQL: unify code for pending work functions

Author: anne-laure-s

src/lib/mina_graphql/mina_graphql.ml

@@ -2198,19 +2198,21 @@ module Queries = struct
         Mina_lib.snark_pool mina |> Network_pool.Snark_pool.resource_pool
         |> Network_pool.Snark_pool.Resource_pool.all_completed_work )
 
+  (* Auxiliary function to fetch all pending work *)
+  let get_pending_work mina =
+    let snark_job_state = Mina_lib.snark_job_state mina in
+    let snark_pool = Mina_lib.snark_pool mina in
+    let fee_opt =
+      Mina_lib.(
+        Option.map (snark_worker_key mina) ~f:(fun _ -> snark_work_fee mina))
+    in
+    Work_selector.pending_work_statements ~snark_pool ~fee_opt snark_job_state
+
   let pending_snark_work =
     field "pendingSnarkWork" ~doc:"List of snark works that are yet to be done"
       ~args:Arg.[]
       ~typ:(non_null @@ list @@ non_null Types.pending_work)
-      ~resolve:(fun { ctx = mina; _ } () ->
-        let snark_job_state = Mina_lib.snark_job_state mina in
-        let snark_pool = Mina_lib.snark_pool mina in
-        let fee_opt =
-          Mina_lib.(
-            Option.map (snark_worker_key mina) ~f:(fun _ -> snark_work_fee mina))
-        in
-        Work_selector.pending_work_statements ~snark_pool ~fee_opt
-          snark_job_state )
+      ~resolve:(fun { ctx = mina; _ } () -> get_pending_work mina)
 
   let pending_snark_work_range =
     field "pendingSnarkWorkRange"
@@ -2233,17 +2235,6 @@ module Queries = struct
           ]
       ~typ:(non_null @@ list @@ non_null Types.pending_work)
       ~resolve:(fun { ctx = mina; _ } () start_idx end_idx ->
-        let pending_work =
-          let snark_job_state = Mina_lib.snark_job_state mina in
-          let snark_pool = Mina_lib.snark_pool mina in
-          let fee_opt =
-            Mina_lib.(
-              Option.map (snark_worker_key mina) ~f:(fun _ ->
-                  snark_work_fee mina ))
-          in
-          Work_selector.pending_work_statements ~snark_pool ~fee_opt
-            snark_job_state
-        in
         let start_idx = Unsigned.UInt32.to_int start_idx in
         (* end_idx is max_int in the default case, in that case we want to take
            only the element at start_idx. If [start_idx, end_idx] is not a
@@ -2253,7 +2244,8 @@ module Queries = struct
           else if start_idx > Unsigned.UInt32.to_int end_idx then 0
           else Unsigned.UInt32.to_int end_idx - start_idx + 1
         in
-        try List.sub ~pos:start_idx ~len @@ pending_work with _ -> [] )
+        try List.sub ~pos:start_idx ~len @@ get_pending_work mina with _ -> []
+        )
 
   module SnarkedLedgerMembership = struct
     let resolve_membership :