compiles with new forgepb proto files

1 files changed, 5 insertions, 1 deletions

diff --git a/generate.go b/generate.go
index e0d8bea..7ab5b07 100644
--- a/generate.go
+++ b/generate.go
@@ -179,8 +179,10 @@ func (pb *Files) makeNewSortfile(pf *File) error {
 		CHILD := s.VarType
 		VARNAME := s.VarName
 
-		pmsg := pf.findMsg(s.MsgName)
+		// log.Printf("TO SORT IterAll() PARENT==VARNAME? PARENT=%s CHILD=%s VARNAME=%s (%v)\n", PARENT, CHILD, VARNAME, pf.ToSort)
+		pmsg := pf.findMsg(PARENT)
 		if pmsg == nil {
+			// log.Printf("FAILED TO ADD SORT pmsg=nil PARENT=%s CHILD=%s VARNAME=%s\n", PARENT, CHILD, VARNAME)
 			return fmt.Errorf("failed to find struct %s", s.MsgName)
 		}
 
@@ -192,6 +194,8 @@ func (pb *Files) makeNewSortfile(pf *File) error {
 			FRUIT := CHILD
 			funcdef = pmsg.addIterAll(wSort, FRUITS, FRUIT)
 			log.Printf("Adding %s\n", funcdef)
+		} else {
+			// log.Printf("FAILED TO ADD SORT IterAll() PARENT != VARNAME PARENT=%s CHILD=%s VARNAME=%s\n", PARENT, CHILD, VARNAME)
 		}
 
 		/*