Add Val method to node type

derekparker · Sep 16, 2024 · 4095f8e · 4095f8e
1 parent 6a4ad62
commit 4095f8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/trie.go b/trie.go
@@ -208,6 +208,11 @@ func (n *node[T]) removeChild(r rune) {
 	}
 }
 
+// Val returns the value of the node.
+func (n *node[T]) Val() T {
+	return n.meta
+}
+
 func findNode[T any](nd *node[T], runes []rune) *node[T] {
 	if nd == nil {
 		return nil

diff --git a/trie_test.go b/trie_test.go
@@ -46,7 +46,7 @@ func TestTrieFind(t *testing.T) {
 		t.Fatal("Could not find node")
 	}
 
-	if n.meta != 1 {
+	if n.Val() != 1 {
 		t.Errorf("Expected 1, got: %d", n.meta)
 	}
 }