summaryrefslogtreecommitdiff
path: root/generate.go
diff options
context:
space:
mode:
Diffstat (limited to 'generate.go')
-rw-r--r--generate.go17
1 files changed, 17 insertions, 0 deletions
diff --git a/generate.go b/generate.go
index 2ef7c6c..2f7f029 100644
--- a/generate.go
+++ b/generate.go
@@ -197,6 +197,23 @@ func (pb *Files) makeNewSortfile(pf *File) error {
*/
}
+ // add Delete()
+ for _, s := range pf.ToSort {
+ PARENT := s.MsgName
+ CHILD := s.VarType
+ VARNAME := s.VarName
+
+ pmsg := pf.findMsg(s.MsgName)
+ if pmsg == nil {
+ return fmt.Errorf("failed to find struct %s", s.MsgName)
+ }
+
+ if PARENT == VARNAME {
+ funcdef := pmsg.addDeleteFunc(wSort, PARENT, VARNAME, CHILD)
+ log.Printf("Adding %s\n", funcdef)
+ }
+ }
+
// add Find() Delete() Append() Insert()
for _, s := range pf.ToSort {
PARENT := s.MsgName