random-port 0.3.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1cd0eda08b292d0482e433b081e984bde31aed33b758faf81815d7e6bebab95a
-  data.tar.gz: cec78b2727717531359e826f1e650981dfc6c0bc37fcb914ce503424824c66ae
+  metadata.gz: c70d46732ca42efae0aceaf8abc0c366b5f96d7742808d1f9b6c20f21ec560b7
+  data.tar.gz: dc873aa7fcecfd86f40ea5cdde11569b863fad043f584b4a463f9fc46a627dc0
 SHA512:
-  metadata.gz: ca08ad8ead3c38bcc22431d5526bad4142ecfdcc3341af242a162872d3caa33261bbe4618d6c350f20b18e6401a32e69ab5da2af2e4631957a30ef679dd26cac
-  data.tar.gz: d349dce4ab4e7171b15526f8b731d95874f3311eebbc247970210e5581d694aee1587e12c5044e52f08e1785dece7441b3ba4263bd7c10cc4983f5efa22b44a9
+  metadata.gz: adb1ecb591ebb526b110b5c478a6a54db907335a2d89ae5f58d542ea9071f85d6a258b4bc046a3ef33fb0a73b5bb32145cb6ec6de3840db108a1cb211f37d565
+  data.tar.gz: d1ab0249446805c014fcc2bdf2ecce929c69149af8f798dedfcc55129461ae567ad9fc912761e8cb4971b36bc4cb1d13decacc26b6e26a14bbe36a9b2fd14500

data/.rultor.yml

@@ -3,7 +3,7 @@ assets:
 install: |-
   export GEM_HOME=~/.ruby
   export GEM_PATH=$GEM_HOME:$GEM_PATH
-  bundle install
+  bundle install --no-color
 release:
   script: |-
     bundle exec rake

data/lib/random-port/pool.rb

@@ -35,6 +35,9 @@ module RandomPort
   #    # to the pool afterwards.
   #  end
   #
+  # You can specify the maximum amount of ports to acquire, using +limit+.
+  # If more acquiring requests will arrive, an exception will be raised.
+  #
   # The class is thread-safe, by default. You can configure it to be
   # not-thread-safe, using optional <tt>sync</tt> argument of the constructor.
   #
@@ -42,10 +45,15 @@ module RandomPort
   # Copyright:: Copyright (c) 2018 Yegor Bugayenko
   # License:: MIT
   class Pool
-    def initialize(sync: false)
+    # If can't acquire by time out.
+    class Timeout < StandardError; end
+
+    # Ctor.
+    def initialize(sync: false, limit: 65_536)
       @ports = []
       @sync = sync
       @monitor = Monitor.new
+      @limit = limit
     end
 
     # Application wide pool of ports
@@ -62,13 +70,21 @@ module RandomPort
     end
 
     # Acquire a new random TCP port.
-    def acquire
+    #
+    # You can specify the amount of seconds to wait until a new port
+    # is available.
+    def acquire(timeout: 4)
+      start = Time.now
       loop do
+        raise Timeout if Time.now > start + timeout
+        next if @ports.count >= @limit
         server = TCPServer.new('127.0.0.1', 0)
         port = server.addr[1]
         server.close
-        next if @ports.include?(port)
-        safe { @ports << port }
+        safe do
+          next if @ports.include?(port)
+          @ports << port
+        end
         return port unless block_given?
         begin
           return yield port

data/random-port.gemspec

@@ -31,7 +31,7 @@ Gem::Specification.new do |s|
   s.rubygems_version = '2.3.3'
   s.required_ruby_version = '>=2.3'
   s.name = 'random-port'
-  s.version = '0.3.2'
+  s.version = '0.4.0'
   s.license = 'MIT'
   s.summary = 'Random TCP port'
   s.description = 'Reserves a random TCP port'

data/test/test_pool.rb

@@ -71,6 +71,14 @@ module RandomPort
       assert_equal(total, numbers.uniq.count)
     end
 
+    def test_raises_when_too_many
+      pool = Pool.new(limit: 1)
+      pool.acquire
+      assert_raises Pool::Timeout do
+        pool.acquire(timeout: 0.1)
+      end
+    end
+
     def test_acquires_unique_numbers_in_no_sync_mode
       total = 25
       pool = Pool.new(sync: false)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: random-port
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.4.0
 platform: ruby
 authors:
 - Yegor Bugayenko
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-08-05 00:00:00.000000000 Z
+date: 2019-08-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: codecov