rails-snowflake 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8f0727305e2203bd51c71c99e038f7902d0abdf9f283bcf0dca93495c5f851bf
-  data.tar.gz: 751f6d5c2e3329ad5be5c6104f8cc164af707181e01f3d428673736980f0539e
+  metadata.gz: 20b88f3269b47cff232d8328e024ae405f53a9c659d2ad56bbe9eb9cc29449ba
+  data.tar.gz: 2e6163a490501f5859d60adeaf31f1ddc87ddea9cba69b0eb7c936aac5801346
 SHA512:
-  metadata.gz: 05771bac80703b818c3c86021624f0d68ce23ceb5c34cd2423de1d2ce65674cc3824540f706d56593b0f97b6bc6afcd7c01b7e542e18a24a625a777323979957
-  data.tar.gz: 287eccd9577239bb6d2543383c3979483f52dc6272e6077681b56f7b86cf0ef94abbe8a11da01e7b90e7254c506d1e732253d048d003b2e79c345fc726ea8c13
+  metadata.gz: 9daf3fd3b436e7833d900d8d83751a2d48024428a4532828a4618f5f19fa7183ba094d967b7eb7b74216d3ca76f495409aef9e07924baae89f07e18c2e1224a3
+  data.tar.gz: 9cfbb4cea7a2d62d5114a5838f64fceca5517641f1797059073255ab1f376fb08fbba276eb04de9bd7d43fb06f38c6e4fa26b6726b1d9aa8556fb3bcf9e18235

data/README.md

@@ -115,8 +115,8 @@ execute("ALTER TABLE table_name ALTER COLUMN id SET DEFAULT timestamp_id('table_
 
 ## Requirements
 
-- **Database**: PostgreSQL (uses PostgreSQL-specific functions)
-- **Rails**: 7.2+ (may work with earlier versions)
+- **Database**: PostgreSQL >= 15
+- **Rails**: 7.1+
 - **Ruby**: 3.2+
 
 ## How it Works

data/Rakefile

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "bundler/setup"
 
 APP_RAKEFILE = File.expand_path("test/dummy/Rakefile", __dir__)

data/lib/rails/snowflake/column_methods.rb

@@ -5,16 +5,14 @@ module Rails
     module ColumnMethods
     def snowflake(name, **options)
       if name == :id && !options[:primary_key]
-        raise ArgumentError, "Cannot use t.snowflake :id directly. Use `create_table` with `id: false` and then `t.snowflake :id, primary_key: true`"
+        raise Error, "Cannot use t.snowflake :id directly. Use `create_table` with `id: false` and then `t.snowflake :id, primary_key: true`"
       end
 
-      table_name = @name
-
-      unless table_name
-        raise ArgumentError, "Could not determine table name for Snowflake column. Make sure you're using it within a `create_table` block."
+      unless @name
+        raise Error, "Could not determine table name for Snowflake column. Make sure you're using it within a `create_table` block."
       end
 
-      options[:default] = -> { "timestamp_id('#{table_name}'::text)" }
+      options[:default] = -> { "timestamp_id('#{@name}'::text)" }
 
       column(name, :bigint, **options)
     end

data/lib/rails/snowflake/database_tasks.rb

@@ -1,40 +1,46 @@
 # frozen_string_literal: true
 
-# Hook into Rails database tasks to ensure snowflake sequences exist
+# Ensures that any tables using a snowflake primary key (:id with timestamp_id default)
+# have their backing <table>_id_seq sequences present. This is necessary for tables
+# created with `id: false` and a subsequent `t.snowflake :id, primary_key: true` where
+# Rails does not auto-create the normal sequence.
 def ensure_snowflake_sequences
-  return unless defined?(ActiveRecord::Base)
-
   begin
-    if ActiveRecord::Base.connection.adapter_name == "PostgreSQL"
-      Rails.logger.debug "Rails::Snowflake: Ensure sequences exist for `timestamp_id` columns"
-      Rails::Snowflake::Id.ensure_id_sequences_exist
+    # Avoid establishing a new connection too early; skip if not connected yet.
+    return unless ActiveRecord::Base.connection_pool.connected?
+
+    adapter = ActiveRecord::Base.connection.adapter_name
+    unless adapter == "PostgreSQL"
+      Rails.logger.debug("Rails::Snowflake: Skipping sequence check (adapter=#{adapter})") if defined?(Rails)
+      return
     end
-  rescue ActiveRecord::NoDatabaseError, ActiveRecord::ConnectionNotEstablished
-    Rails.logger.warn "Rails::Snowflake: Could not ensure sequences: #{e.message}"
-  end
-end
 
-# Enhance existing Rails database tasks by adding our hook to them
-if Rake::Task.task_defined?("db:migrate")
-  Rake::Task["db:migrate"].enhance do
-    ensure_snowflake_sequences
-  end
-end
+    Rails.logger.debug "Rails::Snowflake: Ensuring sequences for timestamp_id columns" if defined?(Rails)
 
-if Rake::Task.task_defined?("db:schema:load")
-  Rake::Task["db:schema:load"].enhance do
-    ensure_snowflake_sequences
+    Rails::Snowflake::Id.ensure_id_sequences_exist
+  rescue ActiveRecord::NoDatabaseError, ActiveRecord::ConnectionNotEstablished => e
+    Rails.logger.debug "Rails::Snowflake: Skipping sequence ensure (#{e.class}: #{e.message})" if defined?(Rails)
+  rescue StandardError => e
+    Rails.logger.warn "Rails::Snowflake: Unexpected error while ensuring sequences: #{e.class}: #{e.message}" if defined?(Rails)
   end
 end
 
-if Rake::Task.task_defined?("db:structure:load")
-  Rake::Task["db:structure:load"].enhance do
-    ensure_snowflake_sequences
-  end
-end
+def enhance_snowflake_db_task(name)
+  return unless Rake::Task.task_defined?(name)
 
-if Rake::Task.task_defined?("db:seed")
-  Rake::Task["db:seed"].enhance do
+  Rake::Task[name].enhance do
     ensure_snowflake_sequences
   end
 end
+
+# Enhance a broad set of lifecycle tasks; migrate first so tests pick it up.
+%w[
+  db:migrate
+  db:setup
+  db:prepare
+  db:reset
+  db:schema:load
+  db:structure:load
+  db:seed
+  db:test:prepare
+].each { |t| enhance_snowflake_db_task(t) }

data/lib/rails/snowflake/generators/install/install_generator.rb

@@ -26,12 +26,6 @@ module Rails
         def create_migration_file
           migration_template "install_snowflake_id.rb.erb", File.join(db_migrate_path, "install_snowflake_id.rb")
         end
-
-        private
-
-        def migration_version
-          "[#{ActiveRecord::VERSION::STRING.to_f}]"
-        end
       end
     end
   end

data/lib/rails/snowflake/generators/install/templates/install_snowflake_id.rb.erb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-class InstallSnowflakeId < ActiveRecord::Migration<%= migration_version %>
+class InstallSnowflakeId < ActiveRecord::Migration[<%= ActiveRecord::VERSION::MAJOR %>.<%= ActiveRecord::VERSION::MINOR %>]
   def up
     # Create the timestamp_id PostgreSQL function
     Rails::Snowflake::Id.define_timestamp_id

data/lib/rails/snowflake/railtie.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Rails
   module Snowflake
     class Railtie < ::Rails::Railtie

data/lib/rails/snowflake/version.rb

@@ -1,5 +1,7 @@
+# frozen_string_literal: true
+
 module Rails
   module Snowflake
-    VERSION = "0.1.0"
+    VERSION = "0.1.1"
   end
 end

data/lib/tasks/rails/snowflake_tasks.rake

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 # desc "Explaining what the task does"
 # task :snowflake_id do
 #   # Task goes here

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-snowflake
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Luiz Eduardo Kowalski
@@ -15,14 +15,20 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.2'
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.2'
+        version: '7.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.2'
 description: A Rails plugin that provides a simple way to generate unique Snowflake
   IDs for your ActiveRecord models.
 email: