type_balancer_rails 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ba34931f5dfe8f33283f9597b3cd1ae8e4cd408c9985822e701806b24b852e9f
-  data.tar.gz: 22ec570ca04de5382e46d0f92e6c45288f0c934bddbc7c87633adb1f055f5db0
+  metadata.gz: 5ebd00e75d4c63af53f5be038f01f920eb6e65dcf3c91ae079a10b20435d62ce
+  data.tar.gz: 67d328f6ae917d27f88455dce176073dd8da0cf8dd7bb2be460c886c7a56086f
 SHA512:
-  metadata.gz: c7465e3fe1519fc3c025121375e4e20445d7929a5f73f2713ad43223655f8f92ec55911c63b27463e4d925d563f7d15973e5e6d273eb284c212ab8aed48658e6
-  data.tar.gz: d6c122262c389fe477eda2453ccfa546fd5c21a2a5f9bcbf40e3c0ef3aed2022402299f9b42c5a3d26bd00ea7adc79509b8dc4ea967a64861d2411bcec8bbe4d
+  metadata.gz: 4a82e84484c322cd4ee02c92950cf3235daa989b21097c845a5e41a454084afe883ee15444729992e7a63b59318508dbd7f89a15d1abe3d371f8de2cc1e1c551
+  data.tar.gz: e427b4224fbed0bd83dcda7dd2ce97683dc1c7df54a2fb673ce15b9d545ee029f87637fb5a8cff685cb7570f7d0e7aeaf3c82e0a5466906773326a2d9d794486

data/lib/type_balancer/rails/active_record_extension.rb

@@ -19,9 +19,15 @@ module TypeBalancer
       end
 
       class_methods do
-        def balance_by_type(options = {})
+        # Accepts either a symbol (type field) or options hash
+        def balance_by_type(type_field = nil, **options)
+          if type_field.is_a?(Hash)
+            options = type_field
+            type_field = options[:type_field]
+          end
+
           self.type_balancer_options = {
-            type_field: options[:type_field] || :type
+            type_field: type_field || options[:type_field] || :type
           }
 
           return [] unless respond_to?(:all)

data/lib/type_balancer/rails/collection_methods.rb

@@ -23,31 +23,43 @@ module TypeBalancer
         records = to_a
         return empty_relation if records.empty?
 
-        # Get type field from options or model configuration
         type_field = fetch_type_field(options)
-
-        # Balance records using TypeBalancer
-        balanced = TypeBalancer.balance(records, type_field: type_field)
+        balanced   = TypeBalancer.balance(records, type_field: type_field)
         return empty_relation if balanced.nil?
 
-        # Handle pagination if requested
-        if options[:page] || options[:per_page]
-          page = (options[:page] || 1).to_i
-          per_page = (options[:per_page] || 20).to_i
-          offset = (page - 1) * per_page
-          balanced = balanced[offset, per_page] || []
-        end
-
-        # Return as relation
-        TestRelation.new(balanced)
+        paged = apply_pagination(balanced, options)
+        build_result(paged)
       end
 
-      def all_records = to_a
-
       private
 
+      def apply_pagination(records, options)
+        return records unless options[:page] || options[:per_page]
+
+        page      = (options[:page] || 1).to_i
+        per_page  = (options[:per_page] || 20).to_i
+        offset    = (page - 1) * per_page
+        records[offset, per_page] || []
+      end
+
+      def build_result(balanced)
+        if klass.respond_to?(:where) && klass != TestModel
+          ids     = balanced.map(&:id)
+          results = klass.where(id: ids).to_a
+          results_by_id = results.index_by(&:id)
+          ordered = ids.map { |id| results_by_id[id] }
+          self.class.new(ordered)
+        else
+          self.class.new(balanced)
+        end
+      end
+
       def empty_relation
-        TestRelation.new([])
+        if klass.respond_to?(:none)
+          klass.none
+        else
+          []
+        end
       end
 
       def fetch_type_field(options)

data/lib/type_balancer/rails/version.rb

@@ -2,6 +2,6 @@
 
 module TypeBalancer
   module Rails
-    VERSION = '0.1.1'
+    VERSION = '0.1.3'
   end
 end

data/lib/type_balancer_rails.rb

@@ -8,3 +8,5 @@ require 'type_balancer'
 require_relative 'type_balancer/rails/version'
 require_relative 'type_balancer/rails'
 require 'type_balancer/rails/collection_methods'
+
+require 'type_balancer/rails/railtie' if defined?(Rails)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: type_balancer_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Carl Smith
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-26 00:00:00.000000000 Z
+date: 2025-04-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord