adjust sql comment ;

x-mod · May 10, 2021 · e9be724 · e9be724
1 parent 5f2bb4a
commit e9be724
Show file tree

Hide file tree

Showing 6 changed files with 55 additions and 6 deletions.
diff --git a/gen/generate.go b/gen/generate.go
@@ -124,6 +124,7 @@ func generateCode(in string, out string, suffix string, objects ...string) error
 			log.Println("gofmt: ", err)
 		}
 	}
+
 	for _, tb := range tables {
 		outfile := filepath.Join(out, fmt.Sprintf("gen.table.%s.go", tb.Name()))
 		wr, err := os.OpenFile(outfile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)

diff --git a/object/object.go b/object/object.go
@@ -37,6 +37,12 @@ type Object struct {
 	Attributes map[string]interface{} `yaml:"attributes,omitempty"`
 }
 
+func (obj *Object) Adjust() {
+	obj.Comment = strings.ReplaceAll(obj.Comment, ";", "")
+	for _, field := range obj.Fields {
+		field.Comment = strings.ReplaceAll(field.Comment, ";", "")
+	}
+}
 func (obj *Object) IsTable() bool {
 	if strings.ToLower(obj.Object) == "table" {
 		return true

diff --git a/parser/parse.go b/parser/parse.go
@@ -19,6 +19,7 @@ func Parse(filename string) ([]*object.Object, error) {
 		if err != nil {
 			return nil, err
 		}
+		obj.Adjust()
 		objs = append(objs, &obj)
 	}
 	return objs, nil

diff --git a/tpl/bindata.go b/tpl/bindata.go