tiny_work_service 1.3.0 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 327188d9aeb076b114aa5915bda2a4a50516b070152df17f8242f9fe09548d67
-  data.tar.gz: fe76da5717bea63e4ab09ff69663ba8209984024fbf50af412b20d96d99f61bf
+  metadata.gz: 3c407366b783cb5229c15a319b4b39d38c1b505b200dde5ea2fa581b2e5d93bc
+  data.tar.gz: 4d30dbe1bbdcf2ce1b91d989f57e99220dbefbcf5665d41da7594901c584e85f
 SHA512:
-  metadata.gz: 7ca1d978cb9aa4a88f61970e3ea675cd58da4a55c20e0fb3121b73d40e15eef84badd27d3ba13b33f06d81e09ab5780e78448c83a58dcfcbfdfe65db9e618774
-  data.tar.gz: ca2f7f1300eb9093ace4657fefa6beba346f4fb9f9284398f4f84ccb7f26324f25b8e6ccd032097d018887511d598b73eaae8dab2015c5441c6c1ec48a5ba6d2
+  metadata.gz: 9d9443292f19c5eda19c346e440a1629ef9d37c9adb71ba6fdf5a1043672b16e728a71e79a1692a2168b00b5da2f1b3762fc1c057433fbe3d5e12146ddcae31b
+  data.tar.gz: 4b981c8db0d0ffad7fe20a744c361b776e66fa5029cc72c8feb95757a2643121012f84d2c5a8a72237140b1014261f58bbbcfe64940e8135b17d3e98e874a2a1

data/lib/tiny_work_service.rb

@@ -1,22 +1,77 @@
+require 'time'
 require 'tiny_tcp_service'
 
 # usage:
-#  s = TinyWorkService.new(1234)
+#  s = TinyWorkService.new(1234, 'TinyWorkService')
 #  s.stop!
 class TinyWorkService
-  def initialize(port, label='TinyWorkService')
+  attr_reader :jobs_enqueued,
+              :jobs_dequeued,
+              :jobs_per_minute,
+              :jobs_per_hour
+
+  def initialize(port, label)
     @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
+    @jobs_per_hour = 0
+
+    # status printing thread
     Thread.new do
+      print "\e[?25l" # hide cursor
       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 "\e[1;1H"
+        puts "#{DateTime.now.iso8601}\e[K"
+        puts "#{@label} on port #{port}\e[K"
+        puts "workers :#{@service.num_clients.to_s.rjust(10)}\e[K"
+        puts "queue   :#{@jobs.length.to_s.rjust(10)}\e[K"
+        puts "jobs/m  :#{@jobs_per_minute.to_s.rjust(10)}\e[K"
+        print "jobs/h  :#{@jobs_per_hour.to_s.rjust(10)}\e[K"
         sleep 0.5
       end
+      print "\e[?25h" # show cursor
+    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
 
@@ -42,13 +97,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: 2.0.0
 platform: ruby
 authors:
 - Jeff Lunt
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-05 00:00:00.000000000 Z
+date: 2023-11-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: tiny_tcp_service