type_balancer 0.1.3 → 0.2.0

data/lib/type_balancer/type_extractor_registry.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module TypeBalancer
+  # Registry to memoize TypeExtractor per type_field in thread/request scope
+  class TypeExtractorRegistry
+    STORAGE_KEY = :type_balancer_extractors
+
+    def self.get(type_field)
+      cache[type_field] ||= TypeExtractor.new(type_field)
+    end
+
+    def self.clear!
+      Thread.current[STORAGE_KEY] = nil
+    end
+
+    def self.cache
+      Thread.current[STORAGE_KEY] ||= {}
+    end
+  end
+end

data/lib/type_balancer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TypeBalancer
-  VERSION = '0.1.3'
+  VERSION = '0.2.0'
 end

data/lib/type_balancer.rb

@@ -1,11 +1,15 @@
 # frozen_string_literal: true
 
 require 'type_balancer/version'
-require 'type_balancer/calculator'
 require_relative 'type_balancer/balancer'
 require_relative 'type_balancer/ratio_calculator'
 require_relative 'type_balancer/batch_processing'
-require 'type_balancer/position_calculator'
+require_relative 'type_balancer/position_calculator'
+require_relative 'type_balancer/type_extractor'
+require_relative 'type_balancer/type_extractor_registry'
+require_relative 'type_balancer/strategies/base_strategy'
+require_relative 'type_balancer/strategies/sliding_window_strategy'
+require_relative 'type_balancer/strategy_factory'
 
 module TypeBalancer
   class Error < StandardError; end
@@ -14,50 +18,62 @@ module TypeBalancer
   class EmptyCollectionError < Error; end
   class InvalidTypeError < Error; end
 
+  # Register default strategies
+  StrategyFactory.register(:sliding_window, Strategies::SlidingWindowStrategy)
+
   # Load Ruby implementations
   require_relative 'type_balancer/distribution_calculator'
   require_relative 'type_balancer/ordered_collection_manager'
-  require_relative 'type_balancer/alternating_filler'
-  require_relative 'type_balancer/sequential_filler'
+  require_relative 'type_balancer/type_extractor'
+  require_relative 'type_balancer/type_extractor_registry'
+  require_relative 'type_balancer/ratio_calculator'
+  require_relative 'type_balancer/position_calculator'
   require_relative 'type_balancer/distributor'
+  require_relative 'type_balancer/sequential_filler'
+  require_relative 'type_balancer/alternating_filler'
+  require_relative 'type_balancer/balancer'
+  require_relative 'type_balancer/batch_processing'
+  require_relative 'type_balancer/calculator'
 
   def self.calculate_positions(total_count:, ratio:, available_items: nil)
-    Distributor.calculate_target_positions(
+    PositionCalculator.calculate_positions(
       total_count: total_count,
       ratio: ratio,
-      available_positions: available_items
+      available_items: available_items
     )
   end
 
-  def self.balance(items, type_field: :type, type_order: nil)
+  def self.balance(items, type_field: :type, type_order: nil, strategy: nil, **strategy_options)
     # Input validation
     raise EmptyCollectionError, 'Collection cannot be empty' if items.empty?
 
-    # Extract and validate types
-    types = extract_types(items, type_field)
-    raise Error, "Invalid type field: #{type_field}" if types.empty?
-
-    # Group items by type
-    items.group_by { |item| extract_type(item, type_field) }
+    # Use centralized extractor
+    extractor = TypeExtractorRegistry.get(type_field)
+    begin
+      types = extractor.extract_types(items)
+      raise Error, "Invalid type field: #{type_field}" if types.empty?
+    rescue Error => e
+      raise Error, "Cannot access type field '#{type_field}': #{e.message}"
+    end
 
-    # Initialize balancer with type order if provided
-    balancer = Balancer.new(types, type_order: type_order)
+    # Create calculator with strategy options
+    calculator = Calculator.new(
+      items,
+      type_field: type_field,
+      types: type_order || types,
+      strategy: strategy,
+      **strategy_options
+    )
 
     # Balance items
-    balancer.call(items)
+    calculator.call
   end
 
+  # Backward compatibility methods
   def self.extract_types(items, type_field)
-    items.map { |item| extract_type(item, type_field) }.uniq
-  end
-
-  def self.extract_type(item, type_field)
-    if item.is_a?(Hash)
-      item[type_field] || item[type_field.to_s]
-    else
-      item.public_send(type_field)
-    end
-  rescue NoMethodError
-    nil
+    TypeExtractorRegistry.get(type_field).extract_types(items)
+  rescue Error
+    # For backward compatibility, return array with nil for inaccessible type fields
+    [nil]
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: type_balancer
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.0
 platform: ruby
 authors:
 - Carl Smith
 bindir: exe
 cert_chain: []
-date: 2025-04-28 00:00:00.000000000 Z
+date: 2025-05-01 00:00:00.000000000 Z
 dependencies: []
 description: Balances types in collections by ensuring each type appears a similar
   number of times
@@ -43,6 +43,7 @@ files:
 - docs/calculate_positions.md
 - docs/quality.md
 - examples/balance_test_data.yml
+- examples/large_scale_balance_test.rb
 - examples/quality.rb
 - lib/type_balancer.rb
 - lib/type_balancer/alternating_filler.rb
@@ -55,7 +56,12 @@ files:
 - lib/type_balancer/position_calculator.rb
 - lib/type_balancer/ratio_calculator.rb
 - lib/type_balancer/sequential_filler.rb
+- lib/type_balancer/strategies.rb
+- lib/type_balancer/strategies/base_strategy.rb
+- lib/type_balancer/strategies/sliding_window_strategy.rb
+- lib/type_balancer/strategy_factory.rb
 - lib/type_balancer/type_extractor.rb
+- lib/type_balancer/type_extractor_registry.rb
 - lib/type_balancer/version.rb
 - type_balancer.gemspec
 homepage: https://github.com/llwebconsulting/type_balancer