make sure to kill the process

yet-another-ai · Feb 6, 2024 · b56c2f4 · b56c2f4
1 parent 19ece48
commit b56c2f4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lib/prolog_mqi/prolog_mqi.rb b/lib/prolog_mqi/prolog_mqi.rb
@@ -8,6 +8,12 @@ def initialize(timeout: 5)
       @process = nil
       @socket = nil
       @running = false
+
+      ObjectSpace.define_finalizer(self, self.class.finalize(@process))
+    end
+
+    def self.finalize(process)
+      proc { process&.kill }
     end
 
     def running?
@@ -99,6 +105,7 @@ def stop
       @stdout.close
       @stderr.close
       @process.kill
+      @process = nil
       @running = false
     end
   end