arql 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5ab8f5f261c0a207fa3d1d44798ec7f86864a0ff28fbb162bae3aa6abdae2311
-  data.tar.gz: 8c8a4238ad33800d565ff9546539213233a0c71f7759d77ad3f273825352f94c
+  metadata.gz: 50c397f035d89506f745002163459930acf16862c71aba29f61e3843da9fefef
+  data.tar.gz: b4dce80cde7add1ec3aafbb54e9d12736bc5e0d661f0f28aebcaf66c3bbc98a3
 SHA512:
-  metadata.gz: 3d4e836e5b0622c66ec1295bac29faa5ec6aa671fe6f88be30be69bb13366b150cf81193085d1618b6f9f63222996b750166e72fa2b944da6a9010f72314315e
-  data.tar.gz: 7f6ba7270d4b9125906dff1d4c7a0d4a22b806a97b2ffbbecec7036524a9d53b0e1b2f7bd3f09d4d966dfc2cb625602631291b6074d0c5c8e22eed6266e7beb2
+  metadata.gz: c60d3e7cd72361b59239c26967079e9c9baa0f28ff16caec7b20074d3a08636670e41afc7290fea95bfe00bc8ca4e78586aa89a996974c37364dc99c1e8c4855
+  data.tar.gz: 64f66abb0e4494b3f2b49b6972af3aab5a5e030f5b9ee9f51e074136a01eeeb2b8647aacad7c8d7501f5ea1c195d9100b93040a634c4db4e552d23d141df046c

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    arql (0.1.4)
+    arql (0.1.5)
       activerecord (~> 6.0.3)
       activesupport (~> 6.0.3)
       mysql2 (~> 0.5.3)

data/lib/arql/definition.rb

@@ -1,4 +1,32 @@
 module Arql
+  module Extension
+    extend ActiveSupport::Concern
+
+    def to_insert_sql
+      self.class.to_insert_sql([self])
+    end
+
+    def to_upsert_sql
+      self.class.to_upsert_sql([self])
+    end
+
+    class_methods do
+      def to_insert_sql(records, batch_size=1)
+        to_sql(records, :skip, batch_size)
+      end
+
+      def to_upsert_sql(records, batch_size=1)
+        to_sql(records, :update, batch_size)
+      end
+
+      def to_sql(records, on_duplicate, batch_size)
+        records.in_groups_of(batch_size, false).map do |group|
+        ActiveRecord::InsertAll.new(self, group.map(&:attributes), on_duplicate: on_duplicate).send(:to_sql) + ';'
+        end.join("\n")
+      end
+    end
+  end
+
   class Definition
     class << self
       def models
@@ -15,6 +43,7 @@ module Arql
               const_name = 'Modul' if const_name == 'Module'
               const_name = 'Clazz' if const_name == 'Class'
               Class.new(ActiveRecord::Base) do
+                include Arql::Extension
                 self.primary_key = pkey
                 self.table_name = table_name
                 self.inheritance_column = nil

data/lib/arql/ext/array.rb

@@ -0,0 +1,15 @@
+class Array
+  def to_insert_sql(batch_size=500)
+    raise 'All element should be an ActiveRecord instance object' unless all? { |e| e.is_a?(ActiveRecord::Base) }
+    group_by(&:class).map do |(klass, records)|
+      klass.to_insert_sql(records, batch_size)
+    end.join("\n")
+  end
+
+  def to_upsert_sql(batch_size=500)
+    raise 'All element should be an ActiveRecord instance object' unless all? { |e| e.is_a?(ActiveRecord::Base) }
+    group_by(&:class).map do |(klass, records)|
+      klass.to_upsert_sql(records, batch_size)
+    end.join("\n")
+  end
+end

data/lib/arql/ext.rb

@@ -0,0 +1 @@
+require 'arql/ext/array'

data/lib/arql/version.rb

@@ -1,3 +1,3 @@
 module Arql
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/arql.rb

@@ -4,6 +4,7 @@ require 'table_print'
 require "arql/version"
 require 'arql/id'
 require 'arql/multi_io'
+require 'arql/ext'
 require "arql/connection"
 require "arql/definition"
 require "arql/repl"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Liu Xiang
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-05-08 00:00:00.000000000 Z
+date: 2020-05-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -178,6 +178,8 @@ files:
 - lib/arql/commands/table.rb
 - lib/arql/connection.rb
 - lib/arql/definition.rb
+- lib/arql/ext.rb
+- lib/arql/ext/array.rb
 - lib/arql/id.rb
 - lib/arql/multi_io.rb
 - lib/arql/repl.rb