Add back options

pokt-network · Jun 14, 2024 · 0f9b3f3 · 0f9b3f3
1 parent 74b04ed
commit 0f9b3f3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/trie_spec.go b/trie_spec.go
@@ -15,11 +15,20 @@ type TrieSpec struct {
 }
 
 // NewTrieSpec returns a new TrieSpec with the given hasher and sumTrie flag
-func NewTrieSpec(hasher hash.Hash, sumTrie bool) TrieSpec {
+func NewTrieSpec(
+	hasher hash.Hash,
+	sumTrie bool,
+	opts ...TrieSpecOption,
+) TrieSpec {
 	spec := TrieSpec{th: *NewTrieHasher(hasher)}
 	spec.ph = &pathHasher{spec.th}
 	spec.vh = &valueHasher{spec.th}
 	spec.sumTrie = sumTrie
+
+	for _, opt := range opts {
+		opt(&spec)
+	}
+
 	return spec
 }