perf: cache cell input

nervosnetwork · Sep 10, 2019 · 4fbdda6 · 4fbdda6
1 parent cefd400
commit 4fbdda6
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 4 deletions.
diff --git a/app/controllers/api/v1/cell_input_lock_scripts_controller.rb b/app/controllers/api/v1/cell_input_lock_scripts_controller.rb
@@ -4,7 +4,7 @@ class CellInputLockScriptsController < ApplicationController
       before_action :validate_query_params
 
       def show
-        cell_input = CellInput.find(params[:id])
+        cell_input = CellInput.cached_find(params[:id])
         lock_script = cell_input.find_lock_script!
 
         render json: LockScriptSerializer.new(lock_script)

diff --git a/app/models/cell_input.rb b/app/models/cell_input.rb
@@ -2,12 +2,18 @@ class CellInput < ApplicationRecord
   belongs_to :ckb_transaction
   belongs_to :block
 
+  after_commit :flush_cache
+
   def find_lock_script!
-    previous_cell_output!.lock_script
+    Rails.cache.fetch(["CellInput", id, "lock_script"]) do
+      previous_cell_output!.lock_script
+    end
   end
 
   def find_type_script!
-    previous_cell_output!.type_script
+    Rails.cache.fetch(["CellInput", id, "type_script"]) do
+      previous_cell_output!.type_script
+    end
   end
 
   def find_cell_output!
@@ -23,6 +29,15 @@ def previous_cell_output
     CellOutput.find_by(tx_hash: tx_hash, cell_index: cell_index)
   end
 
+  def self.cached_find(id)
+    Rails.cache.fetch([name, id]) { find(id) }
+  end
+
+  def flush_cache
+    Rails.cache.delete_matched("CellInput/#{id}*")
+    Rails.cache.delete_matched("previous_cell_output*")
+  end
+
   private
 
   def previous_cell_output!
@@ -31,7 +46,9 @@ def previous_cell_output!
     tx_hash = previous_output["tx_hash"]
     cell_index = previous_output["index"].to_i
 
-    CellOutput.find_by!(tx_hash: tx_hash, cell_index: cell_index)
+    Rails.cache.fetch("previous_cell_output/#{tx_hash}/#{cell_index}") do
+      CellOutput.find_by!(tx_hash: tx_hash, cell_index: cell_index)
+    end
   end
 end
 

diff --git a/app/models/cell_output.rb b/app/models/cell_output.rb
@@ -16,6 +16,8 @@ class CellOutput < ApplicationRecord
 
   attribute :tx_hash, :ckb_hash
 
+  after_commit :flush_cache
+
   def address_hash
     address.address_hash
   end
@@ -25,6 +27,10 @@ def node_output
     type = type_script.present? ? CKB::Types::Script.new(type_script.to_node_lock) : nil
     CKB::Types::Output.new(capacity: capacity.to_i, lock: lock, type: type)
   end
+
+  def flush_cache
+    Rails.cache.delete("previous_cell_output/#{tx_hash}/#{cell_index}")
+  end
 end
 
 # == Schema Information