compact the disksnoop.py example

Signed-off-by: varun-r-mallya <[email protected]>

Author: varun-r-mallya

BCC-Examples/disksnoop.py

@@ -1,11 +1,10 @@
-from vmlinux import struct_request, struct_pt_regs
-from pythonbpf import bpf, section, bpfglobal, compile, map
+from ctypes import c_int32, c_int64, c_uint64
+
+from vmlinux import struct_pt_regs, struct_request
+
+from pythonbpf import bpf, bpfglobal, compile, map, section
 from pythonbpf.helper import ktime
 from pythonbpf.maps import HashMap
-from ctypes import c_int64, c_uint64, c_int32
-
-# Constants
-REQ_WRITE = 1  # from include/linux/blk_types.h
 
 
 @bpf
@@ -17,24 +16,15 @@ def start() -> HashMap:
 @bpf
 @section("kprobe/blk_mq_end_request")
 def trace_completion(ctx: struct_pt_regs) -> c_int64:
-    # Get request pointer from first argument
     req_ptr = ctx.di
     req = struct_request(ctx.di)
-    # Print: data_len, cmd_flags, latency_us
     data_len = req.__data_len
     cmd_flags = req.cmd_flags
-    # Lookup start timestamp
     req_tsp = start.lookup(req_ptr)
     if req_tsp:
-        # Calculate delta in nanoseconds
         delta = ktime() - req_tsp
-
-        # Convert to microseconds for printing
         delta_us = delta // 1000
-
         print(f"{data_len} {cmd_flags:x} {delta_us}\n")
-
-        # Delete the entry
         start.delete(req_ptr)
 
     return c_int64(0)