ridgepole-ext-tidb 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b369c3c9f3078db98f28341fe98396425d7f6b995d9bc931b8d48717c2a52335
-  data.tar.gz: 78bf0c2a62ea2c22618bfb38a5477c6fb674531e765cbb6f2d835b0f013bb68c
+  metadata.gz: 35422c4801686a578e9871993e7ea61443110ef8ce39b2ba31510984824c2559
+  data.tar.gz: 4347c14f4e1990409c665ee39742111f68d214e3ad11fab549222564c5adcfec
 SHA512:
-  metadata.gz: ee91cd5d10bb5d601499587ca3d88ecf5ff04b7eac44cecbccb6eeaed69670fe603c4ac5880958b8d636cc92a9ff09dd1f29ea6c9ebebe3fa817aa13f4cb2134
-  data.tar.gz: f3956f87de294466ca03cf1b34247afbafdbbbaf8cfe6f7d5081bee0cb12cf6c8670ca7098aabc06bc23829ea8d8af41bdbc92b3fd9da2df8f682817fac8af2a
+  metadata.gz: 50330dcb777c17cc030d4569d03988b8343f50a4dbf1537bd21f8203a085cca3b681ff02d7b3ca0ff8158bb1ffb56ba41a1fd1b6fe02c939073b085de1c2bbcb
+  data.tar.gz: 8cf011251797a1d67bad7476dc1c374f00b630306314400dcdac5198d81a82374d41ad00b811e048106f1616eb10ac2daea163380f7ea51651bd95c0a70b6583

data/README.md

@@ -1,19 +1,20 @@
 # Ridgepole::Ext::Tidb
 
 ![Tests](https://github.com/forgxisto/ridgepole-ext-tidb/actions/workflows/test.yml/badge.svg)
-![Ruby Version](https://img.shields.io/badge/ruby-3.1%2B-red)
+![Ruby Version](https://img.shields.io/badge/ruby-3.2%2B-red)
 ![TiDB Compatibility](https://img.shields.io/badge/TiDB-v7.5.0%2B-blue)
 
 TiDBの`AUTO_RANDOM`カラム属性をサポートするRidgepole拡張機能です。この拡張により、TiDBの分散IDジェネレーション機能をSchemafile管理に統合できます。
 
 ## 主な機能
 
-- **AUTO_RANDOM検出**: TiDBのAUTO_RANDOMカラムを自動検出
-- **TiDB判定**: データベースがTiDBかどうかを自動判定
-- **MySQL互換**: mysql2とtrilogyアダプター両方に対応
-- **スキーマダンプ対応**: AUTO_RANDOM属性をRidgefileに正確に出力
-- **冪等性保証**: スキーマ適用の際の差分を正確に計算
-- **Ruby 3.1+ 対応**: 最新のRubyバージョンに完全対応
+- **AUTO_RANDOM適用**: CREATE時に`AUTO_RANDOM(n)`を列定義へ付与し、`AUTO_INCREMENT`を抑止
+- **AUTO_RANDOM_BASE**: テーブルオプションに`AUTO_RANDOM_BASE=<n>`を付与
+- **スキーマダンプ対応**: `create_table`のオプションへ`auto_random:`/`auto_random_base:`を出力（往復一致）
+- **冪等性**: apply→export→diff→applyの繰り返しでも差分ゼロを維持（ALTERは不使用）
+- **TiDB判定**: 接続先がTiDBかどうかを自動判定
+- **MySQL互換**: mysql2 / trilogy アダプターの両方に対応
+- **Ruby 3.2+ 対応**
 
 ## インストール
 
@@ -41,10 +42,7 @@ $ gem install ridgepole-ext-tidb
 
 ```ruby
 require 'ridgepole'
-require 'ridgepole-ext-tidb'
-
-# TiDB拡張をセットアップ（ActiveRecord読み込み後）
-Ridgepole::Ext::Tidb.setup!
+require 'ridgepole/ext_tidb'
 
 # Ridgepoleクライアントを設定
 client = Ridgepole::Client.new({
@@ -61,22 +59,25 @@ client = Ridgepole::Client.new({
 
 ```ruby
 # Schemafile
-create_table "users", id: { type: :bigint, auto_random: true } do |t|
+require "ridgepole/ext_tidb"
+
+# 1) テーブルレベルでAUTO_RANDOMとAUTO_RANDOM_BASEを指定
+create_table "users",
+  id: :bigint,
+  auto_random: 5,
+  auto_random_base: 100_000,
+  options: "DEFAULT CHARSET=utf8mb4" do |t|
   t.string :name, null: false
-  t.string :email, null: false
-  t.timestamps
 end
 
-create_table "posts", force: :cascade do |t|
-  t.bigint :id, auto_random: true, primary_key: true
-  t.bigint :user_id, null: false
+# 2) 手動PK（カラム側でAUTO_RANDOMを指定）
+create_table "events", id: false, options: "DEFAULT CHARSET=utf8mb4" do |t|
+  t.bigint :id, primary_key: true, null: false, auto_random: 6
   t.string :title, null: false
-  t.text :content
-  t.timestamps
 end
 ```
 
-**注意**: 現在の実装では、`auto_random: true`オプションはスキーマダンプ時に出力されますが、`create_table`でのテーブル作成機能は基本実装のみです。実際のテーブル作成は標準のDDLを使用してください。
+出力（export）は`create_table`のオプションに`auto_random:`/`auto_random_base:`を含めて往復一致となります（ALTERは使用しません）。
 
 ### CLI使用例
 
@@ -102,11 +103,10 @@ $ bundle exec ridgepole -c config/database.yml -E development --export -o Schema
 
 ## 動作確認済み環境
 
-- **TiDB**: v7.5.0 (安定版)
-- **Ruby**: 3.1+
-- **ActiveRecord**: 7.0+
-- **Ridgepole**: 3.0.4+
-- **アダプター**: mysql2, trilogy
+- TiDB: v7.5.0 以降
+- ActiveRecord: 7 / 8 系
+- Ridgepole: 3.0.4 以降
+- アダプター: mysql2 / trilogy
 
 ## データベース設定
 
@@ -154,7 +154,8 @@ connection.auto_random_column?('users', 'id')  # => true/false
 
 ### 3. スキーマダンプ対応
 
-既存のAUTO_RANDOMテーブルからSchemafileを生成する際、`auto_random: true`オプションが正しく出力されます。
+既存テーブルからのダンプ時、SHOW CREATE を解析して `create_table` のオプションに
+`auto_random:` と `auto_random_base:` を出力します。apply→export→diff→apply の繰り返しでも差分は発生しません。
 
 ## テスト結果例
 
@@ -213,7 +214,7 @@ CREATE TABLE users (
 
 ### 前提条件
 
-- Ruby 3.1 以上
+- Ruby 3.2 以上
 - TiDB 4.0 以上 (テスト用)
 - Docker (テスト環境用)
 
@@ -227,31 +228,21 @@ $ bundle install
 
 ### テスト実行
 
-```bash
-# 基本機能テスト（TiDBなしでも実行可能）
-$ SKIP_TIDB_TESTS=1 bundle exec rspec
-
-# TiDB統合テスト（Dockerが必要）
-$ docker compose up -d tidb
-$ bundle exec rspec
-
-# Docker環境でのテスト
-$ docker compose run --rm test
-```
-
-### TiDBテスト環境
+このリポジトリのRSpecは、実行時に自動で TiDB コンテナ（docker compose）を起動・停止します。事前の手動起動は不要です。
 
-TiDB 7.5.0を使用したテスト環境が用意されています：
+前提: Docker と docker compose が使用可能で、ポート `14000` が空いていること。
 
 ```bash
-# TiDBを起動
-$ docker compose up -d tidb
+# 統合テスト（TiDBを自動起動）
+$ bundle exec rspec --format documentation
 
-# テストを実行
-$ docker compose run --rm test
-```
+# アダプタを切り替えたい場合（デフォルトは trilogy）
+# ※ mysql2 を使う場合は、別途 mysql2 をインストールしてください
+$ AR_ADAPTER=mysql2 bundle exec rspec --format documentation
 ```
 
+ヒント: コンテナを手動で起動しておきたい場合は `docker compose up -d tidb` を先に実行しても構いません（テストはそのまま動作します）。
+
 ## Contributing
 
 1. このリポジトリをフォーク

data/docker-compose.yml

@@ -1,34 +1,9 @@
 services:
   tidb:
-    image: pingcap/tidb:v7.5.0
-    container_name: ridgepole-tidb-test
+    image: pingcap/tidb:latest
+    command: ["--store=mocktikv", "--log-file="]
     ports:
-      - "14000:4000"
-      - "14080:10080"
-    command:
-      - --store=unistore
-      - --host=0.0.0.0
-      - --path=""
-    healthcheck:
-      test: ["CMD-SHELL", "timeout 1 bash -c '</dev/tcp/127.0.0.1/4000' || exit 1"]
-      interval: 10s
-      timeout: 5s
-      retries: 5
-      start_period: 30s
-    restart: unless-stopped
-
-  test:
-    build: .
-    depends_on:
-      tidb:
-        condition: service_healthy
+      - "14000:4000"   # MySQL protocol
     environment:
-      - TIDB_HOST=tidb
-      - TIDB_PORT=4000
-      - TIDB_USER=root
-      - TIDB_PASSWORD=
-      - TIDB_DATABASE=ridgepole_test
-    working_dir: /app
-    volumes:
-      - .:/app
-    command: bundle exec rspec
+      # TiDB は root 無パスで入れる
+      - TZ=UTC

data/lib/ridgepole/ext_tidb/detector.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    module Detector
+      # TiDB接続の自動検出
+      def tidb?
+        return @tidb_detected if defined?(@tidb_detected)
+
+        @tidb_detected = begin
+          version_info = select_value('SELECT VERSION()')
+          result = version_info&.include?('TiDB') == true
+          Rails.logger.debug "TiDB detection: version=#{version_info}, result=#{result}" if defined?(Rails)
+          result
+        rescue => e
+          Rails.logger.debug "TiDB detection failed: #{e.message}" if defined?(Rails)
+          false
+        end
+      end
+
+      # AUTO_RANDOMカラムの検出
+      def auto_random_column?(table_name, column_name)
+        return false unless tidb?
+
+        # SHOW CREATE TABLEでAUTO_RANDOMを検出
+        result = execute("SHOW CREATE TABLE #{quote_table_name(table_name)}")
+        create_sql = result.first[1] if result.first
+
+        if create_sql
+          # TiDBでのAUTO_RANDOM検出パターン
+          patterns = [
+            /AUTO_RANDOM\(\d+\)/i,
+            /\/\*T!\[auto_rand\] AUTO_RANDOM\(\d+\) \*\//i
+          ]
+
+          patterns.any? { |pattern| create_sql.match?(pattern) }
+        else
+          # フォールバック: INFORMATION_SCHEMA.COLUMNSを確認
+          extra = select_value(<<~SQL)
+            SELECT EXTRA
+            FROM INFORMATION_SCHEMA.COLUMNS
+            WHERE TABLE_SCHEMA = DATABASE()
+              AND TABLE_NAME = #{quote(table_name)}
+              AND COLUMN_NAME = #{quote(column_name)}
+          SQL
+
+          extra&.downcase&.include?('auto_random') == true
+        end
+      rescue => e
+        Rails.logger.debug "AUTO_RANDOM detection failed: #{e.message}" if defined?(Rails)
+        false
+      end
+
+      # AUTO_RANDOM_BASEテーブルオプションの検出
+      def auto_random_base(table_name)
+        return nil unless tidb?
+
+        result = execute("SHOW CREATE TABLE #{quote_table_name(table_name)}")
+        create_sql = result.first[1] if result.first
+
+        if create_sql&.match(/AUTO_RANDOM_BASE=(\d+)/i)
+          $1.to_i
+        end
+      rescue
+        nil
+      end
+    end
+  end
+end

data/lib/ridgepole/ext_tidb/dump_patch.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    module DumpPatch
+      def dump(*args)
+        dsl = super
+
+        conn = ActiveRecord::Base.connection
+        return dsl unless conn.respond_to?(:tidb?) && conn.tidb?
+
+        begin
+          tables = conn.tables
+        rescue StandardError
+          return dsl
+        end
+
+        tables.each do |table|
+          extras = extract_table_auto_random_options_for_dump(conn, table)
+          next if extras.empty?
+
+          # inject into create_table line
+          dsl = inject_table_options_line(dsl, table, extras)
+        end
+
+        dsl
+      end
+
+      private
+
+      def extract_table_auto_random_options_for_dump(conn, table)
+        extras = {}
+        begin
+          row = conn.execute("SHOW CREATE TABLE #{conn.quote_table_name(table)}").first
+          create_sql = row[1] if row
+          return extras unless create_sql
+
+          if (m = create_sql.match(/AUTO_RANDOM\((\d+)\)/i))
+            extras[:auto_random] = m[1].to_i
+          end
+          if (m = create_sql.match(/AUTO_RANDOM_BASE=(\d+)/i))
+            extras[:auto_random_base] = m[1].to_i
+          end
+        rescue StandardError
+        end
+        extras
+      end
+
+      def inject_table_options_line(dsl, table, extras)
+        keyvals = []
+        keyvals << "auto_random: #{extras[:auto_random]}" if extras[:auto_random]
+        keyvals << "auto_random_base: #{extras[:auto_random_base]}" if extras[:auto_random_base]
+        return dsl if keyvals.empty?
+
+        pattern = /(^(\s*)create_table\s+"#{Regexp.escape(table)}",\s*)(.+?)(\s+do\s*\|t\|)/m
+        dsl.sub(pattern) do
+          head = Regexp.last_match(1)
+          indent = Regexp.last_match(2)
+          opts = Regexp.last_match(3)
+          tail = Regexp.last_match(4)
+
+          # 既に同キーがあるなら上書きはせずそのまま
+          already = keyvals.any? { |kv| opts.include?(kv.split(':').first + ':') }
+          if already
+            head + opts + tail
+          else
+            injected = keyvals.join(', ')
+            # id: が先頭にあるならその直後に差し込む。なければ先頭に追加。
+            if opts =~ /(id:\s*[^,]+,\s*)/i
+              head + opts.sub($1, "#{$1}#{injected}, ") + tail
+            else
+              head + "#{injected}, " + opts + tail
+            end
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/ridgepole/ext_tidb/install.rb

@@ -0,0 +1,146 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    module Install
+      # Ridgepole起動時にTiDB検出→各パッチをprepend
+      def self.apply_patches!
+        # Hash#assert_valid_keysを拡張してauto_randomキーを許可
+        extend_hash_assert_valid_keys
+
+        # ActiveRecordが読み込まれている場合は即座に適用
+        if defined?(ActiveRecord::Base)
+          apply_activerecord_patches
+        else
+          # ActiveRecordが後でロードされる場合に備えてフックを設定
+          ActiveSupport.on_load(:active_record) do
+            apply_activerecord_patches
+          end
+        end
+      end
+
+      def self.apply_activerecord_patches
+        extend_connection_adapters
+        # SchemaDumper への直接パッチは不要（dump は DumpPatch が担当）
+        # TableDefinition 拡張も不要（Hash#assert_valid_keys 拡張で回避）
+        install_connection_hook
+        extend_ridgepole_client
+      end
+
+      private
+
+      def self.extend_hash_assert_valid_keys
+        Hash.class_eval do
+          alias_method :assert_valid_keys_without_auto_random, :assert_valid_keys
+          def assert_valid_keys(*valid_keys)
+            # auto_random, auto_random_baseキーを有効なキーとして追加
+            auto_random_keys = [:auto_random, :auto_random_base]
+            auto_random_keys.each do |key|
+              if keys.include?(key) && !valid_keys.include?(key)
+                valid_keys = valid_keys + [key]
+              end
+            end
+            assert_valid_keys_without_auto_random(*valid_keys)
+          end
+        end
+      rescue NameError => e
+        Rails.logger.debug "Could not extend Hash#assert_valid_keys: #{e.message}" if defined?(Rails)
+      end
+
+      def self.extend_connection_adapters
+        return unless defined?(ActiveRecord::ConnectionAdapters)
+
+        # まずは抽象 MySQL アダプタにパッチ（mysql2/trilogy 双方を網羅）
+        begin
+          abstract_mysql = ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter
+
+          abstract_mysql.prepend(Detector) unless abstract_mysql < Detector
+          abstract_mysql.prepend(TableOptions) unless abstract_mysql < TableOptions
+
+          if abstract_mysql.const_defined?(:SchemaCreation)
+            sc = abstract_mysql.const_get(:SchemaCreation)
+            sc.prepend(SchemaCreation) unless sc < SchemaCreation
+          end
+        rescue NameError
+          # 未ロードの環境もあるので無視（接続確立後にロードされる）
+        end
+
+        # 既に具体アダプタがロード済みなら、そちらにも適用（冪等）
+        %w[
+          ActiveRecord::ConnectionAdapters::Mysql2Adapter
+          ActiveRecord::ConnectionAdapters::TrilogyAdapter
+        ].each do |adapter_name|
+          begin
+            adapter_class = Object.const_get(adapter_name)
+            adapter_class.prepend(Detector) unless adapter_class < Detector
+            adapter_class.prepend(TableOptions) unless adapter_class < TableOptions
+            if adapter_class.const_defined?(:SchemaCreation)
+              sc = adapter_class.const_get(:SchemaCreation)
+              sc.prepend(SchemaCreation) unless sc < SchemaCreation
+            end
+          rescue NameError
+            # 未ロードならスキップ
+          end
+        end
+
+        # 名前空間の揺れに備え、ConnectionAdapters 配下の SchemaCreation すべてに適用
+        begin
+          ObjectSpace.each_object(Class) do |klass|
+            name = klass.name rescue nil
+            next unless name && name.start_with?("ActiveRecord::ConnectionAdapters")
+            next unless name.end_with?("::SchemaCreation")
+            klass.prepend(SchemaCreation) unless klass < SchemaCreation
+          end
+        rescue StandardError
+          # noop
+        end
+
+        # AbstractMysqlAdapter のサブクラスにも Detector/TableOptions を適用（ロード順対策）
+        begin
+          if defined?(ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter)
+            ObjectSpace.each_object(Class) do |klass|
+              next unless klass < ActiveRecord::ConnectionAdapters::AbstractMysqlAdapter
+              klass.prepend(Detector) unless klass < Detector
+              klass.prepend(TableOptions) unless klass < TableOptions
+            end
+          end
+        rescue StandardError
+          # noop
+        end
+      end
+
+      # 接続確立後（アダプタ読み込み後）にも確実にパッチを適用するためのフック
+      def self.install_connection_hook
+        return unless defined?(ActiveRecord::Base)
+        return if @establish_hook_installed
+
+        mod = Module.new do
+          def establish_connection(*args)
+            result = super
+            # アダプタが読み込まれた後に再度パッチ適用（冪等）
+            Ridgepole::ExtTidb::Install.extend_connection_adapters
+            result
+          end
+        end
+
+        # class << だとローカル変数がスコープ外になるため、singleton_class で prepend
+        ActiveRecord::Base.singleton_class.prepend(mod)
+
+        @establish_hook_installed = true
+      rescue StandardError => e
+        Rails.logger.debug "Could not install establish_connection hook: #{e.message}" if defined?(Rails)
+      end
+
+      # extend_schema_dumper: dump は DumpPatch に委譲するため不要
+      # extend_table_definition: Hash#assert_valid_keys で未知キーを許すため不要
+
+      def self.extend_ridgepole_client
+        return unless defined?(Ridgepole::Client)
+        return if Ridgepole::Client < Ridgepole::ExtTidb::DumpPatch
+        Ridgepole::Client.prepend(Ridgepole::ExtTidb::DumpPatch)
+      rescue StandardError => e
+        Rails.logger.debug "Could not extend Ridgepole::Client: #{e.message}" if defined?(Rails)
+      end
+    end
+  end
+end

data/lib/ridgepole/ext_tidb/schema_creation.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    module SchemaCreation
+      # この SchemaCreation インスタンスから接続の tidb? を参照できるように
+      def tidb?
+        if instance_variable_defined?(:@conn)
+          conn = instance_variable_get(:@conn)
+          return conn.respond_to?(:tidb?) && conn.tidb?
+        end
+        false
+      end
+      # デフォルト主キーの生成経路でも AUTO_RANDOM を反映
+      def visit_PrimaryKeyDefinition(o)
+        return super unless tidb?
+
+        auto_random_value = o.options.delete(:auto_random)
+        sql = super
+
+        if auto_random_value
+          sql.sub!(/\sAUTO_INCREMENT\b/i, "")
+          bits = (auto_random_value == true ? nil : Integer(auto_random_value) rescue nil)
+          sql << (bits ? " AUTO_RANDOM(#{bits})" : " AUTO_RANDOM")
+        end
+
+        sql
+      end
+
+      # APPLY: カラムに AUTO_RANDOM(n) を追加（PRIMARY KEY の有無に関わらず列側に付与）
+      def visit_ColumnDefinition(o)
+        return super unless tidb?
+
+        # AUTO_RANDOM を取り出し（未知キーassertは Hash 拡張で回避済み）
+        auto_random_value = o.options.delete(:auto_random)
+
+        sql = super
+
+        # テーブルレベル指定のフォールバック（id の自動生成経路で失われた場合に備える）
+        if !auto_random_value && o.respond_to?(:primary_key?) && o.primary_key?
+          if instance_variable_defined?(:@conn)
+            conn = instance_variable_get(:@conn)
+            if conn.instance_variable_defined?(:@tidb_pending_auto_random_pk_bits)
+              auto_random_value = conn.instance_variable_get(:@tidb_pending_auto_random_pk_bits)
+              begin
+                conn.remove_instance_variable(:@tidb_pending_auto_random_pk_bits)
+              rescue StandardError
+              end
+            end
+          end
+        end
+
+        if auto_random_value
+          # 念のため AUTO_INCREMENT を除去
+          sql.sub!(/\sAUTO_INCREMENT\b/i, "")
+
+          bits = (auto_random_value == true ? nil : Integer(auto_random_value) rescue nil)
+          sql << (bits ? " AUTO_RANDOM(#{bits})" : " AUTO_RANDOM")
+        end
+
+        sql
+      end
+
+      private
+
+      def add_column_options!(sql, options)
+        # 独自キーを先に取り出し
+        bits = options.delete(:auto_random)
+
+        # AUTO_RANDOM を付けるなら AUTO_INCREMENT を抑止
+        options[:auto_increment] = false if bits
+
+        super(sql, options)
+
+        if bits
+          unless sql.match?(/\bAUTO_RANDOM\b/i)
+            sql.sub!(/\sAUTO_INCREMENT\b/i, "")
+            sql << " AUTO_RANDOM(#{Integer(bits)})"
+          end
+        end
+
+        sql
+      end
+
+      # 一部の AR バージョンはこちらを呼ぶ場合があるため両方用意
+      def add_column_options(sql, options)
+        add_column_options!(sql, options)
+      end
+    end
+  end
+end

data/lib/ridgepole/ext_tidb/table_options.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    # TiDB 向け create_table 拡張（CREATE 時に完結）
+    # - テーブルレベル auto_random を id 定義へ伝播
+    # - AUTO_RANDOM_BASE をテーブル options に付与
+    module TableOptions
+      # CREATE 時に auto_random / auto_random_base を正しく反映
+      def create_table(table_name, **options, &block)
+        return super unless tidb?
+
+        # 独自キーを取り出す（AR の未知キー検証を回避）
+        auto_random_value = options.delete(:auto_random)
+        auto_random_base_value = options.delete(:auto_random_base)
+
+        # id: false でなければ、テーブルレベル auto_random を id に伝播
+        if auto_random_value && options[:id] != false
+          id_opt = options[:id]
+          id_opts = case id_opt
+                    when Hash
+                      id_opt.dup
+                    when Symbol
+                      { type: id_opt }
+                    when true, nil
+                      {}
+                    else
+                      { type: id_opt }
+                    end
+          id_opts[:auto_random] = auto_random_value
+          id_opts[:auto_increment] = false
+          options[:id] = id_opts
+        end
+
+        # AUTO_RANDOM_BASE を options に付与
+        if auto_random_base_value
+          existing_options = options[:options] || ""
+          if existing_options.present?
+            options[:options] = "#{existing_options} AUTO_RANDOM_BASE=#{auto_random_base_value}"
+          else
+            options[:options] = "AUTO_RANDOM_BASE=#{auto_random_base_value}"
+          end
+        end
+
+        # 主キー経路でビット数が落ちる環境向けのフォールバック用に一時保存
+        if auto_random_value && options[:id] != false
+          begin
+            @tidb_pending_auto_random_pk_bits = auto_random_value
+          rescue StandardError
+          end
+        end
+
+        begin
+          super
+        ensure
+          remove_instance_variable(:@tidb_pending_auto_random_pk_bits) if instance_variable_defined?(:@tidb_pending_auto_random_pk_bits)
+        end
+      end
+
+
+      private
+
+      def build_create_table_options(options)
+        return super unless tidb?
+
+        # AUTO_RANDOM_BASE を options 文字列の先頭に寄せ、ノイズ差分を避ける
+        sql_options = super
+        if sql_options&.include?('AUTO_RANDOM_BASE')
+          parts = sql_options.split(/\s+/)
+          auto_random_parts = parts.select { |part| part.start_with?('AUTO_RANDOM_BASE=') }
+          other_parts = parts.reject { |part| part.start_with?('AUTO_RANDOM_BASE=') }
+          sql_options = (auto_random_parts + other_parts).join(' ')
+        end
+
+        sql_options
+      end
+    end
+  end
+end

data/lib/ridgepole/ext_tidb/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Ridgepole
+  module ExtTidb
+    VERSION = "0.3.0"
+  end
+end

data/lib/ridgepole/ext_tidb.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative 'ext_tidb/version'
+require_relative 'ext_tidb/detector'
+require_relative 'ext_tidb/schema_creation'
+require_relative 'ext_tidb/table_options'
+require_relative 'ext_tidb/dump_patch'
+require_relative 'ext_tidb/install'
+
+module Ridgepole
+  module ExtTidb
+    # エントリポイント - requireで自動的にインストール
+    def self.setup!
+      Install.apply_patches!
+    end
+  end
+end
+
+# 自動インストール
+Ridgepole::ExtTidb.setup!

data/ridgepole-ext-tidb.gemspec

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative 'lib/ridgepole/ext/tidb/version'
+require_relative 'lib/ridgepole/ext_tidb/version'
 
 Gem::Specification.new do |spec|
   spec.name = 'ridgepole-ext-tidb'
-  spec.version = Ridgepole::Ext::Tidb::VERSION
+  spec.version = Ridgepole::ExtTidb::VERSION
   spec.authors = ['ikad']
   spec.email = ['info@forgxisto.com']
 
@@ -12,7 +12,7 @@ Gem::Specification.new do |spec|
   spec.description = "Extends Ridgepole to support TiDB's AUTO_RANDOM column attribute for seamless schema management"
   spec.homepage = 'https://github.com/forgxisto/ridgepole-ext-tidb'
   spec.license = 'MIT'
-  spec.required_ruby_version = '>= 3.1.0'
+  spec.required_ruby_version = '>= 3.2.0'
 
   spec.metadata['allowed_push_host'] = 'https://rubygems.org'
   spec.metadata['homepage_uri'] = spec.homepage
@@ -33,15 +33,15 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   # Dependencies
-  spec.add_dependency 'ridgepole'
+  spec.add_dependency 'ridgepole', ">= 3.0"
 
   # Development dependencies
-  spec.add_development_dependency 'activerecord-trilogy-adapter'
-  spec.add_development_dependency 'mysql2'
   spec.add_development_dependency 'rake'
   spec.add_development_dependency 'rspec'
   spec.add_development_dependency 'rubocop'
   spec.add_development_dependency 'trilogy'
+  spec.add_development_dependency 'activerecord', '>= 8.0'
+  spec.add_development_dependency 'debug'
 
   # Ruby 3.4+ compatibility
   spec.add_development_dependency 'benchmark'

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: ridgepole-ext-tidb
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - ikad
 bindir: exe
 cert_chain: []
-date: 2025-09-06 00:00:00.000000000 Z
+date: 2025-09-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ridgepole
@@ -15,16 +15,16 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '3.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: activerecord-trilogy-adapter
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -38,7 +38,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: mysql2
+  name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -52,7 +52,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -66,7 +66,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rspec
+  name: trilogy
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -80,21 +80,21 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: rubocop
+  name: activerecord
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '8.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '8.0'
 - !ruby/object:Gem::Dependency
-  name: trilogy
+  name: debug
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -179,9 +179,13 @@ files:
 - README.md
 - Rakefile
 - docker-compose.yml
-- lib/ridgepole-ext-tidb.rb
-- lib/ridgepole/ext/tidb.rb
-- lib/ridgepole/ext/tidb/version.rb
+- lib/ridgepole/ext_tidb.rb
+- lib/ridgepole/ext_tidb/detector.rb
+- lib/ridgepole/ext_tidb/dump_patch.rb
+- lib/ridgepole/ext_tidb/install.rb
+- lib/ridgepole/ext_tidb/schema_creation.rb
+- lib/ridgepole/ext_tidb/table_options.rb
+- lib/ridgepole/ext_tidb/version.rb
 - ridgepole-ext-tidb.gemspec
 homepage: https://github.com/forgxisto/ridgepole-ext-tidb
 licenses:
@@ -198,7 +202,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.1.0
+      version: 3.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="

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

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-module Ridgepole
-  module Ext
-    module Tidb
-      VERSION = '0.2.1'
-    end
-  end
-end

data/lib/ridgepole/ext/tidb.rb

@@ -1,266 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'tidb/version'
-
-module Ridgepole
-  module Ext
-    module Tidb
-      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?
-
-        connection = ActiveRecord::Base.connection
-        extend_connection_adapter(connection)
-      end
-
-      def self.extend_connection_adapter(connection)
-        return unless connection
-
-        adapter_class = connection.class
-
-        # 既に拡張済みかチェック
-        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)
-            return false unless tidb?
-
-            # TiDB 7.5.0でのAUTO_RANDOM検出
-            # SHOW CREATE TABLEでCREATE TABLE文を確認
-            result = execute("SHOW CREATE TABLE #{quote_table_name(table_name)}")
-            create_sql = result.first[1] if result.first
-
-            if create_sql
-              # TiDB 7.5.0では AUTO_RANDOM がコメント形式で表示される
-              # 例: /*T![auto_rand] AUTO_RANDOM(5) */
-              if create_sql.include?('AUTO_RANDOM') || create_sql.include?('auto_rand')
-                return true
-              end
-
-              # テーブルオプションにAUTO_RANDOM_BASEが含まれているかチェック
-              if create_sql.include?('AUTO_RANDOM_BASE')
-                return true
-              end
-            end
-
-            # INFORMATION_SCHEMA.COLUMNS の EXTRA を確認（フォールバック）
-            extra = select_value(<<~SQL)
-              SELECT EXTRA
-              FROM INFORMATION_SCHEMA.COLUMNS
-              WHERE TABLE_SCHEMA = DATABASE()
-                AND TABLE_NAME = #{quote(table_name)}
-                AND COLUMN_NAME = #{quote(column_name)}
-            SQL
-
-            if extra&.downcase&.include?('auto_random')
-              return true
-            end
-
-            false
-          rescue => e
-            puts "AUTO_RANDOM detection failed: #{e.message}"
-            false
-          end
-
-          # TiDBかどうかの判定
-          def tidb?
-            # VERSION()関数でTiDBを検出（キャッシュなし）
-            version_info = select_value('SELECT VERSION()')
-            result = version_info&.include?('TiDB') == true
-            Rails.logger.debug "TiDB detection: version=#{version_info}, result=#{result}" if defined?(Rails)
-            result
-          rescue => e
-            Rails.logger.debug "TiDB detection failed: #{e.message}" if defined?(Rails)
-            false
-          end
-
-          # 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フラグを保存
-              auto_random_enabled = options[:id].delete(:auto_random)
-
-              # 通常のcreate_tableを呼び出してテーブル構造を作成
-              create_table_without_auto_random(table_name, **options, &block)
-
-              # 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
-          end
-        end
-
-        puts "✅ Methods added to #{adapter_class}"
-      end
-
-      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')
-
-        # SchemaDumperにもAUTO_RANDOM対応を追加
-        extend_schema_dumper
-        puts "📦 Adapter extension complete"
-      end
-
-      def self.extend_adapter(adapter_name)
-        return unless defined?(ActiveRecord::ConnectionAdapters)
-
-        begin
-          adapter_class = Object.const_get(adapter_name)
-          puts "🔧 Extending #{adapter_name}..."
-        rescue NameError => e
-          # アダプタが利用できない場合はスキップ
-          puts "⚠️  Skipping #{adapter_name}: #{e.message}"
-          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カラムの検出
-          def auto_random_column?(table_name, column_name)
-            return false unless tidb?
-
-            extra = select_value(<<~SQL)
-              SELECT EXTRA
-              FROM INFORMATION_SCHEMA.COLUMNS
-              WHERE TABLE_SCHEMA = DATABASE()
-                AND TABLE_NAME = #{quote(table_name)}
-                AND COLUMN_NAME = #{quote(column_name)}
-            SQL
-
-            extra&.downcase&.include?('auto_random') == true
-          rescue
-            false
-          end
-
-          # TiDBかどうかの判定
-          def tidb?
-            # VERSION()関数でTiDBを検出（キャッシュなし）
-            version_info = select_value('SELECT VERSION()')
-            result = version_info&.include?('TiDB') == true
-            Rails.logger.debug "TiDB detection: version=#{version_info}, result=#{result}" if defined?(Rails)
-            result
-          rescue => e
-            Rails.logger.debug "TiDB detection failed: #{e.message}" if defined?(Rails)
-            false
-          end
-
-          # 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フラグを保存
-              auto_random_enabled = options[:id].delete(:auto_random)
-
-              # 通常のcreate_tableを呼び出してテーブル構造を作成
-              create_table_without_auto_random(table_name, **options, &block)
-
-              # 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
-          end
-        end
-
-        puts "✅ Methods added to #{adapter_name}"
-      end
-
-      def self.extend_schema_dumper
-        return unless defined?(ActiveRecord::SchemaDumper)
-
-        ActiveRecord::SchemaDumper.class_eval do
-          alias_method :prepare_column_options_without_auto_random, :prepare_column_options
-          def prepare_column_options(column)
-            spec = prepare_column_options_without_auto_random(column)
-
-            # TiDB接続でAUTO_RANDOMカラムの場合、auto_randomオプションを追加
-            if @connection.respond_to?(:tidb?) && @connection.tidb? &&
-               @connection.respond_to?(:auto_random_column?) &&
-               @connection.auto_random_column?(@table, column.name)
-              spec[:auto_random] = true
-            end
-
-            spec
-          end
-        end
-      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,16 +0,0 @@
-# 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