leakybucket 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7fcc79980b6028227e6bb9da5cdbad37a08611e0
-  data.tar.gz: b3b788992af595d277bc4c4c9c83be4f9624c50d
+  metadata.gz: 56ec597997603d5c17fef41f9bda0a86d92ea1db
+  data.tar.gz: 49140040673bd82d017ac6993e18f939ad2de476
 SHA512:
-  metadata.gz: b919b59f338b30ac0842f441276efabdaa1d4e8d9b0d5150aa9a989ef79713a7847e4db82caf4f2aeeab6898106711466e064bc1bd57b5cd266d5473ff9eae64
-  data.tar.gz: 8de1698298a0ca5ecd23f1109f4477bc4b6b9fcbd5a2c8fa2cf04cb46bc19ca1aa21b973cc7f392886819120a22d24a4fa464500aa3469c75146f00d71256af7
+  metadata.gz: c51a4b71a5f812c157e6bb5a1662ad68f7703cb9bf6c9a715a5c654ee8fa39755b63f232280ba4291e41f039d63f474d6e77a213cbbdf51dd8d1d03d791294a6
+  data.tar.gz: e7a42103e32ef13723dd83647df233054f89ea6fe4a875b5fd9c012dd16f47d2f70f0c3ccef9872c95c981aa1905285f96b599c8dd0b2c21a641284869b9c2b4

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    leakybucket (0.0.1)
+    leakybucket (0.0.2)
 
 GEM
   remote: https://rubygems.org/

data/lib/leakybucket/bucket.rb

@@ -1,10 +1,11 @@
 class Leakybucket::Bucket
 
-  attr_accessor :limit, :value, :leaking_callback
+  attr_accessor :limit, :value, :leaking_callback, :key
 
-  def initialize(options = {})
-    self.limit = options[:limit] || default_options[:limit]
-    self.value = limit
+  def initialize(options = {}, key = '')
+    self.limit = default_options.merge(options)[:limit]
+    self.value = self.limit
+    self.key = key
   end
 
   def default_options

data/lib/leakybucket/manager.rb

@@ -0,0 +1,26 @@
+require 'securerandom'
+
+class Leakybucket::Manager
+
+  attr_accessor :buckets, :default_options
+
+  def initialize(default_options = {})
+    self.default_options = default_options
+    self.buckets = {}
+  end
+
+  def generate_key
+    SecureRandom.uuid
+  end
+
+  def create_bucket(options = {})
+    key = generate_key
+    b = Leakybucket::Bucket.new(default_options.merge(options), key)
+    buckets[key] = b
+    b
+  end
+
+  def remove_bucket(key)
+    buckets.delete(key)
+  end
+end

data/lib/leakybucket/version.rb

@@ -1,3 +1,3 @@
 module Leakybucket
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/leakybucket.rb

@@ -1,5 +1,6 @@
 require 'leakybucket/version'
 require 'leakybucket/bucket'
+require 'leakybucket/manager'
 
 module Leakybucket
 end

data/spec/{leakybucket_spec.rb → leakybucket_bucket_spec.rb}

@@ -54,3 +54,4 @@ describe Leakybucket::Bucket do
     out.should eql('leaking with value -2')
   end
 end
+

data/spec/leakybucket_manager_spec.rb

@@ -0,0 +1,27 @@
+require 'leakybucket'
+
+describe Leakybucket::Bucket do
+  it 'should create buckets' do
+    m = Leakybucket::Manager.new
+    b1 = m.create_bucket(limit: 5)
+    b1.limit.should eql(5)
+    m.buckets.size.should eql(1)
+    m.buckets[b1.key].should eql(b1)
+
+    b2 = m.create_bucket(limit: 3)
+    b2.limit.should eql(3)
+    m.buckets.size.should eql(2)
+    b1.key.should_not eql(b2.key)
+    m.buckets[b2.key].should eql(b2)
+    m.buckets[b2.key].should_not eql(b1)
+  end
+
+  it 'should remove buckets' do
+    m = Leakybucket::Manager.new
+    b1 = m.create_bucket(limit: 5)
+    m.buckets.size.should eql(1)
+    b2 = m.remove_bucket(b1.key)
+    m.buckets.size.should eql(0)
+    b1.should eq(b2)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: leakybucket
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Lukas Elmer
@@ -69,8 +69,10 @@ files:
 - leakybucket.gemspec
 - lib/leakybucket.rb
 - lib/leakybucket/bucket.rb
+- lib/leakybucket/manager.rb
 - lib/leakybucket/version.rb
-- spec/leakybucket_spec.rb
+- spec/leakybucket_bucket_spec.rb
+- spec/leakybucket_manager_spec.rb
 homepage: https://github.com/lukaselmer/leakybucket
 licenses:
 - MIT
@@ -96,4 +98,5 @@ signing_key:
 specification_version: 4
 summary: Provides an API to manage leaky buckets.
 test_files:
-- spec/leakybucket_spec.rb
+- spec/leakybucket_bucket_spec.rb
+- spec/leakybucket_manager_spec.rb