activerecord-tidb-adapter 0.3.0 → 5.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 442a2dcd175987185739a1fad7a8ef8788d0b13b7263018795ac144a4c10ee03
-  data.tar.gz: e568154643ebebfc8a9dcfd506b18732d52dea1a19b6500b76e9103965fd2cff
+  metadata.gz: 1ec81c5f43cb98b0fe8fb61484ea7b0050a84cb8c7caac6d25e314d58f5f1be0
+  data.tar.gz: 4a99eafd9d2f53885e063cc89af6ecf2afce3506e0f9b118a999e52c5fe93e71
 SHA512:
-  metadata.gz: d01699afa9a6c147317b8b6fd5763bcbf63852bccaa4fe6dd5bac076405bfc02ff98ae177cd1eef39a8aac95493a4e7eac31896260bceb8b78d2c2b3976605c6
-  data.tar.gz: 932f70292fde728c25d21bf7180dbad488ca4b5d7630bfe669b70dbf1dee0165f24b311d284ed0ab8edd1b5ab1b688d28d29e7a1748c8a571101950568373a00
+  metadata.gz: 66909adaf233af63fb1f7f251cb8f11ba4a484c80e7646c9202c5636a7477575880f85927e01403ce26f8ebaa4a88c5eea6ffa1052a5a34d4e8b12d18f5cbb86
+  data.tar.gz: 8ad7289243de4d4f4e1b34d759be041f33c5b02d6b99428e9fc09ad918baa021751d708adc335c126495dce29eae9fca61a63cf7533711c7cd88cb6caf3511a3

data/.buildkite/hooks/pre-command

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+# Place this in /etc/buildkite-agent/hooks/pre-command
+#
+# Needs to be `chown buildkite-agent` and `chmod +x`
+
+# RVM uses unbound variables and will fail without this
+set +u
+
+source /var/lib/buildkite-agent/.rvm/scripts/rvm

data/.buildkite/pipeline.yml

@@ -0,0 +1,7 @@
+env:
+  MYSQL_HOST: 127.0.0.1
+  MYSQL_USER: root
+
+steps:
+  - command: "testing.sh"
+    label: "Run TiDB ActiveRecord Adapter testing 5-2-stable"

data/.github/entrypoint.sh

@@ -0,0 +1,16 @@
+#!/bin/bash
+set -eu
+service supervisor start
+
+while sleep 60; do
+  ps aux |grep tidb |grep -q -v grep
+  TIDB_STATUS=$?
+  ps aux |grep tikv |grep -q -v grep
+  TIKV_STATUS=$?
+  # If the greps above find anything, they exit with 0 status
+  # If they are not both 0, then something is wrong
+  if [ $TIDB_STATUS -ne 0 -o $TIKV_STATUS -ne 0 ]; then
+    echo "One of the processes has already exited."
+    exit 1
+  fi
+done

data/.github/workflows/ci.yml

@@ -0,0 +1,35 @@
+name: activerecord-tidb-adapter ci
+
+on:
+  push:
+    branches:
+      - 5-2-stable
+    paths-ignore:
+      - 'README*.md'
+      - 'docs/**'
+  pull_request:
+    branches:
+      - 5-2-stable
+    paths-ignore:
+      - 'README*.md'
+      - 'docs/**'
+
+jobs:
+  build:
+    if: ${{ !contains(github.event.commits[0].message, '[skip ci]') }}
+    runs-on: ubuntu-latest
+    services:
+      tidb:
+        image: hawkingrei/tind:v5.1.0
+        options: --entrypoint /entrypoint.sh
+        env:
+          TIDB_VERSION: v5.1.0
+        ports: ["4000:4000"]
+    steps:
+    - uses: actions/checkout@v2
+    - uses: ruby/setup-ruby@v1
+      with:
+        ruby-version: 2.7
+        bundler-cache: true
+    - run: sleep 30 && MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle exec rake db:tidb:build
+    - run: sleep 10 && MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle exec rake test:tidb

data/Gemfile

@@ -14,10 +14,14 @@ gem 'pry'
 
 gem 'rubocop', '~> 1.18'
 
-gem 'rails', git: 'https://github.com/tidb-incubator/rails.git', branch: '6-1-stable'
+gem 'rails', git: 'https://github.com/pingcap/rails.git', branch: '5-2-stable'
 
 gem 'byebug', '~> 11.1'
 
 gem 'sqlite3', '~> 1.4'
 
 gem 'pg', '~> 1.2'
+
+gem "mocha", "~> 1.13"
+
+gem "bcrypt", "~> 3.1"

data/Gemfile.lock

@@ -1,100 +1,82 @@
 GIT
-  remote: https://github.com/tidb-incubator/rails.git
-  revision: e377e0dc16e9841a048ec7eab2666c06155b6a0c
-  branch: 6-1-stable
+  remote: https://github.com/pingcap/rails.git
+  revision: a1adbde17b229e9131d40787ee4342c1180afb36
+  branch: 5-2-stable
   specs:
-    actioncable (6.1.4)
-      actionpack (= 6.1.4)
-      activesupport (= 6.1.4)
+    actioncable (5.2.6)
+      actionpack (= 5.2.6)
       nio4r (~> 2.0)
       websocket-driver (>= 0.6.1)
-    actionmailbox (6.1.4)
-      actionpack (= 6.1.4)
-      activejob (= 6.1.4)
-      activerecord (= 6.1.4)
-      activestorage (= 6.1.4)
-      activesupport (= 6.1.4)
-      mail (>= 2.7.1)
-    actionmailer (6.1.4)
-      actionpack (= 6.1.4)
-      actionview (= 6.1.4)
-      activejob (= 6.1.4)
-      activesupport (= 6.1.4)
+    actionmailer (5.2.6)
+      actionpack (= 5.2.6)
+      actionview (= 5.2.6)
+      activejob (= 5.2.6)
       mail (~> 2.5, >= 2.5.4)
       rails-dom-testing (~> 2.0)
-    actionpack (6.1.4)
-      actionview (= 6.1.4)
-      activesupport (= 6.1.4)
-      rack (~> 2.0, >= 2.0.9)
+    actionpack (5.2.6)
+      actionview (= 5.2.6)
+      activesupport (= 5.2.6)
+      rack (~> 2.0, >= 2.0.8)
       rack-test (>= 0.6.3)
       rails-dom-testing (~> 2.0)
-      rails-html-sanitizer (~> 1.0, >= 1.2.0)
-    actiontext (6.1.4)
-      actionpack (= 6.1.4)
-      activerecord (= 6.1.4)
-      activestorage (= 6.1.4)
-      activesupport (= 6.1.4)
-      nokogiri (>= 1.8.5)
-    actionview (6.1.4)
-      activesupport (= 6.1.4)
+      rails-html-sanitizer (~> 1.0, >= 1.0.2)
+    actionview (5.2.6)
+      activesupport (= 5.2.6)
       builder (~> 3.1)
       erubi (~> 1.4)
       rails-dom-testing (~> 2.0)
-      rails-html-sanitizer (~> 1.1, >= 1.2.0)
-    activejob (6.1.4)
-      activesupport (= 6.1.4)
+      rails-html-sanitizer (~> 1.0, >= 1.0.3)
+    activejob (5.2.6)
+      activesupport (= 5.2.6)
       globalid (>= 0.3.6)
-    activemodel (6.1.4)
-      activesupport (= 6.1.4)
-    activerecord (6.1.4)
-      activemodel (= 6.1.4)
-      activesupport (= 6.1.4)
-    activestorage (6.1.4)
-      actionpack (= 6.1.4)
-      activejob (= 6.1.4)
-      activerecord (= 6.1.4)
-      activesupport (= 6.1.4)
+    activemodel (5.2.6)
+      activesupport (= 5.2.6)
+    activerecord (5.2.6)
+      activemodel (= 5.2.6)
+      activesupport (= 5.2.6)
+      arel (>= 9.0)
+    activestorage (5.2.6)
+      actionpack (= 5.2.6)
+      activerecord (= 5.2.6)
       marcel (~> 1.0.0)
-      mini_mime (>= 1.1.0)
-    activesupport (6.1.4)
+    activesupport (5.2.6)
       concurrent-ruby (~> 1.0, >= 1.0.2)
-      i18n (>= 1.6, < 2)
-      minitest (>= 5.1)
-      tzinfo (~> 2.0)
-      zeitwerk (~> 2.3)
-    rails (6.1.4)
-      actioncable (= 6.1.4)
-      actionmailbox (= 6.1.4)
-      actionmailer (= 6.1.4)
-      actionpack (= 6.1.4)
-      actiontext (= 6.1.4)
-      actionview (= 6.1.4)
-      activejob (= 6.1.4)
-      activemodel (= 6.1.4)
-      activerecord (= 6.1.4)
-      activestorage (= 6.1.4)
-      activesupport (= 6.1.4)
-      bundler (>= 1.15.0)
-      railties (= 6.1.4)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+    rails (5.2.6)
+      actioncable (= 5.2.6)
+      actionmailer (= 5.2.6)
+      actionpack (= 5.2.6)
+      actionview (= 5.2.6)
+      activejob (= 5.2.6)
+      activemodel (= 5.2.6)
+      activerecord (= 5.2.6)
+      activestorage (= 5.2.6)
+      activesupport (= 5.2.6)
+      bundler (>= 1.3.0)
+      railties (= 5.2.6)
       sprockets-rails (>= 2.0.0)
-    railties (6.1.4)
-      actionpack (= 6.1.4)
-      activesupport (= 6.1.4)
+    railties (5.2.6)
+      actionpack (= 5.2.6)
+      activesupport (= 5.2.6)
       method_source
-      rake (>= 0.13)
-      thor (~> 1.0)
+      rake (>= 0.8.7)
+      thor (>= 0.19.0, < 2.0)
 
 PATH
   remote: .
   specs:
-    activerecord-tidb-adapter (0.3.0)
-      activerecord (~> 6.1)
+    activerecord-tidb-adapter (5.2.2)
+      activerecord (~> 5.2)
       mysql2
 
 GEM
   remote: https://rubygems.org/
   specs:
+    arel (9.0.0)
     ast (2.4.2)
+    bcrypt (3.1.16)
     builder (3.2.4)
     byebug (11.1.3)
     coderay (1.1.3)
@@ -105,7 +87,7 @@ GEM
       activesupport (>= 5.0)
     i18n (1.8.10)
       concurrent-ruby (~> 1.0)
-    loofah (2.11.0)
+    loofah (2.12.0)
       crass (~> 1.0.2)
       nokogiri (>= 1.5.9)
     mail (2.7.1)
@@ -116,8 +98,11 @@ GEM
     minitest (5.14.4)
     minitest-excludes (2.0.1)
       minitest (~> 5.0)
+    mocha (1.13.0)
     mysql2 (0.5.3)
     nio4r (2.5.8)
+    nokogiri (1.12.3-x86_64-darwin)
+      racc (~> 1.4)
     nokogiri (1.12.3-x86_64-linux)
       racc (~> 1.4)
     parallel (1.20.1)
@@ -134,22 +119,22 @@ GEM
     rails-dom-testing (2.0.3)
       activesupport (>= 4.2.0)
       nokogiri (>= 1.6)
-    rails-html-sanitizer (1.3.0)
+    rails-html-sanitizer (1.4.1)
       loofah (~> 2.3)
     rainbow (3.0.0)
     rake (13.0.6)
     regexp_parser (2.1.1)
     rexml (3.2.5)
-    rubocop (1.18.4)
+    rubocop (1.19.0)
       parallel (~> 1.10)
       parser (>= 3.0.0.0)
       rainbow (>= 2.2.2, < 4.0)
       regexp_parser (>= 1.8, < 3.0)
       rexml
-      rubocop-ast (>= 1.8.0, < 2.0)
+      rubocop-ast (>= 1.9.1, < 2.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (>= 1.4.0, < 3.0)
-    rubocop-ast (1.8.0)
+    rubocop-ast (1.10.0)
       parser (>= 3.0.1.1)
     ruby-progressbar (1.11.0)
     sprockets (4.0.2)
@@ -161,13 +146,13 @@ GEM
       sprockets (>= 3.0.0)
     sqlite3 (1.4.2)
     thor (1.1.0)
-    tzinfo (2.0.4)
-      concurrent-ruby (~> 1.0)
+    thread_safe (0.3.6)
+    tzinfo (1.2.9)
+      thread_safe (~> 0.1)
     unicode-display_width (2.0.0)
     websocket-driver (0.7.5)
       websocket-extensions (>= 0.1.0)
     websocket-extensions (0.1.5)
-    zeitwerk (2.4.2)
 
 PLATFORMS
   x86_64-darwin-18
@@ -175,9 +160,11 @@ PLATFORMS
 
 DEPENDENCIES
   activerecord-tidb-adapter!
+  bcrypt (~> 3.1)
   byebug (~> 11.1)
   minitest (~> 5.0)
   minitest-excludes (~> 2.0)
+  mocha (~> 1.13)
   pg (~> 1.2)
   pry
   rails!

data/README.md

@@ -8,7 +8,7 @@ TiDB adapter for ActiveRecord 5 and 6. This is a lightweight extension of the my
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'activerecord-tidb-adapter'
+gem 'activerecord-tidb-adapter', "~> 5.2.2"
 ```
 
 And then execute:
@@ -44,6 +44,62 @@ development:
 
 * demo repo with rails 6.1.4: https://github.com/hooopo/activerecord-tidb-adapter-demo
 
+## TiDB features
+
+**[Sequence](https://docs.pingcap.com/tidb/stable/sql-statement-create-sequence)**
+
+Sequence as primary key
+
+```ruby
+class TestSeq < ActiveRecord::Migration[6.1]
+  def up
+    # more options: increment, min_value, cycle, cache
+    create_sequence :orders_seq, start: 1024
+    create_table :orders, id: false do |t|
+      t.primary_key :id, default: -> { "nextval(orders_seq)" }
+      t.string :name
+    end
+  end
+
+  def down
+    drop_table :orders
+    drop_sequence :orders_seq
+  end
+end
+```
+
+Generated DDL
+```sql
+mysql> show create table orders_seq\G;
+*************************** 1. row ***************************
+       Sequence: orders_seq
+Create Sequence: CREATE SEQUENCE `orders_seq` start with 1024 minvalue 1 maxvalue 9223372036854775806 increment by 1 nocache nocycle ENGINE=InnoDB
+
+mysql> show create table orders\G;
+*************************** 1. row ***************************
+       Table: orders
+Create Table: CREATE TABLE `orders` (
+  `id` bigint(20) NOT NULL DEFAULT nextval(`activerecord_tidb_adapter_demo_development`.`orders_seq`),
+  `name` varchar(255) COLLATE utf8mb4_general_ci DEFAULT NULL,
+  PRIMARY KEY (`id`) /*T![clustered_index] CLUSTERED */
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci
+
+```
+
+This gem also adds a few helpers to interact with `SEQUENCE`
+
+```ruby
+# Advance sequence and return new value
+ActiveRecord::Base.nextval("numbers")
+
+# Return value most recently obtained with nextval for specified sequence.
+ActiveRecord::Base.lastval("numbers")
+
+# Set sequence's current value
+ActiveRecord::Base.setval("numbers", 1234)
+```
+
+
 ## Setting up local TiDB server
 
 Install [tiup](https://github.com/pingcap/tiup)
@@ -84,16 +140,22 @@ MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle e
 
 ```
 
-run tidb adapter tests
+run tidb adapter tests and activerecord buildin tests
+
+```
+MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle exec rake test:tidb
 
 ```
-MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle exec rake test:idb
 
+run **ONLY** tidb adapter tests using `ONLY_TEST_TIDB` env:
+
+```
+ONLY_TEST_TIDB=1 MYSQL_USER=root MYSQL_HOST=127.0.0.1 MYSQL_PORT=4000 tidb=1 ARCONN=tidb bundle exec rake test:tidb
 ```
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/activerecord-tidb-adapter.
+Bug reports and pull requests are welcome on GitHub at https://github.com/pingcap/activerecord-tidb-adapter.
 
 ## License
 

data/Rakefile

@@ -15,7 +15,7 @@ namespace :test do
   Rake::TestTask.new('tidb') do |t|
     t.libs = ARTest::TiDB.test_load_paths
     t.test_files = test_files
-    t.warning = !!ENV['WARNING']
+    t.warning = !ENV['WARNING'].nil?
     t.verbose = false
   end
 

data/activerecord-tidb-adapter.gemspec

@@ -29,7 +29,7 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
-  spec.add_dependency 'activerecord', '~> 6.1'
+  spec.add_dependency 'activerecord', '~> 5.2'
   spec.add_dependency 'mysql2'
 
   # Uncomment to register a new dependency of your gem

data/lib/active_record/connection_adapters/tidb/database_statements.rb

@@ -0,0 +1,19 @@
+require 'active_record/connection_adapters/abstract/database_statements'
+
+ActiveRecord::ConnectionAdapters::DatabaseStatements.class_eval do 
+  def insert(arel, name = nil, pk = nil, id_value = nil, sequence_name = nil, binds = [])
+    sql, binds = to_sql_and_binds(arel, binds)
+    value = exec_insert(sql, name, binds, pk, sequence_name)
+
+    return id_value if id_value.present?
+    return last_inserted_id(value) if arel.is_a?(String)
+    table_name = arel.ast.relation.table_name
+    pk_def = schema_cache.columns_hash(table_name)[pk]
+    if pk_def&.default_function && pk_def.default_function =~ /nextval/
+      query_value("SELECT #{pk_def.default_function.sub('nextval', 'lastval')}")
+    else
+      last_inserted_id(value)
+    end
+  end
+  alias create insert
+end

data/lib/active_record/connection_adapters/tidb/schema_statements.rb

@@ -0,0 +1,28 @@
+require 'pry'
+require 'active_record/connection_adapters/mysql/schema_statements'
+
+
+ActiveRecord::ConnectionAdapters::MySQL::SchemaStatements.class_eval do 
+  def new_column_from_field(table_name, field)
+    type_metadata = fetch_type_metadata(field[:Type], field[:Extra])
+    if type_metadata.type == :datetime && /\ACURRENT_TIMESTAMP(?:\([0-6]?\))?\z/i.match?(field[:Default])
+      default, default_function = nil, field[:Default]
+    elsif field[:Default].to_s =~ /nextval/i
+      default_function = field[:Default]
+      default = nil
+    else
+      default, default_function = field[:Default], nil
+    end
+
+    ActiveRecord::ConnectionAdapters::MySQL::Column.new(
+      field[:Field],
+      default,
+      type_metadata,
+      field[:Null] == "YES",
+      table_name,
+      default_function,
+      field[:Collation],
+      comment: field[:Comment].presence
+    )
+  end
+end

data/lib/active_record/connection_adapters/tidb/type.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module ActiveRecord
   module Type
     class << self

data/lib/active_record/connection_adapters/tidb_adapter.rb

@@ -1,16 +1,18 @@
 # frozen_string_literal: true
 
-require 'active_record/connection_adapters'
 require 'active_record/connection_adapters/mysql2_adapter'
 require 'active_record/connection_adapters/tidb/setup'
 require_relative '../../version'
+require_relative '../sequence'
+require_relative 'tidb/database_statements'
+require_relative 'tidb/schema_statements'
 
 ActiveRecord::ConnectionAdapters::Tidb.initial_setup
 
 module ActiveRecord
-  module ConnectionHandling #:nodoc:
+  module ConnectionHandling # :nodoc:
     # Establishes a connection to the database that's used by all Active Record objects.
-    def tidb_connection(config) #:nodoc:
+    def tidb_connection(config) # :nodoc:
       config = config.symbolize_keys
       config[:flags] ||= 0
 
@@ -29,8 +31,30 @@ module ActiveRecord
     end
   end
 
+  
+
   module ConnectionAdapters
+    class Mysql2Adapter < AbstractMysqlAdapter
+      ER_BAD_DB_ERROR = 1049
+      ADAPTER_NAME = "Mysql2"
+
+      include MySQL::DatabaseStatements
+
+      class << self
+        def new_client(config)
+          Mysql2::Client.new(config)
+        rescue Mysql2::Error => error
+          if error.error_number == ConnectionAdapters::Mysql2Adapter::ER_BAD_DB_ERROR
+            raise ActiveRecord::NoDatabaseError
+          else
+            raise ActiveRecord::ConnectionNotEstablished, error.message
+          end
+        end
+      end
+    end
+
     class TidbAdapter < Mysql2Adapter
+      include ActiveRecord::Sequence::Adapter
       ADAPTER_NAME = 'Tidb'
 
       def supports_savepoints?
@@ -93,7 +117,7 @@ module ActiveRecord
       end
 
       def self.database_exists?(config)
-        !!ActiveRecord::Base.tidb_connection(config)
+        !ActiveRecord::Base.tidb_connection(config).nil?
       rescue ActiveRecord::NoDatabaseError
         false
       end

data/lib/active_record/sequence/adapter.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Sequence
+    module Adapter
+      def check_sequences
+        select_all(
+          'SELECT * FROM information_schema.sequences ORDER BY sequence_name'
+        ).to_a
+      end
+
+      def create_sequence(name, options = {})
+        increment = options[:increment] || options[:step]
+        name = quote_column_name(name)
+
+        sql = ["CREATE SEQUENCE IF NOT EXISTS #{name}"]
+        sql << "INCREMENT BY #{increment}" if increment
+        sql << "START WITH #{options[:start]}" if options[:start]
+        sql << if options[:cache]
+                 "CACHE #{options[:cache]}"
+               else
+                 'NOCACHE'
+               end
+
+        sql << if options[:cycle]
+                 'CYCLE'
+               else
+                 'NOCYCLE'
+               end
+
+        sql << "MIN_VALUE #{options[:min_value]}" if options[:min_value]
+
+        sql << "COMMENT #{quote(options[:comment].to_s)}" if options[:comment]
+
+        execute(sql.join("\n"))
+      end
+
+      def drop_sequence(name)
+        name = quote_column_name(name)
+        sql = "DROP SEQUENCE IF EXISTS #{name}"
+        execute(sql)
+      end
+
+      def recreate_sequence(name, options = {})
+        drop_sequence(name)
+        create_sequence(name, options)
+      end
+    end
+  end
+end

data/lib/active_record/sequence/command_recorder.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Sequence
+    module CommandRecorder
+      def create_sequence(name, options = {})
+        record(__method__, [name, options])
+      end
+
+      def drop_sequence(name)
+        record(__method__, [name])
+      end
+
+      def invert_create_sequence(args)
+        name, = args
+        [:drop_sequence, [name]]
+      end
+    end
+  end
+end

data/lib/active_record/sequence/model_methods.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Sequence
+    module ModelMethods
+      def nextval(name)
+        name = connection.quote_column_name(name)
+        connection.query_value("SELECT nextval(#{name})")
+      end
+
+      def lastval(name)
+        name = connection.quote_column_name(name)
+        connection.query_value("SELECT lastval(#{name})")
+      end
+
+      def setval(name, value)
+        name = connection.quote_column_name(name)
+        connection.query_value("SELECT setval(#{name}, #{value})")
+      end
+    end
+  end
+end

data/lib/active_record/sequence/schema_dumper.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module Sequence
+    module SchemaDumper
+      def header(stream)
+        super
+        sequences(stream)
+      end
+
+      def sequences(stream)
+        sequences = @connection.check_sequences
+        return if sequences.empty?
+
+        sequences.each do |seq|
+          start_value = seq['START']
+          increment = seq['INCREMENT']
+          cache = seq['CACHE']
+          cache_value = seq['CACHE_VALUE']
+          min_value = seq['MIN_VALUE']
+          cycle = seq['CYCLE']
+          comment = seq['COMMENT']
+
+          options = []
+
+          options << "start: #{start_value}" if start_value && Integer(start_value) != 1
+
+          options << "increment: #{increment}" if increment && Integer(increment) != 1
+
+          options << "cache: #{cache_value}" if cache_value && Integer(cache_value) != 0
+
+          options << "min_value: #{min_value}" if min_value
+
+          options << 'cycle: true' if cycle.to_i != 0
+
+          options << "comment: #{comment.inspect}" if comment.present?
+
+          statement = [
+            'create_sequence',
+            seq['SEQUENCE_NAME'].inspect
+          ].join(' ')
+
+          if options.any?
+            statement << (options.any? ? ", #{options.join(', ')}" : '')
+          end
+
+          stream.puts "  #{statement}"
+        end
+
+        stream.puts
+      end
+    end
+  end
+end

data/lib/active_record/sequence.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'active_support/all'
+require 'active_record'
+require 'active_record/connection_adapters/mysql/schema_dumper'
+require 'active_record/migration/command_recorder'
+require 'active_record/schema_dumper'
+
+module ActiveRecord
+  module Sequence
+    require 'active_record/sequence/command_recorder'
+    require 'active_record/sequence/adapter'
+    require 'active_record/sequence/schema_dumper'
+    require 'active_record/sequence/model_methods'
+  end
+end
+
+ActiveRecord::Migration::CommandRecorder.include(ActiveRecord::Sequence::CommandRecorder)
+
+ActiveRecord::SchemaDumper.prepend(ActiveRecord::Sequence::SchemaDumper)
+ActiveRecord::Base.extend(ActiveRecord::Sequence::ModelMethods)

data/lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveRecord
-  TIDB_ADAPTER_VERSION = '0.3.0'
+  TIDB_ADAPTER_VERSION = '5.2.2'
 end

data/testing.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+set -eo pipefail
+
+bundle config set --local path '/tmp/buildkite-cache' 
+
+echo "Setup gem mirror"
+bundle config mirror.https://rubygems.org https://gems.ruby-china.com 
+
+echo "Bundle install"
+bundle install
+
+echo "Setup database for testing"
+tidb=1 ARCONN=tidb bundle exec rake db:tidb:rebuild && tidb=1 ARCONN=tidb bundle exec rake test:tidb

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-tidb-adapter
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 5.2.2
 platform: ruby
 authors:
 - Hooopo Wang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-08-12 00:00:00.000000000 Z
+date: 2021-08-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '5.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '5.2'
 - !ruby/object:Gem::Dependency
   name: mysql2
   requirement: !ruby/object:Gem::Requirement
@@ -45,6 +45,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".buildkite/hooks/pre-command"
+- ".buildkite/pipeline.yml"
+- ".github/entrypoint.sh"
+- ".github/workflows/ci.yml"
 - ".gitignore"
 - Gemfile
 - Gemfile.lock
@@ -55,12 +59,20 @@ files:
 - bin/console
 - bin/setup
 - config.toml
+- lib/active_record/connection_adapters/tidb/database_statements.rb
 - lib/active_record/connection_adapters/tidb/database_tasks.rb
+- lib/active_record/connection_adapters/tidb/schema_statements.rb
 - lib/active_record/connection_adapters/tidb/setup.rb
 - lib/active_record/connection_adapters/tidb/type.rb
 - lib/active_record/connection_adapters/tidb_adapter.rb
+- lib/active_record/sequence.rb
+- lib/active_record/sequence/adapter.rb
+- lib/active_record/sequence/command_recorder.rb
+- lib/active_record/sequence/model_methods.rb
+- lib/active_record/sequence/schema_dumper.rb
 - lib/activerecord-tidb-adapter.rb
 - lib/version.rb
+- testing.sh
 homepage: https://github.com/pingcap/activerecord-tidb-adapter
 licenses:
 - Apache-2.0