Restructure

重构

data/view/genstruct/common.go

@@ -0,0 +1,179 @@
+package genstruct
+
+import (
+	"fmt"
+	"sort"
+	"strings"
+
+	"github.com/xxjwxc/gormt/data/config"
+	"github.com/xxjwxc/gormt/data/view/cnf"
+	"github.com/xxjwxc/gormt/data/view/generate"
+)
+
+// SetName Setting element name.设置元素名字
+func (e *GenElement) SetName(name string) {
+	e.Name = name
+}
+
+// SetType Setting element type.设置元素类型
+func (e *GenElement) SetType(tp string) {
+	e.Type = tp
+}
+
+// SetNotes Setting element notes.设置注释
+func (e *GenElement) SetNotes(notes string) {
+	e.Notes = strings.Replace(notes, "\n", ",", -1)
+}
+
+// AddTag Add a tag .添加一个tag标记
+func (e *GenElement) AddTag(k string, v string) {
+	if e.Tags == nil {
+		e.Tags = make(map[string][]string)
+	}
+	e.Tags[k] = append(e.Tags[k], v)
+}
+
+// Generate Get the result data.获取结果数据
+func (e *GenElement) Generate() string {
+	tag := ""
+
+	if e.Tags != nil {
+		var ks []string
+		for k := range e.Tags {
+			ks = append(ks, k)
+		}
+		sort.Strings(ks)
+
+		var tags []string
+		for _, v := range ks {
+			tags = append(tags, fmt.Sprintf(`%v:"%v"`, v, strings.Join(e.Tags[v], ";")))
+		}
+		tag = fmt.Sprintf("`%v`", strings.Join(tags, " "))
+	}
+
+	var p generate.PrintAtom
+	if len(e.Notes) > 0 {
+		p.Add(e.Name, e.Type, tag, "// "+e.Notes)
+	} else {
+		p.Add(e.Name, e.Type, tag)
+	}
+
+	return p.Generates()[0]
+}
+
+//////////////////////////////////////////////////////////////////////////////
+// struct
+//////////////////////////////////////////////////////////////////////////////
+
+// SetCreatTableStr Set up SQL create statement, backup use setup create statement, backup use.设置创建语句，备份使用
+func (s *GenStruct) SetCreatTableStr(sql string) {
+	s.SQLBuildStr = sql
+}
+
+// SetStructName Setting the name of struct.设置struct名字
+func (s *GenStruct) SetStructName(name string) {
+	s.Name = name
+}
+
+// SetNotes set the notes.设置注释
+func (s *GenStruct) SetNotes(notes string) {
+	a := strings.Split(notes, "\n")
+	var text []string
+
+	for _, v := range a {
+		if len(v) > 0 {
+			text = append(text, "// "+v)
+		}
+	}
+	s.Notes = strings.Join(text, "\r\n")
+}
+
+// AddElement Add one or more elements.添加一个/或多个元素
+func (s *GenStruct) AddElement(e ...GenElement) {
+	s.Em = append(s.Em, e...)
+}
+
+// Generates Get the result data.获取结果数据
+func (s *GenStruct) Generates() []string {
+	var p generate.PrintAtom
+	if !config.GetSimple() {
+		p.Add("/******sql******")
+		p.Add(s.SQLBuildStr)
+		p.Add("******sql******/")
+	}
+	p.Add(s.Notes)
+	p.Add("type", s.Name, "struct {")
+	for _, v := range s.Em {
+		p.Add(v.Generate())
+	}
+	p.Add("}")
+
+	return p.Generates()
+}
+
+//////////////////////////////////////////////////////////////////////////////
+// package
+//////////////////////////////////////////////////////////////////////////////
+
+// SetPackage Defining package names.定义包名
+func (p *GenPackage) SetPackage(pname string) {
+	p.Name = pname
+}
+
+// AddImport Add import by type.通过类型添加import
+func (p *GenPackage) AddImport(imp string) {
+	if p.Imports == nil {
+		p.Imports = make(map[string]string)
+	}
+	p.Imports[imp] = imp
+}
+
+// AddStruct Add a structure.添加一个结构体
+func (p *GenPackage) AddStruct(st GenStruct) {
+	p.Structs = append(p.Structs, st)
+}
+
+// Generate Get the result data.获取结果数据
+func (p *GenPackage) Generate() string {
+	p.genimport() // auto add import .补充 import
+
+	var pa generate.PrintAtom
+	pa.Add("package", p.Name)
+	// add import
+	if p.Imports != nil {
+		pa.Add("import (")
+		for _, v := range p.Imports {
+			pa.Add(v)
+		}
+		pa.Add(")")
+	}
+	// -----------end
+	// add struct
+	for _, v := range p.Structs {
+		for _, v1 := range v.Generates() {
+			pa.Add(v1)
+		}
+	}
+	// -----------end
+
+	// output.输出
+	strOut := ""
+	for _, v := range pa.Generates() {
+		strOut += v + "\n"
+	}
+
+	return strOut
+}
+
+// compensate and import .获取结果数据
+func (p *GenPackage) genimport() {
+	for _, v := range p.Structs {
+		for _, v1 := range v.Em {
+			if v2, ok := cnf.EImportsHead[v1.Type]; ok {
+				if len(v2) > 0 {
+					p.AddImport(v2)
+				}
+			}
+		}
+	}
+}

data/view/genstruct/def.go

@@ -0,0 +1,24 @@
+package genstruct
+
+// GenElement element of sturct.元素类
+type GenElement struct {
+	Name  string              // Name.元素名
+	Type  string              // Type.类型标记
+	Notes string              // Notes.注释
+	Tags  map[string][]string // tages.标记
+}
+
+// GenStruct struct of IStruct .结构体
+type GenStruct struct {
+	SQLBuildStr string       // Create SQL statements.创建sql语句
+	Name        string       // name.名字
+	Notes       string       // notes.注释
+	Em          []GenElement // em.元素组合
+}
+
+// GenPackage package of IPackage.包体
+type GenPackage struct {
+	Name    string            // name.名字
+	Imports map[string]string // Inclusion term.元素组合
+	Structs []GenStruct       // struct list .结构体组合
+}

data/view/genstruct/def_ifs.go

@@ -0,0 +1,47 @@
+package genstruct
+
+import "github.com/xxjwxc/gormt/data/view/generate"
+
+// IPackage package of interface
+type IPackage interface {
+	generate.IGenerate
+	// Defining package names.定义包名
+	SetPackage(string)
+	// Add import by type.通过类型添加import
+	AddImport(string)
+	// Add a structure.添加一个结构体
+	AddStruct(IStruct)
+}
+
+// IStruct struct of interface
+type IStruct interface {
+	generate.IGenerate
+
+	// Set up SQL create statement, backup use.设置创建语句，备份使用
+	SetCreatTableStr(string)
+
+	// Setting Structural Name设置结构体名字
+	SetStructName(string)
+
+	// setting the notes.设置注释
+	SetNotes(string)
+
+	// add one element.添加一个元素
+	AddElement(...IElement)
+}
+
+// IElement element in stuct of interface.元素类
+type IElement interface {
+	generate.IGenerate
+
+	// setting name of element.设置元素名字
+	SetName(string)
+	// Setting element type.设置元素类型
+	SetType(string)
+
+	// setting notes of element .设置注释
+	SetNotes(string)
+
+	// add one tag.添加一个tag标记
+	AddTag(k string, v string)
+}