initialize RPC input and output entries when augmented (#250)

* initialize RPC input and output entries when augmented * set kind field
openconfig · Sep 21, 2023 · 944052f · 944052f
1 parent 08bc968
commit 944052f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/yang/entry.go b/pkg/yang/entry.go
@@ -1344,8 +1344,22 @@ func (e *Entry) Find(name string) *Entry {
 			_, part = getPrefix(part)
 			switch part {
 			case "input":
+				if e.RPC.Input == nil {
+					e.RPC.Input = &Entry{
+						Name: "input",
+						Kind: InputEntry,
+						Dir:  make(map[string]*Entry),
+					}
+				}
 				e = e.RPC.Input
 			case "output":
+				if e.RPC.Output == nil {
+					e.RPC.Output = &Entry{
+						Name: "output",
+						Kind: OutputEntry,
+						Dir:  make(map[string]*Entry),
+					}
+				}
 				e = e.RPC.Output
 			}
 		default: