Add additional SW internal trace logging

Author: georgeee

src/lib/work_selector/intf.ml

@@ -117,7 +117,8 @@ module type Lib_intf = sig
 
     (** [mark_scheduled t work] Mark [work] as scheduled in [t] *)
     val mark_scheduled :
-         t
+         logger:Logger.t
+      -> t
       -> ( Transaction_witness.t
          , Ledger_proof.Cached.t )
          Snark_work_lib.Work.Single.Spec.t

src/lib/work_selector/random.ml

@@ -1,14 +1,14 @@
 open Core_kernel
 
 module Make (Lib : Intf.Lib_intf) = struct
-  let work ~snark_pool ~fee ~logger:_ (state : Lib.State.t) =
+  let work ~snark_pool ~fee ~logger (state : Lib.State.t) =
     match Lib.State.all_unscheduled_expensive_works ~snark_pool ~fee state with
     | [] ->
         None
     | expensive_work ->
         let i = Random.int (List.length expensive_work) in
         let x = List.nth_exn expensive_work i in
-        Lib.State.mark_scheduled state x ;
+        Lib.State.mark_scheduled ~logger state x ;
         Some x
 end
 

src/lib/work_selector/random_offset.ml

@@ -42,14 +42,14 @@ module Make (Lib : Intf.Lib_intf) = struct
         List.nth_exn expensive_work !offset
   end
 
-  let work ~snark_pool ~fee ~logger:_ (state : Lib.State.t) =
+  let work ~snark_pool ~fee ~logger (state : Lib.State.t) =
     match Lib.State.all_unscheduled_expensive_works ~snark_pool ~fee state with
     | [] ->
         None
     | expensive_work ->
         Offset.update ~new_length:(List.length expensive_work) ;
         let x = Offset.get_nth expensive_work in
-        Lib.State.mark_scheduled state x ;
+        Lib.State.mark_scheduled ~logger state x ;
         Some x
 end
 

src/lib/work_selector/sequence.ml

@@ -1,10 +1,10 @@
 module Make (Lib : Intf.Lib_intf) = struct
-  let work ~snark_pool ~fee ~logger:_ (state : Lib.State.t) =
+  let work ~snark_pool ~fee ~logger (state : Lib.State.t) =
     match Lib.State.all_unscheduled_expensive_works ~snark_pool ~fee state with
     | [] ->
         None
     | x :: _ ->
-        Lib.State.mark_scheduled state x ;
+        Lib.State.mark_scheduled ~logger state x ;
         Some x
 end
 

src/lib/work_selector/work_lib.ml

@@ -76,6 +76,14 @@ module Make (Inputs : Intf.Inputs_intf) = struct
                         List.map ~f:Job_key.of_job t.available_jobs
                         |> Job_key.Set.of_list
                       in
+                      (* Log to internal trace all of the newly available jobs. *)
+                      Job_key.Set.iter new_job_keys ~f:(fun job_key ->
+                          [%log internal] "Snark_work_requested"
+                            ~metadata:
+                              [ ( "work_ids"
+                                , Transaction_snark_work.Statement.compact_json
+                                    job_key )
+                              ] ) ;
                       t.jobs_scheduled <-
                         Job_key.Set.inter t.jobs_scheduled new_job_keys ) ;
                   Deferred.unit )
@@ -84,10 +92,14 @@ module Make (Inputs : Intf.Inputs_intf) = struct
       |> Deferred.don't_wait_for ;
       t
 
-    let mark_scheduled t x =
-      t.jobs_scheduled <-
-        Job_key.Set.add t.jobs_scheduled
-          (One_or_two.map ~f:Work_spec.statement x)
+    let mark_scheduled ~logger t job =
+      let statement = One_or_two.map ~f:Work_spec.statement job in
+      (* Log to internal trace all of the newly available jobs. *)
+      [%log internal] "Snark_work_scheduled"
+        ~metadata:
+          [ ("work_ids", Transaction_snark_work.Statement.compact_json statement)
+          ] ;
+      t.jobs_scheduled <- Job_key.Set.add t.jobs_scheduled statement
 
     let does_not_have_better_fee ~snark_pool ~fee
         (statements : Inputs.Transaction_snark_work.Statement.t) : bool =