diff --git a/app/admin/service/tools/gen.go b/app/admin/service/tools/gen.go index ed5095bf321be9ba5dc1b7f4fe0a3739648f8052..f9223ca41581e9ec44057b86c52eee6a6799a18d 100644 --- a/app/admin/service/tools/gen.go +++ b/app/admin/service/tools/gen.go @@ -335,13 +335,15 @@ func SelectGenTableByName(tableName string) tools.GenTableVO { func SelectDbTableListByNames(tableName []string) []tools.GenTable { var tableVo []tools.GenTable - sql := "select table_name as Table_Name, table_comment as TableComment, create_time as CreateTime, update_time " + - " from information_schema.tables" + - " where table_name NOT LIKE 'qrtz_%' and table_name NOT LIKE 'gen_%' " + - " and table_schema = (select database())" + - " and table_name in( ? )" - err := mysql.MysqlDb().Raw(sql, tableName).Scan(&tableVo).Error - if err != nil { + query := mysql.GetDB(). + Table("information_schema.tables"). + Select("table_name as Table_Name, table_comment as TableComment, create_time as CreateTime, update_time"). + Where("table_name NOT LIKE ?", "qrtz_%"). + Where("table_name NOT LIKE ?", "gen_%"). + Where("table_schema = (SELECT DATABASE())"). + Where("table_name IN ?", tableName) + + if err := query.Scan(&tableVo).Error; err != nil { panic(R.ReturnFailMsg(err.Error())) } return tableVo