tiny_info_service 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ba7ae22876f2c605f0c0bc24eeba58c3fa36600fa6cb4f38314b404ae5a1c441
+  data.tar.gz: d2ef61e2848951688464485b3ef4918f9b612fc6829e5e7cfc9c1b02198d9f6a
+SHA512:
+  metadata.gz: c616b8321f4550335c4e3036d3f9da8b6ffc67e2e053c763a5db14d99fdae49c57a17c544f2713b8c0efbad1717c2b44e1cb1e739934dd386c41a284bf999d13
+  data.tar.gz: d3bbd7d4ceff922a66a3622506fc36fc994927ad48a6dd54d9e30a425e236b8ca053de2da932ed9bf09852f59cbbec8343b433404f4074d500510ea6c7159764

data/lib/tiny_info_service.rb

@@ -0,0 +1,124 @@
+require 'time'
+require 'tiny_tcp_service'
+
+# usage:
+#  s = TinyInfoService.new(
+#    1234,
+#    'TinyInfoService',
+#    2,
+#    {
+#      'uptime': {
+#        'kind': 'shell_cmd',
+#        'shell_cmd': 'uptime'
+#      }
+#    }
+#  )
+#  s.stop!
+class TinyInfoService
+  TIME_MULTIPLIERS = {
+    's' => 1,
+    'm' => 60,
+    'h' => 60 * 60,
+    'd' => 60 * 60 * 24
+  }
+
+  def initialize(port, label, refresh_interval_in_seconds, config)
+    @service = TinyTCPService.new(port)
+    @service.msg_handler = self
+    @label = label
+    @msg_counter = 0
+    @err_counter = 0
+    @label = label
+    @label_and_port = "#{@label}:#{port}"
+    @config = config
+    @config.each do |key, details|
+      @config[key]['cache'] = _parse_cache(details['cache']) if details['cache']
+    end
+
+    @cache = {}
+    @cache_hits = 0
+
+    # status printing thread
+    Thread.new do
+      start_time = Time.now
+      print "\e[?25l" # hide cursor
+      loop do
+        break unless @service.running?
+        sec_runtime = Time.now - start_time
+        human_runtime = "%02d:%02d:%02d" % [sec_runtime / 3600, sec_runtime / 60 % 60, sec_runtime % 60]
+
+        print "\e[1;1H"
+        puts "label/port: #{@label_and_port.rjust(28)}\e[K"
+        puts "time      : #{DateTime.now.iso8601.rjust(28)}\e[K"
+        puts "runtime   : #{human_runtime.rjust(28)}\e[K"
+        puts "clients   : #{@service.num_clients.to_s.rjust(28)}\e[K"
+        puts "msgs      : #{@msg_counter.to_s.rjust(28)}\e[K"
+        puts "errs      : #{@err_counter.to_s.rjust(28)}\e[K"
+
+        sleep refresh_interval_in_seconds
+      end
+      print "\e[?25h" # show cursor
+    end
+  end
+
+  # interface for TinyTCPService
+  def call(m)
+    raise TinyTCPService::BadClient.new("nil message") if m.nil?
+
+    case
+    when m[0] == '?'
+      # return list of keys
+      req = m[1..]
+
+      if @config.keys.include?(req)
+        @msg_counter += 1
+        case @config[req]['kind']
+        when 'shell_cmd'
+          cached_value = @cache[req]['value'] if  @config[req]['cache'] &&
+                                                  @cache.dig(req, 'value') &&
+                                                  Time.now - @cache[req]['time'] < @config[req]['cache']
+
+          if cached_value
+            @cache_hits += 1
+            return cached_value
+          end
+
+          result = `#{@config[req]['shell_cmd']}`
+
+          if $?.exitstatus == 0
+            t = @config[req]['transform']
+            final = t ? eval("result#{t}") : result
+
+            @cache[req] = {
+              'value' => final,
+              'time' => Time.now
+            }
+
+            final
+          else
+            _error!
+          end
+        else
+          _error!
+        end
+      else
+        _error!
+      end
+    else
+      _error!
+    end
+  end
+
+  def join
+    @service.join
+  end
+
+  def _parse_cache(cache_value)
+    cache_value[..-2].to_f * TIME_MULTIPLIERS[cache_value[-1]]
+  end
+
+  def _error!
+    @err_counter += 1
+    'err'
+  end
+end

metadata

@@ -0,0 +1,57 @@
+--- !ruby/object:Gem::Specification
+name: tiny_info_service
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Jeff Lunt
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-01-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: tiny_tcp_service
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: uses the tiny_tcp_service gem to implement a system information service
+email: jefflunt@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tiny_info_service.rb
+homepage: https://github.com/jefflunt/tiny_info_serivce
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.1
+signing_key:
+specification_version: 4
+summary: uses the tiny_tcp_service gem to implement a system information service
+test_files: []