tiny_work_service 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 327188d9aeb076b114aa5915bda2a4a50516b070152df17f8242f9fe09548d67
-  data.tar.gz: fe76da5717bea63e4ab09ff69663ba8209984024fbf50af412b20d96d99f61bf
+  metadata.gz: 64ea8638eba6ebc22945d5c80631513c30d8866519b973e04d66a6a247e408c0
+  data.tar.gz: '099b5c7719d958f0674f1449d8da62d9de10fe66720b28aa0078c78f0c72bbf0'
 SHA512:
-  metadata.gz: 7ca1d978cb9aa4a88f61970e3ea675cd58da4a55c20e0fb3121b73d40e15eef84badd27d3ba13b33f06d81e09ab5780e78448c83a58dcfcbfdfe65db9e618774
-  data.tar.gz: ca2f7f1300eb9093ace4657fefa6beba346f4fb9f9284398f4f84ccb7f26324f25b8e6ccd032097d018887511d598b73eaae8dab2015c5441c6c1ec48a5ba6d2
+  metadata.gz: a68228cd855eeeb8943a20b87a12cb0b62f1bb6b9737251376a388eff7e08e5fa783fd2d0f5150eebb8d5c7237349d8abfe82568d59664c0a7140030b86cce15
+  data.tar.gz: b0a7f962ccac5be3a45c0869bae4432336547a411669b8cf3950751d31a67ac841c47adf13e86def1b046686b188f108e4cb79c3a9c8f7296d5dd8aeeb05bab0

data/lib/tiny_work_service.rb

@@ -4,20 +4,66 @@ require 'tiny_tcp_service'
 #  s = TinyWorkService.new(1234)
 #  s.stop!
 class TinyWorkService
+  attr_reader :jobs_enqueued,
+              :jobs_dequeued,
+              :jobs_per_minute,
+              :jobs_per_hour
+
   def initialize(port, label='TinyWorkService')
     @service = TinyTCPService.new(port)
     @service.msg_handler = self
     @jobs = Queue.new
     @label = label
 
+    @jobs_enqueued = 0
+    @jobs_dequeued = 0
+    @jobs_to_track = Queue.new
+    @jobs_dequeued_tracker = []
+
+    @jobs_per_minute = 0.0
+    @jobs_per_hour = 0.0
+
+    # status printing thread
     Thread.new do
       loop do
         break unless @service.running?
 
-        print "\r#{@label}:#{port} #{@jobs.length.to_s.rjust(6)} jobs #{@service.num_clients.to_s.rjust(4)} workers\e[K"
+        print "\r#{@label}:#{port} jobs:#{@jobs.length.to_s} workers:#{@service.num_clients.to_s} jobs/m:#{@jobs_per_minute} jobs/h:#{@jobs_per_hour}\e[K"
         sleep 0.5
       end
     end
+
+    # update stats thread
+    Thread.new do
+      loop do
+        one_minute_ago = Time.now.to_i - 60
+        one_hour_ago = Time.now.to_i - 3600
+
+        # move jobs_to_track into jobs_dequeued_tracker, threadsafe
+        loop do
+          break if @jobs_to_track.length == 0
+          @jobs_dequeued_tracker << @jobs_to_track.shift
+        end
+
+        # remove job tracking times from older than one_hour_ago
+        loop do
+          break if @jobs_dequeued_tracker.empty? || @jobs_dequeued_tracker.first >= one_hour_ago
+          @jobs_dequeued_tracker.shift
+        end
+
+        counter = 0
+        i = -1
+        loop do
+          break if i.abs > @jobs_dequeued_tracker.length || @jobs_dequeued_tracker[i] < one_minute_ago
+          i -= 1
+          counter += 1
+        end
+        @jobs_per_minute = counter
+        @jobs_per_hour = @jobs_dequeued_tracker.count
+
+        sleep 2
+      end
+    end
   end
 
   # interface for TinyTCPService
@@ -42,13 +88,19 @@ class TinyWorkService
 
   # enqueue a job
   def <<(j)
+    @jobs_enqueued += 1
     @jobs << j
+
+    nil
   end
 
   # return the first job in the work queue, if there is one present
   # otherwise, return nil
   def shift
     return nil if @jobs.empty?
+    @jobs_dequeued += 1
+    @jobs_to_track << Time.now.to_i
+
     @jobs.shift
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tiny_work_service
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-05 00:00:00.000000000 Z
+date: 2023-11-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tiny_tcp_service