feat: parse struct field comment

go-gorm · Feb 13, 2025 · 00b2e49 · 00b2e49
1 parent 14f5dc7
commit 00b2e49
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/generate/query.go b/internal/generate/query.go
@@ -58,12 +58,18 @@ func (b *QueryStructMeta) parseStruct(st interface{}) error {
 		fp = fps
 	}
 	for _, f := range stmt.Schema.Fields {
-		b.appendOrUpdateField(&model.Field{
+		gf := &model.Field{
 			Name:          f.Name,
 			Type:          b.getFieldRealType(f.FieldType),
 			ColumnName:    f.DBName,
 			CustomGenType: fp.GetFieldGenType(f),
-		})
+			ColumnComment: f.Comment,
+		}
+		if gf.ColumnComment == "" {
+			gf.ColumnComment = f.TagSettings["COMMENT"]
+		}
+		gf.MultilineComment = strings.Contains(gf.ColumnComment, "\n")
+		b.appendOrUpdateField(gf)
 	}
 	for _, r := range ParseStructRelationShip(&stmt.Schema.Relationships) {
 		r := r