add column support

支持列名导出

data/view/genstruct/common.go

@@ -133,6 +133,7 @@ func (s *GenStruct) AddElement(e ...GenElement) {
 	s.Em = append(s.Em, e...)
 }
 
+// GenerateTableName generate table name .生成表名
 func (s *GenStruct) GenerateTableName() []string {
 	tmpl, err := template.New("gen_tnf").Parse(genfunc.GetGenTableNameTemp())
 	if err != nil {
@@ -148,6 +149,34 @@ func (s *GenStruct) GenerateTableName() []string {
 	return []string{buf.String()}
 }
 
+// GenerateColumnName generate column name . 生成列名
+func (s *GenStruct) GenerateColumnName() []string {
+	tmpl, err := template.New("gen_tnc").Parse(genfunc.GetGenColumnNameTemp())
+	if err != nil {
+		panic(err)
+	}
+	var data struct {
+		StructName string
+		Em         []struct {
+			ColumnName string
+			StructName string
+		}
+	}
+	data.StructName = s.Name
+	for _, v := range s.Em {
+		data.Em = append(data.Em, struct {
+			ColumnName string
+			StructName string
+		}{ColumnName: v.ColumnName,
+			StructName: v.Name,
+		})
+	}
+
+	var buf bytes.Buffer
+	tmpl.Execute(&buf, data)
+	return []string{buf.String()}
+}
+
 // Generates Get the result data.获取结果数据
 func (s *GenStruct) Generates() []string {
 	var p generate.PrintAtom
@@ -240,6 +269,10 @@ func (p *GenPackage) Generate() string {
 			for _, v1 := range v.GenerateTableName() {
 				pa.Add(v1)
 			}
+
+			for _, v2 := range v.GenerateColumnName() { // add column list
+				pa.Add(v2)
+			}
 		}
 	}
 	// -----------end