enviroblyd 0.4.6 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8467f960365419e9f02f14d3d3186a511787dff28798fd8883ac328a2443e9c
-  data.tar.gz: 2f39dbca93e49cf69ab2bb9aaed76fe89ed2ebd7867baebba640c8ea665082ae
+  metadata.gz: 329d9e7586d58e8b529d2a94d46be76130efb24bf688a4d53efe17dfeaf3e7e0
+  data.tar.gz: 47cf08b7ecf7231460aeed688aa76705abff6cdcfdf770148f9f86f378ee513e
 SHA512:
-  metadata.gz: dc3ea42ebc9125f88ec1828b4ebdf9032ba2482d1db359860b9b7c117ae0cb12eb57369531bd760cb54b2563f64c63e3afb89a304a574c23fa19bc7ff4334214
-  data.tar.gz: 3c9baed9045d458595627b186f5be800abe5da2df36beea47e22fd478bc28e25e984814de9731118040b4a65f16fabea10923d06daaa0f419639ace81f0be67f
+  metadata.gz: f57916ca0d0615cd05cffcf15527f506a8ca60a47cec03f5d9ea4262b4ce0b298ae764cace907c764c4de9307f870659657610629c98e7ebd8d8c81388000623
+  data.tar.gz: a260266cb641e0b7e99d2e35f21cde3fcd82f80570782ce5628c4081c4e6f9773ee35ea4d660514267407ccc000146b3db1b504b60370518653f2b61c13ba719

data/bin/enviroblyd

@@ -3,4 +3,12 @@
 
 require "enviroblyd"
 
-Enviroblyd::Daemon.start
+daemon = Enviroblyd::Daemon.new
+
+trap "INT" do
+  puts "Gracefully shutting down..."
+  daemon&.shutdown
+  exit
+end
+
+daemon.listen

data/lib/enviroblyd/daemon.rb

@@ -6,29 +6,25 @@ class Enviroblyd::Daemon
   MAX_MESSAGE_SIZE = 6000 # bytes
   PORT = ENV.fetch("ENVIROBLYD_PORT", 63106).to_i
 
-  def self.start
+  def initialize
     imds = Enviroblyd::IMDS.new
-    host = imds.private_ipv4
-    daemon = new(host)
-    daemon.listen
-  end
-
-  def initialize(host)
-    @host = host
+    @host = imds.private_ipv4
+    @threads = []
+    @shutdown = false
   end
 
   def listen
-    server = TCPServer.new @host, PORT
+    @server = TCPServer.new @host, PORT
     puts "Listening on #{@host}:#{PORT}"
     Enviroblyd::Web.register
 
-    loop do
-      Thread.start(server.accept) do |client|
+    until @shutdown do
+      @threads << Thread.start(@server.accept) do |client|
         message = client.recv(MAX_MESSAGE_SIZE)
         command = Enviroblyd::Command.new message
 
         unless command.valid?
-          $stderr.puts "Invalid message received: #{message}"
+          puts "Invalid message received: #{message}"
           client.puts "Invalid message"
           next
         end
@@ -38,11 +34,23 @@ class Enviroblyd::Daemon
         command.run
       ensure
         client.close
-        command = nil
-        message = nil
-        client = nil
-        GC.start
       end
+
+      delete_dead_threads
+      GC.start
+      GC.compact
     end
   end
+
+  def shutdown
+    @threads.each(&:join)
+  end
+
+  private
+    def delete_dead_threads
+      @threads.each do |thread|
+        next if thread.alive?
+        @threads.delete thread
+      end
+    end
 end

data/lib/enviroblyd/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Enviroblyd
-  VERSION = "0.4.6"
+  VERSION = "0.5.0"
 end

data/lib/enviroblyd/web.rb

@@ -33,8 +33,8 @@ class Enviroblyd::Web
 
     def http(url, type: Net::HTTP::Get, params: nil, headers: {}, retry_interval: 3, retries: 10, backoff: :exponential, tries: 1)
       if retries <= tries
-        $stderr.puts "Retried #{tries} times. Aborting."
-        exit 1
+        puts "Retried #{url} #{tries} times. Aborting."
+        return
       end
 
       uri = URI(url)
@@ -56,7 +56,7 @@ class Enviroblyd::Web
 
       if response == :retry || (500..599).include?(response.code.to_i)
         sleep_time = (backoff == :exponential) ? (retry_interval * tries) : retry_interval
-        $stderr.puts "Retry #{uri} in #{sleep_time}s"
+        puts "Retry #{uri} in #{sleep_time}s"
         sleep sleep_time
         http(url, type:, params:, retry_interval:, retries:, backoff:, tries: (tries + 1))
       else

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: enviroblyd
 version: !ruby/object:Gem::Version
-  version: 0.4.6
+  version: 0.5.0
 platform: ruby
 authors:
 - Robert Starsi