service/vendor/codecreate/generator/generator.go

121 lines
3.1 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package generator
import (
"bufio"
"fmt"
"io"
"os"
"service/bizcommon/util"
"service/codecreate/consts"
"strings"
"github.com/xuri/excelize/v2"
)
func CreateFileDirectory(genSource *GenSource) {
//在/service/api/proto下准备保存proto的文件夹
//判断是否存在,不存在就创建
path := fmt.Sprintf("%v%v%v/proto", consts.RootPath, consts.ProtoOutPath, genSource.ModuleName)
_, err := os.Stat(path)
if os.IsNotExist(err) {
// mkdir 创建目录mkdirAll 可创建多层级目录
err = os.MkdirAll(path, os.ModePerm)
if err != nil {
fmt.Printf("Create directory fail : %v", err)
}
} else if err != nil {
fmt.Printf("Stat fail :%v", err)
}
}
func GenerateEntity(genSource *GenSource) {
infile, err := os.Open(genSource.InPath)
if err != nil {
fmt.Printf("Open File Err : %v", err)
}
outfile, err := os.OpenFile(genSource.OutPath, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
fmt.Printf("Open File Err : %v", err)
}
entityDefineFile, err := excelize.OpenFile(consts.RootPath + consts.ExcelPath)
if err != nil {
fmt.Printf("Open File Err : %v", err)
}
rows, err := entityDefineFile.GetRows(genSource.EntityName)
if err != nil {
fmt.Printf("Get rows Err : %v", err)
}
defer infile.Close()
defer outfile.Close()
defer entityDefineFile.Close()
reader := bufio.NewReader(infile)
writer := bufio.NewWriter(outfile)
entityDef := CreateEntityDefination(rows)
for {
str, err := reader.ReadString('\n')
str = strings.ReplaceAll(str, consts.EntityNamePlaceHolder, genSource.EntityName)
str = strings.ReplaceAll(str, consts.EntityCNNamePlaceHolder, genSource.EntityCNName)
str = strings.ReplaceAll(str, consts.EntityDefinationPlaceHolder, entityDef)
writer.WriteString(str)
if err == io.EOF {
break
}
}
writer.Flush()
}
func GenerateModule(genSource *GenSource) {
infile, err := os.Open(genSource.InPath)
if err != nil {
fmt.Printf("Open File Err : %v", err)
}
outfile, err := os.OpenFile(genSource.OutPath, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
fmt.Printf("Open File Err : %v", err)
}
defer infile.Close()
defer outfile.Close()
reader := bufio.NewReader(infile)
writer := bufio.NewWriter(outfile)
for {
str, err := reader.ReadString('\n')
str = strings.ReplaceAll(str, consts.EntityNamePlaceHolder, genSource.EntityName)
str = strings.ReplaceAll(str, consts.ModuleNamePlaceHolder, genSource.ModuleName)
str = strings.ReplaceAll(str, consts.EntityCNNamePlaceHolder, genSource.EntityCNName)
str = strings.ReplaceAll(str, consts.ErrCodeSeqPlaceHolder, genSource.ErrCodeSeq)
writer.WriteString(str)
if err == io.EOF {
break
}
}
writer.Flush()
}
func CreateEntityDefination(rows [][]string) string {
var sb strings.Builder
//第四列为ID
row := rows[3]
sb.WriteString(fmt.Sprintf(" Id *%v `json:\"id\" bson:\"_id\"` // %v\n", row[2], row[1]))
//第五列开始为表其他字段
for i := 4; i < len(rows); i++ {
row = rows[i]
sb.WriteString(fmt.Sprintf(" %v *%v `json:\"%v\" bson:\"%v\"` // %v\n", util.UderscoreToUpperCamelCase(row[0]), row[2], row[0], row[0], row[1]))
}
return sb.String()
}