ridgepole-ext-tidb 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f614666ce88942bff9b8d9c8b28288bdb8caef40baaf858e661217cbe473b36b
-  data.tar.gz: b807880de067c8bbd4e50bc926ac282b8ffb6fd6f9bcd3b60ae11dd41deadec2
+  metadata.gz: b369c3c9f3078db98f28341fe98396425d7f6b995d9bc931b8d48717c2a52335
+  data.tar.gz: 78bf0c2a62ea2c22618bfb38a5477c6fb674531e765cbb6f2d835b0f013bb68c
 SHA512:
-  metadata.gz: a38dd256520f28c16538e866f94e6230a16c7e30573428233690c0a1782ed2448f4aa073b2d6a68f742844645a4c3087c3dfb0fe09afa2599caca1e93d5c28d0
-  data.tar.gz: 81deb36f995f80bce6ae820919a2a0db457034ee5f398e9e40bc9c6ee96c6272d0b030df4c860de5fe3064d9f7bf8a8bf9376fcc9c7e2e7f8ca1dc694903ad60
+  metadata.gz: ee91cd5d10bb5d601499587ca3d88ecf5ff04b7eac44cecbccb6eeaed69670fe603c4ac5880958b8d636cc92a9ff09dd1f29ea6c9ebebe3fa817aa13f4cb2134
+  data.tar.gz: f3956f87de294466ca03cf1b34247afbafdbbbaf8cfe6f7d5081bee0cb12cf6c8670ca7098aabc06bc23829ea8d8af41bdbc92b3fd9da2df8f682817fac8af2a

data/lib/ridgepole/ext/tidb/version.rb

@@ -3,7 +3,7 @@
 module Ridgepole
   module Ext
     module Tidb
-      VERSION = '0.2.0'
+      VERSION = '0.2.1'
     end
   end
 end

data/lib/ridgepole/ext/tidb.rb

@@ -8,6 +8,8 @@ module Ridgepole
       def self.setup!
         # SchemaDumperにもAUTO_RANDOM対応を追加
         extend_schema_dumper
+        # Hash#assert_valid_keysを拡張してauto_randomキーを許可
+        extend_hash_assert_valid_keys
       end      # 手動で接続アダプタを拡張するメソッド（外部から呼び出し可能）
       def self.ensure_connection_extended!
         return unless ActiveRecord::Base.connected?
@@ -24,6 +26,11 @@ module Ridgepole
         # 既に拡張済みかチェック
         return if adapter_class.method_defined?(:tidb?)
 
+        # Hash#assert_valid_keysを拡張してauto_randomキーを許可
+        extend_hash_assert_valid_keys
+        # TableDefinitionを拡張して:auto_randomオプションをサポート
+        extend_table_definition
+
         adapter_class.class_eval do
           # AUTO_RANDOMカラムの検出
           def auto_random_column?(table_name, column_name)
@@ -64,7 +71,9 @@ module Ridgepole
           rescue => e
             puts "AUTO_RANDOM detection failed: #{e.message}"
             false
-          end          # TiDBかどうかの判定
+          end
+
+          # TiDBかどうかの判定
           def tidb?
             # VERSION()関数でTiDBを検出（キャッシュなし）
             version_info = select_value('SELECT VERSION()')
@@ -79,27 +88,21 @@ module Ridgepole
           # CREATE TABLE時のAUTO_RANDOM対応
           alias_method :create_table_without_auto_random, :create_table
           def create_table(table_name, **options, &block)
+            # :auto_randomキーを処理する前に、idオプションから取り除く
             if options.dig(:id, :auto_random) && tidb?
-              # AUTO_RANDOMを含むテーブル作成
-              sql = build_create_table_sql_with_auto_random(table_name, **options, &block)
-              execute(sql)
-            else
-              create_table_without_auto_random(table_name, **options, &block)
-            end
-          end
+              # auto_randomフラグを保存
+              auto_random_enabled = options[:id].delete(:auto_random)
 
-          private
-
-          def build_create_table_sql_with_auto_random(table_name, **options, &block)
-            # 簡単な実装 - 実際にはより複雑になる
-            sql = "CREATE TABLE #{quote_table_name(table_name)} ("
+              # 通常のcreate_tableを呼び出してテーブル構造を作成
+              create_table_without_auto_random(table_name, **options, &block)
 
-            if options[:id] && options[:id][:auto_random]
-              sql += "id BIGINT AUTO_RANDOM PRIMARY KEY"
+              # AUTO_RANDOMを有効にするためにALTER TABLEを実行
+              if auto_random_enabled
+                execute("ALTER TABLE #{quote_table_name(table_name)} MODIFY COLUMN id BIGINT AUTO_RANDOM PRIMARY KEY")
+              end
+            else
+              create_table_without_auto_random(table_name, **options, &block)
             end
-
-            sql += ") DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci"
-            sql
           end
         end
 
@@ -108,6 +111,11 @@ module Ridgepole
 
       def self.extend_activerecord_adapters
         puts "📦 Extending ActiveRecord adapters..."
+        # Hash#assert_valid_keysを拡張してauto_randomキーを許可
+        extend_hash_assert_valid_keys
+        # TableDefinitionを拡張して:auto_randomオプションをサポート
+        extend_table_definition
+
         # MySQL系アダプタにAUTO_RANDOMサポートを追加
         extend_adapter('ActiveRecord::ConnectionAdapters::Mysql2Adapter')
         extend_adapter('ActiveRecord::ConnectionAdapters::TrilogyAdapter')
@@ -129,6 +137,11 @@ module Ridgepole
           return
         end
 
+        # Hash#assert_valid_keysを拡張してauto_randomキーを許可
+        extend_hash_assert_valid_keys
+        # TableDefinitionを拡張して:auto_randomオプションをサポート
+        extend_table_definition
+
         # 一時的にputsを外して動作確認
         adapter_class.class_eval do
           # AUTO_RANDOMカラムの検出
@@ -163,27 +176,21 @@ module Ridgepole
           # CREATE TABLE時のAUTO_RANDOM対応
           alias_method :create_table_without_auto_random, :create_table
           def create_table(table_name, **options, &block)
+            # :auto_randomキーを処理する前に、idオプションから取り除く
             if options.dig(:id, :auto_random) && tidb?
-              # AUTO_RANDOMを含むテーブル作成
-              sql = build_create_table_sql_with_auto_random(table_name, **options, &block)
-              execute(sql)
-            else
-              create_table_without_auto_random(table_name, **options, &block)
-            end
-          end
-
-          private
+              # auto_randomフラグを保存
+              auto_random_enabled = options[:id].delete(:auto_random)
 
-          def build_create_table_sql_with_auto_random(table_name, **options, &block)
-            # 簡単な実装 - 実際にはより複雑になる
-            sql = "CREATE TABLE #{quote_table_name(table_name)} ("
+              # 通常のcreate_tableを呼び出してテーブル構造を作成
+              create_table_without_auto_random(table_name, **options, &block)
 
-            if options[:id] && options[:id][:auto_random]
-              sql += "id BIGINT AUTO_RANDOM PRIMARY KEY"
+              # AUTO_RANDOMを有効にするためにALTER TABLEを実行
+              if auto_random_enabled
+                execute("ALTER TABLE #{quote_table_name(table_name)} MODIFY COLUMN id BIGINT AUTO_RANDOM PRIMARY KEY")
+              end
+            else
+              create_table_without_auto_random(table_name, **options, &block)
             end
-
-            sql += ") DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci"
-            sql
           end
         end
 
@@ -211,6 +218,49 @@ module Ridgepole
       rescue NameError
         # SchemaDumperが利用できない場合はスキップ
       end
+
+      def self.extend_table_definition
+        return unless defined?(ActiveRecord::ConnectionAdapters::TableDefinition)
+
+        # TableDefinitionを拡張して:auto_randomオプションをサポート
+        ActiveRecord::ConnectionAdapters::TableDefinition.class_eval do
+          # カラム作成時のオプション検証を拡張
+          alias_method :column_without_auto_random, :column
+          def column(name, type, **options)
+            # :auto_randomオプションが含まれている場合は、それを取り除いて後で処理
+            if options.key?(:auto_random)
+              auto_random_value = options.delete(:auto_random)
+              # カラム定義にauto_randomの情報を保存（後でcreate_tableで使用）
+              @auto_random_columns ||= {}
+              @auto_random_columns[name.to_s] = auto_random_value
+            end
+            column_without_auto_random(name, type, **options)
+          end
+
+          # auto_randomカラムの情報を取得するメソッド
+          def auto_random_columns
+            @auto_random_columns ||= {}
+          end
+        end
+      rescue NameError => e
+        puts "⚠️  Could not extend TableDefinition: #{e.message}"
+      end
+
+      def self.extend_hash_assert_valid_keys
+        # Hashクラスを拡張して、auto_randomキーを有効なキーとして認識させる
+        Hash.class_eval do
+          alias_method :assert_valid_keys_without_auto_random, :assert_valid_keys
+          def assert_valid_keys(*valid_keys)
+            # auto_randomキーが含まれている場合は、それを有効なキーとして追加
+            if keys.include?(:auto_random) && !valid_keys.include?(:auto_random)
+              valid_keys = valid_keys + [:auto_random]
+            end
+            assert_valid_keys_without_auto_random(*valid_keys)
+          end
+        end
+      rescue NameError => e
+        puts "⚠️  Could not extend Hash#assert_valid_keys: #{e.message}"
+      end
     end
   end
 end

data/lib/ridgepole-ext-tidb.rb

@@ -1,4 +1,16 @@
 # frozen_string_literal: true
 
 require_relative 'ridgepole/ext/tidb'
+
+# Ridgepoleが使用される前に拡張を適用
 Ridgepole::Ext::Tidb.setup!
+
+# ActiveRecordアダプタが利用可能になったときに自動拡張を実行
+if defined?(ActiveRecord::Base)
+  Ridgepole::Ext::Tidb.extend_activerecord_adapters
+else
+  # ActiveRecordが後でロードされる場合に備えてフックを設定
+  ActiveSupport.on_load(:active_record) do
+    Ridgepole::Ext::Tidb.extend_activerecord_adapters
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ridgepole-ext-tidb
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - ikad