codegen: add ability to insert comments and blank lines into the generated code
This commit is contained in:
@@ -4,7 +4,6 @@ import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"go/ast"
|
||||
"go/printer"
|
||||
"go/token"
|
||||
"os"
|
||||
|
||||
@@ -37,11 +36,9 @@ var generate_model = &cobra.Command{
|
||||
return ErrNoSuchTable
|
||||
}
|
||||
|
||||
fset := token.NewFileSet()
|
||||
|
||||
if Must(cmd.Flags().GetBool("test")) {
|
||||
file2 := modelgenerate.GenerateModelTestAST(table, modname)
|
||||
PanicIf(printer.Fprint(os.Stdout, fset, file2))
|
||||
PanicIf(modelgenerate.FprintWithComments(os.Stdout, file2))
|
||||
} else {
|
||||
decls := []ast.Decl{
|
||||
&ast.GenDecl{
|
||||
@@ -81,7 +78,7 @@ var generate_model = &cobra.Command{
|
||||
Decls: decls,
|
||||
}
|
||||
|
||||
PanicIf(printer.Fprint(os.Stdout, fset, file))
|
||||
PanicIf(modelgenerate.FprintWithComments(os.Stdout, file))
|
||||
}
|
||||
|
||||
return nil
|
||||
|
||||
Reference in New Issue
Block a user