diff --git a/util/omitempty/omitempty.go b/util/omitempty/omitempty.go index 325d526..801c3d5 100644 --- a/util/omitempty/omitempty.go +++ b/util/omitempty/omitempty.go @@ -12,10 +12,19 @@ import ( "strings" ) -const filePath = "./pkg/database/models.go" +var filePaths = []string{ + "./pkg/database/models.go", + "./pkg/database/calls.sql.go", +} func main() { // Parse the source code + for _, v := range filePaths { + process(v) + } +} + +func process(filePath string) { fset := token.NewFileSet() f, err := parser.ParseFile(fset, filePath, nil, parser.ParseComments) if err != nil {