From 367834ee1623658b0c9f273881951defeec7887a Mon Sep 17 00:00:00 2001
From: rjl493456442 <garyrong0905@gmail.com>
Date: Mon, 12 Oct 2020 12:44:08 +0800
Subject: [PATCH 1/2] trie: polish commit function

---
 trie/stacktrie.go | 23 +++++++++++++++--------
 trie/trie_test.go |  5 ++++-
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/trie/stacktrie.go b/trie/stacktrie.go
index fc653101ae8c..83e5cb391a11 100644
--- a/trie/stacktrie.go
+++ b/trie/stacktrie.go
@@ -17,6 +17,7 @@
 package trie
 
 import (
+	"errors"
 	"fmt"
 	"sync"
 
@@ -26,6 +27,8 @@ import (
 	"github.com/ethereum/go-ethereum/rlp"
 )
 
+var ErrCommitDisabled = errors.New("no database for committing")
+
 var stPool = sync.Pool{
 	New: func() interface{} {
 		return NewStackTrie(nil)
@@ -391,14 +394,18 @@ func (st *StackTrie) Hash() (h common.Hash) {
 	return common.BytesToHash(st.val)
 }
 
-// Commit will commit the current node to database db
-func (st *StackTrie) Commit(db ethdb.KeyValueStore) common.Hash {
-	oldDb := st.db
-	st.db = db
-	defer func() {
-		st.db = oldDb
-	}()
+// Commit will firstly hash the entrie trie if it's still not hashed
+// and then commit all nodes to the assoicated database. Actually most
+// of the trie nodes MAY have been committed already. The main purpose
+// here is to commit the root node.
+//
+// The assoiciated database is expected, otherwise the whole commit
+// functionality should be disabled.
+func (st *StackTrie) Commit() (common.Hash, error) {
+	if st.db == nil {
+		return common.Hash{}, ErrCommitDisabled
+	}
 	st.hash()
 	h := common.BytesToHash(st.val)
-	return h
+	return h, nil
 }
diff --git a/trie/trie_test.go b/trie/trie_test.go
index 03ec0cab8971..539451fbf4dd 100644
--- a/trie/trie_test.go
+++ b/trie/trie_test.go
@@ -831,7 +831,10 @@ func TestCommitSequenceStackTrie(t *testing.T) {
 		// Flush memdb -> disk (sponge)
 		db.Commit(root, false, nil)
 		// And flush stacktrie -> disk
-		stRoot := stTrie.Commit(stTrie.db)
+		stRoot, err := stTrie.Commit()
+		if err != nil {
+			t.Fatalf("Failed to commit stack trie %v", err)
+		}
 		if stRoot != root {
 			t.Fatalf("root wrong, got %x exp %x", stRoot, root)
 		}

From 2871fe8268a2c7c7bcf4300a776bfad33ef12f97 Mon Sep 17 00:00:00 2001
From: rjl493456442 <garyrong0905@gmail.com>
Date: Mon, 12 Oct 2020 14:51:01 +0800
Subject: [PATCH 2/2] trie: fix typo

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

diff --git a/trie/stacktrie.go b/trie/stacktrie.go
index 83e5cb391a11..33fa990077c8 100644
--- a/trie/stacktrie.go
+++ b/trie/stacktrie.go
@@ -395,11 +395,11 @@ func (st *StackTrie) Hash() (h common.Hash) {
 }
 
 // Commit will firstly hash the entrie trie if it's still not hashed
-// and then commit all nodes to the assoicated database. Actually most
+// and then commit all nodes to the associated database. Actually most
 // of the trie nodes MAY have been committed already. The main purpose
 // here is to commit the root node.
 //
-// The assoiciated database is expected, otherwise the whole commit
+// The associated database is expected, otherwise the whole commit
 // functionality should be disabled.
 func (st *StackTrie) Commit() (common.Hash, error) {
 	if st.db == nil {