TRKJ/mayfly-go
2
0
Fork 0
mirror of https://gitee.com/dromara/mayfly-go synced 2025-11-02 23:40:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 由于批量导出数据库不支持 PostgreSQL，非 MySQL 数据库限制使用该功能。

Browse Source
This commit is contained in:
wanli
2023-09-05 14:15:40 +08:00
parent 086dbf278b
commit 0eaff33168
1 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
server/internal/db/api/db.go
View File

@@ -271,31 +271,37 @@ func (d *Db) DumpSql(rc *req.Ctx) {
if len(dbNames) == 1 && len(tablesStr) > 0 { if len(dbNames) == 1 && len(tablesStr) > 0 {
tables = strings.Split(tablesStr, ",") tables = strings.Split(tablesStr, ",")
} }
instance := d.InstanceApp.GetById(db.InstanceId)
writer := gzipResponseWriter{writer: gzip.NewWriter(g.Writer)} writer := gzipResponseWriter{writer: gzip.NewWriter(g.Writer)}
defer writer.Close() defer writer.Close()
for _, dbName := range dbNames { for _, dbName := range dbNames {
d.dumpDb(writer, db, dbName, tables, needStruct, needData) d.dumpHeader(writer, instance, dbName, len(dbNames) > 1)
d.dumpDb(writer, db, instance, dbName, tables, needStruct, needData)
} }
rc.ReqParam = fmt.Sprintf("DB[id=%d, tag=%s, name=%s, databases=%s, tables=%s, dumpType=%s]", db.Id, db.TagPath, db.Name, dbNamesStr, tablesStr, dumpType) rc.ReqParam = fmt.Sprintf("DB[id=%d, tag=%s, name=%s, databases=%s, tables=%s, dumpType=%s]", db.Id, db.TagPath, db.Name, dbNamesStr, tablesStr, dumpType)
} }
func (d *Db) dumpDb(writer gzipResponseWriter, db *entity.Db, dbName string, tables []string, needStruct bool, needData bool) { func (d *Db) dumpHeader(writer gzipResponseWriter, instance *entity.Instance, dbName string, switchDb bool) {
writer.WriteString("-- ----------------------------") writer.WriteString("-- ----------------------------")
writer.WriteString("\n-- 导出平台: mayfly-go") writer.WriteString("\n-- 导出平台: mayfly-go")
writer.WriteString(fmt.Sprintf("\n-- 导出时间: %s ", time.Now().Format("2006-01-02 15:04:05"))) writer.WriteString(fmt.Sprintf("\n-- 导出时间: %s ", time.Now().Format("2006-01-02 15:04:05")))
writer.WriteString(fmt.Sprintf("\n-- 导出数据库: %s ", dbName)) writer.WriteString(fmt.Sprintf("\n-- 导出数据库: %s ", dbName))
writer.WriteString("\n-- ----------------------------\n") writer.WriteString("\n-- ----------------------------\n")
instance := d.InstanceApp.GetById(db.InstanceId) if switchDb {
dbInst := d.DbApp.GetDbConnection(db, instance, dbName) switch instance.Type {
case entity.DbTypeMysql:
switch dbInst.Info.Type { writer.WriteString(fmt.Sprintf("use `%s`;\n", dbName))
case entity.DbTypeMysql: default:
writer.WriteString(fmt.Sprintf("use `%s`;\n", dbName)) biz.IsTrue(false, "数据库类型必须为 %s", entity.DbTypeMysql)
default: }
biz.IsTrue(false, "数据库类型必须为 %s", entity.DbTypeMysql)
} }
}
func (d *Db) dumpDb(writer gzipResponseWriter, db *entity.Db, instance *entity.Instance, dbName string, tables []string, needStruct bool, needData bool) {
dbInst := d.DbApp.GetDbConnection(db, instance, dbName)
dbMeta := dbInst.GetMeta() dbMeta := dbInst.GetMeta()
if len(tables) == 0 { if len(tables) == 0 {
ti := dbMeta.GetTableInfos() ti := dbMeta.GetTableInfos()