Contents

# frozen_string_literal: true

module ActiveRecordDataLoader
  module ActiveRecord
    class EnumValueGenerator
      class << self
        def generator_for(model_class:, ar_column:)
          values = enum_values_for(model_class, ar_column.sql_type)
          -> { values.sample }
        end

        private

        def enum_values_for(model_class, enum_type)
          model_class
            .connection
            .execute("SELECT unnest(enum_range(NULL::#{enum_type}))::text")
            .map(&:values)
            .flatten
            .compact
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
active_record_data_loader-1.0.1	lib/active_record_data_loader/active_record/enum_value_generator.rb
active_record_data_loader-1.0.0	lib/active_record_data_loader/active_record/enum_value_generator.rb
active_record_data_loader-0.1.2	lib/active_record_data_loader/active_record/enum_value_generator.rb
active_record_data_loader-0.1.1	lib/active_record_data_loader/active_record/enum_value_generator.rb