switchman 3.5.20 → 3.5.22

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 88d7a6f4e778e781d7706281f5330b378a7f7982fe7475a95b76693c1549d146
-  data.tar.gz: e8ebfc94e5a57a56a5a44f2924d3f273fd5424211ead5def892c19d5fca0ba7b
+  metadata.gz: fa5da3b192752676ea8513c8fd05d72e347b10979b7b47ada9cdf008941fc2ac
+  data.tar.gz: 1da3c8eba9d967ca1a3edd550ecc7dd8bb1dea6193be3a752a65e95dc0beb54d
 SHA512:
-  metadata.gz: 983b75007895318e8e1dda6022f755c923e3b6d390ce9112037eb202be9384caaa7f7fd9e633c1e200f75bbd5a2346f9871cd60aed71f1b7b7b5a04f6aa63906
-  data.tar.gz: c91f50e916dc069b75fc8260c1ae52cd64e132e8eb68bb22c182a4291b87bf9a03bea390c8797b50b7b477fc7c82af8670b99e0213e4c5b712e097295912d98c
+  metadata.gz: 7ba322eda93832f04595bb93f52153c61db3df1848404d80daf83e4b77b070f45c2d84cc2b6a5ba90c68cd5fbe26745f7ff2932b1a510ad15559f8234dab9bd2
+  data.tar.gz: 2453c5bccf84d20766340fe60512aec89547d27287832806964688c630f2af5d90070932c49e215f4a6ce209dfb701b1afa181ea1abe4fb8a009c03b55dea88e

data/lib/switchman/active_record/calculations.rb

@@ -106,6 +106,45 @@ module Switchman
         compact_grouped_calculation_rows(rows, opts)
       end
 
+      if ::Rails.version >= "7.1"
+        def ids
+          return super unless klass.sharded_primary_key?
+
+          if loaded?
+            result = records.map do |record|
+              Shard.relative_id_for(record._read_attribute(primary_key),
+                                    record.shard,
+                                    Shard.current(klass.connection_class_for_self))
+            end
+            return @async ? Promise::Complete.new(result) : result
+          end
+
+          if has_include?(primary_key)
+            relation = apply_join_dependency.group(primary_key)
+            return relation.ids
+          end
+
+          columns = arel_columns([primary_key])
+          base_shard = Shard.current(klass.connection_class_for_self)
+          activate do |r|
+            relation = r.spawn
+            relation.select_values = columns
+
+            result = if relation.where_clause.contradiction?
+                       ::ActiveRecord::Result.empty
+                     else
+                       skip_query_cache_if_necessary do
+                         klass.connection.select_all(relation, "#{klass.name} Ids", async: @async)
+                       end
+                     end
+
+            result.then do |res|
+              type_cast_pluck_values(res, columns).map { |id| Shard.relative_id_for(id, Shard.current, base_shard) }
+            end
+          end
+        end
+      end
+
       private
 
       def type_cast_calculated_value_switchman(value, column_name, operation)

data/lib/switchman/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Switchman
-  VERSION = "3.5.20"
+  VERSION = "3.5.22"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: switchman
 version: !ruby/object:Gem::Version
-  version: 3.5.20
+  version: 3.5.22
 platform: ruby
 authors:
 - Cody Cutrer
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-02-23 00:00:00.000000000 Z
+date: 2024-02-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord