Issue 419 (#2144)

* feat: bind first lock script to cell_output

Signed-off-by: Miles Zhang <[email protected]>

* feat: use task to update cell output script id

Signed-off-by: Miles Zhang <[email protected]>

---------

Signed-off-by: Miles Zhang <[email protected]>

app/models/ckb_sync/new_node_data_processor.rb

@@ -914,13 +914,13 @@ def prepare_script_ids(outputs)
           local_cache.fetch("NodeData/LockScript/#{output.lock.code_hash}-#{output.lock.hash_type}-#{output.lock.args}") do
             # TODO use LockScript.where(script_hash: output.lock.compute_hash).select(:id)&.first replace search by code_hash, hash_type and args query after script_hash has been filled
             LockScript.where(code_hash: output.lock.code_hash, hash_type: output.lock.hash_type,
-                             args: output.lock.args).take!
+                             args: output.lock.args).order("id asc").first
           end
           if output.type.present?
             local_cache.fetch("NodeData/TypeScript/#{output.type.code_hash}-#{output.type.hash_type}-#{output.type.args}") do
               # TODO use TypeScript.where(script_hash: output.type.compute_hash).select(:id)&.first replace search by code_hash, hash_type and args query after script_hash has been filled
               TypeScript.where(code_hash: output.type.code_hash, hash_type: output.type.hash_type,
-                               args: output.type.args).take!
+                               args: output.type.args).order("id asc").first
             end
           end
         end

lib/tasks/migration/update_cell_output_script_id.rake

@@ -0,0 +1,87 @@
+namespace :migration do
+  desc "Usage: RAILS_ENV=production bundle exec rake migration:update_cell_output_script_id"
+  task update_cell_output_script_id: :environment do
+    ActiveRecord::Base.connection.execute("SET statement_timeout = 0")
+    p "==============lock scripts"
+    duplicate_script_hashes = LockScript.
+      select(:script_hash).
+      group(:script_hash).
+      having("COUNT(*) > 1").
+      pluck(:script_hash)
+    duplicate_script_hashes.each do |hash|
+      LockScript.where(script_hash: hash).each do |script|
+        unless CellOutput.live.where(lock_script_id: script.id).exists?
+          script.destroy
+        end
+      end
+    end; nil
+
+    duplicate_script_hashes.each_with_index do |lock_script_hash, index|
+      p lock_script_hash
+      p index
+      lock_script_ids = LockScript.where(script_hash: lock_script_hash).order("id asc").pluck(:id)
+      base_lock_script_id = lock_script_ids.delete_at(0)
+      lock_script_ids.each do |id|
+        CellOutput.where(lock_script_id: id).in_batches(of: 10000) do |batch|
+          batch.update_all(lock_script_id: base_lock_script_id)
+        end
+      end
+    end; nil
+
+    duplicate_script_hashes = LockScript.
+      select(:script_hash).
+      group(:script_hash).
+      having("COUNT(*) > 1").
+      pluck(:script_hash)
+    duplicate_script_hashes.each do |hash|
+      LockScript.where(script_hash: hash).each do |script|
+        unless CellOutput.live.where(lock_script_id: script.id).exists?
+          script.destroy
+        end
+      end
+    end; nil
+
+    p "==============type scripts"
+    duplicate_type_script_hashes = TypeScript.
+      select(:script_hash).
+      group(:script_hash).
+      having("COUNT(*) > 1").
+      pluck(:script_hash)
+
+    duplicate_type_script_hashes.each do |hash|
+      TypeScript.where(script_hash: hash).each do |script|
+        unless CellOutput.live.where(type_script_id: script.id).exists?
+          script.destroy
+        end
+      end
+    end; nil
+
+    duplicate_type_script_hashes.each_with_index do |type_script_hash, index|
+      p type_script_hash
+      p index
+      type_script_ids = TypeScript.where(script_hash: type_script_hash).order("id asc").pluck(:id)
+      base_type_script_id = type_script_ids.delete_at(0)
+      type_script_ids.each do |id|
+        CellOutput.where(type_script_id: id).in_batches(of: 10000) do |batch|
+          batch.update_all(type_script_id: base_type_script_id)
+        end
+      end
+    end; nil
+
+    duplicate_type_script_hashes = TypeScript.
+      select(:script_hash).
+      group(:script_hash).
+      having("COUNT(*) > 1").
+      pluck(:script_hash)
+
+    duplicate_type_script_hashes.each do |hash|
+      TypeScript.where(script_hash: hash).each do |script|
+        unless CellOutput.live.where(type_script_id: script.id).exists?
+          script.destroy
+        end
+      end
+    end; nil
+
+    puts "done"
+  end
+end