Merge pull request #368 from shaojunda/shaojunda-release-dao-withdraw-fee

[ᚬmaster] release dao withdraw fee

Author: shaojunda

.env

@@ -13,6 +13,7 @@ BLOCK_SAFETY_INTERVAL="10"
 AUTHENTICSYNC_LOOP_INTERVAL="10"
 INAUTHENTICSYNC_LOOP_INTERVAL="1"
 CODE_HASH="0x54811ce986d5c3e57eaafab22cdd080e32209e39590e204a99b32935f835a13c"
+DAO_CODE_HASH="0x0450e0eb0ad787135c1243e5730f72fa3bc8837e232e928c7f4b1efdd6692582"
 INITIAL_BLOCK_REWARD="5_000_000_000_000"
 HASH_RATE_STATISTICAL_INTERVAL="500"
 PROPOSAL_WINDOW = "10"

CHANGELOG.md

@@ -7,7 +7,6 @@ class Address < ApplicationRecord
   has_many :ckb_transactions, through: :account_books
   validates :balance, :cell_consumed, :ckb_transactions_count, numericality: { greater_than_or_equal_to: 0 }, allow_nil: true
 
-  attribute :lock_hash, :ckb_hash
   after_commit :flush_cache
 
   def lock_script

app/models/address.rb

@@ -1,5 +1,6 @@
 class CellOutput < ApplicationRecord
   enum status: { live: 0, dead: 1 }
+  enum cell_type: { normal: 0, dao: 1 }
 
   belongs_to :ckb_transaction
   belongs_to :generated_by, class_name: "CkbTransaction"
@@ -41,6 +42,7 @@ def node_output
 #  cell_index         :integer
 #  generated_by_id    :decimal(30, )
 #  consumed_by_id     :decimal(30, )
+#  cell_type          :integer          default("normal")
 #
 # Indexes
 #

app/models/cell_output.rb

@@ -2,7 +2,7 @@ module CkbSync
   class Api
     include Singleton
 
-    METHOD_NAMES = %w(system_script_out_point dry_run_transaction set_system_script_cell system_script_cell system_script_cell_hash genesis_block get_block_by_number genesis_block_hash get_block_hash get_block get_tip_header get_tip_block_number get_cells_by_lock_hash get_transaction get_live_cell local_node_info get_current_epoch get_epoch_by_number get_peers tx_pool_info get_blockchain_info get_peers_state compute_transaction_hash get_cellbase_output_capacity_details).freeze
+    METHOD_NAMES = %w(system_script_out_point dry_run_transaction set_system_script_cell system_script_cell system_script_cell_hash genesis_block get_block_by_number genesis_block_hash get_block_hash get_block get_tip_header get_tip_block_number get_cells_by_lock_hash get_transaction get_live_cell local_node_info get_current_epoch get_epoch_by_number get_peers tx_pool_info get_blockchain_info get_peers_state compute_transaction_hash get_cellbase_output_capacity_details calculate_dao_maximum_withdraw).freeze
 
     def initialize
       @api = CKB::API.new(host: ENV["CKB_NODE_URL"])

app/models/ckb_sync/api.rb

@@ -234,6 +234,12 @@ def build_cell_outputs(node_outputs, ckb_transaction, addresses)
       end
     end
 
+    def cell_type(type_script)
+      return "normal" if type_script.blank?
+
+      type_script.code_hash == ENV["DAO_CODE_HASH"] ? "dao" : "normal"
+    end
+
     def build_cell_output(ckb_transaction, output, address, cell_index)
       ckb_transaction.cell_outputs.build(
         capacity: output.capacity,
@@ -242,7 +248,8 @@ def build_cell_output(ckb_transaction, output, address, cell_index)
         block: ckb_transaction.block,
         tx_hash: ckb_transaction.tx_hash,
         cell_index: cell_index,
-        generated_by: ckb_transaction
+        generated_by: ckb_transaction,
+        cell_type: cell_type(output.type)
       )
     end
 

app/models/ckb_sync/node_data_processor.rb

@@ -65,10 +65,10 @@ def display_outputs
 #  block_timestamp :decimal(30, )
 #  transaction_fee :decimal(30, )
 #  version         :integer
-#  witnesses       :string           is an Array
 #  created_at      :datetime         not null
 #  updated_at      :datetime         not null
 #  is_cellbase     :boolean          default(FALSE)
+#  witnesses       :jsonb
 #
 # Indexes
 #

app/models/ckb_transaction.rb

@@ -83,7 +83,11 @@ def self.get_epoch_info(epoch)
   end
 
   def self.ckb_transaction_fee(ckb_transaction)
-    ckb_transaction.inputs.sum(:capacity) - ckb_transaction.outputs.sum(:capacity)
+    if ckb_transaction.inputs.dao.present?
+      dao_withdraw_tx_fee(ckb_transaction)
+    else
+      normal_tx_fee(ckb_transaction)
+    end
   end
 
   def self.get_unspent_cells(address_hash)
@@ -143,4 +147,25 @@ def self.update_target_block_miner_address_pending_rewards(current_block)
     miner_address = target_block.miner_address
     Address.decrement_counter(:pending_reward_blocks_count, miner_address.id, touch: true) if miner_address.present?
   end
+
+  def self.normal_tx_fee(ckb_transaction)
+    ckb_transaction.inputs.sum(:capacity) - ckb_transaction.outputs.sum(:capacity)
+  end
+
+  def self.dao_withdraw_tx_fee(ckb_transaction)
+    dao_cells = ckb_transaction.inputs.dao
+    witnesses = ckb_transaction.witnesses
+    deps = ckb_transaction.deps
+    interests =
+      dao_cells.reduce(0) do |memo, dao_cell|
+        witness = witnesses[dao_cell.cell_index]
+        dep = deps[witness["data"].last.hex]
+        out_point = CKB::Types::OutPoint.new(cell: CKB::Types::CellOutPoint.new(tx_hash: dao_cell.tx_hash, index: dao_cell.cell_index))
+        memo + CkbSync::Api.instance.calculate_dao_maximum_withdraw(out_point, dep["block_hash"]).to_i - dao_cell.capacity.to_i
+      end
+
+    ckb_transaction.inputs.sum(:capacity) + interests - ckb_transaction.outputs.sum(:capacity)
+  rescue CKB::RPCError
+    0
+  end
 end

app/utils/ckb_utils.rb

@@ -0,0 +1,5 @@
+class AddCellTypeToCellOutputs < ActiveRecord::Migration[5.2]
+  def change
+    add_column :cell_outputs, :cell_type, :integer, default: 0
+  end
+end

db/migrate/20190809023059_add_cell_type_to_cell_outputs.rb

@@ -0,0 +1,6 @@
+class ChangeWitnessesToJsonbInCkbTransactions < ActiveRecord::Migration[5.2]
+  def change
+    remove_column :ckb_transactions, :witnesses, :jsonb
+    add_column :ckb_transactions, :witnesses, :jsonb
+  end
+end