tiny_tcp_service 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7a9d89a4cf5f6578a0f606fe0c24cb81495fe2c96cfea70074670f2e9a3fd3ac
+  data.tar.gz: 50b6169a75272e14a07fbd17fc2b6575a3c8010bc580776ad456ee3786eca5bf
+SHA512:
+  metadata.gz: d80e1a3243cb5102bc776d19759c0c5ffed178ccbf7b96aaa895bf18144675be725fc0b7df7f42b877358b980fd9e16d5588a39bc4434bd6f5231c90630d23ae
+  data.tar.gz: 73fc25a0c4b58609a711694840c3227717412aca82a84623884aa3b1f7539f8e318cb1310913f139dd4114c9010804d7df6ed48ede08450d4c4433156fdb9ece

data/lib/tiny_tcp_service.rb

@@ -0,0 +1,94 @@
+require 'socket'
+
+# usage:
+#  s = TinyTCPService.new(
+#    1234,
+#    ->(m) { puts m }
+#  )
+#
+#  s.start!     # everything runs in background threads
+#  s.stop!      # gracefully shutdown the server
+class TinyTCPService
+  def initialize(port, msg_handler)
+    @port = port
+    @msg_handler = msg_handler
+
+    @server = TCPServer.new(port)
+    @clients = []
+    @running = false
+    @error_handlers = {}
+  end
+
+  # returns true if the server is running
+  # false otherwise
+  def running?
+    @running
+  end
+
+  # add the error handler and block for the specified class
+  #
+  # you can assume that the local variable name of the error will be `e'
+  def add_error_handler(klass, block)
+    @error_handlers[klass] = block
+  end
+
+  # remove the error handler associated with klass
+  def remove_error_handler(klass)
+    @error_handlers.delete(klass)
+  end
+
+  # returns the number of connected clients
+  def num_clients
+    @clients.length
+  end
+
+  # starts the server
+  def start!
+    return if running?
+    @running = true
+
+    # client accept thread
+    Thread.new do |t|
+      loop do
+        break unless running?
+        @clients << @server.accept
+      end
+
+      @clients.each{|c| c.close if c && !c.closed? }
+      @server.close
+    end
+
+    # service thread
+    Thread.new do |t|
+      loop do
+        break unless running?
+
+        readable, _, errored = IO.select(@clients, nil, @clients, 1)
+        readable&.each do |client|
+          begin
+            @msg_handler.call(client.gets.chomp)
+          rescue => e
+            handler = @error_handlers[e.class]
+
+            if handler
+              handler.call(e)
+            else
+              stop!
+              raise e unless handler
+            end
+          end
+        end
+
+        errored&.each do |client|
+          @clients.delete(client)
+          client.close if client && !client.closed?
+        end
+      end
+    end
+  end
+
+  # stops the server gracefully
+  def stop!
+    @running = false
+  end
+end

metadata

@@ -0,0 +1,43 @@
+--- !ruby/object:Gem::Specification
+name: tiny_tcp_service
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Jeff Lunt
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-10-29 00:00:00.000000000 Z
+dependencies: []
+description: a tiny TCP service with automated client lifecycle
+email: jefflunt@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tiny_tcp_service.rb
+homepage: https://github.com/jefflunt/tiny_tcp_service
+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.3.7
+signing_key:
+specification_version: 4
+summary: a tiny TCP service with automated client lifecycle
+test_files: []