Adding 'server_on_port' method to socketless server maptball/socketless

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/chef_zero/socketless_server_map.rb b/lib/chef_zero/socketless_server_map.rb
index 531dbbb..99351a1 100644
--- a/lib/chef_zero/socketless_server_map.rb
+++ b/lib/chef_zero/socketless_server_map.rb
@@ -33,6 +33,10 @@ module ChefZero
       instance.request(port, request_env)
     end
 
+    def self.server_on_port(port)
+      instance.server_on_port(port)
+    end
+
     MUTEX = Mutex.new
 
     include Singleton
@@ -67,6 +71,10 @@ module ChefZero
       @servers_by_port.key?(port)
     end
 
+    def server_on_port(port)
+      @servers_by_port[port]
+    end
+
     def deregister(port)
       MUTEX.synchronize do
         @servers_by_port.delete(port)