tools_gen_table.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736
  1. /*
  2. * @desc:代码生成功能
  3. * @company:云南省奇讯科技有限公司
  4. * @Author: yixiaohu
  5. * @Date: 2021/7/22 15:30
  6. */
  7. package service
  8. import (
  9. "bufio"
  10. "context"
  11. "database/sql"
  12. "fmt"
  13. "gfast/app/common/global"
  14. comModel "gfast/app/common/model"
  15. comService "gfast/app/common/service"
  16. "gfast/app/system/dao"
  17. "gfast/app/system/model"
  18. "github.com/gogf/gf/database/gdb"
  19. "github.com/gogf/gf/encoding/gjson"
  20. "github.com/gogf/gf/errors/gerror"
  21. "github.com/gogf/gf/frame/g"
  22. "github.com/gogf/gf/os/gfile"
  23. "github.com/gogf/gf/os/gtime"
  24. "github.com/gogf/gf/os/gview"
  25. "github.com/gogf/gf/text/gregex"
  26. "github.com/gogf/gf/text/gstr"
  27. "github.com/gogf/gf/util/gconv"
  28. "io"
  29. "os"
  30. "strings"
  31. )
  32. type toolsGenTable struct{}
  33. var ToolsGenTable = new(toolsGenTable)
  34. // SelectListByPage 查询已导入的数据表
  35. func (s *toolsGenTable) SelectListByPage(param *dao.ToolsGenTableSearchReq) (total int, list []*model.ToolsGenTable, err error) {
  36. model := dao.ToolsGenTable.M
  37. if param != nil {
  38. if param.TableName != "" {
  39. model = model.Where(dao.ToolsGenTable.C.TableName+" like ?", "%"+param.TableName+"%")
  40. }
  41. if param.TableComment != "" {
  42. model = model.Where(dao.ToolsGenTable.C.TableComment+"like ?", "%"+param.TableComment+"%")
  43. }
  44. if param.BeginTime != "" {
  45. model = model.Where(dao.ToolsGenTable.C.CreateTime+" >= ", param.BeginTime)
  46. }
  47. if param.EndTime != "" {
  48. model = model.Where(dao.ToolsGenTable.C.CreateTime+" <= ", param.EndTime)
  49. }
  50. total, err = model.Count()
  51. if err != nil {
  52. g.Log().Error(err)
  53. err = gerror.New("获取总行数失败")
  54. return
  55. }
  56. if param.PageNum == 0 {
  57. param.PageNum = 1
  58. }
  59. if param.PageSize == 0 {
  60. param.PageSize = comModel.PageSize
  61. }
  62. err = model.Page(param.PageNum, param.PageSize).Order(dao.ToolsGenTable.C.TableId + " asc").Scan(&list)
  63. if err != nil {
  64. g.Log().Error(err)
  65. err = gerror.New("获取数据失败")
  66. }
  67. }
  68. return
  69. }
  70. // SelectDbTableList 查询据库表
  71. func (s *toolsGenTable) SelectDbTableList(param *dao.ToolsGenTableSearchReq) (total int, list []*model.ToolsGenTable, err error) {
  72. db := g.DB()
  73. if s.getDbDriver() != "mysql" {
  74. err = gerror.New("代码生成暂时只支持mysql数据库")
  75. return
  76. }
  77. sql := " from information_schema.tables where table_schema = (select database())" +
  78. " and table_name NOT LIKE 'qrtz_%' AND table_name NOT LIKE 'gen_%' and table_name NOT IN (select table_name from " + dao.ToolsGenTable.Table + ") "
  79. if param != nil {
  80. if param.TableName != "" {
  81. sql += gdb.FormatSqlWithArgs(" and lower(table_name) like lower(?)", []interface{}{"%" + param.TableName + "%"})
  82. }
  83. if param.TableComment != "" {
  84. sql += gdb.FormatSqlWithArgs(" and lower(table_comment) like lower(?)", []interface{}{"%" + param.TableComment + "%"})
  85. }
  86. if param.BeginTime != "" {
  87. sql += gdb.FormatSqlWithArgs(" and date_format(create_time,'%y%m%d') >= date_format(?,'%y%m%d') ", []interface{}{param.BeginTime})
  88. }
  89. if param.EndTime != "" {
  90. sql += gdb.FormatSqlWithArgs(" and date_format(create_time,'%y%m%d') <= date_format(?,'%y%m%d') ", []interface{}{param.EndTime})
  91. }
  92. }
  93. countSql := "select count(1) " + sql
  94. total, err = db.GetCount(countSql)
  95. if err != nil {
  96. g.Log().Error(err)
  97. err = gerror.New("读取总表数失败")
  98. return
  99. }
  100. sql = "table_name, table_comment, create_time, update_time " + sql
  101. if param.PageNum == 0 {
  102. param.PageNum = 1
  103. }
  104. if param.PageSize == 0 {
  105. param.PageSize = comModel.PageSize
  106. }
  107. page := (param.PageNum - 1) * param.PageSize
  108. sql += " order by create_time desc,table_name asc limit " + gconv.String(page) + "," + gconv.String(param.PageSize)
  109. err = db.GetScan(&list, "select "+sql)
  110. if err != nil {
  111. g.Log().Error(err)
  112. err = gerror.New("读取数据失败")
  113. }
  114. return
  115. }
  116. //获取数据库驱动类型
  117. func (s *toolsGenTable) getDbDriver() string {
  118. config := g.DB().GetConfig()
  119. return gstr.ToLower(config.Type)
  120. }
  121. // SelectDbTableListByNames 查询数据库中对应的表数据
  122. func (s *toolsGenTable) SelectDbTableListByNames(tableNames []string) ([]*model.ToolsGenTable, error) {
  123. if s.getDbDriver() != "mysql" {
  124. return nil, gerror.New("代码生成只支持mysql数据库")
  125. }
  126. db := g.DB()
  127. sql := "select * from information_schema.tables where table_name NOT LIKE 'qrtz_%' and table_name NOT LIKE 'gen_%' " +
  128. " and table_schema = (select database()) "
  129. if len(tableNames) > 0 {
  130. in := gstr.TrimRight(gstr.Repeat("?,", len(tableNames)), ",")
  131. sql += " and " + gdb.FormatSqlWithArgs("table_name in ("+in+")", gconv.SliceAny(tableNames))
  132. }
  133. var result []*model.ToolsGenTable
  134. err := db.GetScan(&result, sql)
  135. if err != nil {
  136. g.Log().Error(err)
  137. return nil, gerror.New("获取表格信息失败")
  138. }
  139. return result, err
  140. }
  141. // ImportGenTable 导入表结构
  142. func (s *toolsGenTable) ImportGenTable(tableList []*model.ToolsGenTable) error {
  143. if tableList != nil {
  144. tx, err := g.DB().Begin()
  145. if err != nil {
  146. return err
  147. }
  148. for _, table := range tableList {
  149. tableName := table.TableName
  150. s.InitTable(table)
  151. result, err := tx.Model(dao.ToolsGenTable.Table).Insert(table)
  152. if err != nil {
  153. return err
  154. }
  155. tmpId, err := result.LastInsertId()
  156. if err != nil || tmpId <= 0 {
  157. tx.Rollback()
  158. return gerror.New("保存数据失败")
  159. }
  160. table.TableId = tmpId
  161. // 保存列信息
  162. genTableColumns, err := ToolsGenTableColumn.SelectDbTableColumnsByName(tableName)
  163. if err != nil || len(genTableColumns) <= 0 {
  164. tx.Rollback()
  165. return gerror.New("获取列数据失败")
  166. }
  167. for _, column := range genTableColumns {
  168. ToolsGenTableColumn.InitColumnField(column, table)
  169. _, err = tx.Model(dao.ToolsGenTableColumn.Table).Insert(column)
  170. if err != nil {
  171. tx.Rollback()
  172. return gerror.New("保存列数据失败")
  173. }
  174. }
  175. }
  176. return tx.Commit()
  177. } else {
  178. return gerror.New("参数错误")
  179. }
  180. }
  181. // InitTable 初始化表信息
  182. func (s *toolsGenTable) InitTable(table *model.ToolsGenTable) {
  183. table.ClassName = s.ConvertClassName(table.TableName)
  184. table.PackageName = g.Cfg().GetString("gen.packageName")
  185. table.ModuleName = g.Cfg().GetString("gen.moduleName")
  186. table.BusinessName = s.GetBusinessName(table.TableName)
  187. table.FunctionName = strings.ReplaceAll(table.TableComment, "表", "")
  188. table.FunctionAuthor = g.Cfg().GetString("gen.author")
  189. table.TplCategory = "crud"
  190. table.CreateTime = gtime.Now()
  191. table.UpdateTime = table.CreateTime
  192. }
  193. // ConvertClassName 表名转换成类名
  194. func (s *toolsGenTable) ConvertClassName(tableName string) string {
  195. return gstr.CaseCamel(s.removeTablePrefix(tableName))
  196. }
  197. // GetBusinessName 获取业务名
  198. func (s *toolsGenTable) GetBusinessName(tableName string) string {
  199. return s.removeTablePrefix(tableName)
  200. }
  201. //删除表前缀
  202. func (s *toolsGenTable) removeTablePrefix(tableName string) string {
  203. autoRemovePre := g.Cfg().GetBool("gen.autoRemovePre")
  204. tablePrefix := g.Cfg().GetString("gen.tablePrefix")
  205. if autoRemovePre && tablePrefix != "" {
  206. searchList := strings.Split(tablePrefix, ",")
  207. for _, str := range searchList {
  208. if strings.HasPrefix(tableName, str) {
  209. tableName = strings.Replace(tableName, str, "", 1) //注意,只替换一次
  210. }
  211. }
  212. }
  213. return tableName
  214. }
  215. // Delete 删除表信息
  216. func (s *toolsGenTable) Delete(ids []int) error {
  217. tx, err := g.DB().Begin()
  218. if err != nil {
  219. g.Log().Error(err)
  220. return gerror.New("开启删除事务出错")
  221. }
  222. _, err = tx.Model(dao.ToolsGenTable.Table).Where(dao.ToolsGenTable.C.TableId+" in(?)", ids).Delete()
  223. if err != nil {
  224. g.Log().Error(err)
  225. tx.Rollback()
  226. return gerror.New("删除表格数据失败")
  227. }
  228. _, err = tx.Model(dao.ToolsGenTableColumn.Table).Where(dao.ToolsGenTableColumn.C.TableId+" in(?)", ids).Delete()
  229. if err != nil {
  230. g.Log().Error(err)
  231. tx.Rollback()
  232. return gerror.New("删除表格字段数据失败")
  233. }
  234. tx.Commit()
  235. return nil
  236. }
  237. // GetTableInfoByTableId 获取表格信息
  238. func (s *toolsGenTable) GetTableInfoByTableId(tableId int64) (info *model.ToolsGenTable, err error) {
  239. err = dao.ToolsGenTable.WherePri(tableId).Scan(&info)
  240. if err != nil {
  241. g.Log().Error(err)
  242. err = gerror.New("获取表格信息出错")
  243. }
  244. return
  245. }
  246. // SaveEdit 更新表及字段生成信息
  247. func (s *toolsGenTable) SaveEdit(req *dao.ToolsGenTableEditReq) (err error) {
  248. if req == nil {
  249. err = gerror.New("参数错误")
  250. return
  251. }
  252. var table *model.ToolsGenTable
  253. err = dao.ToolsGenTable.Where("table_id=?", req.TableId).Scan(&table)
  254. if err != nil || table == nil {
  255. err = gerror.New("数据不存在")
  256. return
  257. }
  258. if req.TableName != "" {
  259. table.TableName = req.TableName
  260. }
  261. if req.TableComment != "" {
  262. table.TableComment = req.TableComment
  263. }
  264. if req.BusinessName != "" {
  265. table.BusinessName = req.BusinessName
  266. }
  267. if req.ClassName != "" {
  268. table.ClassName = req.ClassName
  269. }
  270. if req.FunctionAuthor != "" {
  271. table.FunctionAuthor = req.FunctionAuthor
  272. }
  273. if req.FunctionName != "" {
  274. table.FunctionName = req.FunctionName
  275. }
  276. if req.ModuleName != "" {
  277. table.ModuleName = req.ModuleName
  278. }
  279. if req.PackageName != "" {
  280. table.PackageName = req.PackageName
  281. }
  282. if req.Remark != "" {
  283. table.Remark = req.Remark
  284. }
  285. if req.TplCategory != "" {
  286. table.TplCategory = req.TplCategory
  287. }
  288. if req.Params != "" {
  289. table.Options = req.Params
  290. }
  291. table.UpdateTime = gtime.Now()
  292. var options g.Map
  293. if req.TplCategory == "tree" {
  294. //树表设置options
  295. options = g.Map{
  296. "treeCode": req.TreeCode,
  297. "treeParentCode": req.TreeParentCode,
  298. "treeName": req.TreeName,
  299. }
  300. table.Options = gconv.String(options)
  301. } else {
  302. table.Options = ""
  303. }
  304. var tx *gdb.TX
  305. tx, err = g.DB().Begin()
  306. if err != nil {
  307. return
  308. }
  309. _, err = tx.Model(dao.ToolsGenTable.Table).Save(table)
  310. if err != nil {
  311. tx.Rollback()
  312. return err
  313. }
  314. //保存列数据
  315. if req.Columns != nil {
  316. for _, column := range req.Columns {
  317. if column.ColumnId > 0 {
  318. var dbColumn *model.ToolsGenTableColumn
  319. err = dao.ToolsGenTableColumn.Where("column_id=?", column.ColumnId).Scan(&dbColumn)
  320. if dbColumn != nil {
  321. dbColumn.ColumnComment = column.ColumnComment
  322. dbColumn.GoType = column.GoType
  323. dbColumn.HtmlType = column.HtmlType
  324. dbColumn.QueryType = column.QueryType
  325. dbColumn.GoField = column.GoField
  326. dbColumn.DictType = column.DictType
  327. dbColumn.IsInsert = column.IsInsert
  328. dbColumn.IsEdit = column.IsEdit
  329. dbColumn.IsList = column.IsList
  330. dbColumn.IsQuery = column.IsQuery
  331. dbColumn.IsRequired = column.IsRequired
  332. if tc, e := options["treeParentCode"]; options != nil && e && tc != "" && tc == dbColumn.HtmlField {
  333. dbColumn.IsQuery = "0"
  334. dbColumn.IsList = "0"
  335. dbColumn.HtmlType = "select"
  336. }
  337. //获取字段关联表信息
  338. if column.LinkLabelName != "" {
  339. dbColumn.LinkTableName = column.LinkTableName
  340. dbColumn.LinkLabelId = column.LinkLabelId
  341. dbColumn.LinkLabelName = column.LinkLabelName
  342. var linkTable *model.ToolsGenTable
  343. err = dao.ToolsGenTable.Where("table_name =?", column.LinkTableName).Scan(&linkTable)
  344. if err != nil {
  345. tx.Rollback()
  346. return
  347. }
  348. dbColumn.LinkTableClass = linkTable.ClassName
  349. dbColumn.LinkTablePackage = linkTable.PackageName
  350. } else {
  351. dbColumn.LinkTableName = ""
  352. dbColumn.LinkTableClass = ""
  353. dbColumn.LinkTablePackage = ""
  354. dbColumn.LinkLabelId = ""
  355. dbColumn.LinkLabelName = ""
  356. }
  357. _, err = tx.Model(dao.ToolsGenTableColumn.Table).Save(dbColumn)
  358. if err != nil {
  359. tx.Rollback()
  360. return
  361. }
  362. }
  363. }
  364. }
  365. }
  366. tx.Commit()
  367. return
  368. }
  369. func (s *toolsGenTable) SelectRecordById(tableId int64) (entityExtend *dao.ToolsGenTableExtend, err error) {
  370. var table *model.ToolsGenTable
  371. table, err = s.GetTableInfoByTableId(tableId)
  372. if err != nil {
  373. return
  374. }
  375. m := gconv.Map(table)
  376. gconv.Struct(m, &entityExtend)
  377. if entityExtend.TplCategory == "tree" {
  378. opt := gjson.New(entityExtend.Options)
  379. entityExtend.TreeParentCode = opt.GetString("treeParentCode")
  380. entityExtend.TreeCode = opt.GetString("treeCode")
  381. entityExtend.TreeName = opt.GetString("treeName")
  382. }
  383. //表字段数据
  384. var columns []*model.ToolsGenTableColumn
  385. columns, err = ToolsGenTableColumn.SelectGenTableColumnListByTableId(tableId)
  386. if err != nil {
  387. return
  388. }
  389. entityExtend.Columns = columns
  390. return
  391. }
  392. func (s *toolsGenTable) GenCode(ids []int, ctx context.Context) (err error) {
  393. //获取当前运行时目录
  394. curDir, err := os.Getwd()
  395. if err != nil {
  396. return gerror.New("获取本地路径失败")
  397. }
  398. frontDir := g.Cfg().GetString("gen.frontDir")
  399. for _, id := range ids {
  400. var genData g.MapStrStr
  401. var extendData *dao.ToolsGenTableExtend
  402. genData, extendData, err = s.GenData(gconv.Int64(id), ctx)
  403. packageName := gstr.SubStr(extendData.PackageName, gstr.Pos(extendData.PackageName, "/"))
  404. businessName := gstr.CaseCamelLower(extendData.BusinessName)
  405. for key, code := range genData {
  406. switch key {
  407. case "controller":
  408. path := strings.Join([]string{curDir, packageName, "/api/", extendData.TableName, ".go"}, "")
  409. err = s.createFile(path, code, false)
  410. case "dao":
  411. path := strings.Join([]string{curDir, packageName, "/dao/", extendData.TableName, ".go"}, "")
  412. err = s.createFile(path, code, false)
  413. case "dao_internal":
  414. path := strings.Join([]string{curDir, packageName, "/dao/internal/", extendData.TableName, ".go"}, "")
  415. err = s.createFile(path, code, true)
  416. case "model":
  417. path := strings.Join([]string{curDir, packageName, "/model/", extendData.TableName, ".go"}, "")
  418. err = s.createFile(path, code, true)
  419. case "router":
  420. path := strings.Join([]string{curDir, packageName, "/router/", extendData.TableName, ".go"}, "")
  421. err = s.createFile(path, code, false)
  422. case "service":
  423. path := strings.Join([]string{curDir, packageName, "/service/", extendData.TableName, ".go"}, "")
  424. err = s.createFile(path, code, false)
  425. case "sql":
  426. path := strings.Join([]string{curDir, "/data/gen_sql/", packageName, "/", extendData.TableName, ".sql"}, "")
  427. hasSql := gfile.Exists(path)
  428. err = s.createFile(path, code, false)
  429. if !hasSql {
  430. //第一次生成则向数据库写入菜单数据
  431. err = s.writeDb(path, ctx)
  432. if err != nil {
  433. return
  434. }
  435. //清除菜单缓存
  436. comService.Cache.New().Remove(global.SysAuthMenu)
  437. }
  438. case "vue":
  439. path := strings.Join([]string{frontDir, "/src/views/", extendData.ModuleName, "/", businessName, "/list/index.vue"}, "")
  440. if gstr.ContainsI(extendData.PackageName, "plugins") {
  441. path = strings.Join([]string{frontDir, "/src/views/plugins/", extendData.ModuleName, "/", businessName, "/list/index.vue"}, "")
  442. }
  443. err = s.createFile(path, code, false)
  444. case "jsApi":
  445. path := strings.Join([]string{frontDir, "/src/api/", extendData.ModuleName, "/", businessName, ".js"}, "")
  446. if gstr.ContainsI(extendData.PackageName, "plugins") {
  447. path = strings.Join([]string{frontDir, "/src/api/plugins/", extendData.ModuleName, "/", businessName, ".js"}, "")
  448. }
  449. err = s.createFile(path, code, false)
  450. }
  451. }
  452. //生成对应的模块路由
  453. err = s.genModuleRouter(curDir, extendData.ModuleName, extendData.PackageName)
  454. }
  455. return
  456. }
  457. // createFile 创建文件
  458. func (s *toolsGenTable) createFile(fileName, data string, cover bool) (err error) {
  459. if !gfile.Exists(fileName) || cover {
  460. var f *os.File
  461. f, err = gfile.Create(fileName)
  462. if err == nil {
  463. f.WriteString(data)
  464. }
  465. f.Close()
  466. }
  467. return
  468. }
  469. // GenData 获取生成数据
  470. func (s *toolsGenTable) GenData(tableId int64, ctx context.Context) (data g.MapStrStr, extendData *dao.ToolsGenTableExtend, err error) {
  471. extendData, err = ToolsGenTable.SelectRecordById(tableId)
  472. if err != nil {
  473. return
  474. }
  475. if extendData == nil {
  476. err = gerror.New("表格数据不存在")
  477. return
  478. }
  479. ToolsGenTableColumn.SetPkColumn(extendData, extendData.Columns)
  480. view := gview.New()
  481. view.SetConfigWithMap(g.Map{
  482. "Paths": g.Cfg().GetString("gen.templatePath"),
  483. "Delimiters": []string{"{{", "}}"},
  484. })
  485. view.BindFuncMap(g.Map{
  486. "UcFirst": func(str string) string {
  487. return gstr.UcFirst(str)
  488. },
  489. "Sum": func(a, b int) int {
  490. return a + b
  491. },
  492. "CaseCamelLower": gstr.CaseCamelLower, //首字母小写驼峰
  493. "CaseCamel": gstr.CaseCamel, //首字母大写驼峰
  494. "HasSuffix": gstr.HasSuffix, //是否存在后缀
  495. "ContainsI": gstr.ContainsI, //是否包含子字符串
  496. "VueTag": func(t string) string {
  497. return t
  498. },
  499. })
  500. //树形菜单选项
  501. tplData := g.Map{"table": extendData}
  502. daoKey := "dao"
  503. daoValue := ""
  504. var tmpDao string
  505. if tmpDao, err = view.Parse(ctx, "go/dao.template", tplData); err == nil {
  506. daoValue = tmpDao
  507. daoValue, err = s.trimBreak(daoValue)
  508. } else {
  509. return
  510. }
  511. daoInternalKey := "dao_internal"
  512. daoInternalValue := ""
  513. var tmpInternalDao string
  514. if tmpInternalDao, err = view.Parse(ctx, "go/dao_internal.template", tplData); err == nil {
  515. daoInternalValue = tmpInternalDao
  516. daoInternalValue, err = s.trimBreak(daoInternalValue)
  517. } else {
  518. return
  519. }
  520. modelKey := "model"
  521. modelValue := ""
  522. var tmpModel string
  523. if tmpModel, err = view.Parse(ctx, "go/model.template", tplData); err == nil {
  524. modelValue = tmpModel
  525. modelValue, err = s.trimBreak(modelValue)
  526. } else {
  527. return
  528. }
  529. controllerKey := "controller"
  530. controllerValue := ""
  531. var tmpController string
  532. if tmpController, err = view.Parse(ctx, "go/controller.template", tplData); err == nil {
  533. controllerValue = tmpController
  534. controllerValue, err = s.trimBreak(controllerValue)
  535. } else {
  536. return
  537. }
  538. serviceKey := "service"
  539. serviceValue := ""
  540. var tmpService string
  541. if tmpService, err = view.Parse(ctx, "go/service.template", tplData); err == nil {
  542. serviceValue = tmpService
  543. serviceValue, err = s.trimBreak(serviceValue)
  544. } else {
  545. return
  546. }
  547. routerKey := "router"
  548. routerValue := ""
  549. var tmpRouter string
  550. if tmpRouter, err = view.Parse(ctx, "go/router.template", tplData); err == nil {
  551. routerValue = tmpRouter
  552. routerValue, err = s.trimBreak(routerValue)
  553. } else {
  554. return
  555. }
  556. sqlKey := "sql"
  557. sqlValue := ""
  558. var tmpSql string
  559. if tmpSql, err = view.Parse(ctx, "sql/sql.template", tplData); err == nil {
  560. sqlValue = tmpSql
  561. sqlValue, err = s.trimBreak(sqlValue)
  562. } else {
  563. return
  564. }
  565. jsApiKey := "jsApi"
  566. jsApiValue := ""
  567. var tmpJsApi string
  568. if tmpJsApi, err = view.Parse(ctx, "js/api.template", tplData); err == nil {
  569. jsApiValue = tmpJsApi
  570. jsApiValue, err = s.trimBreak(jsApiValue)
  571. } else {
  572. return
  573. }
  574. vueKey := "vue"
  575. vueValue := ""
  576. var tmpVue string
  577. tmpFile := "vue/list-vue.template"
  578. if extendData.TplCategory == "tree" {
  579. //树表
  580. tmpFile = "vue/tree-vue.template"
  581. }
  582. if tmpVue, err = view.Parse(ctx, tmpFile, tplData); err == nil {
  583. vueValue = tmpVue
  584. vueValue, err = s.trimBreak(vueValue)
  585. } else {
  586. return
  587. }
  588. data = g.MapStrStr{
  589. daoKey: daoValue,
  590. daoInternalKey: daoInternalValue,
  591. modelKey: modelValue,
  592. controllerKey: controllerValue,
  593. serviceKey: serviceValue,
  594. routerKey: routerValue,
  595. sqlKey: sqlValue,
  596. jsApiKey: jsApiValue,
  597. vueKey: vueValue,
  598. }
  599. return
  600. }
  601. //剔除多余的换行
  602. func (s *toolsGenTable) trimBreak(str string) (rStr string, err error) {
  603. var b []byte
  604. if b, err = gregex.Replace("(([\\s\t]*)\r?\n){2,}", []byte("$2\n"), []byte(str)); err != nil {
  605. return
  606. }
  607. if b, err = gregex.Replace("(([\\s\t]*)/{4}\r?\n)", []byte("$2\n\n"), b); err == nil {
  608. rStr = gconv.String(b)
  609. }
  610. return
  611. }
  612. // 写入菜单数据
  613. func (s *toolsGenTable) writeDb(path string, ctx context.Context) (err error) {
  614. isAnnotation := false
  615. var fi *os.File
  616. fi, err = os.Open(path)
  617. if err != nil {
  618. return
  619. }
  620. defer fi.Close()
  621. br := bufio.NewReader(fi)
  622. var sqlStr []string
  623. now := gtime.Now()
  624. var res sql.Result
  625. var id int64
  626. var tx *gdb.TX
  627. tx, err = g.DB().Ctx(ctx).Begin()
  628. if err != nil {
  629. return
  630. }
  631. for {
  632. bytes, e := br.ReadBytes('\n')
  633. if e == io.EOF {
  634. break
  635. }
  636. str := gstr.Trim(string(bytes))
  637. if str == "" {
  638. continue
  639. }
  640. if strings.Contains(str, "/*") {
  641. isAnnotation = true
  642. }
  643. if isAnnotation {
  644. if strings.Contains(str, "*/") {
  645. isAnnotation = false
  646. }
  647. continue
  648. }
  649. if str == "" || strings.HasPrefix(str, "--") || strings.HasPrefix(str, "#") {
  650. continue
  651. }
  652. if strings.HasSuffix(str, ";") {
  653. if gstr.ContainsI(str, "select") {
  654. if gstr.ContainsI(str, "@now") {
  655. continue
  656. }
  657. if gstr.ContainsI(str, "@parentId") {
  658. id, err = res.LastInsertId()
  659. }
  660. }
  661. sqlStr = append(sqlStr, str)
  662. sql := strings.Join(sqlStr, "")
  663. gstr.ReplaceByArray(sql, []string{"@parentId", gconv.String(id), "@now", now.Format("Y-m-d H:i:s")})
  664. //插入业务
  665. res, err = tx.Exec(sql)
  666. if err != nil {
  667. tx.Rollback()
  668. return
  669. }
  670. sqlStr = nil
  671. } else {
  672. sqlStr = []string{str}
  673. }
  674. }
  675. tx.Commit()
  676. return
  677. }
  678. // GenModuleRouter 生成模块路由
  679. func (s *toolsGenTable) genModuleRouter(curDir, moduleName, packageName string) (err error) {
  680. if gstr.CaseSnake(moduleName) != "system" {
  681. routerFilePath := strings.Join([]string{curDir, "/router/", gstr.CaseSnake(moduleName), ".go"}, "")
  682. if gstr.ContainsI(packageName, "plugins") {
  683. routerFilePath = strings.Join([]string{curDir, "/plugins/router/", gstr.CaseSnake(moduleName), ".go"}, "")
  684. }
  685. code := fmt.Sprintf(`package router%simport _ "%s/router"`, "\n", packageName)
  686. err = s.createFile(routerFilePath, code, false)
  687. }
  688. return
  689. }