diff options
Diffstat (limited to 'parseProtoFile.go')
| -rw-r--r-- | parseProtoFile.go | 33 |
1 files changed, 24 insertions, 9 deletions
diff --git a/parseProtoFile.go b/parseProtoFile.go index b62e5e2..49f382c 100644 --- a/parseProtoFile.go +++ b/parseProtoFile.go @@ -38,18 +38,33 @@ func (pb *Files) findAutogenpb(f *File) error { // log.Info("line:", line) parts := strings.Fields(line) - if strings.Contains(line, "autogenpb:marshal") { - newm := parts[1] - if curmsg != nil { - // log.Info("found marshal", newm) - marshalKeys = append(marshalKeys, newm) + + if strings.Contains(line, "autogenpb:sort") { + if parts[0] == "repeated" { + newm := parts[1] + if curmsg == nil { + log.Info("Error: Found Sort for:", newm, "however, this struct can't be used") + // log.Info("found marshal", newm) + marshalKeys = append(marshalKeys, newm) + } else { + log.Info("Found Sort for:", newm, "in struct", curmsg.Name) + } + } else { + log.Info("Error:", line) + log.Info("Error: can not sort on non repeated fields") } } if strings.Contains(line, "autogenpb:unique") { - newu := parts[1] - newu = cases.Title(language.English, cases.NoLower).String(newu) - // log.Info("found unique field", newu) - uniqueKeys = append(uniqueKeys, newu) + if parts[0] == "repeated" { + // log.Info("Found Unique for:", parts) + newu := parts[1] + newu = cases.Title(language.English, cases.NoLower).String(newu) + log.Info("found unique field", newu, "in struct", curmsg.Name) + // uniqueKeys = append(uniqueKeys, newu) + } else { + log.Info("Error:", line) + log.Info("Error: can not append on non repeated fields") + } } } return nil |