Add GetOrCreate<ListName>Map helper for lists. (#971)

This helps avoid nil checks when there is a need to operate on the map
field directly.

Author: wenovus

exampleoc/oc.go

@@ -851,6 +851,17 @@ type Tstruct struct {
 func (*Tstruct) IsYANGGoStruct() {}
 `,
 			methods: `
+// GetOrCreateListWithKeyMap returns the ordered map field
+// ListWithKey from Tstruct.
+//
+// It initializes the field if not already initialized.
+func (s *Tstruct) GetOrCreateListWithKeyMap() *Tstruct_ListWithKey_OrderedMap {
+	if s.ListWithKey == nil {
+		s.ListWithKey = &Tstruct_ListWithKey_OrderedMap{}
+	}
+	return s.ListWithKey
+}
+
 // AppendNewListWithKey creates a new entry in the ListWithKey
 // ordered map of the Tstruct struct. The keys of the list are
 // populated from the input arguments.
@@ -1548,6 +1559,16 @@ func (t *Tstruct) NewListWithKey(KeyLeafOne string, KeyLeafTwo int8) (*Tstruct_L
 	return t.ListWithKey[key], nil
 }
 
+// GetOrCreateListWithKeyMap returns the list (map) from Tstruct.
+//
+// It initializes the field if not already initialized.
+func (t *Tstruct) GetOrCreateListWithKeyMap() map[Tstruct_ListWithKey_Key]*Tstruct_ListWithKey {
+	if t.ListWithKey == nil {
+		t.ListWithKey = make(map[Tstruct_ListWithKey_Key]*Tstruct_ListWithKey)
+	}
+	return t.ListWithKey
+}
+
 // GetOrCreateListWithKey retrieves the value with the specified keys from
 // the receiver Tstruct. If the entry does not exist, then it is created.
 // It returns the existing or new list member.
@@ -1768,6 +1789,16 @@ func (t *Tstruct) NewListWithKey(KeyLeaf string) (*Tstruct_ListWithKey, error){
 	return t.ListWithKey[key], nil
 }
 
+// GetOrCreateListWithKeyMap returns the list (map) from Tstruct.
+//
+// It initializes the field if not already initialized.
+func (t *Tstruct) GetOrCreateListWithKeyMap() map[string]*Tstruct_ListWithKey {
+	if t.ListWithKey == nil {
+		t.ListWithKey = make(map[string]*Tstruct_ListWithKey)
+	}
+	return t.ListWithKey
+}
+
 // GetOrCreateListWithKey retrieves the value with the specified keys from
 // the receiver Tstruct. If the entry does not exist, then it is created.
 // It returns the existing or new list member.

exampleoc/ocpath.go

@@ -33,6 +33,17 @@ type RoutingPolicy_PolicyDefinition struct {
 	Statement *RoutingPolicy_PolicyDefinition_Statement_OrderedMap
 }
 
+// GetOrCreateStatementMap returns the ordered map field
+// Statement from RoutingPolicy_PolicyDefinition.
+//
+// It initializes the field if not already initialized.
+func (s *RoutingPolicy_PolicyDefinition) GetOrCreateStatementMap() *RoutingPolicy_PolicyDefinition_Statement_OrderedMap {
+	if s.Statement == nil {
+		s.Statement = &RoutingPolicy_PolicyDefinition_Statement_OrderedMap{}
+	}
+	return s.Statement
+}
+
 // AppendNewStatement appends a new policy statement with the given key.
 func (s *RoutingPolicy_PolicyDefinition) AppendNewStatement(key string) (*RoutingPolicy_PolicyDefinition_Statement, error) {
 	if s.Statement == nil {

exampleoc/opstateoc/oc.go

@@ -240,3 +240,12 @@ func TestOrderedMapFromParent(t *testing.T) {
 		},
 	})
 }
+
+func TestOrderedMapFromParentGetOrCreate(t *testing.T) {
+	p := &RoutingPolicy_PolicyDefinition{}
+
+	p.GetOrCreateStatementMap()
+	if p.Statement == nil {
+		t.Errorf("Statement must be initialized")
+	}
+}

gogen/gogen_test.go

@@ -50,6 +50,17 @@ func OrderedMapTypeName(listElemTypeName string) string {
 
 var (
 	goOrderedMapParentMethodsTemplate = mustMakeTemplate("orderedMapParentMethods", `
+// GetOrCreate{{ .ListFieldName }}Map returns the ordered map field
+// {{ .ListFieldName }} from {{ .ParentStructName }}.
+//
+// It initializes the field if not already initialized.
+func (s *{{ .ParentStructName }}) GetOrCreate{{ .ListFieldName }}Map() *{{ .StructName }} {
+	if s.{{ .ListFieldName }} == nil {
+		s.{{ .ListFieldName }} = &{{ .StructName }}{}
+	}
+	return s.{{ .ListFieldName }}
+}
+
 // AppendNew{{ .ListFieldName }} creates a new entry in the {{ .ListFieldName }}
 // ordered map of the {{ .ParentStructName }} struct. The keys of the list are
 // populated from the input arguments.

gogen/internal/gotypes/ordered_map.go

@@ -222,6 +222,16 @@ func (t *OpenconfigListEnumKey_Top_MultiKey) NewEkm(K1 E_OpenconfigListEnumKey_T
 	return t.Ekm[key], nil
 }
 
+// GetOrCreateEkmMap returns the list (map) from OpenconfigListEnumKey_Top_MultiKey.
+//
+// It initializes the field if not already initialized.
+func (t *OpenconfigListEnumKey_Top_MultiKey) GetOrCreateEkmMap() map[OpenconfigListEnumKey_Top_MultiKey_Ekm_Key]*OpenconfigListEnumKey_Top_MultiKey_Ekm {
+	if t.Ekm == nil {
+		t.Ekm = make(map[OpenconfigListEnumKey_Top_MultiKey_Ekm_Key]*OpenconfigListEnumKey_Top_MultiKey_Ekm)
+	}
+	return t.Ekm
+}
+
 // GetOrCreateEkm retrieves the value with the specified keys from
 // the receiver OpenconfigListEnumKey_Top_MultiKey. If the entry does not exist, then it is created.
 // It returns the existing or new list member.
@@ -509,6 +519,16 @@ func (t *OpenconfigListEnumKey_Top_SingleKey) NewEks(K E_OpenconfigListEnumKey_T
 	return t.Eks[key], nil
 }
 
+// GetOrCreateEksMap returns the list (map) from OpenconfigListEnumKey_Top_SingleKey.
+//
+// It initializes the field if not already initialized.
+func (t *OpenconfigListEnumKey_Top_SingleKey) GetOrCreateEksMap() map[E_OpenconfigListEnumKey_Top_SingleKey_Eks_Config_K]*OpenconfigListEnumKey_Top_SingleKey_Eks {
+	if t.Eks == nil {
+		t.Eks = make(map[E_OpenconfigListEnumKey_Top_SingleKey_Eks_Config_K]*OpenconfigListEnumKey_Top_SingleKey_Eks)
+	}
+	return t.Eks
+}
+
 // GetOrCreateEks retrieves the value with the specified keys from
 // the receiver OpenconfigListEnumKey_Top_SingleKey. If the entry does not exist, then it is created.
 // It returns the existing or new list member.

gogen/internal/gotypes/ordered_map_test.go

@@ -197,6 +197,17 @@ func (t *Model) RenameSingleKey(oldK, newK string) error {
 	return nil
 }
 
+// GetOrCreateSingleKeyOrderedMap returns the ordered map field
+// SingleKeyOrdered from Model.
+//
+// It initializes the field if not already initialized.
+func (s *Model) GetOrCreateSingleKeyOrderedMap() *Model_SingleKeyOrdered_OrderedMap {
+	if s.SingleKeyOrdered == nil {
+		s.SingleKeyOrdered = &Model_SingleKeyOrdered_OrderedMap{}
+	}
+	return s.SingleKeyOrdered
+}
+
 // AppendNewSingleKeyOrdered creates a new entry in the SingleKeyOrdered
 // ordered map of the Model struct. The keys of the list are
 // populated from the input arguments.

gogen/ordered_list.go

@@ -197,6 +197,17 @@ func (t *Model) RenameSingleKey(oldK, newK string) error {
 	return nil
 }
 
+// GetOrCreateSingleKeyOrderedMap returns the ordered map field
+// SingleKeyOrdered from Model.
+//
+// It initializes the field if not already initialized.
+func (s *Model) GetOrCreateSingleKeyOrderedMap() *Model_SingleKeyOrdered_OrderedMap {
+	if s.SingleKeyOrdered == nil {
+		s.SingleKeyOrdered = &Model_SingleKeyOrdered_OrderedMap{}
+	}
+	return s.SingleKeyOrdered
+}
+
 // AppendNewSingleKeyOrdered creates a new entry in the SingleKeyOrdered
 // ordered map of the Model struct. The keys of the list are
 // populated from the input arguments.