From 7486437a9c01a008a596e4b8fcfd906be35e1dff Mon Sep 17 00:00:00 2001
From: qianbin <cola.tin.com@gmail.com>
Date: Wed, 12 Jan 2022 01:01:15 +0800
Subject: [PATCH] trie: apply fastNodeEncoder to StackTrie

---
 trie/stacktrie.go | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/trie/stacktrie.go b/trie/stacktrie.go
index 76258c31123c2..33d989ec4ce48 100644
--- a/trie/stacktrie.go
+++ b/trie/stacktrie.go
@@ -376,7 +376,7 @@ func (st *StackTrie) hash() {
 
 	switch st.nodeType {
 	case branchNode:
-		var nodes [17]node
+		var nodes rawFullNode
 		for i, child := range st.children {
 			if child == nil {
 				nodes[i] = nilValueNode
@@ -395,7 +395,7 @@ func (st *StackTrie) hash() {
 		h = newHasher(false)
 		defer returnHasherToPool(h)
 		h.tmp.Reset()
-		if err := rlp.Encode(&h.tmp, nodes); err != nil {
+		if err := frlp.Encode(&h.tmp, nodes); err != nil {
 			panic(err)
 		}
 	case extNode:
@@ -409,14 +409,11 @@ func (st *StackTrie) hash() {
 		} else {
 			valuenode = hashNode(st.children[0].val)
 		}
-		n := struct {
-			Key []byte
-			Val node
-		}{
+		n := &rawShortNode{
 			Key: hexToCompact(st.key),
 			Val: valuenode,
 		}
-		if err := rlp.Encode(&h.tmp, n); err != nil {
+		if err := frlp.Encode(&h.tmp, n); err != nil {
 			panic(err)
 		}
 		returnToPool(st.children[0])