feat: v1.7.2

* feat: 删除数据库备份历史
* refactor dbScheduler
* feat: 从数据库备份历史中恢复数据库
* feat: 删除数据库恢复历史记录
* refactor dbScheuler

Dockerfile

@@ -5,7 +5,7 @@ WORKDIR /mayfly
 
 COPY mayfly_go_web .
 
-RUN yarn config set registry 'https://registry.npm.taobao.org' && \
+RUN yarn config set registry 'https://registry.npmmirror.com' && \
     yarn install && \
     yarn build
 

mayfly_go_web/package.json

@@ -17,7 +17,7 @@
     "countup.js": "^2.7.0",
     "cropperjs": "^1.5.11",
     "echarts": "^5.4.3",
-    "element-plus": "^2.5.2",
+    "element-plus": "^2.5.3",
     "js-base64": "^3.7.5",
     "jsencrypt": "^3.3.2",
     "lodash": "^4.17.21",
@@ -49,13 +49,14 @@
     "@typescript-eslint/parser": "^6.7.4",
     "@vitejs/plugin-vue": "^5.0.3",
     "@vue/compiler-sfc": "^3.4.14",
+    "code-inspector-plugin": "^0.4.5",
     "dotenv": "^16.3.1",
     "eslint": "^8.35.0",
     "eslint-plugin-vue": "^9.19.2",
     "prettier": "^3.1.0",
     "sass": "^1.69.0",
     "typescript": "^5.3.2",
-    "vite": "^5.0.11",
+    "vite": "^5.0.12",
     "vue-eslint-parser": "^9.4.0"
   },
   "browserslist": [

mayfly_go_web/src/assets/iconfont/iconfont.json

@@ -74,6 +74,20 @@
       "font_class": "sqlite",
       "unicode": "e546",
       "unicode_decimal": 58694
+    },
+    {
+      "icon_id": "29340317",
+      "name": "temp-mssql",
+      "font_class": "MSSQLNATIVE",
+      "unicode": "e600",
+      "unicode_decimal": 58880
+    },
+    {
+      "icon_id": "7699332",
+      "name": "gaussdb",
+      "font_class": "gauss",
+      "unicode": "e683",
+      "unicode_decimal": 59011
     }
   ]
 }

mayfly_go_web/src/common/config.ts

@@ -15,7 +15,7 @@ const config = {
     baseWsUrl: `${(window as any).globalConfig.BaseWsUrl || `${location.protocol == 'https:' ? 'wss:' : 'ws:'}//${getBaseApiUrl()}`}/api`,
 
     // 系统版本
-    version: 'v1.7.0',
+    version: 'v1.7.2',
 };
 
 export default config;

mayfly_go_web/src/components/crontab/CrontabInput.vue

@@ -1,5 +1,5 @@
 <template>
-    <el-input v-model="cron" placeholder="可点击左边按钮进行可视化配置">
+    <el-input v-model="cron" placeholder="可点击左边按钮配置">
         <template #prepend>
             <el-button @click="showCron = true" icon="Pointer"></el-button>
         </template>

mayfly_go_web/src/layout/navBars/breadcrumb/search.vue

@@ -1,17 +1,22 @@
 <template>
     <div class="layout-search-dialog">
         <el-dialog v-model="state.isShowSearch" width="300px" destroy-on-close :modal="false" fullscreen :show-close="false">
-            <el-autocomplete v-model="state.menuQuery" :fetch-suggestions="menuSearch" placeholder="菜单搜索"
-                prefix-icon="el-icon-search" ref="layoutMenuAutocompleteRef" @select="onHandleSelect" @blur="onSearchBlur">
+            <el-autocomplete
+                v-model="state.menuQuery"
+                :fetch-suggestions="menuSearch"
+                placeholder="菜单搜索"
+                prefix-icon="el-icon-search"
+                ref="layoutMenuAutocompleteRef"
+                @select="onHandleSelect"
+                @blur="onSearchBlur"
+            >
                 <template #prefix>
                     <el-icon class="el-input__icon">
                         <search />
                     </el-icon>
                 </template>
                 <template #default="{ item }">
-                    <div>
-                        <SvgIcon :name="item.meta.icon" class="mr5" />{{ item.meta.title }}
-                    </div>
+                    <div><SvgIcon :name="item.meta.icon" class="mr5" />{{ item.meta.title }}</div>
                 </template>
             </el-autocomplete>
         </el-dialog>
@@ -23,7 +28,7 @@ import { reactive, ref, nextTick } from 'vue';
 import { useRouter } from 'vue-router';
 import { useRoutesList } from '@/store/routesList';
 
-const layoutMenuAutocompleteRef: any = ref(null);;
+const layoutMenuAutocompleteRef: any = ref(null);
 const router = useRouter();
 const state: any = reactive({
     isShowSearch: false,
@@ -54,8 +59,7 @@ const menuSearch = (queryString: any, cb: any) => {
 const createFilter = (queryString: any) => {
     return (restaurant: any) => {
         return (
-            restaurant.path.toLowerCase().indexOf(queryString.toLowerCase()) > -1 ||
-            restaurant.meta.title.toLowerCase().indexOf(queryString.toLowerCase()) > -1
+            restaurant.path.toLowerCase().indexOf(queryString.toLowerCase()) > -1 || restaurant.meta.title.toLowerCase().indexOf(queryString.toLowerCase()) > -1
         );
     };
 };
@@ -97,7 +101,7 @@ const onSearchBlur = () => {
     closeSearch();
 };
 
-defineExpose({openSearch})
+defineExpose({ openSearch });
 </script>
 
 <style scoped lang="scss">

mayfly_go_web/src/views/home/Home.vue

@@ -73,12 +73,28 @@ const currentTime = computed(() => {
 
 // 初始化数字滚动
 const initNumCountUp = async () => {
-    const res: any = await indexApi.getIndexCount.request();
-    nextTick(() => {
-        new CountUp('mongoNum', res.mongoNum).start();
-        new CountUp('machineNum', res.machineNum).start();
-        new CountUp('dbNum', res.dbNum).start();
-        new CountUp('redisNum', res.redisNum).start();
+    indexApi.machineDashbord.request().then((res: any) => {
+        nextTick(() => {
+            new CountUp('machineNum', res.machineNum).start();
+        });
+    });
+
+    indexApi.dbDashbord.request().then((res: any) => {
+        nextTick(() => {
+            new CountUp('dbNum', res.dbNum).start();
+        });
+    });
+
+    indexApi.redisDashbord.request().then((res: any) => {
+        nextTick(() => {
+            new CountUp('redisNum', res.redisNum).start();
+        });
+    });
+
+    indexApi.mongoDashbord.request().then((res: any) => {
+        nextTick(() => {
+            new CountUp('mongoNum', res.mongoNum).start();
+        });
     });
 };
 

mayfly_go_web/src/views/home/api.ts

@@ -1,6 +1,8 @@
 import Api from '@/common/Api';
 
 export const indexApi = {
-    getIndexCount: Api.newGet("/common/index/count"),
-}
-
+    machineDashbord: Api.newGet('/machines/dashbord'),
+    dbDashbord: Api.newGet('/dbs/dashbord'),
+    redisDashbord: Api.newGet('/redis/dashbord'),
+    mongoDashbord: Api.newGet('/mongos/dashbord'),
+};

mayfly_go_web/src/views/ops/db/DbBackupEdit.vue

@@ -23,7 +23,7 @@
                 </el-form-item>
 
                 <el-form-item prop="name" label="任务名称">
-                    <el-input v-model.number="state.form.name" type="text" placeholder="任务名称"></el-input>
+                    <el-input v-model="state.form.name" type="text" placeholder="任务名称"></el-input>
                 </el-form-item>
                 <el-form-item prop="startTime" label="开始时间">
                     <el-date-picker v-model="state.form.startTime" type="datetime" placeholder="开始时间" />
@@ -101,7 +101,7 @@ const state = reactive({
         id: 0,
         dbId: 0,
         dbNames: '',
-        name: null as any,
+        name: '',
         intervalDay: null,
         startTime: null as any,
         repeated: null as any,

mayfly_go_web/src/views/ops/db/DbBackupHistoryList.vue

@@ -0,0 +1,155 @@
+<template>
+    <div class="db-backup-history">
+        <page-table
+            height="100%"
+            ref="pageTableRef"
+            :page-api="dbApi.getDbBackupHistories"
+            :show-selection="true"
+            v-model:selection-data="state.selectedData"
+            :searchItems="searchItems"
+            :before-query-fn="beforeQueryFn"
+            v-model:query-form="query"
+            :columns="columns"
+        >
+            <template #dbSelect>
+                <el-select v-model="query.dbName" placeholder="请选择数据库" style="width: 200px" filterable clearable>
+                    <el-option v-for="item in props.dbNames" :key="item" :label="`${item}`" :value="item"> </el-option>
+                </el-select>
+            </template>
+
+            <template #tableHeader>
+                <el-button type="primary" icon="back" @click="restoreDbBackupHistory(null)">立即恢复</el-button>
+                <el-button type="danger" icon="delete" @click="deleteDbBackupHistory(null)">删除</el-button>
+            </template>
+
+            <template #action="{ data }">
+                <div>
+                    <el-button @click="restoreDbBackupHistory(data)" type="primary" link>立即恢复</el-button>
+                    <el-button @click="deleteDbBackupHistory(data)" type="danger" link>删除</el-button>
+                </div>
+            </template>
+        </page-table>
+    </div>
+</template>
+
+<script lang="ts" setup>
+import { toRefs, reactive, Ref, ref } from 'vue';
+import { dbApi } from './api';
+import PageTable from '@/components/pagetable/PageTable.vue';
+import { TableColumn } from '@/components/pagetable';
+import { SearchItem } from '@/components/SearchForm';
+import { ElMessage, ElMessageBox } from 'element-plus';
+
+const pageTableRef: Ref<any> = ref(null);
+
+const props = defineProps({
+    dbId: {
+        type: [Number],
+        required: true,
+    },
+    dbNames: {
+        type: [Array<String>],
+        required: true,
+    },
+});
+
+const searchItems = [SearchItem.slot('dbName', '数据库名称', 'dbSelect')];
+
+const columns = [
+    TableColumn.new('dbName', '数据库名称'),
+    TableColumn.new('name', '备份名称'),
+    TableColumn.new('createTime', '创建时间').isTime(),
+    TableColumn.new('lastResult', '恢复结果'),
+    TableColumn.new('lastTime', '恢复时间').isTime(),
+    TableColumn.new('action', '操作').isSlot().setMinWidth(160).fixedRight(),
+];
+
+const emptyQuery = {
+    dbId: 0,
+    dbName: '',
+    pageNum: 1,
+    pageSize: 10,
+};
+
+const state = reactive({
+    data: [],
+    total: 0,
+    query: emptyQuery,
+    /**
+     * 选中的数据
+     */
+    selectedData: [],
+});
+
+const { query } = toRefs(state);
+
+const beforeQueryFn = (query: any) => {
+    query.dbId = props.dbId;
+    return query;
+};
+
+const search = async () => {
+    await pageTableRef.value.search();
+};
+
+const deleteDbBackupHistory = async (data: any) => {
+    let backupHistoryId: string;
+    if (data) {
+        backupHistoryId = data.id;
+    } else if (state.selectedData.length > 0) {
+        backupHistoryId = state.selectedData.map((x: any) => x.id).join(' ');
+    } else {
+        ElMessage.error('请选择需要删除的数据库备份历史');
+        return;
+    }
+    await ElMessageBox.confirm(`确定删除 “数据库备份历史” 吗？`, '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning',
+    });
+    await dbApi.deleteDbBackupHistory.request({ dbId: props.dbId, backupHistoryId: backupHistoryId });
+    await search();
+    ElMessage.success('删除成功');
+};
+
+const restoreDbBackupHistory = async (data: any) => {
+    let backupHistoryId: string;
+    if (data) {
+        backupHistoryId = data.id;
+    } else if (state.selectedData.length > 0) {
+        const pluralDbNames: string[] = [];
+        const dbNames: Map<string, boolean> = new Map();
+        state.selectedData.forEach((item: any) => {
+            if (!dbNames.has(item.dbName)) {
+                dbNames.set(item.dbName, false);
+                return;
+            }
+            if (!dbNames.get(item.dbName)) {
+                dbNames.set(item.dbName, true);
+                pluralDbNames.push(item.dbName);
+            }
+        });
+        if (pluralDbNames.length > 0) {
+            ElMessage.error('多次选择相同数据库：' + pluralDbNames.join(', '));
+            return;
+        }
+        backupHistoryId = state.selectedData.map((x: any) => x.id).join(' ');
+    } else {
+        ElMessage.error('请选择需要恢复的数据库备份历史');
+        return;
+    }
+    await ElMessageBox.confirm(`确定从 “数据库备份历史” 中恢复数据库吗？`, '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning',
+    });
+
+    await dbApi.restoreDbBackupHistory.request({
+        dbId: props.dbId,
+        backupHistoryId: backupHistoryId,
+    });
+    await search();
+    ElMessage.success('成功创建数据库恢复任务');
+};
+</script>
+<style lang="scss"></style>

mayfly_go_web/src/views/ops/db/DbBackupList.vue

@@ -21,6 +21,7 @@
                 <el-button type="primary" icon="plus" @click="createDbBackup()">添加</el-button>
                 <el-button type="primary" icon="video-play" @click="enableDbBackup(null)">启用</el-button>
                 <el-button type="primary" icon="video-pause" @click="disableDbBackup(null)">禁用</el-button>
+                <el-button type="danger" icon="delete" @click="deleteDbBackup(null)">删除</el-button>
             </template>
 
             <template #action="{ data }">
@@ -29,6 +30,7 @@
                     <el-button v-if="!data.enabled" @click="enableDbBackup(data)" type="primary" link>启用</el-button>
                     <el-button v-if="data.enabled" @click="disableDbBackup(data)" type="primary" link>禁用</el-button>
                     <el-button v-if="data.enabled" @click="startDbBackup(data)" type="primary" link>立即备份</el-button>
+                    <el-button @click="deleteDbBackup(data)" type="danger" link>删除</el-button>
                 </div>
             </template>
         </page-table>
@@ -49,7 +51,7 @@ import { dbApi } from './api';
 import PageTable from '@/components/pagetable/PageTable.vue';
 import { TableColumn } from '@/components/pagetable';
 import { SearchItem } from '@/components/SearchForm';
-import { ElMessage } from 'element-plus';
+import { ElMessage, ElMessageBox } from 'element-plus';
 
 const DbBackupEdit = defineAsyncComponent(() => import('./DbBackupEdit.vue'));
 const pageTableRef: Ref<any> = ref(null);
@@ -72,10 +74,10 @@ const columns = [
     TableColumn.new('name', '任务名称'),
     TableColumn.new('startTime', '启动时间').isTime(),
     TableColumn.new('intervalDay', '备份周期'),
-    TableColumn.new('enabled', '是否启用'),
+    TableColumn.new('enabledDesc', '是否启用'),
     TableColumn.new('lastResult', '执行结果'),
     TableColumn.new('lastTime', '执行时间').isTime(),
-    TableColumn.new('action', '操作').isSlot().setMinWidth(180).fixedRight(),
+    TableColumn.new('action', '操作').isSlot().setMinWidth(220).fixedRight(),
 ];
 
 const emptyQuery = {
@@ -168,5 +170,25 @@ const startDbBackup = async (data: any) => {
     await search();
     ElMessage.success('备份任务启动成功');
 };
+
+const deleteDbBackup = async (data: any) => {
+    let backupId: string;
+    if (data) {
+        backupId = data.id;
+    } else if (state.selectedData.length > 0) {
+        backupId = state.selectedData.map((x: any) => x.id).join(' ');
+    } else {
+        ElMessage.error('请选择需要删除的数据库备份任务');
+        return;
+    }
+    await ElMessageBox.confirm(`确定删除 “数据库备份任务” 吗？`, '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning',
+    });
+    await dbApi.deleteDbBackup.request({ dbId: props.dbId, backupId: backupId });
+    await search();
+    ElMessage.success('删除成功');
+};
 </script>
 <style lang="scss"></style>

mayfly_go_web/src/views/ops/db/DbList.vue

@@ -62,8 +62,21 @@
                         <el-dropdown-menu>
                             <el-dropdown-item :command="{ type: 'detail', data }"> 详情 </el-dropdown-item>
                             <el-dropdown-item :command="{ type: 'dumpDb', data }" v-if="supportAction('dumpDb', data.type)"> 导出 </el-dropdown-item>
-                            <el-dropdown-item :command="{ type: 'dbBackup', data }" v-if="supportAction('dbBackup', data.type)"> 备份 </el-dropdown-item>
-                            <el-dropdown-item :command="{ type: 'dbRestore', data }" v-if="supportAction('dbRestore', data.type)"> 恢复 </el-dropdown-item>
+                            <el-dropdown-item :command="{ type: 'backupDb', data }" v-if="actionBtns[perms.backupDb] && supportAction('backupDb', data.type)">
+                                备份任务
+                            </el-dropdown-item>
+                            <el-dropdown-item
+                                :command="{ type: 'backupHistory', data }"
+                                v-if="actionBtns[perms.backupDb] && supportAction('backupDb', data.type)"
+                            >
+                                备份历史
+                            </el-dropdown-item>
+                            <el-dropdown-item
+                                :command="{ type: 'restoreDb', data }"
+                                v-if="actionBtns[perms.restoreDb] && supportAction('restoreDb', data.type)"
+                            >
+                                恢复任务
+                            </el-dropdown-item>
                         </el-dropdown-menu>
                     </template>
                 </el-dropdown>
@@ -131,6 +144,16 @@
             <db-backup-list :dbId="dbBackupDialog.dbId" :dbNames="dbBackupDialog.dbs" />
         </el-dialog>
 
+        <el-dialog
+            width="80%"
+            :title="`${dbBackupHistoryDialog.title} - 数据库备份历史`"
+            :close-on-click-modal="false"
+            :destroy-on-close="true"
+            v-model="dbBackupHistoryDialog.visible"
+        >
+            <db-backup-history-list :dbId="dbBackupHistoryDialog.dbId" :dbNames="dbBackupHistoryDialog.dbs" />
+        </el-dialog>
+
         <el-dialog
             width="80%"
             :title="`${dbRestoreDialog.title} - 数据库恢复`"
@@ -185,6 +208,7 @@ import { getDbDialect } from './dialect/index';
 import { getTagPathSearchItem } from '../component/tag';
 import { SearchItem } from '@/components/SearchForm';
 import DbBackupList from './DbBackupList.vue';
+import DbBackupHistoryList from './DbBackupHistoryList.vue';
 import DbRestoreList from './DbRestoreList.vue';
 
 const DbEdit = defineAsyncComponent(() => import('./DbEdit.vue'));
@@ -193,6 +217,8 @@ const perms = {
     base: 'db',
     saveDb: 'db:save',
     delDb: 'db:del',
+    backupDb: 'db:backup',
+    restoreDb: 'db:restore',
 };
 
 const searchItems = [getTagPathSearchItem(TagResourceTypeEnum.Db.value), SearchItem.slot('instanceId', '实例', 'instanceSelect')];
@@ -208,7 +234,8 @@ const columns = ref([
 ]);
 
 // 该用户拥有的的操作列按钮权限
-const actionBtns = hasPerms([perms.base, perms.saveDb]);
+// const actionBtns = hasPerms([perms.base, perms.saveDb]);
+const actionBtns = hasPerms(Object.values(perms));
 const actionColumn = TableColumn.new('action', '操作').isSlot().setMinWidth(220).fixedRight().alignCenter();
 
 const route = useRoute();
@@ -253,6 +280,13 @@ const state = reactive({
         dbs: [],
         dbId: 0,
     },
+    // 数据库备份历史弹框
+    dbBackupHistoryDialog: {
+        title: '',
+        visible: false,
+        dbs: [],
+        dbId: 0,
+    },
     // 数据库恢复弹框
     dbRestoreDialog: {
         title: '',
@@ -285,7 +319,8 @@ const state = reactive({
     },
 });
 
-const { db, selectionData, query, infoDialog, sqlExecLogDialog, exportDialog, dbEditDialog, dbBackupDialog, dbRestoreDialog } = toRefs(state);
+const { db, selectionData, query, infoDialog, sqlExecLogDialog, exportDialog, dbEditDialog, dbBackupDialog, dbBackupHistoryDialog, dbRestoreDialog } =
+    toRefs(state);
 
 onMounted(async () => {
     if (Object.keys(actionBtns).length > 0) {
@@ -345,11 +380,15 @@ const handleMoreActionCommand = (commond: any) => {
             onDumpDbs(data);
             return;
         }
-        case 'dbBackup': {
+        case 'backupDb': {
             onShowDbBackupDialog(data);
             return;
         }
-        case 'dbRestore': {
+        case 'backupHistory': {
+            onShowDbBackupHistoryDialog(data);
+            return;
+        }
+        case 'restoreDb': {
             onShowDbRestoreDialog(data);
             return;
         }
@@ -402,6 +441,13 @@ const onShowDbBackupDialog = async (row: any) => {
     state.dbBackupDialog.visible = true;
 };
 
+const onShowDbBackupHistoryDialog = async (row: any) => {
+    state.dbBackupHistoryDialog.title = `${row.name}`;
+    state.dbBackupHistoryDialog.dbId = row.id;
+    state.dbBackupHistoryDialog.dbs = row.database.split(' ');
+    state.dbBackupHistoryDialog.visible = true;
+};
+
 const onShowDbRestoreDialog = async (row: any) => {
     state.dbRestoreDialog.title = `${row.name}`;
     state.dbRestoreDialog.dbId = row.id;
@@ -455,7 +501,7 @@ const supportAction = (action: string, dbType: string): boolean => {
     switch (dbType) {
         case DbType.mysql:
         case DbType.mariadb:
-            actions = ['dumpDb', 'dbBackup', 'dbRestore'];
+            actions = ['dumpDb', 'backupDb', 'restoreDb'];
     }
     return actions.includes(action);
 };

mayfly_go_web/src/views/ops/db/DbRestoreEdit.vue

@@ -35,7 +35,13 @@
                         clearable
                         class="w100"
                     >
-                        <el-option v-for="item in state.histories" :key="item.id" :label="item.name" :value="item"> </el-option>
+                        <el-option
+                            v-for="item in state.histories"
+                            :key="item.id"
+                            :label="item.name + (item.binlogFileName ? ' ' : ' 不') + '支持指定时间点恢复'"
+                            :value="item"
+                        >
+                        </el-option>
                     </el-select>
                 </el-form-item>
                 <el-form-item prop="startTime" label="开始时间">
@@ -56,7 +62,7 @@
 <script lang="ts" setup>
 import { onMounted, reactive, ref, watch } from 'vue';
 import { dbApi } from './api';
-import { ElMessage } from 'element-plus';
+import { ElMessage, ElMessageBox } from 'element-plus';
 
 const props = defineProps({
     data: {
@@ -83,20 +89,30 @@ const visible = defineModel<boolean>('visible', {
 });
 
 const validatePointInTime = (rule: any, value: any, callback: any) => {
-    if (!state.histories || state.histories.length == 0) {
-        callback(new Error('数据库没有备份记录'));
-        return;
-    }
-    const history = state.histories[state.histories.length - 1];
-    if (value < new Date(history.createTime)) {
-        callback(new Error('在此之前数据库没有备份记录'));
-        return;
-    }
     if (value > new Date()) {
         callback(new Error('恢复时间点晚于当前时间'));
         return;
     }
-    callback();
+    if (!state.histories || state.histories.length == 0) {
+        callback(new Error('数据库没有备份记录'));
+        return;
+    }
+    let last = null;
+    for (const history of state.histories) {
+        if (!history.binlogFileName || history.binlogFileName.length === 0) {
+            break;
+        }
+        if (new Date(history.createTime) < value) {
+            callback();
+            return;
+        }
+        last = history;
+    }
+    if (!last) {
+        callback(new Error('现有数据库备份不支持指定时间恢复'));
+        return;
+    }
+    callback(last.name + ' 之前的数据库备份不支持指定时间恢复');
 };
 
 const rules = {
@@ -110,7 +126,6 @@ const rules = {
     pointInTime: [
         {
             required: true,
-            // message: '请选择恢复时间点',
             validator: validatePointInTime,
             trigger: ['change', 'blur'],
         },
@@ -146,7 +161,7 @@ const state = reactive({
         id: 0,
         dbId: 0,
         dbName: null as any,
-        intervalDay: 1,
+        intervalDay: 0,
         startTime: null as any,
         repeated: null as any,
         dbBackupId: null as any,
@@ -218,7 +233,8 @@ const init = async (data: any) => {
     } else {
         state.form.dbName = '';
         state.editOrCreate = false;
-        state.form.intervalDay = 1;
+        state.form.intervalDay = 0;
+        state.form.repeated = false;
         state.form.pointInTime = new Date();
         state.form.startTime = new Date();
         state.histories = [];
@@ -237,6 +253,12 @@ const getDbNamesWithoutRestore = async () => {
 const btnOk = async () => {
     restoreForm.value.validate(async (valid: any) => {
         if (valid) {
+            await ElMessageBox.confirm(`确定恢复数据库吗？`, '提示', {
+                confirmButtonText: '确定',
+                cancelButtonText: '取消',
+                type: 'warning',
+            });
+
             if (state.restoreMode == 'point-in-time') {
                 state.form.dbBackupId = 0;
                 state.form.dbBackupHistoryId = 0;
@@ -245,13 +267,14 @@ const btnOk = async () => {
                 state.form.pointInTime = null;
             }
             state.form.repeated = false;
+            state.form.intervalDay = 0;
             const reqForm = { ...state.form };
             let api = dbApi.createDbRestore;
             if (props.data) {
                 api = dbApi.saveDbRestore;
             }
             api.request(reqForm).then(() => {
-                ElMessage.success('保存成功');
+                ElMessage.success('成功创建数据库恢复任务');
                 emit('val-change', state.form);
                 state.btnLoading = true;
                 setTimeout(() => {

mayfly_go_web/src/views/ops/db/DbRestoreList.vue

@@ -21,12 +21,14 @@
                 <el-button type="primary" icon="plus" @click="createDbRestore()">添加</el-button>
                 <el-button type="primary" icon="video-play" @click="enableDbRestore(null)">启用</el-button>
                 <el-button type="primary" icon="video-pause" @click="disableDbRestore(null)">禁用</el-button>
+                <el-button type="danger" icon="delete" @click="deleteDbRestore(null)">删除</el-button>
             </template>
 
             <template #action="{ data }">
                 <el-button @click="showDbRestore(data)" type="primary" link>详情</el-button>
                 <el-button @click="enableDbRestore(data)" v-if="!data.enabled" type="primary" link>启用</el-button>
                 <el-button @click="disableDbRestore(data)" v-if="data.enabled" type="primary" link>禁用</el-button>
+                <el-button @click="deleteDbRestore(data)" type="danger" link>删除</el-button>
             </template>
         </page-table>
 
@@ -49,7 +51,7 @@
                     infoDialog.data.dbBackupHistoryName
                 }}</el-descriptions-item>
                 <el-descriptions-item :span="1" label="开始时间">{{ dateFormat(infoDialog.data.startTime) }}</el-descriptions-item>
-                <el-descriptions-item :span="1" label="是否启用">{{ infoDialog.data.enabled }}</el-descriptions-item>
+                <el-descriptions-item :span="1" label="是否启用">{{ infoDialog.data.enabledDesc }}</el-descriptions-item>
                 <el-descriptions-item :span="1" label="执行时间">{{ dateFormat(infoDialog.data.lastTime) }}</el-descriptions-item>
                 <el-descriptions-item :span="1" label="执行结果">{{ infoDialog.data.lastResult }}</el-descriptions-item>
             </el-descriptions>
@@ -63,7 +65,7 @@ import { dbApi } from './api';
 import PageTable from '@/components/pagetable/PageTable.vue';
 import { TableColumn } from '@/components/pagetable';
 import { SearchItem } from '@/components/SearchForm';
-import { ElMessage } from 'element-plus';
+import { ElMessage, ElMessageBox } from 'element-plus';
 import { dateFormat } from '@/common/utils/date';
 const DbRestoreEdit = defineAsyncComponent(() => import('./DbRestoreEdit.vue'));
 const pageTableRef: Ref<any> = ref(null);
@@ -85,7 +87,7 @@ const searchItems = [SearchItem.slot('dbName', '数据库名称', 'dbSelect')];
 const columns = [
     TableColumn.new('dbName', '数据库名称'),
     TableColumn.new('startTime', '启动时间').isTime(),
-    TableColumn.new('enabled', '是否启用'),
+    TableColumn.new('enabledDesc', '是否启用'),
     TableColumn.new('lastTime', '执行时间').isTime(),
     TableColumn.new('lastResult', '执行结果'),
     TableColumn.new('action', '操作').isSlot().setMinWidth(220).fixedRight().alignCenter(),
@@ -135,19 +137,39 @@ const createDbRestore = async () => {
     state.dbRestoreEditDialog.visible = true;
 };
 
+const deleteDbRestore = async (data: any) => {
+    let restoreId: string;
+    if (data) {
+        restoreId = data.id;
+    } else if (state.selectedData.length > 0) {
+        restoreId = state.selectedData.map((x: any) => x.id).join(' ');
+    } else {
+        ElMessage.error('请选择需要删除的数据库恢复任务');
+        return;
+    }
+    await ElMessageBox.confirm(`确定删除 “数据库恢复任务” 吗？`, '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning',
+    });
+    await dbApi.deleteDbRestore.request({ dbId: props.dbId, restoreId: restoreId });
+    await search();
+    ElMessage.success('删除成功');
+};
+
 const showDbRestore = async (data: any) => {
     state.infoDialog.data = data;
     state.infoDialog.visible = true;
 };
 
 const enableDbRestore = async (data: any) => {
-    let restoreId: String;
+    let restoreId: string;
     if (data) {
         restoreId = data.id;
     } else if (state.selectedData.length > 0) {
         restoreId = state.selectedData.map((x: any) => x.id).join(' ');
     } else {
-        ElMessage.error('请选择需要启用的恢复任务');
+        ElMessage.error('请选择需要启用的数据库恢复任务');
         return;
     }
     await dbApi.enableDbRestore.request({ dbId: props.dbId, restoreId: restoreId });
@@ -156,13 +178,13 @@ const enableDbRestore = async (data: any) => {
 };
 
 const disableDbRestore = async (data: any) => {
-    let restoreId: String;
+    let restoreId: string;
     if (data) {
         restoreId = data.id;
     } else if (state.selectedData.length > 0) {
         restoreId = state.selectedData.map((x: any) => x.id).join(' ');
     } else {
-        ElMessage.error('请选择需要禁用的恢复任务');
+        ElMessage.error('请选择需要禁用的数据库恢复任务');
         return;
     }
     await dbApi.disableDbRestore.request({ dbId: props.dbId, restoreId: restoreId });

mayfly_go_web/src/views/ops/db/DbSqlExecLog.vue

@@ -36,7 +36,7 @@
 </template>
 
 <script lang="ts" setup>
-import { toRefs, watch, reactive, onMounted, Ref, ref } from 'vue';
+import { onMounted, reactive, Ref, ref, toRefs, watch } from 'vue';
 import { dbApi } from './api';
 import { DbSqlExecTypeEnum } from './enums';
 import PageTable from '@/components/pagetable/PageTable.vue';
@@ -120,6 +120,12 @@ const onShowRollbackSql = async (sqlExecLog: any) => {
     const primaryKey = getPrimaryKey(columns);
     const oldValue = JSON.parse(sqlExecLog.oldValue);
 
+    let schema = '';
+    let dbArr = sqlExecLog.db.split('/');
+    if (dbArr.length == 2) {
+        schema = dbArr[1] + '.';
+    }
+
     const rollbackSqls = [];
     if (sqlExecLog.type == DbSqlExecTypeEnum.Update.value) {
         for (let ov of oldValue) {
@@ -130,7 +136,7 @@ const onShowRollbackSql = async (sqlExecLog: any) => {
                 }
                 setItems.push(`${key} = ${wrapValue(ov[key])}`);
             }
-            rollbackSqls.push(`UPDATE ${sqlExecLog.table} SET ${setItems.join(', ')} WHERE ${primaryKey} = ${wrapValue(ov[primaryKey])};`);
+            rollbackSqls.push(`UPDATE ${schema}${sqlExecLog.table} SET ${setItems.join(', ')} WHERE ${primaryKey} = ${wrapValue(ov[primaryKey])};`);
         }
     } else if (sqlExecLog.type == DbSqlExecTypeEnum.Delete.value) {
         const columnNames = columns.map((c: any) => c.columnName);
@@ -139,7 +145,7 @@ const onShowRollbackSql = async (sqlExecLog: any) => {
             for (let column of columnNames) {
                 values.push(wrapValue(ov[column]));
             }
-            rollbackSqls.push(`INSERT INTO ${sqlExecLog.table} (${columnNames.join(', ')}) VALUES (${values.join(', ')});`);
+            rollbackSqls.push(`INSERT INTO ${schema}${sqlExecLog.table} (${columnNames.join(', ')}) VALUES (${values.join(', ')});`);
         }
     }
 
@@ -148,7 +154,7 @@ const onShowRollbackSql = async (sqlExecLog: any) => {
 };
 
 const getPrimaryKey = (columns: any) => {
-    const col = columns.find((c: any) => c.columnKey == 'PRI');
+    const col = columns.find((c: any) => c.isPrimaryKey);
     if (col) {
         return col.columnName;
     }

mayfly_go_web/src/views/ops/db/InstanceEdit.vue

@@ -9,10 +9,19 @@
                         </el-form-item>
                         <el-form-item prop="type" label="类型" required>
                             <el-select @change="changeDbType" style="width: 100%" v-model="form.type" placeholder="请选择数据库类型">
-                                <el-option v-for="dt in dbTypes" :key="dt.type" :value="dt.type" :label="dt.label">
-                                    <SvgIcon :name="getDbDialect(dt.type).getInfo().icon" :size="18" />
-                                    {{ dt.label }}
+                                <el-option
+                                    v-for="(dbTypeAndDialect, key) in getDbDialectMap()"
+                                    :key="key"
+                                    :value="dbTypeAndDialect[0]"
+                                    :label="dbTypeAndDialect[1].getInfo().name"
+                                >
+                                    <SvgIcon :name="dbTypeAndDialect[1].getInfo().icon" :size="20" />
+                                    {{ dbTypeAndDialect[1].getInfo().name }}
                                 </el-option>
+
+                                <template #prefix>
+                                    <SvgIcon :name="getDbDialect(form.type).getInfo().icon" :size="20" />
+                                </template>
                             </el-select>
                         </el-form-item>
                         <el-form-item v-if="form.type !== DbType.sqlite" prop="host" label="host" required>
@@ -95,7 +104,7 @@ import { ElMessage } from 'element-plus';
 import { notBlank } from '@/common/assert';
 import { RsaEncrypt } from '@/common/rsa';
 import SshTunnelSelect from '../component/SshTunnelSelect.vue';
-import { DbType, getDbDialect } from './dialect';
+import { DbType, getDbDialect, getDbDialectMap } from './dialect';
 import SvgIcon from '@/components/svgIcon/index.vue';
 
 const props = defineProps({
@@ -153,33 +162,6 @@ const rules = {
 
 const dbForm: any = ref(null);
 
-const dbTypes = [
-    {
-        type: 'mysql',
-        label: 'mysql',
-    },
-    {
-        type: 'mariadb',
-        label: 'mariadb',
-    },
-    {
-        type: 'postgres',
-        label: 'postgres',
-    },
-    {
-        type: 'dm',
-        label: '达梦',
-    },
-    {
-        type: 'oracle',
-        label: 'oracle',
-    },
-    {
-        type: 'sqlite',
-        label: 'sqlite',
-    },
-];
-
 const state = reactive({
     dialogVisible: false,
     tabActiveName: 'basic',
@@ -196,17 +178,17 @@ const state = reactive({
         remark: '',
         sshTunnelMachineId: null as any,
     },
-    subimtForm: {},
+    submitForm: {},
     // 原密码
     pwd: '',
     // 原用户名
     oldUserName: null,
 });
 
-const { dialogVisible, tabActiveName, form, subimtForm, pwd } = toRefs(state);
+const { dialogVisible, tabActiveName, form, submitForm, pwd } = toRefs(state);
 
-const { isFetching: saveBtnLoading, execute: saveInstanceExec } = dbApi.saveInstance.useApi(subimtForm);
-const { isFetching: testConnBtnLoading, execute: testConnExec } = dbApi.testConn.useApi(subimtForm);
+const { isFetching: saveBtnLoading, execute: saveInstanceExec } = dbApi.saveInstance.useApi(submitForm);
+const { isFetching: testConnBtnLoading, execute: testConnExec } = dbApi.testConn.useApi(submitForm);
 
 watch(props, (newValue: any) => {
     state.dialogVisible = newValue.visible;
@@ -218,7 +200,7 @@ watch(props, (newValue: any) => {
         state.form = { ...newValue.data };
         state.oldUserName = state.form.username;
     } else {
-        state.form = { port: null } as any;
+        state.form = { port: null, type: DbType.mysql } as any;
         state.oldUserName = null;
     }
 });
@@ -249,7 +231,7 @@ const testConn = async () => {
             return false;
         }
 
-        state.subimtForm = await getReqForm();
+        state.submitForm = await getReqForm();
         await testConnExec();
         ElMessage.success('连接成功');
     });
@@ -270,7 +252,7 @@ const btnOk = async () => {
             return false;
         }
 
-        state.subimtForm = await getReqForm();
+        state.submitForm = await getReqForm();
         await saveInstanceExec();
         ElMessage.success('保存成功');
         emit('val-change', state.form);

mayfly_go_web/src/views/ops/db/SqlExec.vue

@@ -151,12 +151,22 @@
                 </div>
             </Pane>
         </Splitpanes>
+        <db-table-op
+            :title="tableCreateDialog.title"
+            :active-name="tableCreateDialog.activeName"
+            :dbId="tableCreateDialog.dbId"
+            :db="tableCreateDialog.db"
+            :dbType="tableCreateDialog.dbType"
+            :data="tableCreateDialog.data"
+            v-model:visible="tableCreateDialog.visible"
+            @submit-sql="onSubmitEditTableSql"
+        />
     </div>
 </template>
 
 <script lang="ts" setup>
-import { defineAsyncComponent, onBeforeUnmount, onMounted, reactive, ref, toRefs } from 'vue';
-import { ElMessage, ElMessageBox } from 'element-plus';
+import { defineAsyncComponent, h, onBeforeUnmount, onMounted, reactive, ref, toRefs } from 'vue';
+import { ElCheckbox, ElMessage, ElMessageBox } from 'element-plus';
 import { formatByteSize } from '@/common/utils/format';
 import { DbInst, registerDbCompletionItemProvider, TabInfo, TabType } from './db';
 import { NodeType, TagTreeNode } from '../component/tag';
@@ -165,12 +175,13 @@ import { dbApi } from './api';
 import { dispposeCompletionItemProvider } from '@/components/monaco/completionItemProvider';
 import SvgIcon from '@/components/svgIcon/index.vue';
 import { ContextmenuItem } from '@/components/contextmenu';
-import { DbType, getDbDialect } from './dialect/index';
+import { getDbDialect, schemaDbTypes } from './dialect/index';
 import { sleep } from '@/common/utils/loading';
 import { TagResourceTypeEnum } from '@/common/commonEnum';
 import { Pane, Splitpanes } from 'splitpanes';
 import { useEventListener } from '@vueuse/core';
 
+const DbTableOp = defineAsyncComponent(() => import('./component/table/DbTableOp.vue'));
 const DbSqlEditor = defineAsyncComponent(() => import('./component/sqleditor/DbSqlEditor.vue'));
 const DbTableDataOp = defineAsyncComponent(() => import('./component/table/DbTableDataOp.vue'));
 const DbTablesOp = defineAsyncComponent(() => import('./component/table/DbTablesOp.vue'));
@@ -218,21 +229,25 @@ const nodeClickChangeDb = (nodeData: TagTreeNode) => {
     }
 };
 
-// tagpath 节点类型
-const NodeTypeTagPath = new NodeType(TagTreeNode.TagPath).withLoadNodesFunc(async (parentNode: TagTreeNode) => {
-    const dbInfoRes = await dbApi.dbs.request({ tagPath: parentNode.key });
-    const dbInfos = dbInfoRes.list;
-    if (!dbInfos) {
-        return [];
-    }
+const ContextmenuItemRefresh = new ContextmenuItem('refresh', '刷新').withIcon('RefreshRight').withOnClick((data: any) => reloadNode(data.key));
 
-    // 防止过快加载会出现一闪而过，对眼睛不好
-    await sleep(100);
-    return dbInfos?.map((x: any) => {
-        x.tagPath = parentNode.key;
-        return new TagTreeNode(`${parentNode.key}.${x.id}`, x.name, NodeTypeDbInst).withParams(x);
-    });
-});
+// tagpath 节点类型
+const NodeTypeTagPath = new NodeType(TagTreeNode.TagPath)
+    .withLoadNodesFunc(async (parentNode: TagTreeNode) => {
+        const dbInfoRes = await dbApi.dbs.request({ tagPath: parentNode.key });
+        const dbInfos = dbInfoRes.list;
+        if (!dbInfos) {
+            return [];
+        }
+
+        // 防止过快加载会出现一闪而过，对眼睛不好
+        await sleep(100);
+        return dbInfos?.map((x: any) => {
+            x.tagPath = parentNode.key;
+            return new TagTreeNode(`${parentNode.key}.${x.id}`, x.name, NodeTypeDbInst).withParams(x);
+        });
+    })
+    .withContextMenuItems([ContextmenuItemRefresh]);
 
 // 数据库实例节点类型
 const NodeTypeDbInst = new NodeType(SqlExecNodeType.DbInst).withLoadNodesFunc((parentNode: TagTreeNode) => {
@@ -255,12 +270,12 @@ const NodeTypeDbInst = new NodeType(SqlExecNodeType.DbInst).withLoadNodesFunc((p
 
 // 数据库节点
 const NodeTypeDb = new NodeType(SqlExecNodeType.Db)
-    .withContextMenuItems([new ContextmenuItem('reloadTables', '刷新').withIcon('RefreshRight').withOnClick((data: any) => reloadNode(data.key))])
+    .withContextMenuItems([ContextmenuItemRefresh])
     .withLoadNodesFunc(async (parentNode: TagTreeNode) => {
         const params = parentNode.params;
+        params.parentKey = parentNode.key;
         // pg类数据库会多一层schema
-        if (params.type == DbType.postgresql || params.type === DbType.dm || params.type === DbType.oracle) {
-            const params = parentNode.params;
+        if (schemaDbTypes.includes(params.type)) {
             const { id, db } = params;
             const schemaNames = await dbApi.pgSchemas.request({ id, db });
             return schemaNames.map((sn: any) => {
@@ -277,23 +292,29 @@ const NodeTypeDb = new NodeType(SqlExecNodeType.Db)
     .withNodeClickFunc(nodeClickChangeDb);
 
 const NodeTypeTables = (params: any) => {
+    let tableKey = `${params.id}.${params.db}.table-menu`;
+    let sqlKey = getSqlMenuNodeKey(params.id, params.db);
     return [
-        new TagTreeNode(`${params.id}.${params.db}.table-menu`, '表', NodeTypeTableMenu).withParams(params).withIcon(TableIcon),
-        new TagTreeNode(getSqlMenuNodeKey(params.id, params.db), 'SQL', NodeTypeSqlMenu).withParams(params).withIcon(SqlIcon),
+        new TagTreeNode(`${params.id}.${params.db}.table-menu`, '表', NodeTypeTableMenu).withParams({ ...params, key: tableKey }).withIcon(TableIcon),
+        new TagTreeNode(sqlKey, 'SQL', NodeTypeSqlMenu).withParams({ ...params, key: sqlKey }).withIcon(SqlIcon),
     ];
 };
 
 // postgres schema模式
 const NodeTypePostgresSchema = new NodeType(SqlExecNodeType.PgSchema)
-    .withContextMenuItems([new ContextmenuItem('reloadTables', '刷新').withIcon('RefreshRight').withOnClick((data: any) => reloadNode(data.key))])
-    .withLoadNodesFunc(async (parentNode: TagTreeNode) => NodeTypeTables(parentNode.params))
+    .withContextMenuItems([ContextmenuItemRefresh])
+    .withLoadNodesFunc(async (parentNode: TagTreeNode) => {
+        const params = parentNode.params;
+        params.parentKey = parentNode.key;
+        return NodeTypeTables(params);
+    })
     .withNodeClickFunc(nodeClickChangeDb);
 
 // 数据库表菜单节点
 const NodeTypeTableMenu = new NodeType(SqlExecNodeType.TableMenu)
     .withContextMenuItems([
-        new ContextmenuItem('reloadTables', '刷新').withIcon('RefreshRight').withOnClick((data: any) => reloadNode(data.key)),
-
+        ContextmenuItemRefresh,
+        new ContextmenuItem('createTable', '创建表').withIcon('Plus').withOnClick((data: any) => onEditTable(data)),
         new ContextmenuItem('tablesOp', '表操作').withIcon('Setting').withOnClick((data: any) => {
             const params = data.params;
             addTablesOpTab({ id: params.id, db: params.db, type: params.type, nodeKey: data.key });
@@ -301,7 +322,7 @@ const NodeTypeTableMenu = new NodeType(SqlExecNodeType.TableMenu)
     ])
     .withLoadNodesFunc(async (parentNode: TagTreeNode) => {
         const params = parentNode.params;
-        let { id, db } = params;
+        let { id, db, type } = params;
         // 获取当前库的所有表信息
         let tables = await DbInst.getInst(id).loadTables(db, state.reloadStatus);
         state.reloadStatus = false;
@@ -309,11 +330,15 @@ const NodeTypeTableMenu = new NodeType(SqlExecNodeType.TableMenu)
         const tablesNode = tables.map((x: any) => {
             const tableSize = x.dataLength + x.indexLength;
             dbTableSize += tableSize;
-            return new TagTreeNode(`${id}.${db}.${x.tableName}`, x.tableName, NodeTypeTable)
+            const key = `${id}.${db}.${x.tableName}`;
+            return new TagTreeNode(key, x.tableName, NodeTypeTable)
                 .withIsLeaf(true)
                 .withParams({
                     id,
                     db,
+                    type,
+                    key: key,
+                    parentKey: parentNode.key,
                     tableName: x.tableName,
                     tableComment: x.tableComment,
                     size: tableSize == 0 ? '' : formatByteSize(tableSize, 1),
@@ -339,22 +364,23 @@ const NodeTypeSqlMenu = new NodeType(SqlExecNodeType.SqlMenu)
         return sqls.map((x: any) => {
             return new TagTreeNode(`${id}.${db}.${x.name}`, x.name, NodeTypeSql)
                 .withIsLeaf(true)
-                .withParams({
-                    id,
-                    db,
-                    dbs,
-                    sqlName: x.name,
-                })
+                .withParams({ id, db, dbs, sqlName: x.name })
                 .withIcon(SqlIcon);
         });
     })
     .withNodeClickFunc(nodeClickChangeDb);
 
 // 表节点类型
-const NodeTypeTable = new NodeType(SqlExecNodeType.Table).withNodeClickFunc((nodeData: TagTreeNode) => {
-    const params = nodeData.params;
-    loadTableData({ id: params.id, nodeKey: nodeData.key }, params.db, params.tableName);
-});
+const NodeTypeTable = new NodeType(SqlExecNodeType.Table)
+    .withContextMenuItems([
+        new ContextmenuItem('copyTable', '复制表').withIcon('copyDocument').withOnClick((data: any) => onCopyTable(data)),
+        new ContextmenuItem('editTable', '编辑表').withIcon('edit').withOnClick((data: any) => onEditTable(data)),
+        new ContextmenuItem('delTable', '删除表').withIcon('Delete').withOnClick((data: any) => onDeleteTable(data)),
+    ])
+    .withNodeClickFunc((nodeData: TagTreeNode) => {
+        const params = nodeData.params;
+        loadTableData({ id: params.id, nodeKey: nodeData.key }, params.db, params.tableName);
+    });
 
 // sql模板节点类型
 const NodeTypeSql = new NodeType(SqlExecNodeType.Sql)
@@ -384,9 +410,19 @@ const state = reactive({
         loading: true,
         version: '',
     },
+    tableCreateDialog: {
+        visible: false,
+        title: '',
+        activeName: '',
+        dbId: 0,
+        db: '',
+        dbType: '',
+        data: {},
+        parentKey: '',
+    },
 });
 
-const { nowDbInst } = toRefs(state);
+const { nowDbInst, tableCreateDialog } = toRefs(state);
 
 const serverInfoReqParam = ref({
     instanceId: 0,
@@ -407,7 +443,7 @@ onBeforeUnmount(() => {
  * 设置editor高度和数据表高度
  */
 const setHeight = () => {
-    state.dataTabsTableHeight = window.innerHeight - 270 + 'px';
+    state.dataTabsTableHeight = window.innerHeight - 253 + 'px';
     state.tablesOpHeight = window.innerHeight - 225 + 'px';
 };
 
@@ -602,6 +638,85 @@ const reloadNode = (nodeKey: string) => {
     tagTreeRef.value.reloadNode(nodeKey);
 };
 
+const onEditTable = async (data: any) => {
+    let { db, id, tableName, tableComment, type, parentKey, key } = data.params;
+    // data.label就是表名
+    if (tableName) {
+        state.tableCreateDialog.title = '修改表';
+        let indexs = await dbApi.tableIndex.request({ id, db, tableName });
+        let columns = await dbApi.columnMetadata.request({ id, db, tableName });
+        let row = { tableName, tableComment };
+        state.tableCreateDialog.data = { edit: true, row, indexs, columns };
+        state.tableCreateDialog.parentKey = parentKey;
+    } else {
+        state.tableCreateDialog.title = '创建表';
+        state.tableCreateDialog.data = { edit: false, row: {} };
+        state.tableCreateDialog.parentKey = key;
+    }
+
+    state.tableCreateDialog.visible = true;
+    state.tableCreateDialog.activeName = '1';
+    state.tableCreateDialog.dbId = id;
+    state.tableCreateDialog.db = db;
+    state.tableCreateDialog.dbType = type;
+};
+
+const onDeleteTable = async (data: any) => {
+    let { db, id, tableName, parentKey } = data.params;
+    await ElMessageBox.confirm(`此操作是永久性且无法撤销，确定删除【${tableName}】? `, '提示', {
+        confirmButtonText: '确定',
+        cancelButtonText: '取消',
+        type: 'warning',
+    });
+    // 执行sql
+    dbApi.sqlExec.request({ id, db, sql: `drop table ${tableName}` }).then(() => {
+        ElMessage.success('删除成功');
+        setTimeout(() => {
+            parentKey && reloadNode(parentKey);
+        }, 1000);
+    });
+};
+
+const onCopyTable = async (data: any) => {
+    let { db, id, tableName, parentKey } = data.params;
+
+    let checked = ref(false);
+
+    // 弹出确认框，并选择是否复制数据
+    await ElMessageBox({
+        title: `复制表【${tableName}】`,
+        type: 'warning',
+        //  icon: markRaw(Delete),
+        message: () =>
+            h(ElCheckbox, {
+                label: '是否复制数据?',
+                modelValue: checked.value,
+                'onUpdate:modelValue': (val: boolean | string | number) => {
+                    if (typeof val === 'boolean') {
+                        checked.value = val;
+                    }
+                },
+            }),
+        callback: (action: string) => {
+            if (action === 'confirm') {
+                // 执行sql
+                dbApi.copyTable.request({ id, db, tableName, copyData: checked.value }).then(() => {
+                    ElMessage.success('复制成功');
+                    setTimeout(() => {
+                        parentKey && reloadNode(parentKey);
+                    }, 1000);
+                });
+            }
+        },
+    });
+};
+
+const onSubmitEditTableSql = () => {
+    state.tableCreateDialog.visible = false;
+    state.tableCreateDialog.data = { edit: false, row: {} };
+    reloadNode(state.tableCreateDialog.parentKey);
+};
+
 /**
  * 获取当前操作的数据库信息
  */

mayfly_go_web/src/views/ops/db/SyncTaskEdit.vue

@@ -47,6 +47,7 @@
                                 v-model:db-id="form.srcDbId"
                                 v-model:db-name="form.srcDbName"
                                 v-model:tag-path="form.srcTagPath"
+                                v-model:db-type="form.srcDbType"
                                 @select-db="onSelectSrcDb"
                             />
                         </el-form-item>
@@ -181,7 +182,7 @@ import { ElMessage } from 'element-plus';
 import DbSelectTree from '@/views/ops/db/component/DbSelectTree.vue';
 import MonacoEditor from '@/components/monaco/MonacoEditor.vue';
 import { DbInst, registerDbCompletionItemProvider } from '@/views/ops/db/db';
-import { getDbDialect } from '@/views/ops/db/dialect';
+import {DbType, getDbDialect} from '@/views/ops/db/dialect'
 import CrontabInput from '@/components/crontab/CrontabInput.vue';
 
 const props = defineProps({
@@ -227,6 +228,7 @@ type FormData = {
     taskCron: string;
     srcDbId?: number;
     srcDbName?: string;
+    srcDbType?: string;
     srcTagPath?: string;
     targetDbId?: number;
     targetDbName?: string;
@@ -245,7 +247,7 @@ const basicFormData = {
     targetDbId: -1,
     dataSql: 'select * from',
     pageSize: 1000,
-    updField: 'id',
+    updField: '',
     updFieldVal: '0',
     fieldMap: [{ src: 'a', target: 'b' }],
     status: 1,
@@ -302,6 +304,7 @@ watch(dialogVisible, async (newValue: boolean) => {
         // 初始化实例
         db.databases = db.database?.split(' ').sort() || [];
         state.srcDbInst = DbInst.getOrNewInst(db);
+        state.form.srcDbType = state.srcDbInst.type
     }
 
     //  初始化target数据源
@@ -396,8 +399,8 @@ const handleGetSrcFields = async () => {
     }
 
     // 判断sql是否是查询语句
-    if (!/^select/i.test(state.form.dataSql!)) {
-        let msg = 'sql语句错误，请输入查询语句';
+    if (!/^select/i.test(state.form.dataSql.trim()!)) {
+        let msg = 'sql语句错误，请输入select语句';
         ElMessage.warning(msg);
         return;
     }
@@ -410,10 +413,16 @@ const handleGetSrcFields = async () => {
     }
 
     // 执行sql
+    // oracle的分页关键字不一样
+    let limit = ' limit 1'
+    if(state.form.srcDbType === DbType.oracle){
+      limit = ' where rownum <= 1'
+    }
+    
     const res = await dbApi.sqlExec.request({
         id: state.form.srcDbId,
         db: state.form.srcDbName,
-        sql: state.form.dataSql.trim() + ' limit 1',
+        sql: `select * from (${state.form.dataSql}) t ${limit}`
     });
 
     if (!res.columns) {

mayfly_go_web/src/views/ops/db/api.ts

@@ -11,6 +11,7 @@ export const dbApi = {
     tableInfos: Api.newGet('/dbs/{id}/t-infos'),
     tableIndex: Api.newGet('/dbs/{id}/t-index'),
     tableDdl: Api.newGet('/dbs/{id}/t-create-ddl'),
+    copyTable: Api.newPost('/dbs/{id}/copy-table'),
     columnMetadata: Api.newGet('/dbs/{id}/c-metadata'),
     pgSchemas: Api.newGet('/dbs/{id}/pg/schemas'),
     // 获取表即列提示
@@ -48,16 +49,20 @@ export const dbApi = {
     // 获取数据库备份列表
     getDbBackups: Api.newGet('/dbs/{dbId}/backups'),
     createDbBackup: Api.newPost('/dbs/{dbId}/backups'),
+    deleteDbBackup: Api.newDelete('/dbs/{dbId}/backups/{backupId}'),
     getDbNamesWithoutBackup: Api.newGet('/dbs/{dbId}/db-names-without-backup'),
     enableDbBackup: Api.newPut('/dbs/{dbId}/backups/{backupId}/enable'),
     disableDbBackup: Api.newPut('/dbs/{dbId}/backups/{backupId}/disable'),
     startDbBackup: Api.newPut('/dbs/{dbId}/backups/{backupId}/start'),
     saveDbBackup: Api.newPut('/dbs/{dbId}/backups/{id}'),
     getDbBackupHistories: Api.newGet('/dbs/{dbId}/backup-histories'),
+    restoreDbBackupHistory: Api.newPost('/dbs/{dbId}/backup-histories/{backupHistoryId}/restore'),
+    deleteDbBackupHistory: Api.newDelete('/dbs/{dbId}/backup-histories/{backupHistoryId}'),
 
-    // 获取数据库备份列表
+    // 获取数据库恢复列表
     getDbRestores: Api.newGet('/dbs/{dbId}/restores'),
     createDbRestore: Api.newPost('/dbs/{dbId}/restores'),
+    deleteDbRestore: Api.newDelete('/dbs/{dbId}/restores/{restoreId}'),
     getDbNamesWithoutRestore: Api.newGet('/dbs/{dbId}/db-names-without-restore'),
     enableDbRestore: Api.newPut('/dbs/{dbId}/restores/{restoreId}/enable'),
     disableDbRestore: Api.newPut('/dbs/{dbId}/restores/{restoreId}/disable'),

mayfly_go_web/src/views/ops/db/component/DbSelectTree.vue

@@ -19,7 +19,7 @@ import { NodeType, TagTreeNode } from '@/views/ops/component/tag';
 import { dbApi } from '@/views/ops/db/api';
 import { sleep } from '@/common/utils/loading';
 import SvgIcon from '@/components/svgIcon/index.vue';
-import { DbType, getDbDialect } from '@/views/ops/db/dialect';
+import { getDbDialect, noSchemaTypes } from '@/views/ops/db/dialect';
 import TagTreeResourceSelect from '../../component/TagTreeResourceSelect.vue';
 import { computed } from 'vue';
 
@@ -33,9 +33,12 @@ const props = defineProps({
     tagPath: {
         type: String,
     },
+    dbType: {
+        type: String,
+    },
 });
 
-const emits = defineEmits(['update:dbName', 'update:tagPath', 'update:dbId', 'selectDb']);
+const emits = defineEmits(['update:dbName', 'update:tagPath', 'update:dbId', 'update:dbType', 'selectDb']);
 
 /**
  * 树节点类型
@@ -87,8 +90,8 @@ const NodeTypeTagPath = new NodeType(TagTreeNode.TagPath).withLoadNodesFunc(asyn
 });
 
 /**  mysql类型的数据库，没有schema层 */
-const mysqlType = (type: string) => {
-    return type === DbType.mysql;
+const noSchemaType = (type: string) => {
+    return noSchemaTypes.includes(type);
 };
 
 // 数据库实例节点类型
@@ -96,7 +99,7 @@ const NodeTypeDbInst = new NodeType(SqlExecNodeType.DbInst).withLoadNodesFunc((p
     const params = parentNode.params;
     const dbs = params.database.split(' ')?.sort();
     let fn: NodeType;
-    if (mysqlType(params.type)) {
+    if (noSchemaType(params.type)) {
         fn = MysqlNodeTypes;
     } else {
         fn = PgNodeTypes;
@@ -114,7 +117,7 @@ const NodeTypeDbInst = new NodeType(SqlExecNodeType.DbInst).withLoadNodesFunc((p
                 db: x,
             })
             .withIcon(DbIcon);
-        if (mysqlType(params.type)) {
+        if (noSchemaType(params.type)) {
             tagTreeNode.isLeaf = true;
         }
         return tagTreeNode;
@@ -150,6 +153,7 @@ const changeNode = (nodeData: TagTreeNode) => {
     emits('update:dbName', params.db);
     emits('update:dbId', params.id);
     emits('update:tagPath', params.tagPath);
+    emits('update:dbType', params.type);
     emits('selectDb', params);
 };
 </script>

mayfly_go_web/src/views/ops/db/component/sqleditor/DbSqlEditor.vue

@@ -128,12 +128,12 @@
 </template>
 
 <script lang="ts" setup>
-import { h, nextTick, onMounted, reactive, toRefs, ref, unref } from 'vue';
+import { h, nextTick, onMounted, reactive, ref, toRefs, unref } from 'vue';
 import { getToken } from '@/common/utils/storage';
 import { notBlank } from '@/common/assert';
 import { format as sqlFormatter } from 'sql-formatter';
 import config from '@/common/config';
-import { ElMessage, ElMessageBox } from 'element-plus';
+import { ElMessage, ElMessageBox, ElNotification } from 'element-plus';
 
 import * as monaco from 'monaco-editor/esm/vs/editor/editor.api';
 import { editor } from 'monaco-editor';
@@ -146,11 +146,10 @@ import MonacoEditor from '@/components/monaco/MonacoEditor.vue';
 import { joinClientParams } from '@/common/request';
 import { buildProgressProps } from '@/components/progress-notify/progress-notify';
 import ProgressNotify from '@/components/progress-notify/progress-notify.vue';
-import { ElNotification } from 'element-plus';
 import syssocket from '@/common/syssocket';
 import SvgIcon from '@/components/svgIcon/index.vue';
 import { getDbDialect } from '../../dialect';
-import { Splitpanes, Pane } from 'splitpanes';
+import { Pane, Splitpanes } from 'splitpanes';
 
 const emits = defineEmits(['saveSqlSuccess']);
 
@@ -357,6 +356,7 @@ const onRunSql = async (newTab = false) => {
         const colAndData: any = data.value;
         if (!colAndData.res || colAndData.res.length === 0) {
             ElMessage.warning('未查询到结果集');
+            return;
         }
 
         // 要实时响应，故需要用索引改变数据才生效

mayfly_go_web/src/views/ops/db/component/table/ColumnFormItem.vue

@@ -3,6 +3,7 @@
         <el-input
             v-if="dataType == DataType.String"
             :ref="(el: any) => focus && el?.focus()"
+            :disabled="disabled"
             @blur="handleBlur"
             :class="`w100 mb4 ${showEditorIcon ? 'string-input-container-show-icon' : ''}`"
             input-style="text-align: center; height: 26px;"
@@ -16,6 +17,7 @@
     <el-input
         v-else-if="dataType == DataType.Number"
         :ref="(el: any) => focus && el?.focus()"
+        :disabled="disabled"
         @blur="handleBlur"
         class="w100 mb4"
         input-style="text-align: center; height: 26px;"
@@ -28,6 +30,7 @@
     <el-date-picker
         v-else-if="dataType == DataType.Date"
         :ref="(el: any) => focus && el?.focus()"
+        :disabled="disabled"
         @change="emit('blur')"
         @blur="handleBlur"
         class="edit-time-picker mb4"
@@ -43,6 +46,7 @@
     <el-date-picker
         v-else-if="dataType == DataType.DateTime"
         :ref="(el: any) => focus && el?.focus()"
+        :disabled="disabled"
         @change="handleBlur"
         @blur="handleBlur"
         class="edit-time-picker mb4"
@@ -58,6 +62,7 @@
     <el-time-picker
         v-else-if="dataType == DataType.Time"
         :ref="(el: any) => focus && el?.focus()"
+        :disabled="disabled"
         @change="handleBlur"
         @blur="handleBlur"
         class="edit-time-picker mb4"
@@ -71,7 +76,7 @@
 </template>
 
 <script lang="ts" setup>
-import { Ref, ref, computed } from 'vue';
+import { computed, ref, Ref } from 'vue';
 import { ElInput } from 'element-plus';
 import { DataType } from '../../dialect/index';
 import SvgIcon from '@/components/svgIcon/index.vue';
@@ -83,11 +88,13 @@ export interface ColumnFormItemProps {
     focus?: boolean; // 是否获取焦点
     placeholder?: string;
     columnName?: string;
+    disabled?: boolean;
 }
 
 const props = withDefaults(defineProps<ColumnFormItemProps>(), {
     focus: false,
     dataType: DataType.String,
+    disabled: false,
 });
 
 const emit = defineEmits(['update:modelValue', 'blur']);

mayfly_go_web/src/views/ops/db/component/table/DbTableData.vue

@@ -46,14 +46,6 @@
                                             <b :title="column.remark" class="el-text" style="cursor: pointer">
                                                 {{ column.title }}
                                             </b>
-
-                                            <span>
-                                                <SvgIcon
-                                                    color="var(--el-color-primary)"
-                                                    v-if="column.title == nowSortColumn?.columnName"
-                                                    :name="nowSortColumn?.order == 'asc' ? 'top' : 'bottom'"
-                                                ></SvgIcon>
-                                            </span>
                                         </div>
 
                                         <!-- 字段备注信息 -->
@@ -71,6 +63,13 @@
                                             {{ column.title }}
                                         </b>
                                     </div>
+
+                                    <!-- 字段列右部分内容 -->
+                                    <div class="column-right">
+                                        <span v-if="column.title == nowSortColumn?.columnName">
+                                            <SvgIcon color="var(--el-color-primary)" :name="nowSortColumn?.order == 'asc' ? 'top' : 'bottom'"></SvgIcon>
+                                        </span>
+                                    </div>
                                 </div>
                             </div>
                         </div>
@@ -715,9 +714,13 @@ const submitUpdateFields = async () => {
     const db = state.db;
     let res = '';
     const dbDialect = getDbDialect(dbInst.type);
-
+    let schema = '';
+    let dbArr = db.split('/');
+    if (dbArr.length == 2) {
+        schema = dbInst.wrapName(dbArr[1]) + '.';
+    }
     for (let updateRow of cellUpdateMap.values()) {
-        let sql = `UPDATE ${dbInst.wrapName(state.table)} SET `;
+        let sql = `UPDATE ${schema}${dbInst.wrapName(state.table)} SET `;
         const rowData = updateRow.rowData;
         // 主键列信息
         const primaryKey = await dbInst.loadTableColumn(db, state.table);
@@ -870,7 +873,13 @@ defineExpose({
         position: absolute;
         top: -5px;
         padding: 2px;
-        height: 12px;
+    }
+
+    .column-right {
+        position: absolute;
+        top: 2px;
+        right: 0;
+        padding: 2px;
     }
 }
 </style>

mayfly_go_web/src/views/ops/db/component/table/DbTableDataOp.vue

@@ -158,21 +158,52 @@
             @data-delete="onRefresh"
         ></db-table-data>
 
-        <el-row type="flex" class="mt5" justify="center">
-            <el-pagination
-                small
-                :total="count"
-                @size-change="handleSizeChange"
-                @current-change="pageChange()"
-                layout="prev, pager, next, total, sizes, jumper"
-                v-model:current-page="pageNum"
-                v-model:page-size="pageSize"
-                :page-sizes="pageSizes"
-            ></el-pagination>
+        <el-row type="flex" class="mt5" :gutter="10" justify="space-between" style="user-select: none">
+            <el-col :span="12">
+                <el-text
+                    id="copyValue"
+                    style="color: var(--el-color-info-light-3)"
+                    class="is-truncated font12 mt5"
+                    @click="copyToClipboard(sql)"
+                    :title="sql"
+                    >{{ sql }}</el-text
+                >
+            </el-col>
+            <el-col :span="12">
+                <el-row :gutter="10" justify="left">
+                    <el-link class="op-page" :underline="false" @click="pageNum = 1" :disabled="pageNum == 1" icon="DArrowLeft" title="首页" />
+                    <el-link class="op-page" :underline="false" @click="pageNum = --pageNum || 1" :disabled="pageNum == 1" icon="Back" title="上一页" />
+                    <div class="op-page">
+                        <el-input-number
+                            style="width: 50px"
+                            :controls="false"
+                            :min="1"
+                            v-model="state.setPageNum"
+                            size="small"
+                            @blur="handleSetPageNum"
+                            @keydown.enter="handleSetPageNum"
+                        />
+                    </div>
+                    <el-link class="op-page" :underline="false" @click="++pageNum" :disabled="datas.length < pageSize" icon="Right" />
+                    <el-link class="op-page" :underline="false" @click="handleEndPage" :disabled="datas.length < pageSize" icon="DArrowRight" />
+                    <div style="width: 90px" class="op-page ml10">
+                        <el-select size="small" :default-first-option="true" v-model="pageSize" @change="handleSizeChange">
+                            <el-option
+                                style="font-size: 12px; height: 24px; line-height: 24px"
+                                v-for="(op, i) in pageSizes"
+                                :key="i"
+                                :label="op + '条/页'"
+                                :value="op"
+                            />
+                        </el-select>
+                    </div>
+
+                    <el-button @click="handleCount" :loading="state.counting" class="ml10" text bg size="small">
+                        {{ state.showTotal ? `${state.total} 条` : 'count' }}
+                    </el-button>
+                </el-row>
+            </el-col>
         </el-row>
-        <div style="font-size: 12px; padding: 0 10px; color: #606266">
-            <span>{{ state.sql }}</span>
-        </div>
 
         <el-dialog v-model="conditionDialog.visible" :title="conditionDialog.title" width="420px">
             <el-row>
@@ -211,13 +242,14 @@
                     class="w100 mb5"
                     :prop="column.columnName"
                     :label="column.columnName"
-                    :required="column.nullable != 'YES' && column.columnKey != 'PRI'"
+                    :required="column.nullable != 'YES' && !column.isPrimaryKey && !column.isIdentity"
                 >
                     <ColumnFormItem
                         v-model="addDataDialog.data[`${column.columnName}`]"
                         :data-type="dbDialect.getDataType(column.columnType)"
                         :placeholder="`${column.columnType}  ${column.columnComment}`"
                         :column-name="column.columnName"
+                        :disabled="column.isIdentity"
                     />
                 </el-form-item>
             </el-form>
@@ -241,6 +273,7 @@ import { DbDialect, getDbDialect } from '@/views/ops/db/dialect';
 import SvgIcon from '@/components/svgIcon/index.vue';
 import ColumnFormItem from './ColumnFormItem.vue';
 import { useEventListener, useStorage } from '@vueuse/core';
+import { copyToClipboard } from '@/common/utils/string';
 
 const props = defineProps({
     dbId: {
@@ -289,7 +322,10 @@ const state = reactive({
         defaultPageSize * 40,
         defaultPageSize * 80,
     ],
-    count: 0,
+    setPageNum: 0,
+    total: 0,
+    showTotal: false,
+    counting: false,
     selectionDatas: [] as any,
     condPopVisible: false,
     columnNameSearch: '',
@@ -313,7 +349,7 @@ const state = reactive({
     dbDialect: {} as DbDialect,
 });
 
-const { datas, condition, loading, columns, pageNum, pageSize, pageSizes, count, hasUpdatedFileds, conditionDialog, addDataDialog, dbDialect } = toRefs(state);
+const { datas, condition, loading, columns, pageNum, pageSize, pageSizes, sql, hasUpdatedFileds, conditionDialog, addDataDialog, dbDialect } = toRefs(state);
 
 watch(
     () => props.tableHeight,
@@ -346,18 +382,19 @@ const onRefresh = async () => {
     await selectData();
 };
 
-/**
- * 数据tab修改页数
- */
-const pageChange = async () => {
-    await selectData();
-};
+watch(
+    () => state.pageNum,
+    async () => {
+        await selectData();
+    }
+);
 
 /**
  * 单表数据信息查询数据
  */
 const selectData = async () => {
     state.loading = true;
+    state.setPageNum = state.pageNum;
     const dbInst = getNowDbInst();
     const db = props.dbName;
     const table = props.tableName;
@@ -370,16 +407,10 @@ const selectData = async () => {
             state.columns = columns;
         }
 
-        const countRes = await dbInst.runSql(db, dbInst.getDefaultCountSql(table, state.condition));
-        state.count = parseInt(countRes.res[0].count || countRes.res[0].COUNT || 0);
-        let sql = dbInst.getDefaultSelectSql(table, state.condition, state.orderBy, state.pageNum, state.pageSize);
+        let sql = dbInst.getDefaultSelectSql(db, table, state.condition, state.orderBy, state.pageNum, state.pageSize);
         state.sql = sql;
-        if (state.count > 0) {
-            const colAndData: any = await dbInst.runSql(db, sql);
-            state.datas = colAndData.res;
-        } else {
-            state.datas = [];
-        }
+        const colAndData: any = await dbInst.runSql(db, sql);
+        state.datas = colAndData.res;
     } finally {
         state.loading = false;
     }
@@ -391,6 +422,33 @@ const handleSizeChange = async (size: any) => {
     await selectData();
 };
 
+const handleEndPage = async () => {
+    await handleCount();
+    state.pageNum = Math.ceil(state.total / state.pageSize);
+    await selectData();
+};
+
+const handleSetPageNum = async () => {
+    state.pageNum = state.setPageNum;
+    await selectData();
+};
+const handleCount = async () => {
+    state.counting = true;
+
+    try {
+        const db = props.dbName;
+        const table = props.tableName;
+        const dbInst = getNowDbInst();
+        const countRes = await dbInst.runSql(db, dbInst.getDefaultCountSql(table, state.condition));
+        state.total = parseInt(countRes.res[0].count || countRes.res[0].COUNT || 0);
+        state.showTotal = true;
+    } catch (e) {
+        /* empty */
+    }
+
+    state.counting = false;
+};
+
 // 完整的条件,每次选中后会重置条件框内容，故需要这个变量在获取建议时将文本框内容保存
 let completeCond = '';
 // 是否存在列建议
@@ -566,7 +624,13 @@ const addRow = async () => {
             }
             let columnNames = Object.keys(obj).join(',');
             let values = Object.values(obj).join(',');
-            let sql = `INSERT INTO ${dbInst.wrapName(props.tableName)} (${columnNames}) VALUES (${values});`;
+            // 获取schema
+            let schema = '';
+            let arr = props.dbName?.split('/');
+            if (arr && arr.length > 1) {
+                schema = dbInst.wrapName(arr[1]) + '.';
+            }
+            let sql = `INSERT INTO ${schema}${dbInst.wrapName(props.tableName)} (${columnNames}) VALUES (${values});`;
             dbInst.promptExeSql(props.dbName, sql, null, () => {
                 closeAddDataDialog();
                 onRefresh();
@@ -579,4 +643,8 @@ const addRow = async () => {
 };
 </script>
 
-<style lang="scss"></style>
+<style lang="scss">
+.op-page {
+    margin-left: 5px;
+}
+</style>

mayfly_go_web/src/views/ops/db/component/table/DbTableOp.vue

@@ -56,7 +56,7 @@
                                         v-else-if="item.prop === 'auto_increment'"
                                         size="small"
                                         v-model="scope.row.auto_increment"
-                                        :disabled="dbType === DbType.postgresql"
+                                        :disabled="disableEditIncr()"
                                     />
 
                                     <el-input v-else-if="item.prop === 'remark'" size="small" v-model="scope.row.remark" />
@@ -99,9 +99,7 @@
                                     <el-checkbox v-if="item.prop === 'unique'" size="small" v-model="scope.row.unique" @change="indexChanges(scope.row)">
                                     </el-checkbox>
 
-                                    <el-select v-if="item.prop === 'indexType'" disabled size="small" v-model="scope.row.indexType">
-                                        <el-option v-for="typeValue in indexTypeList" :key="typeValue" :value="typeValue">{{ typeValue }}</el-option>
-                                    </el-select>
+                                    <el-input v-if="item.prop === 'indexType'" disabled size="small" v-model="scope.row.indexType" />
 
                                     <el-input v-if="item.prop === 'indexComment'" size="small" v-model="scope.row.indexComment"> </el-input>
 
@@ -158,7 +156,7 @@ const props = defineProps({
 //定义事件
 const emit = defineEmits(['update:visible', 'cancel', 'val-change', 'submit-sql']);
 
-const dbDialect = getDbDialect(props.dbType);
+let dbDialect = getDbDialect(props.dbType);
 
 type ColName = {
     prop: string;
@@ -172,7 +170,6 @@ const state = reactive({
     btnloading: false,
     activeName: '1',
     columnTypeList: dbDialect.getInfo().columnTypes,
-    indexTypeList: ['BTREE', 'NORMAL'], // mysql索引类型详解 http://c.biancheng.net/view/7897.html
     tableData: {
         fields: {
             colNames: [
@@ -268,10 +265,11 @@ const state = reactive({
     },
 });
 
-const { dialogVisible, btnloading, activeName, indexTypeList, tableData } = toRefs(state);
+const { dialogVisible, btnloading, activeName, tableData } = toRefs(state);
 
 watch(props, async (newValue) => {
     state.dialogVisible = newValue.visible;
+    dbDialect = getDbDialect(newValue.dbType);
 });
 
 const cancel = () => {
@@ -407,7 +405,7 @@ const genSql = () => {
         } else if (state.activeName === '2') {
             // 修改索引
             let changeData = filterChangedData(state.tableData.indexs.oldIndexs, state.tableData.indexs.res, 'indexName');
-            return dbDialect.getModifyIndexSql(data.tableName, changeData);
+            return dbDialect.getModifyIndexSql(data, data.tableName, changeData);
         }
     }
 };
@@ -417,28 +415,8 @@ const reset = () => {
     formRef.value.resetFields();
     state.tableData.tableName = '';
     state.tableData.tableComment = '';
-    state.tableData.fields.res = [
-        {
-            name: '',
-            type: '',
-            value: '',
-            length: '',
-            numScale: '',
-            notNull: false,
-            pri: false,
-            auto_increment: false,
-            remark: '',
-        },
-    ];
-    state.tableData.indexs.res = [
-        {
-            indexName: '',
-            columnNames: [],
-            unique: false,
-            indexType: 'BTREE',
-            indexComment: '',
-        },
-    ];
+    state.tableData.fields.res = [];
+    state.tableData.indexs.res = [];
 };
 
 const indexChanges = (row: any) => {
@@ -459,6 +437,21 @@ const indexChanges = (row: any) => {
     row.indexComment = `${tableData.value.tableName}表(${name.replaceAll('_', ',')})${commentSuffix}`;
 };
 
+const disableEditIncr = () => {
+    if (DbType.postgresql === props.dbType) {
+        return true;
+    }
+
+    // 如果是mssql则不能修改自增
+    if (props.data?.edit) {
+        if (DbType.mssql === props.dbType) {
+            return true;
+        }
+    }
+
+    return false;
+};
+
 watch(
     () => props.data,
     (newValue: any) => {
@@ -491,8 +484,8 @@ watch(
                     length,
                     numScale: a.numScale,
                     notNull: a.nullable !== 'YES',
-                    pri: a.columnKey === 'PRI',
-                    auto_increment: a.columnKey === 'PRI' /*a.extra?.indexOf('auto_increment') > -1*/,
+                    pri: a.isPrimaryKey,
+                    auto_increment: a.isIdentity /*a.extra?.indexOf('auto_increment') > -1*/,
                     remark: a.columnComment,
                 };
                 state.tableData.fields.res.push(data);
@@ -512,7 +505,7 @@ watch(
                     let data = {
                         indexName: a.indexName,
                         columnNames: a.columnName?.split(','),
-                        unique: a.nonUnique === 0 || false,
+                        unique: a.isUnique || false,
                         indexType: a.indexType,
                         indexComment: a.indexComment,
                     };

mayfly_go_web/src/views/ops/db/component/table/DbTablesOp.vue

@@ -68,9 +68,7 @@
                 <template #default="scope">
                     <el-link @click.prevent="showColumns(scope.row)" type="primary">字段</el-link>
                     <el-link class="ml5" @click.prevent="showTableIndex(scope.row)" type="success">索引</el-link>
-                    <el-link class="ml5" v-if="tableCreateDialog.enableEditTypes.indexOf(dbType) > -1" @click.prevent="openEditTable(scope.row)" type="warning"
-                        >编辑表</el-link
-                    >
+                    <el-link class="ml5" v-if="editDbTypes.indexOf(dbType) > -1" @click.prevent="openEditTable(scope.row)" type="warning">编辑表</el-link>
                     <el-link class="ml5" @click.prevent="showCreateDdl(scope.row)" type="info">DDL</el-link>
                 </template>
             </el-table-column>
@@ -127,7 +125,7 @@ import SqlExecBox from '../sqleditor/SqlExecBox';
 import config from '@/common/config';
 import { joinClientParams } from '@/common/request';
 import { isTrue } from '@/common/assert';
-import { compatibleMysql, DbType } from '../../dialect/index';
+import { compatibleMysql, DbType, editDbTypes } from '../../dialect/index';
 
 const DbTableOp = defineAsyncComponent(() => import('./DbTableOp.vue'));
 
@@ -181,7 +179,6 @@ const state = reactive({
         visible: false,
         activeName: '1',
         type: '',
-        enableEditTypes: [DbType.mysql, DbType.mariadb, DbType.postgresql, DbType.dm, DbType.oracle, DbType.sqlite], // 支持"编辑表"的数据库类型
         data: {
             // 修改表时，传递修改数据
             edit: false,

mayfly_go_web/src/views/ops/db/db.ts

@@ -7,6 +7,10 @@ import { editor, languages, Position } from 'monaco-editor';
 
 import { registerCompletionItemProvider } from '@/components/monaco/completionItemProvider';
 import { DbDialect, EditorCompletionItem, getDbDialect } from './dialect';
+import { type RemovableRef, useLocalStorage } from '@vueuse/core';
+
+const hintsStorage: RemovableRef<Map<string, any>> = useLocalStorage('db-table-hints', new Map());
+const tableStorage: RemovableRef<Map<string, any>> = useLocalStorage('db-tables', new Map());
 
 const dbInstCache: Map<number, DbInst> = new Map();
 
@@ -58,14 +62,15 @@ export class DbInst {
         if (!dbName) {
             throw new Error('dbName不能为空');
         }
-        let db = this.dbs.get(dbName);
+        let key = `${this.id}_${dbName}`;
+        let db = this.dbs.get(key);
         if (db) {
             return db;
         }
         console.info(`new db -> dbId: ${this.id}, dbName: ${dbName}`);
         db = new Db();
         db.name = dbName;
-        this.dbs.set(dbName, db);
+        this.dbs.set(key, db);
         return db;
     }
 
@@ -77,17 +82,22 @@ export class DbInst {
      */
     async loadTables(dbName: string, reload?: boolean) {
         const db = this.getDb(dbName);
-        // 优先从 table map中获取
-        let tables = db.tables;
+        let key = this.dbTablesKey(dbName);
+        let tables = tableStorage.value.get(key);
+        // 优先从 table 缓存中获取
         if (!reload && tables) {
+            db.tables = tables;
             return tables;
         }
         // 重置列信息缓存与表提示信息
         db.columnsMap?.clear();
-        db.tableHints = null;
         console.log(`load tables -> dbName: ${dbName}`);
         tables = await dbApi.tableInfos.request({ id: this.id, db: dbName });
+        tableStorage.value.set(key, tables);
         db.tables = tables;
+
+        // 异步加载表提示信息
+        this.loadDbHints(dbName, true).then(() => {});
         return tables;
     }
 
@@ -169,18 +179,30 @@ export class DbInst {
         return this.getDb(dbName).getColumn(table, columnName);
     }
 
+    dbTableHintsKey(dbName: string) {
+        return `db-table-hints_${this.id}_${dbName}`;
+    }
+
+    dbTablesKey(dbName: string) {
+        return `db-tables_${this.id}_${dbName}`;
+    }
+
     /**
      * 获取库信息提示
      */
-    async loadDbHints(dbName: string) {
+    async loadDbHints(dbName: string, reload?: boolean) {
         const db = this.getDb(dbName);
-        if (db.tableHints) {
-            return db.tableHints;
+        let key = this.dbTableHintsKey(dbName);
+        let hints = hintsStorage.value.get(key);
+        if (!reload && hints) {
+            db.tableHints = hints;
+            return hints;
         }
         console.log(`load db-hits -> dbName: ${dbName}`);
-        const hits = await dbApi.hintTables.request({ id: this.id, db: db.name });
-        db.tableHints = hits;
-        return hits;
+        hints = await dbApi.hintTables.request({ id: this.id, db: db.name });
+        db.tableHints = hints;
+        hintsStorage.value.set(key, hints);
+        return hints;
     }
 
     /**
@@ -225,8 +247,8 @@ export class DbInst {
     };
 
     // 获取指定表的默认查询sql
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number = DbInst.DefaultLimit) {
-        return getDbDialect(this.type).getDefaultSelectSql(table, condition, orderBy, pageNum, limit);
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number = DbInst.DefaultLimit) {
+        return getDbDialect(this.type).getDefaultSelectSql(db, table, condition, orderBy, pageNum, limit);
     }
 
     /**
@@ -275,6 +297,7 @@ export class DbInst {
             sql,
             dbId: this.id,
             db,
+            dbType: getDbDialect(this.type).getInfo().formatSqlDialect,
             runSuccessCallback: successFunc,
             cancelCallback: cancelFunc,
         });
@@ -363,7 +386,7 @@ export class DbInst {
             return value;
         }
         if (!dbDialect) {
-            return `${value}`;
+            return `'${value}'`;
         }
         return dbDialect.wrapStrValue(columnType, value);
     }
@@ -441,7 +464,7 @@ class Db {
     getColumn(table: string, columnName: string = '') {
         const cols = this.getColumns(table);
         if (!columnName) {
-            const col = cols.find((c: any) => c.columnKey == 'PRI');
+            const col = cols.find((c: any) => c.isPrimaryKey);
             return col || cols[0];
         }
         return cols.find((c: any) => c.columnName == columnName);

mayfly_go_web/src/views/ops/db/dialect/dm_dialect.ts

@@ -366,6 +366,7 @@ class DMDialect implements DbDialect {
         };
 
         dmDialectInfo = {
+            name: 'DM',
             icon: 'iconfont icon-db-dm',
             defaultPort: 5236,
             formatSqlDialect: 'plsql',
@@ -375,12 +376,12 @@ class DMDialect implements DbDialect {
         return dmDialectInfo;
     }
 
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
         return `SELECT * FROM "${table}" ${condition ? 'WHERE ' + condition : ''} ${orderBy ? orderBy : ''} ${this.getPageSql(pageNum, limit)};`;
     }
 
     getPageSql(pageNum: number, limit: number) {
-        return ` OFFSET ${(pageNum - 1) * limit} LIMIT ${limit};`;
+        return ` OFFSET ${(pageNum - 1) * limit} LIMIT ${limit}`;
     }
 
     getDefaultRows(): RowDefinition[] {
@@ -620,7 +621,7 @@ class DMDialect implements DbDialect {
         return dropPkSql + modifySql + dropSql + renameSql + addPkSql + commentSql;
     }
 
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
         // 不能直接修改索引名或字段、需要先删后加
         let dropIndexNames: string[] = [];
         let addIndexs: any[] = [];

mayfly_go_web/src/views/ops/db/dialect/gauss_dialect.ts

@@ -0,0 +1,17 @@
+import { PostgresqlDialect } from '@/views/ops/db/dialect/postgres_dialect';
+import { DialectInfo } from '@/views/ops/db/dialect/index';
+
+let gsDialectInfo: DialectInfo;
+export class GaussDialect extends PostgresqlDialect {
+    getInfo(): DialectInfo {
+        if (gsDialectInfo) {
+            return gsDialectInfo;
+        }
+
+        gsDialectInfo = {} as DialectInfo;
+        Object.assign(gsDialectInfo, super.getInfo());
+        gsDialectInfo.icon = 'iconfont icon-gauss';
+        gsDialectInfo.name = 'GaussDB';
+        return gsDialectInfo;
+    }
+}

mayfly_go_web/src/views/ops/db/dialect/index.ts

@@ -4,6 +4,8 @@ import { DMDialect } from '@/views/ops/db/dialect/dm_dialect';
 import { OracleDialect } from '@/views/ops/db/dialect/oracle_dialect';
 import { MariadbDialect } from '@/views/ops/db/dialect/mariadb_dialect';
 import { SqliteDialect } from '@/views/ops/db/dialect/sqlite_dialect';
+import { MssqlDialect } from '@/views/ops/db/dialect/mssql_dialect';
+import { GaussDialect } from '@/views/ops/db/dialect/gauss_dialect';
 
 export interface sqlColumnType {
     udtName: string;
@@ -80,6 +82,11 @@ export const ColumnTypeSubscript = {
 
 // 数据库基础信息
 export interface DialectInfo {
+    /**
+     * 数据库类型label
+     */
+    name: string;
+
     /**
      * 图标
      */
@@ -110,11 +117,21 @@ export const DbType = {
     mysql: 'mysql',
     mariadb: 'mariadb',
     postgresql: 'postgres',
+    gauss: 'gauss',
     dm: 'dm', // 达梦
     oracle: 'oracle',
     sqlite: 'sqlite',
+    mssql: 'mssql', // ms sqlserver
 };
 
+// mysql兼容的数据库
+export const noSchemaTypes = [DbType.mysql, DbType.mariadb, DbType.sqlite];
+
+// 有schema层的数据库
+export const schemaDbTypes = [DbType.postgresql, DbType.gauss, DbType.dm, DbType.oracle, DbType.mssql];
+
+export const editDbTypes = [...noSchemaTypes, ...schemaDbTypes];
+
 export const compatibleMysql = (dbType: string): boolean => {
     switch (dbType) {
         case DbType.mysql:
@@ -133,13 +150,14 @@ export interface DbDialect {
 
     /**
      * 获取默认查询sql
+     * @param db  数据库信息
      * @param table  表名
      * @param condition 条件
      * @param orderBy 排序
      * @param pageNum  页数
      * @param limit  条数
      */
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number): string;
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number): string;
 
     getPageSql(pageNum: number, limit: number): string;
 
@@ -175,10 +193,11 @@ export interface DbDialect {
 
     /**
      * 生成编辑索引sql
+     * @param tableData 表数据，包含表名、列数据、索引数据
      * @param tableName   表名
      * @param changeData  改变数据
      */
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string;
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string;
 
     /** 通过数据库字段类型，返回基本数据类型 */
     getDataType(columnType: string): DataType;
@@ -188,30 +207,29 @@ export interface DbDialect {
 }
 
 let mysqlDialect = new MysqlDialect();
-let mariadbDialect = new MariadbDialect();
-let postgresDialect = new PostgresqlDialect();
-let dmDialect = new DMDialect();
-let oracleDialect = new OracleDialect();
-let sqliteDialect = new SqliteDialect();
 
-export const getDbDialect = (dbType: string | undefined): DbDialect => {
-    if (!dbType) {
-        return mysqlDialect;
-    }
-    switch (dbType) {
-        case DbType.mysql:
-            return mysqlDialect;
-        case DbType.mariadb:
-            return mariadbDialect;
-        case DbType.postgresql:
-            return postgresDialect;
-        case DbType.dm:
-            return dmDialect;
-        case DbType.oracle:
-            return oracleDialect;
-        case DbType.sqlite:
-            return sqliteDialect;
-        default:
-            throw new Error('不支持的数据库');
-    }
+let dbType2DialectMap: Map<string, DbDialect> = new Map();
+
+export const registerDbDialect = (dbType: string, dd: DbDialect) => {
+    dbType2DialectMap.set(dbType, dd);
 };
+
+export const getDbDialectMap = () => {
+    return dbType2DialectMap;
+};
+
+export const getDbDialect = (dbType: string): DbDialect => {
+    return dbType2DialectMap.get(dbType) || mysqlDialect;
+};
+
+(function () {
+    console.log('init register db dialect');
+    registerDbDialect(DbType.mysql, mysqlDialect);
+    registerDbDialect(DbType.mariadb, new MariadbDialect());
+    registerDbDialect(DbType.postgresql, new PostgresqlDialect());
+    registerDbDialect(DbType.gauss, new GaussDialect());
+    registerDbDialect(DbType.dm, new DMDialect());
+    registerDbDialect(DbType.oracle, new OracleDialect());
+    registerDbDialect(DbType.sqlite, new SqliteDialect());
+    registerDbDialect(DbType.mssql, new MssqlDialect());
+})();

mayfly_go_web/src/views/ops/db/dialect/mariadb_dialect.ts

@@ -12,6 +12,7 @@ class MariadbDialect extends MysqlDialect implements DbDialect {
 
         mariadbDialectInfo = {} as DialectInfo;
         Object.assign(mariadbDialectInfo, super.getInfo());
+        mariadbDialectInfo.name = 'MariaDB';
         mariadbDialectInfo.icon = 'iconfont icon-mariadb';
         return mariadbDialectInfo;
     }

mayfly_go_web/src/views/ops/db/dialect/mssql_dialect.ts

@@ -0,0 +1,405 @@
+import { DbInst } from '../db';
+import { commonCustomKeywords, DataType, DbDialect, DialectInfo, EditorCompletion, EditorCompletionItem, IndexDefinition, RowDefinition } from './index';
+import { language as sqlLanguage } from 'monaco-editor/esm/vs/basic-languages/sql/sql.js';
+
+export { MSSQL_TYPE_LIST, MssqlDialect };
+
+// 参考官方文档：https://docs.microsoft.com/zh-cn/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver15
+const MSSQL_TYPE_LIST = [
+    //精确数字
+    'bigint',
+    'numeric',
+    'bit',
+    'smallint',
+    'decimal',
+    'smallmoney',
+    'int',
+    'tinyint',
+    'money',
+    // 近似数字
+    'float',
+    'real',
+    // 日期和时间
+    'date',
+    'datetimeoffset',
+    'datetime2',
+    'smalldatetime',
+    'datetime',
+    'time',
+    // 字符串
+    'char',
+    'varchar',
+    'text',
+    'nchar',
+    'nvarchar',
+    'ntext',
+    'binary',
+    'varbinary',
+
+    // 其他
+    'cursor',
+    'rowversion',
+    'hierarchyid',
+    'uniqueidentifier',
+    'sql_variant',
+    'xml',
+    'table',
+    // 空间几何类型 参照 https://learn.microsoft.com/zh-cn/sql/t-sql/spatial-geometry/spatial-types-geometry-transact-sql?view=sql-server-ver15
+    'geometry',
+    // 空间地理类型 参照 https://learn.microsoft.com/zh-cn/sql/t-sql/spatial-geography/spatial-types-geography?view=sql-server-ver15
+    'geography',
+];
+// 函数参考官方文档 https://learn.microsoft.com/zh-cn/sql/t-sql/functions/functions?view=sql-server-ver15
+
+let mssqlDialectInfo: DialectInfo;
+
+const customKeywords: EditorCompletionItem[] = [
+    {
+        label: 'select top ',
+        description: 'keyword',
+        insertText: 'select top 100 * from',
+    },
+    {
+        label: 'select page ',
+        description: 'keyword',
+        insertText: 'SELECT *, 0 AS _ORDER_F_ FROM table_name \n ORDER BY _ORDER_F_ \n OFFSET 0 ROWS FETCH NEXT 25 ROWS ONLY;',
+    },
+];
+
+const fixedLengthTypes = [
+    'int',
+    'bigint',
+    'smallint',
+    'tinyint',
+    'float',
+    'real',
+    'datetime',
+    'smalldatetime',
+    'date',
+    'time',
+    'datetime2',
+    'datetimeoffset',
+    'bit',
+    'uniqueidentifier',
+    'geometry',
+    'geography',
+];
+
+class MssqlDialect implements DbDialect {
+    getInfo(): DialectInfo {
+        if (mssqlDialectInfo) {
+            return mssqlDialectInfo;
+        }
+
+        let { keywords, operators, builtinVariables, builtinFunctions } = sqlLanguage;
+        let functions = builtinFunctions.map((a: string): EditorCompletionItem => ({ label: a, insertText: `${a}()`, description: 'func' }));
+
+        let excludeKeywords = new Set(operators);
+        let editorCompletions: EditorCompletion = {
+            keywords: keywords
+                .filter((a: string) => !excludeKeywords.has(a)) // 移除已存在的operator、function
+                .map((a: string): EditorCompletionItem => ({ label: a, description: 'keyword' }))
+                .concat(customKeywords)
+                .concat(commonCustomKeywords.map((a): EditorCompletionItem => ({ label: a, description: 'keyword' }))),
+            operators: operators.map((a: string): EditorCompletionItem => ({ label: a, description: 'operator' })),
+            functions,
+            variables: builtinVariables.map((a: string): EditorCompletionItem => ({ label: a, description: 'var' })),
+        };
+
+        mssqlDialectInfo = {
+            name: 'MSSQL',
+            icon: 'iconfont icon-MSSQLNATIVE',
+            defaultPort: 1433,
+            formatSqlDialect: 'transactsql',
+            columnTypes: MSSQL_TYPE_LIST.map((a) => ({ udtName: a, dataType: a, desc: '', space: '' })),
+            editorCompletions,
+        };
+        return mssqlDialectInfo;
+    }
+
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+        let schema = db.split('/')[1];
+        return `SELECT *, 0 AS _MAY_ORDER_F_ FROM ${this.quoteIdentifier(schema)}.${this.quoteIdentifier(table)} ${condition ? 'WHERE ' + condition : ''} ${
+            orderBy ? orderBy + ', _MAY_ORDER_F_' : 'order by _MAY_ORDER_F_'
+        } ${this.getPageSql(pageNum, limit)};`.toUpperCase();
+    }
+
+    getPageSql(pageNum: number, limit: number) {
+        return ` offset ${(pageNum - 1) * limit} rows fetch next ${limit} rows only`.toUpperCase();
+    }
+
+    getDefaultRows(): RowDefinition[] {
+        return [
+            { name: 'id', type: 'bigint', length: '', numScale: '', value: '', notNull: true, pri: true, auto_increment: true, remark: '主键ID' },
+            { name: 'creator_id', type: 'bigint', length: '20', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '创建人id' },
+            {
+                name: 'creator',
+                type: 'varchar',
+                length: '100',
+                numScale: '',
+                value: '',
+                notNull: true,
+                pri: false,
+                auto_increment: false,
+                remark: '创建人姓名',
+            },
+            {
+                name: 'create_time',
+                type: 'datetime2',
+                length: '',
+                numScale: '',
+                value: 'CURRENT_TIMESTAMP',
+                notNull: true,
+                pri: false,
+                auto_increment: false,
+                remark: '创建时间',
+            },
+            { name: 'updator_id', type: 'bigint', length: '20', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '修改人id' },
+            {
+                name: 'updator',
+                type: 'varchar',
+                length: '100',
+                numScale: '',
+                value: '',
+                notNull: true,
+                pri: false,
+                auto_increment: false,
+                remark: '修改人姓名',
+            },
+            {
+                name: 'update_time',
+                type: 'datetime2',
+                length: '',
+                numScale: '',
+                value: 'CURRENT_TIMESTAMP',
+                notNull: true,
+                pri: false,
+                auto_increment: false,
+                remark: '修改时间',
+            },
+        ];
+    }
+
+    getDefaultIndex(): IndexDefinition {
+        return {
+            indexName: '',
+            columnNames: [],
+            unique: false,
+            indexType: 'NONCLUSTERED',
+            indexComment: '',
+        };
+    }
+
+    quoteIdentifier = (name: string) => {
+        return `[${name}]`;
+    };
+
+    genColumnBasicSql(cl: any): string {
+        let val = cl.value ? (cl.value === 'CURRENT_TIMESTAMP' ? cl.value : `'${cl.value}'`) : '';
+        let defVal = val ? `DEFAULT ${val}` : '';
+        // mssql哪些字段允许有长度
+        let length = '';
+        if (!fixedLengthTypes.includes(cl.type)) {
+            length = cl.length ? `(${cl.length})` : '';
+        }
+        return ` ${this.quoteIdentifier(cl.name)} ${cl.type}${length} ${cl.auto_increment ? 'IDENTITY(1,1)' : ''} ${defVal} ${cl.notNull ? 'NOT NULL' : 'NULL'} `;
+    }
+    getCreateTableSql(data: any): string {
+        let schema = data.db.split('/')[1];
+
+        // 创建表结构
+        let pks = [] as string[];
+        let fields: string[] = [];
+        let fieldComments: string[] = [];
+        data.fields.res.forEach((item: any) => {
+            item.name && fields.push(this.genColumnBasicSql(item));
+            item.remark &&
+                fieldComments.push(
+                    `EXECUTE sp_addextendedproperty N'MS_Description', N'${item.remark}', N'SCHEMA', N'${schema}', N'TABLE', N'${data.tableName}', N'COLUMN', N'${item.name}'`
+                );
+            if (item.pri) {
+                pks.push(`${this.quoteIdentifier(item.name)}`);
+            }
+        });
+
+        let baseTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(data.tableName)}`;
+
+        // 建表语句
+        let createTable = `CREATE TABLE ${baseTable}
+                ( ${fields.join(',')}
+                  ${pks.length > 0 ? `, PRIMARY KEY CLUSTERED (${pks.join(',')})` : ''}
+                );`;
+
+        let createIndexSql = this.getCreateIndexSql(data);
+
+        // 表注释
+        if (data.tableComment) {
+            createTable += ` EXECUTE sp_addextendedproperty N'MS_Description', N'${data.tableComment}', N'SCHEMA', N'${schema}', N'TABLE', N'${data.tableName}';`;
+        }
+
+        return createTable + createIndexSql + fieldComments.join(';');
+    }
+
+    getCreateIndexSql(data: any): string {
+        // CREATE UNIQUE NONCLUSTERED INDEX [aaa]
+        // ON [dbo].[无标题] (
+        //   [id],
+        //   [name]
+        // )
+        let schema = data.db.split('/')[1];
+        let baseTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(data.tableName)}`;
+
+        let indexComment = [] as string[];
+
+        // 创建索引
+        let sql: string[] = [];
+        data.indexs.res.forEach((a: any) => {
+            let columnNames = a.columnNames.map((b: string) => `${this.quoteIdentifier(b)}`);
+            sql.push(` CREATE ${a.unique ? 'UNIQUE' : ''} NONCLUSTERED INDEX ${this.quoteIdentifier(a.indexName)} on ${baseTable} (${columnNames.join(',')})`);
+            if (a.indexComment) {
+                indexComment.push(
+                    `EXECUTE sp_addextendedproperty N'MS_Description', N'${a.indexComment}', N'SCHEMA', N'${schema}', N'TABLE', N'${data.tableName}', N'INDEX', N'${a.indexName}'`
+                );
+            }
+        });
+
+        return sql.join(';') + ';' + indexComment.join(';');
+    }
+
+    getModifyColumnSql(tableData: any, tableName: string, changeData: { del: RowDefinition[]; add: RowDefinition[]; upd: RowDefinition[] }): string {
+        // sql执行顺序
+        // 1. 删除字段
+        // 2. 添加字段
+        // 3. 修改字段名字
+        // 4. 修改字段类型
+        // 5. 修改字段注释
+        // 6. 添加字段注释
+
+        let schema = tableData.db.split('/')[1];
+        let baseTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableName)}`;
+
+        let delSql = '';
+        let addArr = [] as string[];
+        let renameArr = [] as string[];
+        let updArr = [] as string[];
+        let changeCommentArr = [] as string[];
+        let addCommentArr = [] as string[];
+
+        if (changeData.del.length > 0) {
+            delSql = `ALTER TABLE ${baseTable} DROP ${changeData.del.map((a) => 'COLUMN ' + this.quoteIdentifier(a.name)).join(',')};`;
+        }
+        if (changeData.add.length > 0) {
+            changeData.add.forEach((a) => {
+                addArr.push(` ALTER TABLE ${baseTable} ADD COLUMN ${this.genColumnBasicSql(a)}`);
+                if (a.remark) {
+                    addCommentArr.push(
+                        `EXECUTE sp_addextendedproperty N'MS_Description', N'${a.remark}', N'SCHEMA', N'${schema}', N'TABLE', N'${tableName}', N'COLUMN', N'${a.name}'`
+                    );
+                }
+            });
+        }
+
+        if (changeData.upd.length > 0) {
+            changeData.upd.forEach((a) => {
+                if (a.oldName && a.name !== a.oldName) {
+                    renameArr.push(` EXEC sp_rename '${baseTable}.${this.quoteIdentifier(a.oldName)}', '${a.name}', 'COLUMN' `);
+                } else {
+                    updArr.push(` ALTER TABLE ${baseTable} ALTER COLUMN ${this.genColumnBasicSql(a)} `);
+                }
+                if (a.remark) {
+                    changeCommentArr.push(`IF ((SELECT COUNT(*) FROM fn_listextendedproperty('MS_Description',
+'SCHEMA', N'${schema}',
+'TABLE', N'${tableName}',
+'COLUMN', N'${a.name}')) > 0)
+  EXEC sp_updateextendedproperty
+'MS_Description', N'${a.remark}',
+'SCHEMA', N'${schema}',
+'TABLE', N'${tableName}',
+'COLUMN', N'${a.name}'
+ELSE
+  EXEC sp_addextendedproperty
+'MS_Description', N'${a.remark}',
+'SCHEMA', N'${schema}',
+'TABLE', N'${tableName}',
+'COLUMN',N'${a.name}'`);
+                }
+            });
+        }
+
+        return (
+            delSql +
+            (addArr.length > 0 ? addArr.join(';') + ';' : '') +
+            (renameArr.length > 0 ? renameArr.join(';') + ';' : '') +
+            (updArr.length > 0 ? updArr.join(';') + ';' : '') +
+            (changeCommentArr.length > 0 ? changeCommentArr.join(';') + ';' : '') +
+            (addCommentArr.length > 0 ? addCommentArr.join(';') + ';' : '')
+        );
+    }
+
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+        let schema = tableData.db.split('/')[1];
+        let baseTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableName)}`;
+
+        let dropArr = [] as string[];
+        let addArr = [] as string[];
+        let commentArr = [] as string[];
+
+        const pushDrop = (a: any) => {
+            dropArr.push(` DROP INDEX ${this.quoteIdentifier(a.indexName)} ON ${baseTable} `);
+        };
+        const pushAdd = (a: any) => {
+            addArr.push(
+                ` CREATE ${a.unique ? 'UNIQUE' : ''} NONCLUSTERED INDEX ${this.quoteIdentifier(a.indexName)} ON ${baseTable} (${a.columnNames.map((b: string) => this.quoteIdentifier(b)).join(',')}) `
+            );
+            if (a.indexComment) {
+                commentArr.push(
+                    ` EXEC sp_addextendedproperty N'MS_Description', N'${a.indexComment}', N'SCHEMA', N'${schema}', N'TABLE', N'${tableName}', N'INDEX', N'${a.indexName}' `
+                );
+            }
+        };
+
+        if (changeData.upd.length > 0) {
+            changeData.upd.forEach((a) => {
+                pushDrop(a);
+                pushAdd(a);
+            });
+        }
+
+        if (changeData.del.length > 0) {
+            changeData.del.forEach((a) => {
+                pushDrop(a);
+            });
+        }
+
+        if (changeData.add.length > 0) {
+            changeData.add.forEach((a) => pushAdd(a));
+        }
+        let dropSql = dropArr.join(';');
+        let addSql = addArr.join(';');
+        let commentSql = commentArr.join(';');
+        return dropSql + ';' + addSql + ';' + commentSql + ';';
+    }
+
+    getDataType(columnType: string): DataType {
+        if (DbInst.isNumber(columnType)) {
+            return DataType.Number;
+        }
+        // 日期时间类型
+        if (/datetime|timestamp/gi.test(columnType)) {
+            return DataType.DateTime;
+        }
+        // 日期类型
+        if (/date/gi.test(columnType)) {
+            return DataType.Date;
+        }
+        // 时间类型
+        if (/time/gi.test(columnType)) {
+            return DataType.Time;
+        }
+        return DataType.String;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-unused-vars
+    wrapStrValue(columnType: string, value: string): string {
+        return `'${value}'`;
+    }
+}

mayfly_go_web/src/views/ops/db/dialect/mysql_dialect.ts

@@ -104,6 +104,7 @@ class MysqlDialect implements DbDialect {
         };
 
         mysqlDialectInfo = {
+            name: 'MySQL',
             icon: 'iconfont icon-op-mysql',
             defaultPort: 3306,
             formatSqlDialect: 'mysql',
@@ -113,7 +114,7 @@ class MysqlDialect implements DbDialect {
         return mysqlDialectInfo;
     }
 
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
         return `SELECT * FROM ${this.quoteIdentifier(table)} ${condition ? 'WHERE ' + condition : ''} ${orderBy ? orderBy : ''} ${this.getPageSql(
             pageNum,
             limit
@@ -252,7 +253,7 @@ class MysqlDialect implements DbDialect {
         return sql + arr.join(',') + ';';
     }
 
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
         // 搜集修改和删除的索引，添加到drop index xx
         // 收集新增和修改的索引，添加到ADD xx
         // ALTER TABLE `test1`

mayfly_go_web/src/views/ops/db/dialect/oracle_dialect.ts

@@ -92,7 +92,35 @@ const replaceFunctions: EditorCompletionItem[] = [
     { label: 'NVL2', insertText: 'NVL2(x,value1,value2)', description: '如果x非空，返回value1，否则返回value2' },
 ];
 
-const addCustomKeywords = ['ROWNUM', 'DUAL'];
+const addCustomKeywords: EditorCompletionItem[] = [
+    {
+        label: 'ROWNUM',
+        description: 'keyword',
+        insertText: 'ROWNUM',
+    },
+    {
+        label: 'DUAL',
+        description: 'keyword',
+        insertText: 'DUAL',
+    },
+    // 分页代码块
+    {
+        label: 'SELECT ROWNUM',
+        description: 'code block',
+        insertText: 'SELECT * from table_name where rownum <= 10',
+    },
+    {
+        label: 'SELECT PAGE',
+        description: 'code block',
+        insertText: ` SELECT * FROM
+    (
+      SELECT t.*, ROWNUM AS rn
+      FROM table_name t
+      WHERE ROWNUM <= 25
+    )
+  WHERE rn > 0 \n`,
+    },
+];
 
 let oracleDialectInfo: DialectInfo;
 class OracleDialect implements DbDialect {
@@ -104,6 +132,7 @@ class OracleDialect implements DbDialect {
         let { keywords, operators, builtinVariables } = sqlLanguage;
         let functionNames = replaceFunctions.map((a) => a.label);
         let excludeKeywords = new Set(functionNames.concat(operators));
+        excludeKeywords.add('SELECT');
 
         let editorCompletions: EditorCompletion = {
             keywords: keywords
@@ -118,21 +147,14 @@ class OracleDialect implements DbDialect {
                         })
                     )
                 )
-                .concat(
-                    // 加上自定义的关键字
-                    addCustomKeywords.map(
-                        (a): EditorCompletionItem => ({
-                            label: a,
-                            description: 'keyword',
-                        })
-                    )
-                ),
+                .concat(addCustomKeywords),
             operators: operators.map((a: string): EditorCompletionItem => ({ label: a, description: 'operator' })),
             functions: replaceFunctions,
             variables: builtinVariables.map((a: string): EditorCompletionItem => ({ label: a, description: 'var' })),
         };
 
         oracleDialectInfo = {
+            name: 'Oracle',
             icon: 'iconfont icon-oracle',
             defaultPort: 1521,
             formatSqlDialect: 'plsql',
@@ -142,7 +164,7 @@ class OracleDialect implements DbDialect {
         return oracleDialectInfo;
     }
 
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
         return `
         SELECT *
         FROM (
@@ -399,7 +421,7 @@ class OracleDialect implements DbDialect {
         return dropPkSql + modifySql + dropSql + renameSql + addPkSql + commentSql;
     }
 
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
         // 不能直接修改索引名或字段、需要先删后加
         let dropIndexNames: string[] = [];
         let addIndexs: any[] = [];

mayfly_go_web/src/views/ops/db/dialect/postgres_dialect.ts

@@ -123,6 +123,7 @@ class PostgresqlDialect implements DbDialect {
         };
 
         pgDialectInfo = {
+            name: 'PostgreSQL',
             icon: 'iconfont icon-op-postgres',
             defaultPort: 5432,
             formatSqlDialect: 'postgresql',
@@ -132,7 +133,7 @@ class PostgresqlDialect implements DbDialect {
         return pgDialectInfo;
     }
 
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
         return `SELECT * FROM ${this.quoteIdentifier(table)} ${condition ? 'WHERE ' + condition : ''} ${orderBy ? orderBy : ''} ${this.getPageSql(
             pageNum,
             limit
@@ -228,7 +229,7 @@ class PostgresqlDialect implements DbDialect {
             let marks = false;
             if (this.matchType(cl.type, ['char', 'time', 'date', 'text'])) {
                 // 默认值是now()的time或date不需要加引号
-                if (cl.value.toLowerCase() === 'pg_systimestamp()' && this.matchType(cl.type, ['time', 'date'])) {
+                if (['pg_systimestamp()', 'current_timestamp'].includes(cl.value.toLowerCase()) && this.matchType(cl.type, ['time', 'date'])) {
                     marks = false;
                 } else {
                     marks = true;
@@ -365,7 +366,7 @@ class PostgresqlDialect implements DbDialect {
         return dropPkSql + modifySql + dropSql + renameSql + addPkSql + commentSql;
     }
 
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
         // 不能直接修改索引名或字段、需要先删后加
         let dropIndexNames: string[] = [];
         let addIndexs: any[] = [];

mayfly_go_web/src/views/ops/db/dialect/sqlite_dialect.ts

@@ -123,6 +123,7 @@ class SqliteDialect implements DbDialect {
         };
 
         sqliteDialectInfo = {
+            name: 'Sqlite',
             icon: 'iconfont icon-sqlite',
             defaultPort: 0,
             formatSqlDialect: 'sql',
@@ -132,7 +133,7 @@ class SqliteDialect implements DbDialect {
         return sqliteDialectInfo;
     }
 
-    getDefaultSelectSql(table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
+    getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) {
         return `SELECT * FROM ${this.quoteIdentifier(table)} ${condition ? 'WHERE ' + condition : ''} ${orderBy ? orderBy : ''} ${this.getPageSql(
             pageNum,
             limit
@@ -145,7 +146,7 @@ class SqliteDialect implements DbDialect {
 
     getDefaultRows(): RowDefinition[] {
         return [
-            { name: 'id', type: 'bigint', length: '20', numScale: '', value: '', notNull: true, pri: true, auto_increment: true, remark: '主键ID' },
+            { name: 'id', type: 'integer', length: '', numScale: '', value: '', notNull: true, pri: true, auto_increment: true, remark: '主键ID' },
             { name: 'creator_id', type: 'bigint', length: '20', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '创建人id' },
             {
                 name: 'creator',
@@ -284,7 +285,7 @@ class SqliteDialect implements DbDialect {
         return sql.join(';') + ';';
     }
 
-    getModifyIndexSql(tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
+    getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string {
         // sqlite创建索引需要先删除再创建
         // CREATE INDEX "main"."aa1" ON "t_sys_resource" ( "ui_path" );
 

mayfly_go_web/vite.config.ts

@@ -2,6 +2,7 @@ import vue from '@vitejs/plugin-vue';
 import { resolve } from 'path';
 import type { UserConfig } from 'vite';
 import { loadEnv } from './src/common/utils/viteBuild';
+import { CodeInspectorPlugin } from 'code-inspector-plugin';
 
 const pathResolve = (dir: string): any => {
     return resolve(__dirname, '.', dir);
@@ -14,7 +15,12 @@ const alias: Record<string, string> = {
 };
 
 const viteConfig: UserConfig = {
-    plugins: [vue()],
+    plugins: [
+        vue(),
+        CodeInspectorPlugin({
+            bundler: 'vite',
+        }),
+    ],
     root: process.cwd(),
     resolve: {
         alias,

server/config.yml.example

@@ -37,22 +37,19 @@ sqlite:
 #   password: 111049
 #   db: 0
 log:
-   # 日志等级, debug, info, warn, error
+  # 日志等级, debug, info, warn, error
   level: info
   # 日志格式类型, text/json
   type: text
   # 是否记录方法调用栈信息
   add-source: false
+  # 日志文件配置
   # file:
-  #   path: ./
+  #   path: ./log
   #   name: mayfly-go.log
-db:
-  backup-path: ./backup
-  mysqlutil-path:
-    mysql: ./mysqlutil/bin/mysql
-    mysqldump: ./mysqlutil/bin/mysqldump
-    mysqlbinlog: ./mysqlutil/bin/mysqlbinlog
-  mariadbutil-path:
-    mysql: ./mariadbutil/bin/mariadb
-    mysqldump: ./mariadbutil/bin/mariadb-dump
-    mysqlbinlog: ./mariadbutil/bin/mariadb-binlog
+  #   # 日志文件的最大大小（以兆字节为单位）。当日志文件大小达到该值时，将触发切割操作
+  #   max-size: 500
+  #   # 根据文件名中的时间戳，设置保留旧日志文件的最大天数
+  #   max-age: 60
+  #   # 是否使用 gzip 压缩方式压缩轮转后的日志文件
+  #   compress: true

server/go.mod

@@ -10,31 +10,34 @@ require (
 	github.com/gin-gonic/gin v1.9.1
 	github.com/glebarez/sqlite v1.10.0
 	github.com/go-gormigrate/gormigrate/v2 v2.1.0
-	github.com/go-ldap/ldap/v3 v3.4.5
+	github.com/go-ldap/ldap/v3 v3.4.6
 	github.com/go-playground/locales v0.14.1
 	github.com/go-playground/universal-translator v0.18.1
 	github.com/go-playground/validator/v10 v10.14.0
 	github.com/go-sql-driver/mysql v1.7.1
 	github.com/golang-jwt/jwt/v5 v5.2.0
-	github.com/google/uuid v1.3.0
+	github.com/google/uuid v1.3.1
 	github.com/gorilla/websocket v1.5.1
 	github.com/kanzihuang/vitess/go/vt/sqlparser v0.0.0-20231018071450-ac8d9f0167e9
 	github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230712084735-068dc2aee82d
+	github.com/microsoft/go-mssqldb v1.6.0
 	github.com/mojocn/base64Captcha v1.3.6 // 验证码
 	github.com/pkg/errors v0.9.1
 	github.com/pkg/sftp v1.13.6
 	github.com/pquerna/otp v1.4.0
 	github.com/redis/go-redis/v9 v9.4.0
 	github.com/robfig/cron/v3 v3.0.1 // 定时任务
-	github.com/sijms/go-ora/v2 v2.8.6
+	github.com/sijms/go-ora/v2 v2.8.7
 	github.com/stretchr/testify v1.8.4
 	go.mongodb.org/mongo-driver v1.13.1 // mongo
 	golang.org/x/crypto v0.18.0 // ssh
 	golang.org/x/oauth2 v0.15.0
+	golang.org/x/sync v0.6.0
+	gopkg.in/natefinch/lumberjack.v2 v2.2.1
 	gopkg.in/yaml.v3 v3.0.1
 	// gorm
 	gorm.io/driver/mysql v1.5.2
-	gorm.io/gorm v1.25.5
+	gorm.io/gorm v1.25.6
 
 )
 
@@ -50,8 +53,10 @@ require (
 	github.com/gabriel-vasile/mimetype v1.4.2 // indirect
 	github.com/gin-contrib/sse v0.1.0 // indirect
 	github.com/glebarez/go-sqlite v1.21.2 // indirect
-	github.com/go-asn1-ber/asn1-ber v1.5.4 // indirect
+	github.com/go-asn1-ber/asn1-ber v1.5.5 // indirect
 	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 // indirect
+	github.com/golang-sql/sqlexp v0.1.0 // indirect
 	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
 	github.com/golang/glog v1.0.0 // indirect
 	github.com/golang/protobuf v1.5.3 // indirect
@@ -80,10 +85,9 @@ require (
 	github.com/xdg-go/stringprep v1.0.4 // indirect
 	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
 	golang.org/x/arch v0.3.0 // indirect
-	golang.org/x/exp v0.0.0-20230519143937-03e91628a987
+	golang.org/x/exp v0.0.0-20230519143937-03e91628a987 // indirect
 	golang.org/x/image v0.13.0 // indirect
 	golang.org/x/net v0.19.0 // indirect
-	golang.org/x/sync v0.1.0 // indirect
 	golang.org/x/sys v0.16.0 // indirect
 	golang.org/x/text v0.14.0 // indirect
 	google.golang.org/appengine v1.6.7 // indirect

server/initialize/initialize.go

@@ -1,11 +1,45 @@
 package initialize
 
 import (
-	dbInit "mayfly-go/internal/db/init"
-	machineInit "mayfly-go/internal/machine/init"
+	"mayfly-go/pkg/biz"
+	"mayfly-go/pkg/ioc"
 )
 
-func InitOther() {
-	machineInit.Init()
-	dbInit.Init()
+// 初始化ioc函数
+type InitIocFunc func()
+
+// 初始化函数
+type InitFunc func()
+
+var (
+	initIocFuncs = make([]InitIocFunc, 0)
+	initFuncs    = make([]InitFunc, 0)
+)
+
+// 添加初始化ioc函数，由各个默认自行添加
+func AddInitIocFunc(initIocFunc InitIocFunc) {
+	initIocFuncs = append(initIocFuncs, initIocFunc)
+}
+
+// 添加初始化函数，由各个默认自行添加
+func AddInitFunc(initFunc InitFunc) {
+	initFuncs = append(initFuncs, initFunc)
+}
+
+// 系统启动时，调用各个模块的初始化函数
+func InitOther() {
+	// 调用各个默认ioc组件注册初始化，优先调用ioc初始化注册函数和注入函数（可能在后续的InitFunc中需要用到依赖实例）
+	for _, initIocFunc := range initIocFuncs {
+		initIocFunc()
+	}
+	initIocFuncs = nil
+
+	// 为所有注册的实例注入其依赖的其他组件实例
+	biz.ErrIsNil(ioc.InjectComponents())
+
+	// 调用各个默认的初始化函数
+	for _, initFunc := range initFuncs {
+		go initFunc()
+	}
+	initFuncs = nil
 }

server/initialize/router.go

@@ -3,15 +3,6 @@ package initialize
 import (
 	"fmt"
 	"io/fs"
-	auth_router "mayfly-go/internal/auth/router"
-	common_router "mayfly-go/internal/common/router"
-	db_router "mayfly-go/internal/db/router"
-	machine_router "mayfly-go/internal/machine/router"
-	mongo_router "mayfly-go/internal/mongo/router"
-	msg_router "mayfly-go/internal/msg/router"
-	redis_router "mayfly-go/internal/redis/router"
-	sys_router "mayfly-go/internal/sys/router"
-	tag_router "mayfly-go/internal/tag/router"
 	"mayfly-go/pkg/config"
 	"mayfly-go/pkg/middleware"
 	"mayfly-go/static"
@@ -20,6 +11,18 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+// 初始化路由函数
+type InitRouterFunc func(router *gin.RouterGroup)
+
+var (
+	initRouterFuncs = make([]InitRouterFunc, 0)
+)
+
+// 添加初始化路由函数，由各个默认自行添加
+func AddInitRouterFunc(initRouterFunc InitRouterFunc) {
+	initRouterFuncs = append(initRouterFuncs, initRouterFunc)
+}
+
 func InitRouter() *gin.Engine {
 	// server配置
 	serverConfig := config.Conf.Server
@@ -43,20 +46,11 @@ func InitRouter() *gin.Engine {
 
 	// 设置路由组
 	api := router.Group(serverConfig.ContextPath + "/api")
-	{
-		common_router.Init(api)
-
-		auth_router.Init(api)
-
-		sys_router.Init(api)
-		msg_router.Init(api)
-
-		tag_router.Init(api)
-		machine_router.Init(api)
-		db_router.Init(api)
-		redis_router.Init(api)
-		mongo_router.Init(api)
+	// 调用所有模块注册的初始化路由函数
+	for _, initRouterFunc := range initRouterFuncs {
+		initRouterFunc(api)
 	}
+	initRouterFuncs = nil
 
 	return router
 }

server/initialize/savelog.go

@@ -1,10 +0,0 @@
-package initialize
-
-import (
-	sysapp "mayfly-go/internal/sys/application"
-	"mayfly-go/pkg/req"
-)
-
-func InitSaveLogFunc() req.SaveLogFunc {
-	return sysapp.GetSyslogApp().SaveFromReq
-}

server/initialize/terminate.go

@@ -1,10 +1,20 @@
 package initialize
 
-import (
-	dbInit "mayfly-go/internal/db/init"
+// 系统进程退出终止函数
+type TerminateFunc func()
+
+var (
+	terminateFuncs = make([]TerminateFunc, 0)
 )
 
+// 添加系统退出终止时执行的函数，由各个默认自行添加
+func AddTerminateFunc(terminateFunc TerminateFunc) {
+	terminateFuncs = append(terminateFuncs, terminateFunc)
+}
+
 // 终止进程服务后的一些操作
 func Terminate() {
-	dbInit.Terminate()
+	for _, terminateFunc := range terminateFuncs {
+		terminateFunc()
+	}
 }

server/internal/auth/api/account_login.go

@@ -25,8 +25,8 @@ import (
 )
 
 type AccountLogin struct {
-	AccountApp sysapp.Account
-	MsgApp     msgapp.Msg
+	AccountApp sysapp.Account `inject:""`
+	MsgApp     msgapp.Msg     `inject:""`
 }
 
 /**   用户账号密码登录   **/

server/internal/auth/api/ldap_login.go

@@ -28,8 +28,8 @@ import (
 )
 
 type LdapLogin struct {
-	AccountApp sysapp.Account
-	MsgApp     msgapp.Msg
+	AccountApp sysapp.Account `inject:""`
+	MsgApp     msgapp.Msg     `inject:""`
 }
 
 // @router /auth/ldap/enabled [get]

server/internal/auth/api/oauth2_login.go

@@ -28,9 +28,9 @@ import (
 )
 
 type Oauth2Login struct {
-	Oauth2App  application.Oauth2
-	AccountApp sysapp.Account
-	MsgApp     msgapp.Msg
+	Oauth2App  application.Oauth2 `inject:""`
+	AccountApp sysapp.Account     `inject:""`
+	MsgApp     msgapp.Msg         `inject:""`
 }
 
 func (a *Oauth2Login) OAuth2Login(rc *req.Ctx) {

server/internal/auth/application/application.go

@@ -1,11 +1,12 @@
 package application
 
-import "mayfly-go/internal/auth/infrastructure/persistence"
-
-var (
-	authApp = newAuthApp(persistence.GetOauthAccountRepo())
+import (
+	"mayfly-go/internal/auth/infrastructure/persistence"
+	"mayfly-go/pkg/ioc"
 )
 
-func GetAuthApp() Oauth2 {
-	return authApp
+func InitIoc() {
+	persistence.Init()
+
+	ioc.Register(new(oauth2AppImpl), ioc.WithComponentName("Oauth2App"))
 }

server/internal/auth/application/oauth2.go

@@ -14,27 +14,21 @@ type Oauth2 interface {
 	Unbind(accountId uint64)
 }
 
-func newAuthApp(oauthAccountRepo repository.Oauth2Account) Oauth2 {
-	return &oauth2AppImpl{
-		oauthAccountRepo: oauthAccountRepo,
-	}
-}
-
 type oauth2AppImpl struct {
-	oauthAccountRepo repository.Oauth2Account
+	Oauth2AccountRepo repository.Oauth2Account `inject:""`
 }
 
 func (a *oauth2AppImpl) GetOAuthAccount(condition *entity.Oauth2Account, cols ...string) error {
-	return a.oauthAccountRepo.GetBy(condition, cols...)
+	return a.Oauth2AccountRepo.GetBy(condition, cols...)
 }
 
 func (a *oauth2AppImpl) BindOAuthAccount(e *entity.Oauth2Account) error {
 	if e.Id == 0 {
-		return a.oauthAccountRepo.Insert(context.Background(), e)
+		return a.Oauth2AccountRepo.Insert(context.Background(), e)
 	}
-	return a.oauthAccountRepo.UpdateById(context.Background(), e)
+	return a.Oauth2AccountRepo.UpdateById(context.Background(), e)
 }
 
 func (a *oauth2AppImpl) Unbind(accountId uint64) {
-	a.oauthAccountRepo.DeleteByCond(context.Background(), &entity.Oauth2Account{AccountId: accountId})
+	a.Oauth2AccountRepo.DeleteByCond(context.Background(), &entity.Oauth2Account{AccountId: accountId})
 }

server/internal/auth/infrastructure/persistence/persistence.go

@@ -1,11 +1,9 @@
 package persistence
 
-import "mayfly-go/internal/auth/domain/repository"
-
-var (
-	authAccountRepo = newAuthAccountRepo()
+import (
+	"mayfly-go/pkg/ioc"
 )
 
-func GetOauthAccountRepo() repository.Oauth2Account {
-	return authAccountRepo
+func Init() {
+	ioc.Register(newAuthAccountRepo(), ioc.WithComponentName("Oauth2AccountRepo"))
 }

server/internal/auth/init/init.go

@@ -0,0 +1,12 @@
+package init
+
+import (
+	"mayfly-go/initialize"
+	"mayfly-go/internal/auth/application"
+	"mayfly-go/internal/auth/router"
+)
+
+func init() {
+	initialize.AddInitIocFunc(application.InitIoc)
+	initialize.AddInitRouterFunc(router.Init)
+}

server/internal/auth/router/router.go

@@ -2,30 +2,22 @@ package router
 
 import (
 	"mayfly-go/internal/auth/api"
-	"mayfly-go/internal/auth/application"
-	msgapp "mayfly-go/internal/msg/application"
-	sysapp "mayfly-go/internal/sys/application"
+	"mayfly-go/pkg/biz"
+	"mayfly-go/pkg/ioc"
 	"mayfly-go/pkg/req"
 
 	"github.com/gin-gonic/gin"
 )
 
 func Init(router *gin.RouterGroup) {
-	accountLogin := &api.AccountLogin{
-		AccountApp: sysapp.GetAccountApp(),
-		MsgApp:     msgapp.GetMsgApp(),
-	}
+	accountLogin := new(api.AccountLogin)
+	biz.ErrIsNil(ioc.Inject(accountLogin))
 
-	ldapLogin := &api.LdapLogin{
-		AccountApp: sysapp.GetAccountApp(),
-		MsgApp:     msgapp.GetMsgApp(),
-	}
+	ldapLogin := new(api.LdapLogin)
+	biz.ErrIsNil(ioc.Inject(ldapLogin))
 
-	oauth2Login := &api.Oauth2Login{
-		Oauth2App:  application.GetAuthApp(),
-		AccountApp: sysapp.GetAccountApp(),
-		MsgApp:     msgapp.GetMsgApp(),
-	}
+	oauth2Login := new(api.Oauth2Login)
+	biz.ErrIsNil(ioc.Inject(oauth2Login))
 
 	rg := router.Group("/auth")
 

server/internal/common/api/index.go

@@ -1,36 +0,0 @@
-package api
-
-import (
-	"mayfly-go/internal/common/consts"
-	dbapp "mayfly-go/internal/db/application"
-	machineapp "mayfly-go/internal/machine/application"
-	mongoapp "mayfly-go/internal/mongo/application"
-	redisapp "mayfly-go/internal/redis/application"
-	tagapp "mayfly-go/internal/tag/application"
-	"mayfly-go/pkg/req"
-	"mayfly-go/pkg/utils/collx"
-)
-
-type Index struct {
-	TagApp     tagapp.TagTree
-	MachineApp machineapp.Machine
-	DbApp      dbapp.Db
-	RedisApp   redisapp.Redis
-	MongoApp   mongoapp.Mongo
-}
-
-func (i *Index) Count(rc *req.Ctx) {
-	accountId := rc.GetLoginAccount().Id
-
-	mongoNum := len(i.TagApp.GetAccountResourceCodes(accountId, consts.TagResourceTypeMongo, ""))
-	machienNum := len(i.TagApp.GetAccountResourceCodes(accountId, consts.TagResourceTypeMachine, ""))
-	dbNum := len(i.TagApp.GetAccountResourceCodes(accountId, consts.TagResourceTypeDb, ""))
-	redisNum := len(i.TagApp.GetAccountResourceCodes(accountId, consts.TagResourceTypeRedis, ""))
-
-	rc.ResData = collx.M{
-		"mongoNum":   mongoNum,
-		"machineNum": machienNum,
-		"dbNum":      dbNum,
-		"redisNum":   redisNum,
-	}
-}

server/internal/common/init/init.go

@@ -0,0 +1,10 @@
+package init
+
+import (
+	"mayfly-go/initialize"
+	"mayfly-go/internal/common/router"
+)
+
+func init() {
+	initialize.AddInitRouterFunc(router.Init)
+}

server/internal/common/router/index.go

@@ -1,28 +0,0 @@
-package router
-
-import (
-	"mayfly-go/internal/common/api"
-	dbapp "mayfly-go/internal/db/application"
-	machineapp "mayfly-go/internal/machine/application"
-	mongoapp "mayfly-go/internal/mongo/application"
-	redisapp "mayfly-go/internal/redis/application"
-	tagapp "mayfly-go/internal/tag/application"
-	"mayfly-go/pkg/req"
-
-	"github.com/gin-gonic/gin"
-)
-
-func InitIndexRouter(router *gin.RouterGroup) {
-	index := router.Group("common/index")
-	i := &api.Index{
-		TagApp:     tagapp.GetTagTreeApp(),
-		MachineApp: machineapp.GetMachineApp(),
-		DbApp:      dbapp.GetDbApp(),
-		RedisApp:   redisapp.GetRedisApp(),
-		MongoApp:   mongoapp.GetMongoApp(),
-	}
-	{
-		// 首页基本信息统计
-		req.NewGet("count", i.Count).Group(index)
-	}
-}

server/internal/common/router/router.go

@@ -4,5 +4,4 @@ import "github.com/gin-gonic/gin"
 
 func Init(router *gin.RouterGroup) {
 	InitCommonRouter(router)
-	InitIndexRouter(router)
 }

server/internal/db/api/dashbord.go

@@ -0,0 +1,23 @@
+package api
+
+import (
+	"mayfly-go/internal/common/consts"
+	"mayfly-go/internal/db/application"
+	tagapp "mayfly-go/internal/tag/application"
+	"mayfly-go/pkg/req"
+	"mayfly-go/pkg/utils/collx"
+)
+
+type Dashbord struct {
+	TagTreeApp tagapp.TagTree `inject:""`
+	DbApp      application.Db `inject:""`
+}
+
+func (m *Dashbord) Dashbord(rc *req.Ctx) {
+	accountId := rc.GetLoginAccount().Id
+	dbNum := len(m.TagTreeApp.GetAccountResourceCodes(accountId, consts.TagResourceTypeDb, ""))
+
+	rc.ResData = collx.M{
+		"dbNum": dbNum,
+	}
+}

server/internal/db/api/db.go

@@ -32,11 +32,11 @@ import (
 )
 
 type Db struct {
-	InstanceApp  application.Instance
-	DbApp        application.Db
-	DbSqlExecApp application.DbSqlExec
-	MsgApp       msgapp.Msg
-	TagApp       tagapp.TagTree
+	InstanceApp  application.Instance  `inject:"DbInstanceApp"`
+	DbApp        application.Db        `inject:""`
+	DbSqlExecApp application.DbSqlExec `inject:""`
+	MsgApp       msgapp.Msg            `inject:""`
+	TagApp       tagapp.TagTree        `inject:"TagTreeApp"`
 }
 
 // @router /api/dbs [get]
@@ -355,7 +355,7 @@ func (d *Db) dumpDb(writer *gzipWriter, dbId uint64, dbName string, tables []str
 			writer.WriteString("BEGIN;\n")
 		}
 		insertSql := "INSERT INTO %s VALUES (%s);\n"
-		dbMeta.WalkTableRecord(table, func(record map[string]any, columns []*dbi.QueryColumn) error {
+		dbConn.WalkTableRows(context.TODO(), table, func(record map[string]any, columns []*dbi.QueryColumn) error {
 			var values []string
 			writer.TryFlush()
 			for _, column := range columns {
@@ -462,6 +462,20 @@ func (d *Db) GetSchemas(rc *req.Ctx) {
 	rc.ResData = res
 }
 
+func (d *Db) CopyTable(rc *req.Ctx) {
+	form := &form.DbCopyTableForm{}
+	copy := ginx.BindJsonAndCopyTo[*dbi.DbCopyTable](rc.GinCtx, form, new(dbi.DbCopyTable))
+
+	conn, err := d.DbApp.GetDbConn(form.Id, form.Db)
+	biz.ErrIsNilAppendErr(err, "拷贝表失败: %s")
+
+	err = conn.GetDialect().CopyTable(copy)
+	if err != nil {
+		logx.Errorf("拷贝表失败: %s", err.Error())
+	}
+	biz.ErrIsNilAppendErr(err, "拷贝表失败: %s")
+}
+
 func getDbId(g *gin.Context) uint64 {
 	dbId, _ := strconv.Atoi(g.Param("dbId"))
 	biz.IsTrue(dbId > 0, "dbId错误")

server/internal/db/api/db_backup.go

@@ -9,13 +9,16 @@ import (
 	"mayfly-go/pkg/biz"
 	"mayfly-go/pkg/ginx"
 	"mayfly-go/pkg/req"
+	"mayfly-go/pkg/utils/timex"
 	"strconv"
 	"strings"
+	"time"
 )
 
 type DbBackup struct {
-	DbBackupApp *application.DbBackupApp
-	DbApp       application.Db
+	backupApp  *application.DbBackupApp  `inject:"DbBackupApp"`
+	dbApp      application.Db            `inject:"DbApp"`
+	restoreApp *application.DbRestoreApp `inject:"DbRestoreApp"`
 }
 
 // todo: 鉴权，避免未经授权进行数据库备份和恢复
@@ -25,13 +28,13 @@ type DbBackup struct {
 func (d *DbBackup) GetPageList(rc *req.Ctx) {
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
 	biz.IsTrue(dbId > 0, "无效的 dbId: %v", dbId)
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 
-	queryCond, page := ginx.BindQueryAndPage[*entity.DbJobQuery](rc.GinCtx, new(entity.DbJobQuery))
+	queryCond, page := ginx.BindQueryAndPage[*entity.DbBackupQuery](rc.GinCtx, new(entity.DbBackupQuery))
 	queryCond.DbInstanceId = db.InstanceId
 	queryCond.InDbNames = strings.Fields(db.Database)
-	res, err := d.DbBackupApp.GetPageList(queryCond, page, new([]vo.DbBackup))
+	res, err := d.backupApp.GetPageList(queryCond, page, new([]vo.DbBackup))
 	biz.ErrIsNilAppendErr(err, "获取数据库备份任务失败: %v")
 	rc.ResData = res
 }
@@ -48,23 +51,22 @@ func (d *DbBackup) Create(rc *req.Ctx) {
 
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
 	biz.IsTrue(dbId > 0, "无效的 dbId: %v", dbId)
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "instanceId")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "instanceId")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
-
 	jobs := make([]*entity.DbBackup, 0, len(dbNames))
 	for _, dbName := range dbNames {
 		job := &entity.DbBackup{
-			DbJobBaseImpl: entity.NewDbBJobBase(db.InstanceId, entity.DbJobTypeBackup),
-			Enabled:       true,
-			Repeated:      backupForm.Repeated,
-			StartTime:     backupForm.StartTime,
-			Interval:      backupForm.Interval,
-			Name:          backupForm.Name,
+			DbInstanceId: db.InstanceId,
+			DbName:       dbName,
+			Enabled:      true,
+			Repeated:     backupForm.Repeated,
+			StartTime:    backupForm.StartTime,
+			Interval:     backupForm.Interval,
+			Name:         backupForm.Name,
 		}
-		job.DbName = dbName
 		jobs = append(jobs, job)
 	}
-	biz.ErrIsNilAppendErr(d.DbBackupApp.Create(rc.MetaCtx, jobs), "添加数据库备份任务失败: %v")
+	biz.ErrIsNilAppendErr(d.backupApp.Create(rc.MetaCtx, jobs), "添加数据库备份任务失败: %v")
 }
 
 // Update 保存数据库备份任务
@@ -74,17 +76,17 @@ func (d *DbBackup) Update(rc *req.Ctx) {
 	ginx.BindJsonAndValid(rc.GinCtx, backupForm)
 	rc.ReqParam = backupForm
 
-	job := entity.NewDbJob(entity.DbJobTypeBackup).(*entity.DbBackup)
+	job := &entity.DbBackup{}
 	job.Id = backupForm.Id
 	job.Name = backupForm.Name
 	job.StartTime = backupForm.StartTime
 	job.Interval = backupForm.Interval
-	biz.ErrIsNilAppendErr(d.DbBackupApp.Update(rc.MetaCtx, job), "保存数据库备份任务失败: %v")
+	biz.ErrIsNilAppendErr(d.backupApp.Update(rc.MetaCtx, job), "保存数据库备份任务失败: %v")
 }
 
-func (d *DbBackup) walk(rc *req.Ctx, fn func(ctx context.Context, backupId uint64) error) error {
-	idsStr := ginx.PathParam(rc.GinCtx, "backupId")
-	biz.NotEmpty(idsStr, "backupId 为空")
+func (d *DbBackup) walk(rc *req.Ctx, paramName string, fn func(ctx context.Context, id uint64) error) error {
+	idsStr := ginx.PathParam(rc.GinCtx, paramName)
+	biz.NotEmpty(idsStr, paramName+" 为空")
 	rc.ReqParam = idsStr
 	ids := strings.Fields(idsStr)
 	for _, v := range ids {
@@ -104,28 +106,28 @@ func (d *DbBackup) walk(rc *req.Ctx, fn func(ctx context.Context, backupId uint6
 // Delete 删除数据库备份任务
 // @router /api/dbs/:dbId/backups/:backupId [DELETE]
 func (d *DbBackup) Delete(rc *req.Ctx) {
-	err := d.walk(rc, d.DbBackupApp.Delete)
+	err := d.walk(rc, "backupId", d.backupApp.Delete)
 	biz.ErrIsNilAppendErr(err, "删除数据库备份任务失败: %v")
 }
 
 // Enable 启用数据库备份任务
 // @router /api/dbs/:dbId/backups/:backupId/enable [PUT]
 func (d *DbBackup) Enable(rc *req.Ctx) {
-	err := d.walk(rc, d.DbBackupApp.Enable)
+	err := d.walk(rc, "backupId", d.backupApp.Enable)
 	biz.ErrIsNilAppendErr(err, "启用数据库备份任务失败: %v")
 }
 
 // Disable 禁用数据库备份任务
 // @router /api/dbs/:dbId/backups/:backupId/disable [PUT]
 func (d *DbBackup) Disable(rc *req.Ctx) {
-	err := d.walk(rc, d.DbBackupApp.Disable)
+	err := d.walk(rc, "backupId", d.backupApp.Disable)
 	biz.ErrIsNilAppendErr(err, "禁用数据库备份任务失败: %v")
 }
 
 // Start 禁用数据库备份任务
 // @router /api/dbs/:dbId/backups/:backupId/start [PUT]
 func (d *DbBackup) Start(rc *req.Ctx) {
-	err := d.walk(rc, d.DbBackupApp.Start)
+	err := d.walk(rc, "backupId", d.backupApp.StartNow)
 	biz.ErrIsNilAppendErr(err, "运行数据库备份任务失败: %v")
 }
 
@@ -133,10 +135,10 @@ func (d *DbBackup) Start(rc *req.Ctx) {
 // @router /api/dbs/:dbId/db-names-without-backup [GET]
 func (d *DbBackup) GetDbNamesWithoutBackup(rc *req.Ctx) {
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "instance_id", "database")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "instance_id", "database")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 	dbNames := strings.Fields(db.Database)
-	dbNamesWithoutBackup, err := d.DbBackupApp.GetDbNamesWithoutBackup(db.InstanceId, dbNames)
+	dbNamesWithoutBackup, err := d.backupApp.GetDbNamesWithoutBackup(db.InstanceId, dbNames)
 	biz.ErrIsNilAppendErr(err, "获取未配置定时备份的数据库名称失败: %v")
 	rc.ResData = dbNamesWithoutBackup
 }
@@ -146,13 +148,74 @@ func (d *DbBackup) GetDbNamesWithoutBackup(rc *req.Ctx) {
 func (d *DbBackup) GetHistoryPageList(rc *req.Ctx) {
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
 	biz.IsTrue(dbId > 0, "无效的 dbId: %v", dbId)
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 
-	queryCond, page := ginx.BindQueryAndPage[*entity.DbBackupHistoryQuery](rc.GinCtx, new(entity.DbBackupHistoryQuery))
-	queryCond.DbInstanceId = db.InstanceId
-	queryCond.InDbNames = strings.Fields(db.Database)
-	res, err := d.DbBackupApp.GetHistoryPageList(queryCond, page, new([]vo.DbBackupHistory))
+	backupHistoryCond, page := ginx.BindQueryAndPage[*entity.DbBackupHistoryQuery](rc.GinCtx, new(entity.DbBackupHistoryQuery))
+	backupHistoryCond.DbInstanceId = db.InstanceId
+	backupHistoryCond.InDbNames = strings.Fields(db.Database)
+	backupHistories := make([]*vo.DbBackupHistory, 0, page.PageSize)
+	res, err := d.backupApp.GetHistoryPageList(backupHistoryCond, page, &backupHistories)
 	biz.ErrIsNilAppendErr(err, "获取数据库备份历史失败: %v")
+	historyIds := make([]uint64, 0, len(backupHistories))
+	for _, history := range backupHistories {
+		historyIds = append(historyIds, history.Id)
+	}
+	restores := make([]*entity.DbRestore, 0, page.PageSize)
+	if err := d.restoreApp.GetRestoresEnabled(&restores, historyIds...); err != nil {
+		biz.ErrIsNilAppendErr(err, "获取数据库备份恢复记录失败")
+	}
+	for _, history := range backupHistories {
+		for _, restore := range restores {
+			if restore.DbBackupHistoryId == history.Id {
+				history.LastStatus = restore.LastStatus
+				history.LastResult = restore.LastResult
+				history.LastTime = restore.LastTime
+				break
+			}
+		}
+	}
 	rc.ResData = res
 }
+
+// RestoreHistories 删除数据库备份历史
+// @router /api/dbs/:dbId/backup-histories/:backupHistoryId/restore [POST]
+func (d *DbBackup) RestoreHistories(rc *req.Ctx) {
+	pm := ginx.PathParam(rc.GinCtx, "backupHistoryId")
+	biz.NotEmpty(pm, "backupHistoryId 为空")
+	idsStr := strings.Fields(pm)
+	ids := make([]uint64, 0, len(idsStr))
+	for _, s := range idsStr {
+		id, err := strconv.ParseUint(s, 10, 64)
+		biz.ErrIsNilAppendErr(err, "从数据库备份历史恢复数据库失败: %v")
+		ids = append(ids, id)
+	}
+	histories := make([]*entity.DbBackupHistory, 0, len(ids))
+	err := d.backupApp.GetHistories(ids, &histories)
+	biz.ErrIsNilAppendErr(err, "添加数据库恢复任务失败: %v")
+	restores := make([]*entity.DbRestore, 0, len(histories))
+	now := time.Now()
+	for _, history := range histories {
+		job := &entity.DbRestore{
+			DbInstanceId:        history.DbInstanceId,
+			DbName:              history.DbName,
+			Enabled:             true,
+			Repeated:            false,
+			StartTime:           now,
+			Interval:            0,
+			PointInTime:         timex.NewNullTime(time.Time{}),
+			DbBackupId:          history.DbBackupId,
+			DbBackupHistoryId:   history.Id,
+			DbBackupHistoryName: history.Name,
+		}
+		restores = append(restores, job)
+	}
+	biz.ErrIsNilAppendErr(d.restoreApp.Create(rc.MetaCtx, restores), "添加数据库恢复任务失败: %v")
+}
+
+// DeleteHistories 删除数据库备份历史
+// @router /api/dbs/:dbId/backup-histories/:backupHistoryId [DELETE]
+func (d *DbBackup) DeleteHistories(rc *req.Ctx) {
+	err := d.walk(rc, "backupHistoryId", d.backupApp.DeleteHistory)
+	biz.ErrIsNilAppendErr(err, "删除数据库备份历史失败: %v")
+}

server/internal/db/api/db_data_sync.go

@@ -1,7 +1,6 @@
 package api
 
 import (
-	"context"
 	"encoding/base64"
 	"mayfly-go/internal/db/api/form"
 	"mayfly-go/internal/db/api/vo"
@@ -15,11 +14,10 @@ import (
 	"strings"
 
 	"github.com/gin-gonic/gin"
-	"github.com/google/uuid"
 )
 
 type DataSyncTask struct {
-	DataSyncTaskApp application.DataSyncTask
+	DataSyncTaskApp application.DataSyncTask `inject:"DbDataSyncTaskApp"`
 }
 
 func (d *DataSyncTask) Tasks(rc *req.Ctx) {
@@ -47,13 +45,6 @@ func (d *DataSyncTask) SaveTask(rc *req.Ctx) {
 	task.DataSql = sql
 	form.DataSql = sql
 
-	key := task.TaskKey
-	// 判断key为空就生成随机key
-	if key == "" {
-		key = uuid.New().String()
-		task.TaskKey = key
-	}
-
 	rc.ReqParam = form
 	biz.ErrIsNil(d.DataSyncTaskApp.Save(rc.MetaCtx, task))
 }
@@ -73,7 +64,7 @@ func (d *DataSyncTask) DeleteTask(rc *req.Ctx) {
 func (d *DataSyncTask) ChangeStatus(rc *req.Ctx) {
 	form := &form.DataSyncTaskStatusForm{}
 	task := ginx.BindJsonAndCopyTo[*entity.DataSyncTask](rc.GinCtx, form, new(entity.DataSyncTask))
-	_ = d.DataSyncTaskApp.UpdateById(context.Background(), task)
+	_ = d.DataSyncTaskApp.UpdateById(rc.MetaCtx, task)
 
 	if task.Status == entity.DataSyncTaskStatusEnable {
 		task, err := d.DataSyncTaskApp.GetById(new(entity.DataSyncTask), task.Id)
@@ -89,7 +80,7 @@ func (d *DataSyncTask) ChangeStatus(rc *req.Ctx) {
 func (d *DataSyncTask) Run(rc *req.Ctx) {
 	taskId := getTaskId(rc.GinCtx)
 	rc.ReqParam = taskId
-	d.DataSyncTaskApp.RunCronJob(taskId)
+	_ = d.DataSyncTaskApp.RunCronJob(taskId)
 }
 
 func (d *DataSyncTask) Stop(rc *req.Ctx) {
@@ -99,7 +90,7 @@ func (d *DataSyncTask) Stop(rc *req.Ctx) {
 	task := new(entity.DataSyncTask)
 	task.Id = taskId
 	task.RunningState = entity.DataSyncTaskRunStateStop
-	_ = d.DataSyncTaskApp.UpdateById(context.Background(), task)
+	_ = d.DataSyncTaskApp.UpdateById(rc.MetaCtx, task)
 }
 
 func (d *DataSyncTask) GetTask(rc *req.Ctx) {

server/internal/db/api/db_restore.go

@@ -14,8 +14,8 @@ import (
 )
 
 type DbRestore struct {
-	DbRestoreApp *application.DbRestoreApp
-	DbApp        application.Db
+	restoreApp *application.DbRestoreApp `inject:"DbRestoreApp"`
+	dbApp      application.Db            `inject:"DbApp"`
 }
 
 // GetPageList 获取数据库恢复任务
@@ -23,14 +23,14 @@ type DbRestore struct {
 func (d *DbRestore) GetPageList(rc *req.Ctx) {
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
 	biz.IsTrue(dbId > 0, "无效的 dbId: %v", dbId)
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "db_instance_id", "database")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 
 	var restores []vo.DbRestore
-	queryCond, page := ginx.BindQueryAndPage[*entity.DbJobQuery](rc.GinCtx, new(entity.DbJobQuery))
+	queryCond, page := ginx.BindQueryAndPage[*entity.DbRestoreQuery](rc.GinCtx, new(entity.DbRestoreQuery))
 	queryCond.DbInstanceId = db.InstanceId
 	queryCond.InDbNames = strings.Fields(db.Database)
-	res, err := d.DbRestoreApp.GetPageList(queryCond, page, &restores)
+	res, err := d.restoreApp.GetPageList(queryCond, page, &restores)
 	biz.ErrIsNilAppendErr(err, "获取数据库恢复任务失败: %v")
 	rc.ResData = res
 }
@@ -44,11 +44,12 @@ func (d *DbRestore) Create(rc *req.Ctx) {
 
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
 	biz.IsTrue(dbId > 0, "无效的 dbId: %v", dbId)
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "instanceId")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "instanceId")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 
 	job := &entity.DbRestore{
-		DbJobBaseImpl:       entity.NewDbBJobBase(db.InstanceId, entity.DbJobTypeRestore),
+		DbInstanceId:        db.InstanceId,
+		DbName:              restoreForm.DbName,
 		Enabled:             true,
 		Repeated:            restoreForm.Repeated,
 		StartTime:           restoreForm.StartTime,
@@ -58,8 +59,11 @@ func (d *DbRestore) Create(rc *req.Ctx) {
 		DbBackupHistoryId:   restoreForm.DbBackupHistoryId,
 		DbBackupHistoryName: restoreForm.DbBackupHistoryName,
 	}
-	job.DbName = restoreForm.DbName
-	biz.ErrIsNilAppendErr(d.DbRestoreApp.Create(rc.MetaCtx, job), "添加数据库恢复任务失败: %v")
+	biz.ErrIsNilAppendErr(d.restoreApp.Create(rc.MetaCtx, job), "添加数据库恢复任务失败: %v")
+}
+
+func (d *DbRestore) createWithBackupHistory(backupHistoryIds string) {
+
 }
 
 // Update 保存数据库恢复任务
@@ -73,7 +77,7 @@ func (d *DbRestore) Update(rc *req.Ctx) {
 	job.Id = restoreForm.Id
 	job.StartTime = restoreForm.StartTime
 	job.Interval = restoreForm.Interval
-	biz.ErrIsNilAppendErr(d.DbRestoreApp.Update(rc.MetaCtx, job), "保存数据库恢复任务失败: %v")
+	biz.ErrIsNilAppendErr(d.restoreApp.Update(rc.MetaCtx, job), "保存数据库恢复任务失败: %v")
 }
 
 func (d *DbRestore) walk(rc *req.Ctx, fn func(ctx context.Context, restoreId uint64) error) error {
@@ -98,21 +102,21 @@ func (d *DbRestore) walk(rc *req.Ctx, fn func(ctx context.Context, restoreId uin
 // Delete 删除数据库恢复任务
 // @router /api/dbs/:dbId/restores/:restoreId [DELETE]
 func (d *DbRestore) Delete(rc *req.Ctx) {
-	err := d.walk(rc, d.DbRestoreApp.Delete)
+	err := d.walk(rc, d.restoreApp.Delete)
 	biz.ErrIsNilAppendErr(err, "删除数据库恢复任务失败: %v")
 }
 
 // Enable 启用数据库恢复任务
 // @router /api/dbs/:dbId/restores/:restoreId/enable [PUT]
 func (d *DbRestore) Enable(rc *req.Ctx) {
-	err := d.walk(rc, d.DbRestoreApp.Enable)
+	err := d.walk(rc, d.restoreApp.Enable)
 	biz.ErrIsNilAppendErr(err, "启用数据库恢复任务失败: %v")
 }
 
 // Disable 禁用数据库恢复任务
 // @router /api/dbs/:dbId/restores/:restoreId/disable [PUT]
 func (d *DbRestore) Disable(rc *req.Ctx) {
-	err := d.walk(rc, d.DbRestoreApp.Disable)
+	err := d.walk(rc, d.restoreApp.Disable)
 	biz.ErrIsNilAppendErr(err, "禁用数据库恢复任务失败: %v")
 }
 
@@ -120,21 +124,21 @@ func (d *DbRestore) Disable(rc *req.Ctx) {
 // @router /api/dbs/:dbId/db-names-without-backup [GET]
 func (d *DbRestore) GetDbNamesWithoutRestore(rc *req.Ctx) {
 	dbId := uint64(ginx.PathParamInt(rc.GinCtx, "dbId"))
-	db, err := d.DbApp.GetById(new(entity.Db), dbId, "instance_id", "database")
+	db, err := d.dbApp.GetById(new(entity.Db), dbId, "instance_id", "database")
 	biz.ErrIsNilAppendErr(err, "获取数据库信息失败: %v")
 	dbNames := strings.Fields(db.Database)
-	dbNamesWithoutRestore, err := d.DbRestoreApp.GetDbNamesWithoutRestore(db.InstanceId, dbNames)
+	dbNamesWithoutRestore, err := d.restoreApp.GetDbNamesWithoutRestore(db.InstanceId, dbNames)
 	biz.ErrIsNilAppendErr(err, "获取未配置定时备份的数据库名称失败: %v")
 	rc.ResData = dbNamesWithoutRestore
 }
 
-// 获取数据库备份历史
+// GetHistoryPageList 获取数据库备份历史
 // @router /api/dbs/:dbId/restores/:restoreId/histories [GET]
 func (d *DbRestore) GetHistoryPageList(rc *req.Ctx) {
 	queryCond := &entity.DbRestoreHistoryQuery{
 		DbRestoreId: uint64(ginx.PathParamInt(rc.GinCtx, "restoreId")),
 	}
-	res, err := d.DbRestoreApp.GetHistoryPageList(queryCond, ginx.GetPageParam(rc.GinCtx), new([]vo.DbRestoreHistory))
+	res, err := d.restoreApp.GetHistoryPageList(queryCond, ginx.GetPageParam(rc.GinCtx), new([]vo.DbRestoreHistory))
 	biz.ErrIsNilAppendErr(err, "获取数据库备份历史失败: %v")
 	rc.ResData = res
 }

server/internal/db/api/db_sql.go

@@ -10,7 +10,7 @@ import (
 )
 
 type DbSql struct {
-	DbSqlApp application.DbSql
+	DbSqlApp application.DbSql `inject:""`
 }
 
 // @router /api/db/:dbId/sql [post]

server/internal/db/api/db_sql_exec.go

@@ -9,7 +9,7 @@ import (
 )
 
 type DbSqlExec struct {
-	DbSqlExecApp application.DbSqlExec
+	DbSqlExecApp application.DbSqlExec `inject:""`
 }
 
 func (d *DbSqlExec) DbSqlExecs(rc *req.Ctx) {

server/internal/db/api/form/db.go

@@ -23,3 +23,11 @@ type DbSqlExecForm struct {
 	Sql    string `binding:"required" json:"sql"` // 执行sql
 	Remark string `json:"remark"`                 // 执行备注
 }
+
+// 数据库复制表
+type DbCopyTableForm struct {
+	Id        uint64 `binding:"required" json:"id"`
+	Db        string `binding:"required" json:"db" `
+	TableName string `binding:"required" json:"tableName"`
+	CopyData  bool   `json:"copyData"` // 是否复制数据
+}

server/internal/db/api/instance.go

@@ -16,8 +16,8 @@ import (
 )
 
 type Instance struct {
-	InstanceApp application.Instance
-	DbApp       application.Db
+	InstanceApp application.Instance `inject:"DbInstanceApp"`
+	DbApp       application.Db       `inject:""`
 }
 
 // Instances 获取数据库实例信息

server/internal/db/api/vo/db_backup.go

@@ -2,37 +2,50 @@ package vo
 
 import (
 	"encoding/json"
+	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/pkg/utils/timex"
 	"time"
 )
 
 // DbBackup 数据库备份任务
 type DbBackup struct {
-	Id           uint64         `json:"id"`
-	DbName       string         `json:"dbName"`               // 数据库名
-	CreateTime   time.Time      `json:"createTime"`           // 创建时间
-	StartTime    time.Time      `json:"startTime"`            // 开始时间
-	Interval     time.Duration  `json:"-"`                    // 间隔时间
-	IntervalDay  uint64         `json:"intervalDay" gorm:"-"` // 间隔天数
-	Enabled      bool           `json:"enabled"`              // 是否启用
-	LastTime     timex.NullTime `json:"lastTime"`             // 最近一次执行时间
-	LastStatus   string         `json:"lastStatus"`           // 最近一次执行状态
-	LastResult   string         `json:"lastResult"`           // 最近一次执行结果
-	DbInstanceId uint64         `json:"dbInstanceId"`         // 数据库实例ID
-	Name         string         `json:"name"`                 // 备份任务名称
+	Id           uint64             `json:"id"`
+	DbName       string             `json:"dbName"`               // 数据库名
+	CreateTime   time.Time          `json:"createTime"`           // 创建时间
+	StartTime    time.Time          `json:"startTime"`            // 开始时间
+	Interval     time.Duration      `json:"-"`                    // 间隔时间
+	IntervalDay  uint64             `json:"intervalDay" gorm:"-"` // 间隔天数
+	Enabled      bool               `json:"enabled"`              // 是否启用
+	EnabledDesc  string             `json:"enabledDesc"`          // 启用状态描述
+	LastTime     timex.NullTime     `json:"lastTime"`             // 最近一次执行时间
+	LastStatus   entity.DbJobStatus `json:"lastStatus"`           // 最近一次执行状态
+	LastResult   string             `json:"lastResult"`           // 最近一次执行结果
+	DbInstanceId uint64             `json:"dbInstanceId"`         // 数据库实例ID
+	Name         string             `json:"name"`                 // 备份任务名称
 }
 
 func (backup *DbBackup) MarshalJSON() ([]byte, error) {
 	type dbBackup DbBackup
 	backup.IntervalDay = uint64(backup.Interval / time.Hour / 24)
+	if len(backup.EnabledDesc) == 0 {
+		if backup.Enabled {
+			backup.EnabledDesc = "任务已启用"
+		} else {
+			backup.EnabledDesc = "任务已禁用"
+		}
+	}
 	return json.Marshal((*dbBackup)(backup))
 }
 
 // DbBackupHistory 数据库备份历史
 type DbBackupHistory struct {
-	Id         uint64    `json:"id"`
-	DbBackupId uint64    `json:"dbBackupId"`
-	CreateTime time.Time `json:"createTime"`
-	DbName     string    `json:"dbName"` // 数据库名称
-	Name       string    `json:"name"`   // 备份历史名称
+	Id             uint64             `json:"id"`
+	DbBackupId     uint64             `json:"dbBackupId"`
+	CreateTime     time.Time          `json:"createTime"`
+	DbName         string             `json:"dbName"` // 数据库名称
+	Name           string             `json:"name"`   // 备份历史名称
+	BinlogFileName string             `json:"binlogFileName"`
+	LastTime       timex.NullTime     `json:"lastTime" gorm:"-"`   // 最近一次恢复时间
+	LastStatus     entity.DbJobStatus `json:"lastStatus" gorm:"-"` // 最近一次恢复状态
+	LastResult     string             `json:"lastResult" gorm:"-"` // 最近一次恢复结果
 }

server/internal/db/api/vo/db_restore.go

@@ -14,6 +14,7 @@ type DbRestore struct {
 	Interval            time.Duration  `json:"-"`                    // 间隔时间
 	IntervalDay         uint64         `json:"intervalDay" gorm:"-"` // 间隔天数
 	Enabled             bool           `json:"enabled"`              // 是否启用
+	EnabledDesc         string         `json:"enabledDesc"`          // 启用状态描述
 	LastTime            timex.NullTime `json:"lastTime"`             // 最近一次执行时间
 	LastStatus          string         `json:"lastStatus"`           // 最近一次执行状态
 	LastResult          string         `json:"lastResult"`           // 最近一次执行结果
@@ -27,6 +28,13 @@ type DbRestore struct {
 func (restore *DbRestore) MarshalJSON() ([]byte, error) {
 	type dbBackup DbRestore
 	restore.IntervalDay = uint64(restore.Interval / time.Hour / 24)
+	if len(restore.EnabledDesc) == 0 {
+		if restore.Enabled {
+			restore.EnabledDesc = "任务已启用"
+		} else {
+			restore.EnabledDesc = "任务已禁用"
+		}
+	}
 	return json.Marshal((*dbBackup)(restore))
 }
 

server/internal/db/application/application.go

@@ -2,91 +2,50 @@ package application
 
 import (
 	"fmt"
-	"mayfly-go/internal/db/domain/repository"
 	"mayfly-go/internal/db/infrastructure/persistence"
-	tagapp "mayfly-go/internal/tag/application"
+	"mayfly-go/pkg/ioc"
 	"sync"
 )
 
-var (
-	instanceApp  Instance
-	dbApp        Db
-	dbSqlExecApp DbSqlExec
-	dbSqlApp     DbSql
-	dbBackupApp  *DbBackupApp
-	dbRestoreApp *DbRestoreApp
-	dbBinlogApp  *DbBinlogApp
-	dataSyncApp  DataSyncTask
-)
+func InitIoc() {
+	persistence.Init()
+
+	ioc.Register(new(instanceAppImpl), ioc.WithComponentName("DbInstanceApp"))
+	ioc.Register(new(dbAppImpl), ioc.WithComponentName("DbApp"))
+	ioc.Register(new(dbSqlExecAppImpl), ioc.WithComponentName("DbSqlExecApp"))
+	ioc.Register(new(dbSqlAppImpl), ioc.WithComponentName("DbSqlApp"))
+	ioc.Register(new(dataSyncAppImpl), ioc.WithComponentName("DbDataSyncTaskApp"))
+
+	ioc.Register(newDbScheduler(), ioc.WithComponentName("DbScheduler"))
+	ioc.Register(new(DbBackupApp), ioc.WithComponentName("DbBackupApp"))
+	ioc.Register(new(DbRestoreApp), ioc.WithComponentName("DbRestoreApp"))
+	ioc.Register(newDbBinlogApp(), ioc.WithComponentName("DbBinlogApp"))
+}
 
 func Init() {
 	sync.OnceFunc(func() {
-		repositories := &repository.Repositories{
-			Instance:       persistence.GetInstanceRepo(),
-			Backup:         persistence.NewDbBackupRepo(),
-			BackupHistory:  persistence.NewDbBackupHistoryRepo(),
-			Restore:        persistence.NewDbRestoreRepo(),
-			RestoreHistory: persistence.NewDbRestoreHistoryRepo(),
-			Binlog:         persistence.NewDbBinlogRepo(),
-			BinlogHistory:  persistence.NewDbBinlogHistoryRepo(),
-		}
-		var err error
-		instanceRepo := persistence.GetInstanceRepo()
-		instanceApp = newInstanceApp(instanceRepo)
-		dbApp = newDbApp(persistence.GetDbRepo(), persistence.GetDbSqlRepo(), instanceApp, tagapp.GetTagTreeApp())
-		dbSqlExecApp = newDbSqlExecApp(persistence.GetDbSqlExecRepo())
-		dbSqlApp = newDbSqlApp(persistence.GetDbSqlRepo())
-		dataSyncApp = newDataSyncApp(persistence.GetDataSyncTaskRepo(), persistence.GetDataSyncLogRepo())
-
-		scheduler, err := newDbScheduler(repositories)
-		if err != nil {
-			panic(fmt.Sprintf("初始化 dbScheduler 失败: %v", err))
-		}
-		dbBackupApp, err = newDbBackupApp(repositories, dbApp, scheduler)
-		if err != nil {
+		if err := GetDbBackupApp().Init(); err != nil {
 			panic(fmt.Sprintf("初始化 dbBackupApp 失败: %v", err))
 		}
-		dbRestoreApp, err = newDbRestoreApp(repositories, dbApp, scheduler)
-		if err != nil {
+		if err := GetDbRestoreApp().Init(); err != nil {
 			panic(fmt.Sprintf("初始化 dbRestoreApp 失败: %v", err))
 		}
-		dbBinlogApp, err = newDbBinlogApp(repositories, dbApp, scheduler)
-		if err != nil {
-			panic(fmt.Sprintf("初始化 dbBinlogApp 失败: %v", err))
-		}
-
-		dataSyncApp.InitCronJob()
+		GetDataSyncTaskApp().InitCronJob()
 	})()
 }
 
-func GetInstanceApp() Instance {
-	return instanceApp
-}
-
-func GetDbApp() Db {
-	return dbApp
-}
-
-func GetDbSqlApp() DbSql {
-	return dbSqlApp
-}
-
-func GetDbSqlExecApp() DbSqlExec {
-	return dbSqlExecApp
-}
-
 func GetDbBackupApp() *DbBackupApp {
-	return dbBackupApp
+	return ioc.Get[*DbBackupApp]("DbBackupApp")
 }
 
 func GetDbRestoreApp() *DbRestoreApp {
-	return dbRestoreApp
+	return ioc.Get[*DbRestoreApp]("DbRestoreApp")
 }
 
 func GetDbBinlogApp() *DbBinlogApp {
-	return dbBinlogApp
+	return ioc.Get[*DbBinlogApp]("DbBinlogApp")
 }
 
 func GetDataSyncTaskApp() DataSyncTask {
-	return dataSyncApp
+	return ioc.Get[DataSyncTask]("DbDataSyncTaskApp")
 }

server/internal/db/application/db.go

@@ -40,22 +40,17 @@ type Db interface {
 	GetDbConnByInstanceId(instanceId uint64) (*dbi.DbConn, error)
 }
 
-func newDbApp(dbRepo repository.Db, dbSqlRepo repository.DbSql, dbInstanceApp Instance, tagApp tagapp.TagTree) Db {
-	app := &dbAppImpl{
-		dbSqlRepo:     dbSqlRepo,
-		dbInstanceApp: dbInstanceApp,
-		tagApp:        tagApp,
-	}
-	app.Repo = dbRepo
-	return app
-}
-
 type dbAppImpl struct {
 	base.AppImpl[*entity.Db, repository.Db]
 
-	dbSqlRepo     repository.DbSql
-	dbInstanceApp Instance
-	tagApp        tagapp.TagTree
+	dbSqlRepo     repository.DbSql `inject:"DbSqlRepo"`
+	dbInstanceApp Instance         `inject:"DbInstanceApp"`
+	tagApp        tagapp.TagTree   `inject:"TagTreeApp"`
+}
+
+// 注入DbRepo
+func (d *dbAppImpl) InjectDbRepo(repo repository.Db) {
+	d.Repo = repo
 }
 
 // 分页获取数据库信息列表
@@ -103,9 +98,13 @@ func (d *dbAppImpl) SaveDb(ctx context.Context, dbEntity *entity.Db, tagIds ...u
 	// 比较新旧数据库列表，需要将移除的数据库相关联的信息删除
 	_, delDb, _ := collx.ArrayCompare(newDbs, oldDbs)
 
-	for _, v := range delDb {
+	// 先简单关闭可能存在的旧库连接（可能改了关联标签导致DbConn.Info.TagPath与修改后的标签不一致、导致操作权限校验出错）
+	for _, v := range oldDbs {
 		// 关闭数据库连接
 		dbm.CloseDb(dbEntity.Id, v)
+	}
+
+	for _, v := range delDb {
 		// 删除该库关联的所有sql记录
 		d.dbSqlRepo.DeleteByCond(ctx, &entity.DbSql{DbId: dbId, Db: v})
 	}
@@ -155,7 +154,7 @@ func (d *dbAppImpl) GetDbConn(dbId uint64, dbName string) (*dbi.DbConn, error) {
 
 		checkDb := dbName
 		// 兼容pgsql/dm db/schema模式
-		if dbi.DbTypePostgres.Equal(instance.Type) || dbi.DbTypeDM.Equal(instance.Type) || dbi.DbTypeOracle.Equal(instance.Type) {
+		if dbi.DbTypePostgres.Equal(instance.Type) || dbi.DbTypeGauss.Equal(instance.Type) || dbi.DbTypeDM.Equal(instance.Type) || dbi.DbTypeOracle.Equal(instance.Type) || dbi.DbTypeMssql.Equal(instance.Type) {
 			ss := strings.Split(dbName, "/")
 			if len(ss) > 1 {
 				checkDb = ss[0]

server/internal/db/application/db_backup.go

@@ -3,36 +3,36 @@ package application
 import (
 	"context"
 	"encoding/binary"
-	"github.com/google/uuid"
+	"errors"
+	"fmt"
+	"gorm.io/gorm"
 	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/internal/db/domain/repository"
+	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/model"
+	"sync"
+
+	"github.com/google/uuid"
 )
 
-func newDbBackupApp(repositories *repository.Repositories, dbApp Db, scheduler *dbScheduler) (*DbBackupApp, error) {
-	var jobs []*entity.DbBackup
-	if err := repositories.Backup.ListToDo(&jobs); err != nil {
-		return nil, err
-	}
-	if err := scheduler.AddJob(context.Background(), false, entity.DbJobTypeBackup, jobs); err != nil {
-		return nil, err
-	}
-	app := &DbBackupApp{
-		backupRepo:        repositories.Backup,
-		instanceRepo:      repositories.Instance,
-		backupHistoryRepo: repositories.BackupHistory,
-		dbApp:             dbApp,
-		scheduler:         scheduler,
-	}
-	return app, nil
+type DbBackupApp struct {
+	scheduler         *dbScheduler               `inject:"DbScheduler"`
+	backupRepo        repository.DbBackup        `inject:"DbBackupRepo"`
+	backupHistoryRepo repository.DbBackupHistory `inject:"DbBackupHistoryRepo"`
+	restoreRepo       repository.DbRestore       `inject:"DbRestoreRepo"`
+	dbApp             Db                         `inject:"DbApp"`
+	mutex             sync.Mutex
 }
 
-type DbBackupApp struct {
-	backupRepo        repository.DbBackup
-	instanceRepo      repository.Instance
-	backupHistoryRepo repository.DbBackupHistory
-	dbApp             Db
-	scheduler         *dbScheduler
+func (app *DbBackupApp) Init() error {
+	var jobs []*entity.DbBackup
+	if err := app.backupRepo.ListToDo(&jobs); err != nil {
+		return err
+	}
+	if err := app.scheduler.AddJob(context.Background(), jobs); err != nil {
+		return err
+	}
+	return nil
 }
 
 func (app *DbBackupApp) Close() {
@@ -40,32 +40,111 @@ func (app *DbBackupApp) Close() {
 }
 
 func (app *DbBackupApp) Create(ctx context.Context, jobs []*entity.DbBackup) error {
-	return app.scheduler.AddJob(ctx, true /* 保存到数据库 */, entity.DbJobTypeBackup, jobs)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.backupRepo.AddJob(ctx, jobs); err != nil {
+		return err
+	}
+	return app.scheduler.AddJob(ctx, jobs)
 }
 
 func (app *DbBackupApp) Update(ctx context.Context, job *entity.DbBackup) error {
-	return app.scheduler.UpdateJob(ctx, job)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.backupRepo.UpdateById(ctx, job); err != nil {
+		return err
+	}
+	_ = app.scheduler.UpdateJob(ctx, job)
+	return nil
 }
 
 func (app *DbBackupApp) Delete(ctx context.Context, jobId uint64) error {
 	// todo: 删除数据库备份历史文件
-	return app.scheduler.RemoveJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.scheduler.RemoveJob(ctx, entity.DbJobTypeBackup, jobId); err != nil {
+		return err
+	}
+	history := &entity.DbBackupHistory{
+		DbBackupId: jobId,
+	}
+	err := app.backupHistoryRepo.GetBy(history, "name")
+	switch {
+	default:
+		return err
+	case err == nil:
+		return fmt.Errorf("数据库备份存在历史记录【%s】，无法删除该任务", history.Name)
+	case errors.Is(err, gorm.ErrRecordNotFound):
+	}
+	if err := app.backupRepo.DeleteById(ctx, jobId); err != nil {
+		return err
+	}
+	return nil
 }
 
 func (app *DbBackupApp) Enable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.EnableJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.backupRepo
+	job := &entity.DbBackup{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if job.IsEnabled() {
+		return nil
+	}
+	if job.IsExpired() {
+		return errors.New("任务已过期")
+	}
+	_ = app.scheduler.EnableJob(ctx, job)
+	if err := repo.UpdateEnabled(ctx, jobId, true); err != nil {
+		logx.Errorf("数据库备份任务已启用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }
 
 func (app *DbBackupApp) Disable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.DisableJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.backupRepo
+	job := &entity.DbBackup{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if !job.IsEnabled() {
+		return nil
+	}
+	_ = app.scheduler.DisableJob(ctx, entity.DbJobTypeBackup, jobId)
+	if err := repo.UpdateEnabled(ctx, jobId, false); err != nil {
+		logx.Errorf("数据库恢复任务已禁用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }
 
-func (app *DbBackupApp) Start(ctx context.Context, jobId uint64) error {
-	return app.scheduler.StartJobNow(ctx, entity.DbJobTypeBackup, jobId)
+func (app *DbBackupApp) StartNow(ctx context.Context, jobId uint64) error {
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	job := &entity.DbBackup{}
+	if err := app.backupRepo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if !job.IsEnabled() {
+		return errors.New("任务未启用")
+	}
+	_ = app.scheduler.StartJobNow(ctx, job)
+	return nil
 }
 
 // GetPageList 分页获取数据库备份任务
-func (app *DbBackupApp) GetPageList(condition *entity.DbJobQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
+func (app *DbBackupApp) GetPageList(condition *entity.DbBackupQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
 	return app.backupRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
 }
 
@@ -76,7 +155,11 @@ func (app *DbBackupApp) GetDbNamesWithoutBackup(instanceId uint64, dbNames []str
 
 // GetHistoryPageList 分页获取数据库备份历史
 func (app *DbBackupApp) GetHistoryPageList(condition *entity.DbBackupHistoryQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
-	return app.backupHistoryRepo.GetHistories(condition, pageParam, toEntity, orderBy...)
+	return app.backupHistoryRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
+}
+
+func (app *DbBackupApp) GetHistories(backupHistoryIds []uint64, toEntity any) error {
+	return app.backupHistoryRepo.GetHistories(backupHistoryIds, toEntity)
 }
 
 func NewIncUUID() (uuid.UUID, error) {
@@ -99,3 +182,41 @@ func NewIncUUID() (uuid.UUID, error) {
 
 	return uid, nil
 }
+
+func (app *DbBackupApp) DeleteHistory(ctx context.Context, historyId uint64) (retErr error) {
+	// todo: 删除数据库备份历史文件
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	ok, err := app.backupHistoryRepo.UpdateDeleting(true, historyId)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		_, err = app.backupHistoryRepo.UpdateDeleting(false, historyId)
+		if err == nil {
+			return
+		}
+		if retErr == nil {
+			retErr = err
+			return
+		}
+		retErr = fmt.Errorf("%w, %w", retErr, err)
+	}()
+	if !ok {
+		return errors.New("正在从备份历史中恢复数据库")
+	}
+	job := &entity.DbBackupHistory{}
+	if err := app.backupHistoryRepo.GetById(job, historyId); err != nil {
+		return err
+	}
+	conn, err := app.dbApp.GetDbConnByInstanceId(job.DbInstanceId)
+	if err != nil {
+		return err
+	}
+	dbProgram := conn.GetDialect().GetDbProgram()
+	if err := dbProgram.RemoveBackupHistory(ctx, job.DbBackupId, job.Uuid); err != nil {
+		return err
+	}
+	return app.backupHistoryRepo.DeleteById(ctx, historyId)
+}

server/internal/db/application/db_binlog.go

@@ -11,32 +11,24 @@ import (
 )
 
 type DbBinlogApp struct {
-	binlogRepo        repository.DbBinlog
-	binlogHistoryRepo repository.DbBinlogHistory
-	backupRepo        repository.DbBackup
-	backupHistoryRepo repository.DbBackupHistory
-	dbApp             Db
-	context           context.Context
-	cancel            context.CancelFunc
-	waitGroup         sync.WaitGroup
-	scheduler         *dbScheduler
+	scheduler  *dbScheduler        `inject:"DbScheduler"`
+	binlogRepo repository.DbBinlog `inject:"DbBinlogRepo"`
+	backupRepo repository.DbBackup `inject:"DbBackupRepo"`
+
+	context   context.Context
+	cancel    context.CancelFunc
+	waitGroup sync.WaitGroup
 }
 
-func newDbBinlogApp(repositories *repository.Repositories, dbApp Db, scheduler *dbScheduler) (*DbBinlogApp, error) {
+func newDbBinlogApp() *DbBinlogApp {
 	ctx, cancel := context.WithCancel(context.Background())
 	svc := &DbBinlogApp{
-		binlogRepo:        repositories.Binlog,
-		binlogHistoryRepo: repositories.BinlogHistory,
-		backupRepo:        repositories.Backup,
-		backupHistoryRepo: repositories.BackupHistory,
-		dbApp:             dbApp,
-		scheduler:         scheduler,
-		context:           ctx,
-		cancel:            cancel,
+		context: ctx,
+		cancel:  cancel,
 	}
 	svc.waitGroup.Add(1)
 	go svc.run()
-	return svc, nil
+	return svc
 }
 
 func (app *DbBinlogApp) run() {
@@ -54,7 +46,7 @@ func (app *DbBinlogApp) run() {
 		if app.closed() {
 			break
 		}
-		if err := app.scheduler.AddJob(app.context, false, entity.DbJobTypeBinlog, jobs); err != nil {
+		if err := app.scheduler.AddJob(app.context, jobs); err != nil {
 			logx.Error("DbBinlogApp: 添加 BINLOG 同步任务失败: ", err.Error())
 		}
 		timex.SleepWithContext(app.context, entity.BinlogDownloadInterval)

server/internal/db/application/db_data_sync.go

@@ -14,7 +14,12 @@ import (
 	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/model"
 	"mayfly-go/pkg/scheduler"
+	"regexp"
+	"strconv"
+	"strings"
 	"time"
+
+	"github.com/google/uuid"
 )
 
 type DataSyncTask interface {
@@ -38,17 +43,20 @@ type DataSyncTask interface {
 	GetTaskLogList(condition *entity.DataSyncLogQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error)
 }
 
-func newDataSyncApp(dataSyncRepo repository.DataSyncTask, dataSyncLogRepo repository.DataSyncLog) DataSyncTask {
-	app := new(dataSyncAppImpl)
-	app.Repo = dataSyncRepo
-	app.dataSyncLogRepo = dataSyncLogRepo
-	return app
-}
-
 type dataSyncAppImpl struct {
 	base.AppImpl[*entity.DataSyncTask, repository.DataSyncTask]
 
-	dataSyncLogRepo repository.DataSyncLog
+	dbDataSyncLogRepo repository.DataSyncLog `inject:"DbDataSyncLogRepo"`
+
+	dbApp Db `inject:"DbApp"`
+}
+
+var (
+	dateTimeReg = regexp.MustCompile(`^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$`)
+)
+
+func (app *dataSyncAppImpl) InjectDbDataSyncTaskRepo(repo repository.DataSyncTask) {
+	app.Repo = repo
 }
 
 func (app *dataSyncAppImpl) GetPageList(condition *entity.DataSyncTaskQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
@@ -58,15 +66,22 @@ func (app *dataSyncAppImpl) GetPageList(condition *entity.DataSyncTaskQuery, pag
 func (app *dataSyncAppImpl) Save(ctx context.Context, taskEntity *entity.DataSyncTask) error {
 	var err error
 	if taskEntity.Id == 0 {
+		// 新建时生成key
+		taskEntity.TaskKey = uuid.New().String()
 		err = app.Insert(ctx, taskEntity)
 	} else {
 		err = app.UpdateById(ctx, taskEntity)
 	}
+
 	if err != nil {
 		return err
 	}
 
-	app.AddCronJob(taskEntity)
+	task, err := app.GetById(new(entity.DataSyncTask), taskEntity.Id)
+	if err != nil {
+		return err
+	}
+	app.AddCronJob(task)
 	return nil
 }
 
@@ -85,8 +100,10 @@ func (app *dataSyncAppImpl) AddCronJob(taskEntity *entity.DataSyncTask) {
 
 	// 根据状态添加新的任务
 	if taskEntity.Status == entity.DataSyncTaskStatusEnable {
+		taskId := taskEntity.Id
 		scheduler.AddFunByKey(key, taskEntity.TaskCron, func() {
-			if err := app.RunCronJob(taskEntity.Id); err != nil {
+			logx.Infof("开始执行同步任务: %d", taskId)
+			if err := app.RunCronJob(taskId); err != nil {
 				logx.Errorf("定时执行数据同步任务失败: %s", err.Error())
 			}
 		})
@@ -126,7 +143,23 @@ func (app *dataSyncAppImpl) RunCronJob(id uint64) error {
 		updSql := ""
 		orderSql := ""
 		if task.UpdFieldVal != "0" && task.UpdFieldVal != "" && task.UpdField != "" {
-			updSql = fmt.Sprintf("and %s > '%s'", task.UpdField, task.UpdFieldVal)
+			srcConn, _ := app.dbApp.GetDbConn(uint64(task.SrcDbId), task.SrcDbName)
+
+			task.UpdFieldVal = strings.Trim(task.UpdFieldVal, " ")
+			// 把UpdFieldVal尝试转为int，如果可以转为int，则不添加引号，否则添加引号
+			if _, err := strconv.Atoi(task.UpdFieldVal); err != nil {
+				updSql = fmt.Sprintf("and %s > '%s'", task.UpdField, task.UpdFieldVal)
+			} else {
+				updSql = fmt.Sprintf("and %s > %s", task.UpdField, task.UpdFieldVal)
+			}
+
+			// 如果是oracle且数据类型是时间类型，则需要加上to_date函数
+			if srcConn.Info.Type == dbi.DbTypeOracle {
+				// 用正则判断数据类型是时间
+				if dateTimeReg.MatchString(task.UpdFieldVal) {
+					updSql = fmt.Sprintf("and %s > to_date('%s','yyyy-mm-dd hh24:mi:ss')", task.UpdField, task.UpdFieldVal)
+				}
+			}
 			orderSql = "order by " + task.UpdField + " asc "
 		}
 		// 组装查询sql
@@ -153,13 +186,13 @@ func (app *dataSyncAppImpl) doDataSync(sql string, task *entity.DataSyncTask) (*
 	}
 
 	// 获取源数据库连接
-	srcConn, err := GetDbApp().GetDbConn(uint64(task.SrcDbId), task.SrcDbName)
+	srcConn, err := app.dbApp.GetDbConn(uint64(task.SrcDbId), task.SrcDbName)
 	if err != nil {
 		return syncLog, errorx.NewBiz("连接源数据库失败: %s", err.Error())
 	}
 
 	// 获取目标数据库连接
-	targetConn, err := GetDbApp().GetDbConn(uint64(task.TargetDbId), task.TargetDbName)
+	targetConn, err := app.dbApp.GetDbConn(uint64(task.TargetDbId), task.TargetDbName)
 	if err != nil {
 		return syncLog, errorx.NewBiz("连接目标数据库失败: %s", err.Error())
 	}
@@ -197,8 +230,8 @@ func (app *dataSyncAppImpl) doDataSync(sql string, task *entity.DataSyncTask) (*
 			// 遍历columns 取task.UpdField的字段类型
 			updFieldType = dbi.DataTypeString
 			for _, column := range columns {
-				if column.Name == task.UpdField {
-					updFieldType = srcDialect.GetDataType(column.Type)
+				if strings.EqualFold(strings.ToLower(column.Name), strings.ToLower(task.UpdField)) {
+					updFieldType = srcDialect.GetDataConverter().GetDataType(column.Type)
 					break
 				}
 			}
@@ -207,7 +240,7 @@ func (app *dataSyncAppImpl) doDataSync(sql string, task *entity.DataSyncTask) (*
 		total++
 		result = append(result, row)
 		if total%batchSize == 0 {
-			if err := app.srcData2TargetDb(result, fieldMap, updFieldType, task, srcDialect, targetConn, targetDbTx); err != nil {
+			if err := app.srcData2TargetDb(result, fieldMap, columns, updFieldType, task, srcDialect, targetConn, targetDbTx); err != nil {
 				return err
 			}
 
@@ -229,7 +262,7 @@ func (app *dataSyncAppImpl) doDataSync(sql string, task *entity.DataSyncTask) (*
 
 	// 处理剩余的数据
 	if len(result) > 0 {
-		if err := app.srcData2TargetDb(result, fieldMap, updFieldType, task, srcDialect, targetConn, targetDbTx); err != nil {
+		if err := app.srcData2TargetDb(result, fieldMap, queryColumns, updFieldType, task, srcDialect, targetConn, targetDbTx); err != nil {
 			targetDbTx.Rollback()
 			return syncLog, err
 		}
@@ -249,10 +282,16 @@ func (app *dataSyncAppImpl) doDataSync(sql string, task *entity.DataSyncTask) (*
 	return syncLog, nil
 }
 
-func (app *dataSyncAppImpl) srcData2TargetDb(srcRes []map[string]any, fieldMap []map[string]string, updFieldType dbi.DataType, task *entity.DataSyncTask, srcDialect dbi.Dialect, targetDbConn *dbi.DbConn, targetDbTx *sql.Tx) error {
-	var data = make([]map[string]any, 0)
+func (app *dataSyncAppImpl) srcData2TargetDb(srcRes []map[string]any, fieldMap []map[string]string, columns []*dbi.QueryColumn, updFieldType dbi.DataType, task *entity.DataSyncTask, srcDialect dbi.Dialect, targetDbConn *dbi.DbConn, targetDbTx *sql.Tx) error {
 
-	// 遍历res，组装插入sql
+	// 遍历src字段列表，取出字段对应的类型
+	var srcColumnTypes = make(map[string]string)
+	for _, column := range columns {
+		srcColumnTypes[column.Name] = column.Type
+	}
+
+	// 遍历res，组装数据
+	var data = make([]map[string]any, 0)
 	for _, record := range srcRes {
 		var rowData = make(map[string]any)
 		// 遍历字段映射, target字段的值为src字段取值
@@ -265,18 +304,23 @@ func (app *dataSyncAppImpl) srcData2TargetDb(srcRes []map[string]any, fieldMap [
 
 		data = append(data, rowData)
 	}
+	// 解决字段大小写问题
+	updFieldVal := srcRes[len(srcRes)-1][strings.ToUpper(task.UpdField)]
+	if updFieldVal == "" || updFieldVal == nil {
+		updFieldVal = srcRes[len(srcRes)-1][strings.ToLower(task.UpdField)]
+	}
 
-	updFieldVal := fmt.Sprintf("%v", srcRes[len(srcRes)-1][task.UpdField])
-	updFieldVal = srcDialect.FormatStrData(updFieldVal, updFieldType)
-	task.UpdFieldVal = updFieldVal
+	task.UpdFieldVal = srcDialect.GetDataConverter().FormatData(updFieldVal, updFieldType)
 
 	// 获取目标库字段数组
 	targetWrapColumns := make([]string, 0)
 	// 获取源库字段数组
 	srcColumns := make([]string, 0)
+	srcFieldTypes := make(map[string]dbi.DataType)
 	for _, item := range fieldMap {
 		targetField := item["target"]
 		srcField := item["target"]
+		srcFieldTypes[srcField] = srcDialect.GetDataConverter().GetDataType(srcColumnTypes[item["src"]])
 		targetWrapColumns = append(targetWrapColumns, targetDbConn.Info.Type.QuoteIdentifier(targetField))
 		srcColumns = append(srcColumns, srcField)
 	}
@@ -286,7 +330,9 @@ func (app *dataSyncAppImpl) srcData2TargetDb(srcRes []map[string]any, fieldMap [
 	for _, record := range data {
 		rawValue := make([]any, 0)
 		for _, column := range srcColumns {
-			rawValue = append(rawValue, record[column])
+			// 某些情况，如oracle，需要转换时间类型的字符串为time类型
+			res := srcDialect.GetDataConverter().ParseData(record[column], srcFieldTypes[column])
+			rawValue = append(rawValue, res)
 		}
 		values = append(values, rawValue)
 	}
@@ -328,7 +374,7 @@ func (app *dataSyncAppImpl) endRunning(taskEntity *entity.DataSyncTask, log *ent
 }
 
 func (app *dataSyncAppImpl) saveLog(log *entity.DataSyncLog) {
-	app.dataSyncLogRepo.Save(context.Background(), log)
+	app.dbDataSyncLogRepo.Save(context.Background(), log)
 }
 
 func (app *dataSyncAppImpl) InitCronJob() {
@@ -374,5 +420,5 @@ func (app *dataSyncAppImpl) InitCronJob() {
 }
 
 func (app *dataSyncAppImpl) GetTaskLogList(condition *entity.DataSyncLogQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
-	return app.dataSyncLogRepo.GetTaskLogList(condition, pageParam, toEntity, orderBy...)
+	return app.dbDataSyncLogRepo.GetTaskLogList(condition, pageParam, toEntity, orderBy...)
 }

server/internal/db/application/db_restore.go

@@ -2,71 +2,131 @@ package application
 
 import (
 	"context"
+	"errors"
 	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/internal/db/domain/repository"
+	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/model"
+	"sync"
 )
 
-func newDbRestoreApp(repositories *repository.Repositories, dbApp Db, scheduler *dbScheduler) (*DbRestoreApp, error) {
-	var jobs []*entity.DbRestore
-	if err := repositories.Restore.ListToDo(&jobs); err != nil {
-		return nil, err
-	}
-	if err := scheduler.AddJob(context.Background(), false, entity.DbJobTypeRestore, jobs); err != nil {
-		return nil, err
-	}
-	app := &DbRestoreApp{
-		restoreRepo:        repositories.Restore,
-		instanceRepo:       repositories.Instance,
-		backupHistoryRepo:  repositories.BackupHistory,
-		restoreHistoryRepo: repositories.RestoreHistory,
-		binlogHistoryRepo:  repositories.BinlogHistory,
-		dbApp:              dbApp,
-		scheduler:          scheduler,
-	}
-	return app, nil
+type DbRestoreApp struct {
+	scheduler          *dbScheduler                `inject:"DbScheduler"`
+	restoreRepo        repository.DbRestore        `inject:"DbRestoreRepo"`
+	restoreHistoryRepo repository.DbRestoreHistory `inject:"DbRestoreHistoryRepo"`
+	mutex              sync.Mutex
 }
 
-type DbRestoreApp struct {
-	restoreRepo        repository.DbRestore
-	instanceRepo       repository.Instance
-	backupHistoryRepo  repository.DbBackupHistory
-	restoreHistoryRepo repository.DbRestoreHistory
-	binlogHistoryRepo  repository.DbBinlogHistory
-	dbApp              Db
-	scheduler          *dbScheduler
+func (app *DbRestoreApp) Init() error {
+	var jobs []*entity.DbRestore
+	if err := app.restoreRepo.ListToDo(&jobs); err != nil {
+		return err
+	}
+	if err := app.scheduler.AddJob(context.Background(), jobs); err != nil {
+		return err
+	}
+	return nil
 }
 
 func (app *DbRestoreApp) Close() {
 	app.scheduler.Close()
 }
 
-func (app *DbRestoreApp) Create(ctx context.Context, job *entity.DbRestore) error {
-	return app.scheduler.AddJob(ctx, true /* 保存到数据库 */, entity.DbJobTypeRestore, job)
+func (app *DbRestoreApp) Create(ctx context.Context, jobs any) error {
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.restoreRepo.AddJob(ctx, jobs); err != nil {
+		return err
+	}
+	_ = app.scheduler.AddJob(ctx, jobs)
+	return nil
 }
 
 func (app *DbRestoreApp) Update(ctx context.Context, job *entity.DbRestore) error {
-	return app.scheduler.UpdateJob(ctx, job)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.restoreRepo.UpdateById(ctx, job); err != nil {
+		return err
+	}
+	_ = app.scheduler.UpdateJob(ctx, job)
+	return nil
 }
 
 func (app *DbRestoreApp) Delete(ctx context.Context, jobId uint64) error {
 	// todo: 删除数据库恢复历史文件
-	return app.scheduler.RemoveJob(ctx, entity.DbJobTypeRestore, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.scheduler.RemoveJob(ctx, entity.DbJobTypeRestore, jobId); err != nil {
+		return err
+	}
+	history := &entity.DbRestoreHistory{
+		DbRestoreId: jobId,
+	}
+	if err := app.restoreHistoryRepo.DeleteByCond(ctx, history); err != nil {
+		return err
+	}
+	if err := app.restoreRepo.DeleteById(ctx, jobId); err != nil {
+		return err
+	}
+	return nil
 }
 
 func (app *DbRestoreApp) Enable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.EnableJob(ctx, entity.DbJobTypeRestore, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.restoreRepo
+	job := &entity.DbRestore{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if job.IsEnabled() {
+		return nil
+	}
+	if job.IsExpired() {
+		return errors.New("任务已过期")
+	}
+	_ = app.scheduler.EnableJob(ctx, job)
+	if err := repo.UpdateEnabled(ctx, jobId, true); err != nil {
+		logx.Errorf("数据库恢复任务已启用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }
 
 func (app *DbRestoreApp) Disable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.DisableJob(ctx, entity.DbJobTypeRestore, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.restoreRepo
+	job := &entity.DbRestore{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if !job.IsEnabled() {
+		return nil
+	}
+	_ = app.scheduler.DisableJob(ctx, entity.DbJobTypeRestore, jobId)
+	if err := repo.UpdateEnabled(ctx, jobId, false); err != nil {
+		logx.Errorf("数据库恢复任务已禁用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }
 
 // GetPageList 分页获取数据库恢复任务
-func (app *DbRestoreApp) GetPageList(condition *entity.DbJobQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
+func (app *DbRestoreApp) GetPageList(condition *entity.DbRestoreQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
 	return app.restoreRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
 }
 
+// GetRestoresEnabled 获取数据库恢复任务
+func (app *DbRestoreApp) GetRestoresEnabled(toEntity any, backupHistoryId ...uint64) error {
+	return app.restoreRepo.GetEnabledRestores(toEntity, backupHistoryId...)
+}
+
 // GetDbNamesWithoutRestore 获取未配置定时恢复的数据库名称
 func (app *DbRestoreApp) GetDbNamesWithoutRestore(instanceId uint64, dbNames []string) ([]string, error) {
 	return app.restoreRepo.GetDbNamesWithoutRestore(instanceId, dbNames)

server/internal/db/application/db_scheduler.go

@@ -4,10 +4,10 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"gorm.io/gorm"
 	"mayfly-go/internal/db/dbm/dbi"
 	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/internal/db/domain/repository"
-	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/runner"
 	"reflect"
 	"sync"
@@ -21,58 +21,34 @@ const (
 type dbScheduler struct {
 	mutex              sync.Mutex
 	runner             *runner.Runner[entity.DbJob]
-	dbApp              Db
-	backupRepo         repository.DbBackup
-	backupHistoryRepo  repository.DbBackupHistory
-	restoreRepo        repository.DbRestore
-	restoreHistoryRepo repository.DbRestoreHistory
-	binlogRepo         repository.DbBinlog
-	binlogHistoryRepo  repository.DbBinlogHistory
-	binlogTimes        map[uint64]time.Time
+	dbApp              Db                          `inject:"DbApp"`
+	backupRepo         repository.DbBackup         `inject:"DbBackupRepo"`
+	backupHistoryRepo  repository.DbBackupHistory  `inject:"DbBackupHistoryRepo"`
+	restoreRepo        repository.DbRestore        `inject:"DbRestoreRepo"`
+	restoreHistoryRepo repository.DbRestoreHistory `inject:"DbRestoreHistoryRepo"`
+	binlogRepo         repository.DbBinlog         `inject:"DbBinlogRepo"`
+	binlogHistoryRepo  repository.DbBinlogHistory  `inject:"DbBinlogHistoryRepo"`
 }
 
-func newDbScheduler(repositories *repository.Repositories) (*dbScheduler, error) {
-	scheduler := &dbScheduler{
-		dbApp:              dbApp,
-		backupRepo:         repositories.Backup,
-		backupHistoryRepo:  repositories.BackupHistory,
-		restoreRepo:        repositories.Restore,
-		restoreHistoryRepo: repositories.RestoreHistory,
-		binlogRepo:         repositories.Binlog,
-		binlogHistoryRepo:  repositories.BinlogHistory,
-	}
+func newDbScheduler() *dbScheduler {
+	scheduler := &dbScheduler{}
 	scheduler.runner = runner.NewRunner[entity.DbJob](maxRunning, scheduler.runJob,
 		runner.WithScheduleJob[entity.DbJob](scheduler.scheduleJob),
 		runner.WithRunnableJob[entity.DbJob](scheduler.runnableJob),
+		runner.WithUpdateJob[entity.DbJob](scheduler.updateJob),
 	)
-	return scheduler, nil
+	return scheduler
 }
 
 func (s *dbScheduler) scheduleJob(job entity.DbJob) (time.Time, error) {
 	return job.Schedule()
 }
 
-func (s *dbScheduler) repo(typ entity.DbJobType) repository.DbJob {
-	switch typ {
-	case entity.DbJobTypeBackup:
-		return s.backupRepo
-	case entity.DbJobTypeRestore:
-		return s.restoreRepo
-	case entity.DbJobTypeBinlog:
-		return s.binlogRepo
-	default:
-		panic(errors.New(fmt.Sprintf("无效的数据库任务类型: %v", typ)))
-	}
-}
-
 func (s *dbScheduler) UpdateJob(ctx context.Context, job entity.DbJob) error {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	if err := s.repo(job.GetJobType()).UpdateById(ctx, job); err != nil {
-		return err
-	}
-	_ = s.runner.UpdateOrAdd(ctx, job)
+	_ = s.runner.Update(ctx, job)
 	return nil
 }
 
@@ -80,28 +56,20 @@ func (s *dbScheduler) Close() {
 	s.runner.Close()
 }
 
-func (s *dbScheduler) AddJob(ctx context.Context, saving bool, jobType entity.DbJobType, jobs any) error {
+func (s *dbScheduler) AddJob(ctx context.Context, jobs any) error {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	if saving {
-		if err := s.repo(jobType).AddJob(ctx, jobs); err != nil {
-			return err
-		}
-	}
-
 	reflectValue := reflect.ValueOf(jobs)
 	switch reflectValue.Kind() {
 	case reflect.Array, reflect.Slice:
 		reflectLen := reflectValue.Len()
 		for i := 0; i < reflectLen; i++ {
 			job := reflectValue.Index(i).Interface().(entity.DbJob)
-			job.SetJobType(jobType)
 			_ = s.runner.Add(ctx, job)
 		}
 	default:
 		job := jobs.(entity.DbJob)
-		job.SetJobType(jobType)
 		_ = s.runner.Add(ctx, job)
 	}
 	return nil
@@ -112,29 +80,16 @@ func (s *dbScheduler) RemoveJob(ctx context.Context, jobType entity.DbJobType, j
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	if err := s.repo(jobType).DeleteById(ctx, jobId); err != nil {
+	if err := s.runner.Remove(ctx, entity.FormatJobKey(jobType, jobId)); err != nil {
 		return err
 	}
-	_ = s.runner.Remove(ctx, entity.FormatJobKey(jobType, jobId))
 	return nil
 }
 
-func (s *dbScheduler) EnableJob(ctx context.Context, jobType entity.DbJobType, jobId uint64) error {
+func (s *dbScheduler) EnableJob(ctx context.Context, job entity.DbJob) error {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	repo := s.repo(jobType)
-	job := entity.NewDbJob(jobType)
-	if err := repo.GetById(job, jobId); err != nil {
-		return err
-	}
-	if job.IsEnabled() {
-		return nil
-	}
-	job.SetEnabled(true)
-	if err := repo.UpdateEnabled(ctx, jobId, true); err != nil {
-		return err
-	}
 	_ = s.runner.Add(ctx, job)
 	return nil
 }
@@ -143,37 +98,19 @@ func (s *dbScheduler) DisableJob(ctx context.Context, jobType entity.DbJobType,
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	repo := s.repo(jobType)
-	job := entity.NewDbJob(jobType)
-	if err := repo.GetById(job, jobId); err != nil {
-		return err
-	}
-	if !job.IsEnabled() {
-		return nil
-	}
-	if err := repo.UpdateEnabled(ctx, jobId, false); err != nil {
-		return err
-	}
-	_ = s.runner.Remove(ctx, job.GetKey())
+	_ = s.runner.Remove(ctx, entity.FormatJobKey(jobType, jobId))
 	return nil
 }
 
-func (s *dbScheduler) StartJobNow(ctx context.Context, jobType entity.DbJobType, jobId uint64) error {
+func (s *dbScheduler) StartJobNow(ctx context.Context, job entity.DbJob) error {
 	s.mutex.Lock()
 	defer s.mutex.Unlock()
 
-	job := entity.NewDbJob(jobType)
-	if err := s.repo(jobType).GetById(job, jobId); err != nil {
-		return err
-	}
-	if !job.IsEnabled() {
-		return errors.New("任务未启用")
-	}
 	_ = s.runner.StartNow(ctx, job)
 	return nil
 }
 
-func (s *dbScheduler) backupMysql(ctx context.Context, job entity.DbJob) error {
+func (s *dbScheduler) backup(ctx context.Context, dbProgram dbi.DbProgram, job entity.DbJob) error {
 	id, err := NewIncUUID()
 	if err != nil {
 		return err
@@ -185,19 +122,14 @@ func (s *dbScheduler) backupMysql(ctx context.Context, job entity.DbJob) error {
 		DbInstanceId: backup.DbInstanceId,
 		DbName:       backup.DbName,
 	}
-	conn, err := s.dbApp.GetDbConnByInstanceId(backup.DbInstanceId)
-	if err != nil {
-		return err
-	}
-	dbProgram := conn.GetDialect().GetDbProgram()
 	binlogInfo, err := dbProgram.Backup(ctx, history)
 	if err != nil {
 		return err
 	}
 	now := time.Now()
-	name := backup.Name
-	if len(name) == 0 {
-		name = backup.DbName
+	name := backup.DbName
+	if len(backup.Name) > 0 {
+		name = fmt.Sprintf("%s-%s", backup.DbName, backup.Name)
 	}
 	history.Name = fmt.Sprintf("%s[%s]", name, now.Format(time.DateTime))
 	history.CreateTime = now
@@ -211,43 +143,59 @@ func (s *dbScheduler) backupMysql(ctx context.Context, job entity.DbJob) error {
 	return nil
 }
 
-func (s *dbScheduler) restoreMysql(ctx context.Context, job entity.DbJob) error {
+func (s *dbScheduler) restore(ctx context.Context, dbProgram dbi.DbProgram, job entity.DbJob) error {
 	restore := job.(*entity.DbRestore)
-	conn, err := s.dbApp.GetDbConnByInstanceId(restore.DbInstanceId)
-	if err != nil {
-		return err
-	}
-	dbProgram := conn.GetDialect().GetDbProgram()
 	if restore.PointInTime.Valid {
-		latestBinlogSequence, earliestBackupSequence := int64(-1), int64(-1)
-		binlogHistory, ok, err := s.binlogHistoryRepo.GetLatestHistory(restore.DbInstanceId)
-		if err != nil {
-			return err
-		}
-		if ok {
-			latestBinlogSequence = binlogHistory.Sequence
-		} else {
-			backupHistory, ok, err := s.backupHistoryRepo.GetEarliestHistory(restore.DbInstanceId)
-			if err != nil {
-				return err
-			}
-			if !ok {
-				return nil
-			}
-			earliestBackupSequence = backupHistory.BinlogSequence
-		}
-		binlogFiles, err := dbProgram.FetchBinlogs(ctx, true, earliestBackupSequence, latestBinlogSequence)
-		if err != nil {
-			return err
-		}
-		if err := s.binlogHistoryRepo.InsertWithBinlogFiles(ctx, restore.DbInstanceId, binlogFiles); err != nil {
+		//if enabled, err := dbProgram.CheckBinlogEnabled(ctx); err != nil {
+		//	return err
+		//} else if !enabled {
+		//	return errors.New("数据库未启用 BINLOG")
+		//}
+		//if enabled, err := dbProgram.CheckBinlogRowFormat(ctx); err != nil {
+		//	return err
+		//} else if !enabled {
+		//	return errors.New("数据库未启用 BINLOG 行模式")
+		//}
+		//
+		//latestBinlogSequence, earliestBackupSequence := int64(-1), int64(-1)
+		//binlogHistory, ok, err := s.binlogHistoryRepo.GetLatestHistory(restore.DbInstanceId)
+		//if err != nil {
+		//	return err
+		//}
+		//if ok {
+		//	latestBinlogSequence = binlogHistory.Sequence
+		//} else {
+		//	backupHistory, ok, err := s.backupHistoryRepo.GetEarliestHistory(restore.DbInstanceId)
+		//	if err != nil {
+		//		return err
+		//	}
+		//	if !ok {
+		//		return nil
+		//	}
+		//	earliestBackupSequence = backupHistory.BinlogSequence
+		//}
+		//binlogFiles, err := dbProgram.FetchBinlogs(ctx, true, earliestBackupSequence, latestBinlogSequence)
+		//if err != nil {
+		//	return err
+		//}
+		//if err := s.binlogHistoryRepo.InsertWithBinlogFiles(ctx, restore.DbInstanceId, binlogFiles); err != nil {
+		//	return err
+		//}
+		if err := s.fetchBinlog(ctx, dbProgram, job.GetInstanceId(), true); err != nil {
 			return err
 		}
 		if err := s.restorePointInTime(ctx, dbProgram, restore); err != nil {
 			return err
 		}
 	} else {
-		if err := s.restoreBackupHistory(ctx, dbProgram, restore); err != nil {
+		backupHistory := &entity.DbBackupHistory{}
+		if err := s.backupHistoryRepo.GetById(backupHistory, restore.DbBackupHistoryId); err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				err = errors.New("备份历史已删除")
+			}
+			return err
+		}
+		if err := s.restoreBackupHistory(ctx, dbProgram, backupHistory); err != nil {
 			return err
 		}
 	}
@@ -262,76 +210,108 @@ func (s *dbScheduler) restoreMysql(ctx context.Context, job entity.DbJob) error
 	return nil
 }
 
-func (s *dbScheduler) runJob(ctx context.Context, job entity.DbJob) {
-	job.SetLastStatus(entity.DbJobRunning, nil)
-	if err := s.repo(job.GetJobType()).UpdateLastStatus(ctx, job); err != nil {
-		logx.Errorf("failed to update job status: %v", err)
-		return
-	}
+//func (s *dbScheduler) updateLastStatus(ctx context.Context, job entity.DbJob) error {
+//	switch typ := job.GetJobType(); typ {
+//	case entity.DbJobTypeBackup:
+//		return s.backupRepo.UpdateLastStatus(ctx, job)
+//	case entity.DbJobTypeRestore:
+//		return s.restoreRepo.UpdateLastStatus(ctx, job)
+//	case entity.DbJobTypeBinlog:
+//		return s.binlogRepo.UpdateLastStatus(ctx, job)
+//	default:
+//		panic(fmt.Errorf("无效的数据库任务类型: %v", typ))
+//	}
+//}
 
-	var errRun error
+func (s *dbScheduler) updateJob(ctx context.Context, job entity.DbJob) error {
 	switch typ := job.GetJobType(); typ {
 	case entity.DbJobTypeBackup:
-		errRun = s.backupMysql(ctx, job)
+		return s.backupRepo.UpdateById(ctx, job)
 	case entity.DbJobTypeRestore:
-		errRun = s.restoreMysql(ctx, job)
+		return s.restoreRepo.UpdateById(ctx, job)
 	case entity.DbJobTypeBinlog:
-		errRun = s.fetchBinlogMysql(ctx, job)
+		return s.binlogRepo.UpdateById(ctx, job)
 	default:
-		errRun = errors.New(fmt.Sprintf("无效的数据库任务类型: %v", typ))
-	}
-	status := entity.DbJobSuccess
-	if errRun != nil {
-		status = entity.DbJobFailed
-	}
-	job.SetLastStatus(status, errRun)
-	if err := s.repo(job.GetJobType()).UpdateLastStatus(ctx, job); err != nil {
-		logx.Errorf("failed to update job status: %v", err)
-		return
+		return fmt.Errorf("无效的数据库任务类型: %v", typ)
 	}
 }
 
-func (s *dbScheduler) runnableJob(job entity.DbJob, next runner.NextJobFunc[entity.DbJob]) bool {
+func (s *dbScheduler) runJob(ctx context.Context, job entity.DbJob) error {
+	//job.SetLastStatus(entity.DbJobRunning, nil)
+	//if err := s.updateLastStatus(ctx, job); err != nil {
+	//	logx.Errorf("failed to update job status: %v", err)
+	//	return
+	//}
+
+	//var errRun error
+	conn, err := s.dbApp.GetDbConnByInstanceId(job.GetInstanceId())
+	if err != nil {
+		return err
+	}
+	dbProgram := conn.GetDialect().GetDbProgram()
+	switch typ := job.GetJobType(); typ {
+	case entity.DbJobTypeBackup:
+		return s.backup(ctx, dbProgram, job)
+	case entity.DbJobTypeRestore:
+		return s.restore(ctx, dbProgram, job)
+	case entity.DbJobTypeBinlog:
+		return s.fetchBinlog(ctx, dbProgram, job.GetInstanceId(), false)
+	default:
+		return fmt.Errorf("无效的数据库任务类型: %v", typ)
+	}
+	//status := entity.DbJobSuccess
+	//if errRun != nil {
+	//	status = entity.DbJobFailed
+	//}
+	//job.SetLastStatus(status, errRun)
+	//if err := s.updateLastStatus(ctx, job); err != nil {
+	//	logx.Errorf("failed to update job status: %v", err)
+	//	return
+	//}
+}
+
+func (s *dbScheduler) runnableJob(job entity.DbJob, next runner.NextJobFunc[entity.DbJob]) (bool, error) {
+	if job.IsExpired() {
+		return false, runner.ErrJobExpired
+	}
 	const maxCountByInstanceId = 4
 	const maxCountByDbName = 1
 	var countByInstanceId, countByDbName int
-	jobBase := job.GetJobBase()
 	for item, ok := next(); ok; item, ok = next() {
-		itemBase := item.GetJobBase()
-		if jobBase.DbInstanceId == itemBase.DbInstanceId {
+		if job.GetInstanceId() == item.GetInstanceId() {
 			countByInstanceId++
 			if countByInstanceId >= maxCountByInstanceId {
-				return false
+				return false, nil
 			}
 
 			if relatedToBinlog(job.GetJobType()) {
 				// todo: 恢复数据库前触发 BINLOG 同步，BINLOG 同步完成后才能恢复数据库
 				if relatedToBinlog(item.GetJobType()) {
-					return false
+					return false, nil
 				}
 			}
 
 			if job.GetDbName() == item.GetDbName() {
 				countByDbName++
 				if countByDbName >= maxCountByDbName {
-					return false
+					return false, nil
 				}
 			}
 		}
 	}
-	return true
+	return true, nil
 }
 
 func relatedToBinlog(typ entity.DbJobType) bool {
 	return typ == entity.DbJobTypeRestore || typ == entity.DbJobTypeBinlog
 }
 
-func (s *dbScheduler) restorePointInTime(ctx context.Context, program dbi.DbProgram, job *entity.DbRestore) error {
+func (s *dbScheduler) restorePointInTime(ctx context.Context, dbProgram dbi.DbProgram, job *entity.DbRestore) error {
 	binlogHistory, err := s.binlogHistoryRepo.GetHistoryByTime(job.DbInstanceId, job.PointInTime.Time)
 	if err != nil {
 		return err
 	}
-	position, err := program.GetBinlogEventPositionAtOrAfterTime(ctx, binlogHistory.FileName, job.PointInTime.Time)
+	position, err := dbProgram.GetBinlogEventPositionAtOrAfterTime(ctx, binlogHistory.FileName, job.PointInTime.Time)
 	if err != nil {
 		return err
 	}
@@ -360,22 +340,63 @@ func (s *dbScheduler) restorePointInTime(ctx context.Context, program dbi.DbProg
 		TargetPosition:  target.Position,
 		TargetTime:      job.PointInTime.Time,
 	}
-	if err := program.RestoreBackupHistory(ctx, backupHistory.DbName, backupHistory.DbBackupId, backupHistory.Uuid); err != nil {
+	if err := dbProgram.ReplayBinlog(ctx, job.DbName, job.DbName, restoreInfo); err != nil {
 		return err
 	}
-	return program.ReplayBinlog(ctx, job.DbName, job.DbName, restoreInfo)
+	if err := s.restoreBackupHistory(ctx, dbProgram, backupHistory); err != nil {
+		return err
+	}
+	// 由于 ReplayBinlog 未记录 BINLOG 事件，系统自动备份，避免数据丢失
+	backup := &entity.DbBackup{
+		DbInstanceId: backupHistory.DbInstanceId,
+		DbName:       backupHistory.DbName,
+		Enabled:      true,
+		Repeated:     false,
+		StartTime:    time.Now(),
+		Interval:     0,
+		Name:         "系统备份",
+	}
+	backup.Id = backupHistory.DbBackupId
+	if err := s.backup(ctx, dbProgram, backup); err != nil {
+		return err
+	}
+	return nil
 }
 
-func (s *dbScheduler) restoreBackupHistory(ctx context.Context, program dbi.DbProgram, job *entity.DbRestore) error {
-	backupHistory := &entity.DbBackupHistory{}
-	if err := s.backupHistoryRepo.GetById(backupHistory, job.DbBackupHistoryId); err != nil {
+func (s *dbScheduler) restoreBackupHistory(ctx context.Context, program dbi.DbProgram, backupHistory *entity.DbBackupHistory) (retErr error) {
+	ok, err := s.backupHistoryRepo.UpdateRestoring(true, backupHistory.Id)
+	if err != nil {
 		return err
 	}
+	defer func() {
+		_, err = s.backupHistoryRepo.UpdateRestoring(false, backupHistory.Id)
+		if err == nil {
+			return
+		}
+		if retErr == nil {
+			retErr = err
+			return
+		}
+		retErr = fmt.Errorf("%w, %w", retErr, err)
+	}()
+	if !ok {
+		return errors.New("关联的数据库备份历史已删除")
+	}
 	return program.RestoreBackupHistory(ctx, backupHistory.DbName, backupHistory.DbBackupId, backupHistory.Uuid)
 }
 
-func (s *dbScheduler) fetchBinlogMysql(ctx context.Context, backup entity.DbJob) error {
-	instanceId := backup.GetJobBase().DbInstanceId
+func (s *dbScheduler) fetchBinlog(ctx context.Context, dbProgram dbi.DbProgram, instanceId uint64, downloadLatestBinlogFile bool) error {
+	if enabled, err := dbProgram.CheckBinlogEnabled(ctx); err != nil {
+		return err
+	} else if !enabled {
+		return errors.New("数据库未启用 BINLOG")
+	}
+	if enabled, err := dbProgram.CheckBinlogRowFormat(ctx); err != nil {
+		return err
+	} else if !enabled {
+		return errors.New("数据库未启用 BINLOG 行模式")
+	}
+
 	latestBinlogSequence, earliestBackupSequence := int64(-1), int64(-1)
 	binlogHistory, ok, err := s.binlogHistoryRepo.GetLatestHistory(instanceId)
 	if err != nil {
@@ -393,14 +414,9 @@ func (s *dbScheduler) fetchBinlogMysql(ctx context.Context, backup entity.DbJob)
 		}
 		earliestBackupSequence = backupHistory.BinlogSequence
 	}
-	conn, err := s.dbApp.GetDbConnByInstanceId(instanceId)
+	binlogFiles, err := dbProgram.FetchBinlogs(ctx, downloadLatestBinlogFile, earliestBackupSequence, latestBinlogSequence)
 	if err != nil {
 		return err
 	}
-	dbProgram := conn.GetDialect().GetDbProgram()
-	binlogFiles, err := dbProgram.FetchBinlogs(ctx, false, earliestBackupSequence, latestBinlogSequence)
-	if err == nil {
-		err = s.binlogHistoryRepo.InsertWithBinlogFiles(ctx, instanceId, binlogFiles)
-	}
-	return nil
+	return s.binlogHistoryRepo.InsertWithBinlogFiles(ctx, instanceId, binlogFiles)
 }

server/internal/db/application/db_sql.go

@@ -14,8 +14,7 @@ type dbSqlAppImpl struct {
 	base.AppImpl[*entity.DbSql, repository.DbSql]
 }
 
-func newDbSqlApp(dbSqlRepo repository.DbSql) DbSql {
-	app := new(dbSqlAppImpl)
-	app.Repo = dbSqlRepo
-	return app
+// 注入DbSqlRepo
+func (d *dbSqlAppImpl) InjectDbSqlRepo(repo repository.DbSql) {
+	d.Repo = repo
 }

server/internal/db/application/db_sql_exec.go

@@ -9,6 +9,7 @@ import (
 	"mayfly-go/internal/db/domain/repository"
 	"mayfly-go/pkg/contextx"
 	"mayfly-go/pkg/errorx"
+	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/model"
 	"mayfly-go/pkg/utils/jsonx"
 	"strconv"
@@ -56,14 +57,8 @@ type DbSqlExec interface {
 	GetPageList(condition *entity.DbSqlExecQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error)
 }
 
-func newDbSqlExecApp(dbExecSqlRepo repository.DbSqlExec) DbSqlExec {
-	return &dbSqlExecAppImpl{
-		dbSqlExecRepo: dbExecSqlRepo,
-	}
-}
-
 type dbSqlExecAppImpl struct {
-	dbSqlExecRepo repository.DbSqlExec
+	dbSqlExecRepo repository.DbSqlExec `inject:"DbSqlExecRepo"`
 }
 
 func createSqlExecRecord(ctx context.Context, execSqlReq *DbSqlExecReq) *entity.DbSqlExec {
@@ -93,8 +88,14 @@ func (d *dbSqlExecAppImpl) Exec(ctx context.Context, execSqlReq *DbSqlExecReq) (
 			// 如果配置为0，则不校验分页参数
 			maxCount := config.GetDbQueryMaxCount()
 			if maxCount != 0 {
-				// 兼容oracle rownum分页
-				if !strings.Contains(lowerSql, "limit") && !strings.Contains(lowerSql, "rownum") {
+
+				if !strings.Contains(lowerSql, "limit") &&
+					// 兼容oracle rownum分页
+					!strings.Contains(lowerSql, "rownum") &&
+					// 兼容mssql offset分页
+					!strings.Contains(lowerSql, "offset") &&
+					// 兼容mssql top 分页  with result as ({query sql}) select top 100 * from result
+					!strings.Contains(lowerSql, " top ") {
 					// 判断是不是count语句
 					if !strings.Contains(lowerSql, "count(") {
 						return nil, errorx.NewBiz("请完善分页信息后执行")
@@ -169,7 +170,9 @@ func doSelect(ctx context.Context, selectStmt *sqlparser.Select, execSqlReq *DbS
 		len(strings.Split(selectExprsStr, ",")) > 1 {
 		// 如果配置为0，则不校验分页参数
 		maxCount := config.GetDbQueryMaxCount()
-		if maxCount != 0 {
+		// 哪些数据库跳过校验
+		skipped := dbi.DbTypeOracle == execSqlReq.DbConn.Info.Type || dbi.DbTypeMssql == execSqlReq.DbConn.Info.Type
+		if maxCount != 0 && !skipped {
 			limit := selectStmt.Limit
 			if limit == nil {
 				return nil, errorx.NewBiz("请完善分页信息后执行")
@@ -208,6 +211,9 @@ func doUpdate(ctx context.Context, update *sqlparser.Update, execSqlReq *DbSqlEx
 	tableStr := sqlparser.String(update.TableExprs)
 	// 可能使用别名，故空格切割
 	tableName := strings.Split(tableStr, " ")[0]
+	if strings.Contains(tableName, ".") {
+		tableName = strings.Split(tableName, ".")[1]
+	}
 	where := sqlparser.String(update.Where)
 	if len(where) == 0 {
 		return nil, errorx.NewBiz("SQL[%s]未执行. 请完善 where 条件后再执行", execSqlReq.Sql)
@@ -227,14 +233,26 @@ func doUpdate(ctx context.Context, update *sqlparser.Update, execSqlReq *DbSqlEx
 
 	updateColumnsAndPrimaryKey := strings.Join(updateColumns, ",") + "," + primaryKey
 	// 查询要更新字段数据的旧值，以及主键值
-	selectSql := fmt.Sprintf("SELECT %s FROM %s %s LIMIT 200", updateColumnsAndPrimaryKey, tableStr, where)
-	_, res, err := dbConn.QueryContext(ctx, selectSql)
-	if err == nil {
-		dbSqlExec.OldValue = jsonx.ToStr(res)
-	} else {
-		dbSqlExec.OldValue = err.Error()
+	selectSql := fmt.Sprintf("SELECT %s FROM %s %s", updateColumnsAndPrimaryKey, tableStr, where)
+
+	// WalkQuery查出最多200条数据
+	maxRec := 200
+	nowRec := 0
+	res := make([]map[string]any, 0)
+	err = dbConn.WalkQueryRows(ctx, selectSql, func(row map[string]any, columns []*dbi.QueryColumn) error {
+		nowRec++
+		res = append(res, row)
+		if nowRec == maxRec {
+			return errorx.NewBiz(fmt.Sprintf("超出更新最大查询条数限制: %d", maxRec))
+		}
+		return nil
+	})
+
+	if err != nil {
+		logx.Warn(err.Error())
 	}
 
+	dbSqlExec.OldValue = jsonx.ToStr(res)
 	dbSqlExec.Table = tableName
 	dbSqlExec.Type = entity.DbSqlExecTypeUpdate
 

server/internal/db/application/instance.go

@@ -30,16 +30,15 @@ type Instance interface {
 	GetDatabases(entity *entity.DbInstance) ([]string, error)
 }
 
-func newInstanceApp(instanceRepo repository.Instance) Instance {
-	app := new(instanceAppImpl)
-	app.Repo = instanceRepo
-	return app
-}
-
 type instanceAppImpl struct {
 	base.AppImpl[*entity.DbInstance, repository.Instance]
 }
 
+// 注入DbInstanceRepo
+func (app *instanceAppImpl) InjectDbInstanceRepo(repo repository.Instance) {
+	app.Repo = repo
+}
+
 // GetPageList 分页获取数据库实例
 func (app *instanceAppImpl) GetPageList(condition *entity.InstanceQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
 	return app.GetRepo().GetInstanceList(condition, pageParam, toEntity, orderBy...)

server/internal/db/dbm/dbi/conn.go

@@ -77,6 +77,11 @@ func (d *DbConn) WalkQueryRows(ctx context.Context, querySql string, walkFn Walk
 	return walkQueryRows(ctx, d.db, querySql, walkFn, args...)
 }
 
+// 游标方式遍历指定表的结果集, walkFn返回error不为nil, 则跳出遍历
+func (d *DbConn) WalkTableRows(ctx context.Context, tableName string, walkFn WalkQueryRowsFunc) error {
+	return d.WalkQueryRows(ctx, fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
+}
+
 // 执行 update, insert, delete，建表等sql
 // 返回影响条数和错误
 func (d *DbConn) Exec(sql string, args ...any) (int64, error) {
@@ -122,6 +127,11 @@ func (d *DbConn) GetDialect() Dialect {
 	return d.Info.Meta.GetDialect(d)
 }
 
+// 返回数据库连接状态
+func (d *DbConn) Stats(ctx context.Context, execSql string, args ...any) sql.DBStats {
+	return d.db.Stats()
+}
+
 // 关闭连接
 func (d *DbConn) Close() {
 	if d.db != nil {
@@ -163,7 +173,12 @@ func walkQueryRows(ctx context.Context, db *sql.DB, selectSql string, walkFn Wal
 	// 这里表示一行所有列的值，用[]byte表示
 	values := make([][]byte, lenCols)
 	for k, colType := range colTypes {
-		cols[k] = &QueryColumn{Name: colType.Name(), Type: colType.DatabaseTypeName()}
+		// 处理字段名，如果为空，则命名为匿名列
+		colName := colType.Name()
+		if colName == "" {
+			colName = fmt.Sprintf("<anonymous%d>", k+1)
+		}
+		cols[k] = &QueryColumn{Name: colName, Type: colType.DatabaseTypeName()}
 		// 这里scans引用values，把数据填充到[]byte里
 		scans[k] = &values[k]
 	}
@@ -177,7 +192,7 @@ func walkQueryRows(ctx context.Context, db *sql.DB, selectSql string, walkFn Wal
 		rowData := make(map[string]any, lenCols)
 		// 把values中的数据复制到row中
 		for i, v := range values {
-			rowData[colTypes[i].Name()] = valueConvert(v, colTypes[i])
+			rowData[cols[i].Name] = valueConvert(v, colTypes[i])
 		}
 		if err = walkFn(rowData, cols); err != nil {
 			logx.Error("游标遍历查询结果集出错,退出遍历: %s", err.Error())

server/internal/db/dbm/dbi/db_program.go

@@ -8,6 +8,9 @@ import (
 )
 
 type DbProgram interface {
+	CheckBinlogEnabled(ctx context.Context) (bool, error)
+	CheckBinlogRowFormat(ctx context.Context) (bool, error)
+
 	Backup(ctx context.Context, backupHistory *entity.DbBackupHistory) (*entity.BinlogInfo, error)
 
 	FetchBinlogs(ctx context.Context, downloadLatestBinlogFile bool, earliestBackupSequence, latestBinlogSequence int64) ([]*entity.BinlogFile, error)
@@ -16,6 +19,8 @@ type DbProgram interface {
 
 	RestoreBackupHistory(ctx context.Context, dbName string, dbBackupId uint64, dbBackupHistoryUuid string) error
 
+	RemoveBackupHistory(ctx context.Context, dbBackupId uint64, dbBackupHistoryUuid string) error
+
 	GetBinlogEventPositionAtOrAfterTime(ctx context.Context, binlogName string, targetTime time.Time) (position int64, parseErr error)
 }
 

server/internal/db/dbm/dbi/db_type.go

@@ -14,9 +14,11 @@ const (
 	DbTypeMysql    DbType = "mysql"
 	DbTypeMariadb  DbType = "mariadb"
 	DbTypePostgres DbType = "postgres"
+	DbTypeGauss    DbType = "gauss"
 	DbTypeDM       DbType = "dm"
 	DbTypeOracle   DbType = "oracle"
 	DbTypeSqlite   DbType = "sqlite"
+	DbTypeMssql    DbType = "mssql"
 )
 
 func ToDbType(dbType string) DbType {
@@ -42,8 +44,10 @@ func (dbType DbType) QuoteIdentifier(name string) string {
 	switch dbType {
 	case DbTypeMysql, DbTypeMariadb:
 		return quoteIdentifier(name, "`")
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		return quoteIdentifier(name, `"`)
+	case DbTypeMssql:
+		return fmt.Sprintf("[%s]", name)
 	default:
 		return quoteIdentifier(name, `"`)
 	}
@@ -53,7 +57,7 @@ func (dbType DbType) RemoveQuote(name string) string {
 	switch dbType {
 	case DbTypeMysql, DbTypeMariadb:
 		return removeQuote(name, "`")
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		return removeQuote(name, `"`)
 	default:
 		return removeQuote(name, `"`)
@@ -66,7 +70,7 @@ func (dbType DbType) QuoteLiteral(literal string) string {
 		literal = strings.ReplaceAll(literal, `\`, `\\`)
 		literal = strings.ReplaceAll(literal, `'`, `''`)
 		return "'" + literal + "'"
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		return pq.QuoteLiteral(literal)
 	default:
 		return pq.QuoteLiteral(literal)
@@ -77,7 +81,7 @@ func (dbType DbType) MetaDbName() string {
 	switch dbType {
 	case DbTypeMysql, DbTypeMariadb:
 		return ""
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		return "postgres"
 	case DbTypeDM:
 		return ""
@@ -90,7 +94,7 @@ func (dbType DbType) Dialect() sqlparser.Dialect {
 	switch dbType {
 	case DbTypeMysql, DbTypeMariadb:
 		return sqlparser.MysqlDialect{}
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		return sqlparser.PostgresDialect{}
 	default:
 		return sqlparser.PostgresDialect{}
@@ -118,7 +122,7 @@ func (dbType DbType) StmtSetForeignKeyChecks(check bool) string {
 		} else {
 			return "SET FOREIGN_KEY_CHECKS = 0;\n"
 		}
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		// not currently supported postgres
 		return ""
 	default:
@@ -130,7 +134,7 @@ func (dbType DbType) StmtUseDatabase(dbName string) string {
 	switch dbType {
 	case DbTypeMysql, DbTypeMariadb:
 		return fmt.Sprintf("USE %s;\n", dbType.QuoteIdentifier(dbName))
-	case DbTypePostgres:
+	case DbTypePostgres, DbTypeGauss:
 		// not currently supported postgres
 		return ""
 	default:

server/internal/db/dbm/dbi/dialect.go

@@ -42,7 +42,8 @@ type Column struct {
 	ColumnName    string  `json:"columnName"`    // 列名
 	ColumnType    string  `json:"columnType"`    // 列类型
 	ColumnComment string  `json:"columnComment"` // 列备注
-	ColumnKey     string  `json:"columnKey"`     // 是否为主键，逐渐的话值钱为PRI
+	IsPrimaryKey  bool    `json:"isPrimaryKey"`  // 是否为主键
+	IsIdentity    bool    `json:"isIdentity"`    // 是否自增
 	ColumnDefault string  `json:"columnDefault"` // 默认值
 	Nullable      string  `json:"nullable"`      // 是否可为null
 	NumScale      string  `json:"numScale"`      // 小数点
@@ -56,12 +57,33 @@ type Index struct {
 	IndexType    string `json:"indexType"`    // 索引类型
 	IndexComment string `json:"indexComment"` // 备注
 	SeqInIndex   int    `json:"seqInIndex"`
-	NonUnique    int    `json:"nonUnique"`
+	IsUnique     bool   `json:"isUnique"`
+}
+
+type DbCopyTable struct {
+	Id        uint64 `json:"id"`
+	Db        string `json:"db" `
+	TableName string `json:"tableName"`
+	CopyData  bool   `json:"copyData"` // 是否复制数据
+}
+
+// 数据转换器
+type DataConverter interface {
+	// 获取数据对应的类型
+	// @param dbColumnType 数据库原始列类型，如varchar等
+	GetDataType(dbColumnType string) DataType
+
+	// 根据数据类型格式化指定数据
+	FormatData(dbColumnValue any, dataType DataType) string
+
+	// 根据数据类型解析数据为符合要求的指定类型等
+	ParseData(dbColumnValue any, dataType DataType) any
 }
 
 // -----------------------------------元数据接口定义------------------------------------------
 // 数据库方言、元信息接口（表、列、获取表数据等元信息）
 type Dialect interface {
+
 	// 获取数据库服务实例信息
 	GetDbServer() (*DbServer, error)
 
@@ -83,9 +105,6 @@ type Dialect interface {
 	// 获取建表ddl
 	GetTableDDL(tableName string) (string, error)
 
-	// WalkTableRecord 遍历指定表的数据
-	WalkTableRecord(tableName string, walkFn WalkQueryRowsFunc) error
-
 	GetSchemas() ([]string, error)
 
 	// GetDbProgram 获取数据库程序模块，用于数据库备份与恢复
@@ -94,9 +113,10 @@ type Dialect interface {
 	// 批量保存数据
 	BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error)
 
-	GetDataType(dbColumnType string) DataType
+	// 获取数据转换器用于解析格式化列数据等
+	GetDataConverter() DataConverter
 
-	FormatStrData(dbColumnValue string, dataType DataType) string
+	CopyTable(copy *DbCopyTable) error
 }
 
 // ------------------------- 元数据sql操作 -------------------------

server/internal/db/dbm/dbi/meta.go

@@ -2,6 +2,20 @@ package dbi
 
 import "database/sql"
 
+var (
+	metas map[DbType]Meta = make(map[DbType]Meta)
+)
+
+// 注册数据库类型与dbmeta
+func Register(dt DbType, meta Meta) {
+	metas[dt] = meta
+}
+
+// 根据数据库类型获取对应的Meta
+func GetMeta(dt DbType) Meta {
+	return metas[dt]
+}
+
 // 数据库元信息获取，如获取sql.DB、Dialect等
 type Meta interface {
 	// 根据数据库信息获取sql.DB

server/internal/db/dbm/dbi/metasql/dm_meta.sql

@@ -36,7 +36,7 @@ ORDER BY a.object_name
 select
     a.index_name as INDEX_NAME,
     a.index_type as INDEX_TYPE,
-    case when a.uniqueness = 'UNIQUE' then 1 else 0 end as NON_UNIQUE,
+    case when a.uniqueness = 'UNIQUE' then 1 else 0 end as IS_UNIQUE,
     indexdef(b.object_id,1) as INDEX_DEF,
     c.column_name as COLUMN_NAME,
     c.column_position as SEQ_IN_INDEX,
@@ -64,22 +64,26 @@ select a.table_name
        b.comments                                                                          as COLUMN_COMMENT,
        a.data_default                                                                      as COLUMN_DEFAULT,
        a.data_scale                                                                        as NUM_SCALE,
-       case when t.COL_NAME = a.column_name then 'PRI' else '' end                         as COLUMN_KEY
+       case when t.COL_NAME = a.column_name then 1 else 0 end as IS_IDENTITY,
+       case when t2.constraint_type = 'P' then 1 else 0 end   as IS_PRIMARY_KEY
 from all_tab_columns a
          left join user_col_comments b
-                   on b.owner = (SELECT SF_GET_SCHEMA_NAME_BY_ID(CURRENT_SCHID)) and b.table_name = a.table_name and
-                      a.column_name = b.column_name
-         left join (select b.owner, b.table_name, a.name COL_NAME
+                   on b.owner = (SELECT SF_GET_SCHEMA_NAME_BY_ID(CURRENT_SCHID))
+                       and b.table_name = a.table_name
+                       and a.column_name = b.column_name
+         left join (select b.owner, b.TABLE_NAME, a.NAME as COL_NAME
                     from SYS.SYSCOLUMNS a,
-                         all_tables b,
-                         sys.sysobjects c,
-                         sys.sysobjects d
+                         SYS.all_tables b,
+                         SYS.SYSOBJECTS c
                     where a.INFO2 & 0x01 = 0x01
-   and a.id=c.id and d.type$ = 'SCH' and d.id = c.schid
-   and b.owner =  (SELECT SF_GET_SCHEMA_NAME_BY_ID(CURRENT_SCHID))
-   and c.schid = ( select id from sys.sysobjects where type$ = 'SCH' and name = (SELECT SF_GET_SCHEMA_NAME_BY_ID(CURRENT_SCHID)))
-   and c.name = b.table_name) t
-                   on t.table_name = a.table_name
+                      and a.ID = c.ID
+                      and c.NAME = b.TABLE_NAME) t
+                   on t.table_name = a.table_name and t.owner = a.owner
+         left join (select uc.OWNER, uic.column_name, uic.table_name, uc.constraint_type
+                    from user_ind_columns uic
+                             left join user_constraints uc on uic.index_name = uc.index_name) t2
+                   on t2.table_name = t.table_name and a.column_name = t2.column_name
 where a.owner = (SELECT SF_GET_SCHEMA_NAME_BY_ID(CURRENT_SCHID))
   and a.table_name in (%s)
-order by a.table_name, a.column_id
+order by a.table_name,
+         a.column_id

server/internal/db/dbm/dbi/metasql/mssql_meta.sql

@@ -0,0 +1,209 @@
+--MSSQL_DBS 数据库名信息
+SELECT name AS dbname
+FROM sys.databases
+WHERE owner_sid = SUSER_SID()
+  and name not in ('master', 'tempdb', 'model', 'msdb')
+---------------------------------------
+--MSSQL_TABLE_DETAIL 查询表名和表注释
+SELECT t.name   AS tableName,
+       ep.value AS tableComment
+FROM sys.tables t
+         left OUTER JOIN sys.schemas ss on t.schema_id = ss.schema_id
+         LEFT OUTER JOIN
+     sys.extended_properties ep ON ep.major_id = t.object_id AND ep.minor_id = 0 AND ep.class = 1
+WHERE ss.name = ?
+  and t.name = ?
+---------------------------------------
+--MSSQL_DB_SCHEMAS 数据库下所有schema
+SELECT a.SCHEMA_NAME
+FROM information_schema.schemata a
+where a.catalog_name = DB_NAME()
+  and (a.SCHEMA_NAME in ('dbo', 'guest') or a.SCHEMA_NAME not like 'db_%')
+  and a.SCHEMA_NAME not in ('sys', 'INFORMATION_SCHEMA')
+---------------------------------------
+--MSSQL_TABLE_INFO 表详细信息
+SELECT t.name        AS tableName,
+       ss.name AS tableSchema,
+       c.value       AS tableComment,
+       p.rows        AS tableRows,
+       0             AS dataLength,
+       0             AS indexLength,
+       t.create_date AS createTime
+FROM sys.tables t
+         left OUTER JOIN sys.schemas ss on t.schema_id = ss.schema_id
+         left OUTER JOIN sys.partitions p ON t.object_id = p.object_id AND p.index_id = 1
+         left OUTER JOIN sys.extended_properties c ON t.object_id = c.major_id AND c.minor_id = 0 AND c.class = 1
+where ss.name = ?
+ORDER BY t.name DESC;
+---------------------------------------
+--MSSQL_INDEX_INFO 索引信息
+SELECT ind.name                          AS indexName,
+       col.name                          AS columnName,
+       CASE
+           WHEN ind.is_primary_key = 1 THEN 'CLUSTERED'
+           ELSE 'NON-CLUSTERED'
+           END                           AS indexType,
+       IIF(ind.is_unique = 'true', 1, 0) AS isUnique,
+       ic.key_ordinal                    AS seqInIndex,
+       idx.value                         AS indexComment
+FROM sys.indexes ind
+         LEFT JOIN sys.tables t on t.object_id = ind.object_id
+         LEFT JOIN sys.schemas ss on t.schema_id = ss.schema_id
+         LEFT JOIN
+     sys.index_columns ic ON ind.object_id = ic.object_id AND ind.index_id = ic.index_id
+         LEFT JOIN
+     sys.columns col ON ind.object_id = col.object_id AND ic.column_id = col.column_id
+         LEFT JOIN
+     sys.extended_properties idx ON ind.object_id = idx.major_id AND ind.index_id = idx.minor_id AND idx.class = 7
+WHERE ss.name = ?
+  and ind.name is not null
+  and t.name = ?
+---------------------------------------
+--MSSQL_COLUMN_MA 列信息元数据
+SELECT t.name   AS TABLE_NAME,
+       c.name   AS COLUMN_NAME,
+       CASE
+           WHEN c.is_nullable = 1 THEN 'YES'
+           ELSE 'NO'
+           END  AS NULLABLE,
+       tp.name +
+       CASE
+           WHEN tp.name IN ('char', 'varchar', 'nchar', 'nvarchar') THEN '(' + CASE
+                                                                                   WHEN c.max_length = -1 THEN 'max'
+                                                                                   ELSE CAST(c.max_length AS NVARCHAR(255)) END +
+                                                                         ')'
+           WHEN tp.name IN ('numeric', 'decimal') THEN '(' + CAST(c.precision AS NVARCHAR(255)) + ',' +
+                                                       CAST(c.scale AS NVARCHAR(255)) + ')'
+           ELSE ''
+           END  AS COLUMN_TYPE,
+       ep.value AS COLUMN_COMMENT,
+       COLUMN_DEFAULT = CASE
+                            WHEN c.default_object_id IS NOT NULL THEN object_definition(c.default_object_id)
+                            ELSE ''
+           END,
+       c.scale  AS NUM_SCALE,
+       IS_IDENTITY = COLUMNPROPERTY(c.object_id, c.name, 'IsIdentity'),
+       IS_PRIMARY_KEY = CASE
+                            WHEN (SELECT COUNT(*)
+                                  FROM sys.index_columns ic
+                                           INNER JOIN sys.indexes i
+                                                      ON ic.index_id = i.index_id AND ic.object_id = i.object_id
+                                  WHERE ic.object_id = c.object_id
+                                    AND ic.column_id = c.column_id
+                                    AND i.is_primary_key = 1) > 0 THEN 1
+                            ELSE 0
+           END
+FROM sys.tables t
+         INNER JOIN sys.schemas ss on t.schema_id = ss.schema_id
+         INNER JOIN
+     sys.columns c ON t.object_id = c.object_id
+         INNER JOIN
+     sys.types tp ON c.system_type_id = tp.system_type_id AND c.user_type_id = tp.user_type_id
+         LEFT JOIN
+     sys.extended_properties ep ON t.object_id = ep.major_id AND c.column_id = ep.minor_id AND ep.class = 1
+WHERE ss.name = ?
+  and t.name in (%s)
+ORDER BY t.name, c.column_id
+---------------------------------------
+--MSSQL_TABLE_DDL 建表ddl
+declare
+@tabname varchar(50)
+set @tabname= ? --表名
+if ( object_id('tempdb.dbo.#t') is not null)
+begin
+DROP TABLE #t
+end
+select 'create table [' + so.name + '] (' + o.list + ')'
+           + CASE
+                 WHEN tc.Constraint_Name IS NULL THEN ''
+                 ELSE 'ALTER TABLE ' + so.Name + ' ADD CONSTRAINT ' + tc.Constraint_Name + ' PRIMARY KEY ' +
+                      ' (' + LEFT(j.List, Len(j.List)-1) + ')' END
+    TABLE_DDL
+into #t
+from sysobjects so
+    cross apply
+    (SELECT
+    ' \n ['+ column_name +'] ' +
+    data_type + case data_type
+    when 'sql_variant' then ''
+    when 'text' then ''
+    when 'ntext' then ''
+    when 'xml' then ''
+    when 'decimal' then '(' + cast (numeric_precision as varchar) + ', ' + cast (numeric_scale as varchar) + ')'
+    else coalesce ('('+ case when character_maximum_length = -1 then 'MAX' else cast (character_maximum_length as varchar) end +')', '') end + ' ' +
+    case when exists (
+    select id from syscolumns
+    where object_name(id)=so.name
+    and name = column_name
+    and columnproperty(id, name, 'IsIdentity') = 1
+    ) then
+    'IDENTITY(' +
+    cast (ident_seed(so.name) as varchar) + ',' +
+    cast (ident_incr(so.name) as varchar) + ')'
+    else ''
+    end + ' ' +
+    (case when IS_NULLABLE = 'No' then 'NOT ' else '' end ) + 'NULL ' +
+    case when information_schema.columns.COLUMN_DEFAULT IS NOT NULL THEN 'DEFAULT '+ information_schema.columns.COLUMN_DEFAULT ELSE '' END + ', '
+    from information_schema.columns where table_name = so.name
+    order by ordinal_position
+    FOR XML PATH ('')) o (list)
+    left join
+    information_schema.table_constraints tc
+on tc.Table_name = so.Name
+    AND tc.Constraint_Type = 'PRIMARY KEY'
+    cross apply
+    (select '[' + Column_Name + '], '
+    FROM information_schema.key_column_usage kcu
+    WHERE kcu.Constraint_Name = tc.Constraint_Name
+    ORDER BY
+    ORDINAL_POSITION
+    FOR XML PATH ('')) j (list)
+where xtype = 'U'
+  AND name =@tabname
+
+select (
+           case
+               when (select count(a.constraint_type)
+                     from information_schema.table_constraints a
+                              inner join information_schema.constraint_column_usage b
+                                         on a.constraint_name = b.constraint_name
+                     where a.constraint_type = 'PRIMARY KEY'--主键
+                       and a.table_name = @tabname) = 1 then replace(table_ddl
+                   , ', )ALTER TABLE'
+                   , ')' + CHAR (13)+'ALTER TABLE')
+               else SUBSTRING(table_ddl
+                        , 1
+                        , len(table_ddl) - 3) + ')' end
+           ) as TableDDL
+from #t
+
+drop table #t
+---------------------------------------
+--MSSQL_TABLE_INDEX_DDL 建索引ddl
+DECLARE
+@TableName NVARCHAR(255)
+SET @TableName = ?;
+
+SELECT 'CREATE ' +
+       CASE
+           WHEN i.is_primary_key = 1 THEN 'CLUSTERED '
+           WHEN i.type_desc = 'HEAP' THEN ''
+           ELSE 'NONCLUSTERED '
+           END +
+       'INDEX ' + i.name + ' ON ' + t.name + ' (' +
+       STUFF((SELECT ',' + c.name +
+                     CASE
+                         WHEN ic.is_descending_key = 1 THEN ' DESC'
+                         ELSE ' ASC'
+                         END
+              FROM sys.index_columns ic
+                       INNER JOIN
+                   sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id
+              WHERE ic.object_id = i.object_id
+                AND ic.index_id = i.index_id
+              ORDER BY ic.key_ordinal
+                 FOR XML PATH (''), TYPE).value('.', 'NVARCHAR(MAX)'), 1, 1, '') + ');' AS IndexDDL
+FROM sys.tables t
+         INNER JOIN
+     sys.indexes i ON t.object_id = i.object_id
+WHERE t.name = @TableName;

server/internal/db/dbm/dbi/metasql/mysql_meta.sql

@@ -30,7 +30,7 @@ SELECT
   index_name indexName,
   column_name columnName,
   index_type indexType,
-  non_unique nonUnique,
+  IF(non_unique, 0, 1) isUnique,
   SEQ_IN_INDEX seqInIndex,
   INDEX_COMMENT indexComment
 FROM
@@ -46,24 +46,25 @@ ORDER BY
   SEQ_IN_INDEX asc
 ---------------------------------------
 --MYSQL_COLUMN_MA 列信息元数据
-SELECT
-  table_name tableName,
-  column_name columnName,
-  column_type columnType,
-  column_default columnDefault,
-  column_comment columnComment,
-  column_key columnKey,
-  extra extra,
-  is_nullable nullable,
-  NUMERIC_SCALE numScale
-from
-  information_schema.columns
-WHERE
-  table_schema = (
-    SELECT
-      database ()
-  )
-  AND table_name in (%s)
-ORDER BY
-  tableName,
-  ordinal_position
+SELECT table_name     tableName,
+       column_name    columnName,
+       column_type    columnType,
+       column_default columnDefault,
+       column_comment columnComment,
+       CASE
+           WHEN column_key = 'PRI' THEN
+               1
+           ELSE 0
+           END AS     isPrimaryKey,
+       CASE
+           WHEN extra LIKE '%%auto_increment%%' THEN
+               1
+           ELSE 0
+           END AS     isIdentity,
+       is_nullable    nullable,
+       NUMERIC_SCALE  numScale
+FROM information_schema.COLUMNS
+WHERE table_schema = (SELECT DATABASE())
+  AND table_name IN (%s)
+ORDER BY table_name,
+         ordinal_position

server/internal/db/dbm/dbi/metasql/oracle_meta.sql

@@ -21,9 +21,9 @@ ORDER BY a.TABLE_NAME
 SELECT ai.INDEX_NAME                          AS INDEX_NAME,
        ai.INDEX_TYPE                          AS INDEX_TYPE,
        CASE
-           WHEN ai.uniqueness = 'UNIQUE' THEN 'NO'
-           ELSE 'YES'
-           END                                AS NON_UNIQUE,
+           WHEN ai.uniqueness = 'UNIQUE' THEN 1
+           ELSE 0
+           END AS IS_UNIQUE,
        (SELECT LISTAGG(column_name, ', ') WITHIN GROUP (ORDER BY column_position)
         FROM ALL_IND_COLUMNS aic
         WHERE aic.INDEX_NAME = ai.INDEX_NAME
@@ -53,9 +53,8 @@ SELECT a.TABLE_NAME                                              as TABLE_NAME,
        b.COMMENTS                                                as COLUMN_COMMENT,
        a.DATA_DEFAULT                                            as COLUMN_DEFAULT,
        a.DATA_SCALE                                              as NUM_SCALE,
-       CASE
-           WHEN d.pri IS NOT NULL THEN 'PRI'
-           END as COLUMN_KEY
+       CASE WHEN d.pri IS NOT NULL THEN 1 ELSE 0 END         as IS_PRIMARY_KEY,
+       CASE WHEN a.IDENTITY_COLUMN = 'YES' THEN 1 ELSE 0 END as IS_IDENTITY
 FROM all_tab_columns a
          LEFT JOIN all_col_comments b
                    on a.OWNER = b.OWNER AND a.TABLE_NAME = b.TABLE_NAME AND a.COLUMN_NAME = b.COLUMN_NAME

server/internal/db/dbm/dbi/metasql/pgsql_meta.sql

@@ -35,7 +35,7 @@ order by c.relname
 SELECT
     indexname AS "indexName",
     'BTREE' AS "IndexType",
-    case when indexdef like 'CREATE UNIQUE INDEX%%' then 0 else 1 end as "nonUnique",
+    case when indexdef like 'CREATE UNIQUE INDEX%%' then 1 else 0 end as "isUnique",
     obj_description(b.oid, 'pg_class') AS "indexComment",
     indexdef AS "indexDef",
     c.attname AS "columnName",
@@ -47,18 +47,21 @@ WHERE a.schemaname = (select current_schema())
   AND a.tablename = '%s';
 ---------------------------------------
 --PGSQL_COLUMN_MA 表列信息
-SELECT
-    table_name AS "tableName",
-    column_name AS "columnName",
-    is_nullable AS "nullable",
+SELECT a.table_name                                                                           AS "tableName",
+       a.column_name                                                                          AS "columnName",
+       a.is_nullable                                                                          AS "nullable",
     case when character_maximum_length > 0 then concat(udt_name, '(',character_maximum_length,')') else udt_name end  AS "columnType",
-    column_default as "columnDefault",
-    numeric_scale  AS "numScale",
-    case when column_default like 'nextval%%' then 'PRI' else '' end "columnKey",
-    col_description((table_schema || '.' || table_name)::regclass, ordinal_position) AS "columnComment"
-FROM information_schema.columns
-WHERE table_schema = (select current_schema()) and table_name in (%s)
-order by table_name, ordinal_position
+       a.column_default                                                                       as "columnDefault",
+       a.numeric_scale                                                                        AS "numScale",
+       case when a.column_default like 'nextval%%' then 1 else 0 end                             "isIdentity",
+       case when b.column_name is not null then 1 else 0 end                                     "isPrimaryKey",
+       col_description((a.table_schema || '.' || a.table_name)::regclass, a.ordinal_position) AS "columnComment"
+FROM information_schema.columns a
+         left join information_schema.key_column_usage b
+                   on a.table_schema = b.table_schema and b.table_name = a.table_name and b.column_name = a.column_name
+WHERE a.table_schema = (select current_schema())
+  and a.table_name in (%s)
+order by a.table_name, a.ordinal_position
 ---------------------------------------
 --PGSQL_TABLE_DDL_FUNC 表ddl函数
  CREATE OR REPLACE FUNCTION showcreatetable(namespace character varying, tablename character varying)

server/internal/db/dbm/dbm.go

@@ -4,11 +4,12 @@ import (
 	"fmt"
 	"mayfly-go/internal/common/consts"
 	"mayfly-go/internal/db/dbm/dbi"
-	"mayfly-go/internal/db/dbm/dm"
-	"mayfly-go/internal/db/dbm/mysql"
-	"mayfly-go/internal/db/dbm/oracle"
-	"mayfly-go/internal/db/dbm/postgres"
-	"mayfly-go/internal/db/dbm/sqlite"
+	_ "mayfly-go/internal/db/dbm/dm"
+	_ "mayfly-go/internal/db/dbm/mssql"
+	_ "mayfly-go/internal/db/dbm/mysql"
+	_ "mayfly-go/internal/db/dbm/oracle"
+	_ "mayfly-go/internal/db/dbm/postgres"
+	_ "mayfly-go/internal/db/dbm/sqlite"
 	"mayfly-go/internal/machine/mcm"
 	"mayfly-go/pkg/cache"
 	"mayfly-go/pkg/logx"
@@ -39,23 +40,6 @@ func init() {
 
 var mutex sync.Mutex
 
-func getDbMetaByType(dt dbi.DbType) dbi.Meta {
-	switch dt {
-	case dbi.DbTypeMysql, dbi.DbTypeMariadb:
-		return mysql.GetMeta()
-	case dbi.DbTypePostgres:
-		return postgres.GetMeta()
-	case dbi.DbTypeDM:
-		return dm.GetMeta()
-	case dbi.DbTypeOracle:
-		return oracle.GetMeta()
-	case dbi.DbTypeSqlite:
-		return sqlite.GetMeta()
-	default:
-		panic(fmt.Sprintf("invalid database type: %s", dt))
-	}
-}
-
 // 从缓存中获取数据库连接信息，若缓存中不存在则会使用回调函数获取dbInfo进行连接并缓存
 func GetDbConn(dbId uint64, database string, getDbInfo func() (*dbi.DbInfo, error)) (*dbi.DbConn, error) {
 	connId := dbi.GetDbConnId(dbId, database)
@@ -92,7 +76,7 @@ func GetDbConn(dbId uint64, database string, getDbInfo func() (*dbi.DbInfo, erro
 
 // 使用指定dbInfo信息进行连接
 func Conn(di *dbi.DbInfo) (*dbi.DbConn, error) {
-	return di.Conn(getDbMetaByType(di.Type))
+	return di.Conn(dbi.GetMeta(di.Type))
 }
 
 // 根据实例id获取连接

server/internal/db/dbm/dm/dialect.go

@@ -1,7 +1,6 @@
 package dm
 
 import (
-	"context"
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
@@ -9,10 +8,13 @@ import (
 	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/utils/anyx"
 	"mayfly-go/pkg/utils/collx"
+	"mayfly-go/pkg/utils/stringx"
 	"regexp"
 	"strings"
 	"time"
 
+	"github.com/kanzihuang/vitess/go/vt/sqlparser"
+
 	_ "gitee.com/chunanyong/dm"
 )
 
@@ -99,7 +101,8 @@ func (dd *DMDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
 			ColumnType:    anyx.ConvString(re["COLUMN_TYPE"]),
 			ColumnComment: anyx.ConvString(re["COLUMN_COMMENT"]),
 			Nullable:      anyx.ConvString(re["NULLABLE"]),
-			ColumnKey:     anyx.ConvString(re["COLUMN_KEY"]),
+			IsPrimaryKey:  anyx.ConvInt(re["IS_PRIMARY_KEY"]) == 1,
+			IsIdentity:    anyx.ConvInt(re["IS_IDENTITY"]) == 1,
 			ColumnDefault: anyx.ConvString(re["COLUMN_DEFAULT"]),
 			NumScale:      anyx.ConvString(re["NUM_SCALE"]),
 		})
@@ -116,7 +119,7 @@ func (dd *DMDialect) GetPrimaryKey(tablename string) (string, error) {
 		return "", errorx.NewBiz("[%s] 表不存在", tablename)
 	}
 	for _, v := range columns {
-		if v.ColumnKey == "PRI" {
+		if v.IsPrimaryKey {
 			return v.ColumnName, nil
 		}
 	}
@@ -138,7 +141,7 @@ func (dd *DMDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
 			ColumnName:   anyx.ConvString(re["COLUMN_NAME"]),
 			IndexType:    anyx.ConvString(re["INDEX_TYPE"]),
 			IndexComment: anyx.ConvString(re["INDEX_COMMENT"]),
-			NonUnique:    anyx.ConvInt(re["NON_UNIQUE"]),
+			IsUnique:     anyx.ConvInt(re["IS_UNIQUE"]) == 1,
 			SeqInIndex:   anyx.ConvInt(re["SEQ_IN_INDEX"]),
 		})
 	}
@@ -230,10 +233,6 @@ func (dd *DMDialect) GetTableDDL(tableName string) (string, error) {
 	return builder.String(), nil
 }
 
-func (dd *DMDialect) WalkTableRecord(tableName string, walkFn dbi.WalkQueryRowsFunc) error {
-	return dd.dc.WalkQueryRows(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
-}
-
 // 获取DM当前连接的库可访问的schemaNames
 func (dd *DMDialect) GetSchemas() ([]string, error) {
 	sql := dbi.GetLocalSql(DM_META_FILE, DM_DB_SCHEMAS)
@@ -253,24 +252,16 @@ func (dd *DMDialect) GetDbProgram() dbi.DbProgram {
 	panic("implement me")
 }
 
-func (dd *DMDialect) GetDataType(dbColumnType string) dbi.DataType {
-	if regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`).MatchString(dbColumnType) {
-		return dbi.DataTypeNumber
-	}
+var (
+	// 数字类型
+	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
 	// 日期时间类型
-	if regexp.MustCompile(`(?i)datetime|timestamp`).MatchString(dbColumnType) {
-		return dbi.DataTypeDateTime
-	}
+	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
 	// 日期类型
-	if regexp.MustCompile(`(?i)date`).MatchString(dbColumnType) {
-		return dbi.DataTypeDate
-	}
+	dateRegexp = regexp.MustCompile(`(?i)date`)
 	// 时间类型
-	if regexp.MustCompile(`(?i)time`).MatchString(dbColumnType) {
-		return dbi.DataTypeTime
-	}
-	return dbi.DataTypeString
-}
+	timeRegexp = regexp.MustCompile(`(?i)time`)
+)
 
 func (dd *DMDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
 	// 执行批量insert sql
@@ -297,17 +288,86 @@ func (dd *DMDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string,
 	return int64(effRows), nil
 }
 
-func (dd *DMDialect) FormatStrData(dbColumnValue string, dataType dbi.DataType) string {
+func (dd *DMDialect) GetDataConverter() dbi.DataConverter {
+	return new(DataConverter)
+}
+
+type DataConverter struct {
+}
+
+func (dd *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	if datetimeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	if dateRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDate
+	}
+	if timeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dd *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	str := anyx.ToString(dbColumnValue)
 	switch dataType {
 	case dbi.DataTypeDateTime: // "2024-01-02T22:08:22.275697+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.DateTime)
 	case dbi.DataTypeDate: // "2024-01-02T00:00:00+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.DateOnly)
 	case dbi.DataTypeTime: // "0000-01-01T22:08:22.275688+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.TimeOnly)
 	}
+	return str
+}
+
+func (dd *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
 	return dbColumnValue
 }
+
+func (dd *DMDialect) CopyTable(copy *dbi.DbCopyTable) error {
+	tableName := copy.TableName
+	ddl, err := dd.GetTableDDL(tableName)
+	if err != nil {
+		return err
+	}
+	// 生成新表名,为老表明+_copy_时间戳
+	newTableName := tableName + "_copy_" + time.Now().Format("20060102150405")
+
+	// 替换新表名
+	ddl = strings.ReplaceAll(ddl, fmt.Sprintf("\"%s\"", strings.ToUpper(tableName)), fmt.Sprintf("\"%s\"", strings.ToUpper(newTableName)))
+	// 去除空格换行
+	ddl = stringx.TrimSpaceAndBr(ddl)
+	sqls, err := sqlparser.SplitStatementToPieces(ddl, sqlparser.WithDialect(dd.dc.Info.Type.Dialect()))
+	for _, sql := range sqls {
+		_, _ = dd.dc.Exec(sql)
+	}
+
+	// 复制数据
+	if copy.CopyData {
+		go func() {
+			// 设置允许填充自增列之后，显示指定列名可以插入自增列
+			_, _ = dd.dc.Exec(fmt.Sprintf("set identity_insert \"%s\" on", newTableName))
+			// 获取列名
+			columns, _ := dd.GetColumns(tableName)
+			columnArr := make([]string, 0)
+			for _, column := range columns {
+				columnArr = append(columnArr, fmt.Sprintf("\"%s\"", column.ColumnName))
+			}
+			columnStr := strings.Join(columnArr, ",")
+			// 插入新数据并显示指定列
+			_, _ = dd.dc.Exec(fmt.Sprintf("insert into \"%s\" (%s) select %s from \"%s\"", newTableName, columnStr, columnStr, tableName))
+
+			// 执行完成后关闭允许填充自增列
+			_, _ = dd.dc.Exec(fmt.Sprintf("set identity_insert \"%s\" off", newTableName))
+		}()
+	}
+
+	return err
+}

server/internal/db/dbm/dm/meta.go

@@ -6,19 +6,10 @@ import (
 	"mayfly-go/internal/db/dbm/dbi"
 	"net/url"
 	"strings"
-	"sync"
 )
 
-var (
-	meta dbi.Meta
-	once sync.Once
-)
-
-func GetMeta() dbi.Meta {
-	once.Do(func() {
-		meta = new(DmMeta)
-	})
-	return meta
+func init() {
+	dbi.Register(dbi.DbTypeDM, new(DmMeta))
 }
 
 type DmMeta struct {

server/internal/db/dbm/mssql/dialect.go

@@ -0,0 +1,425 @@
+package mssql
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"mayfly-go/internal/db/dbm/dbi"
+	"mayfly-go/pkg/errorx"
+	"mayfly-go/pkg/logx"
+	"mayfly-go/pkg/utils/anyx"
+	"mayfly-go/pkg/utils/collx"
+	"regexp"
+	"strings"
+	"time"
+)
+
+const (
+	MSSQL_META_FILE           = "metasql/mssql_meta.sql"
+	MSSQL_DBS_KEY             = "MSSQL_DBS"
+	MSSQL_DB_SCHEMAS_KEY      = "MSSQL_DB_SCHEMAS"
+	MSSQL_TABLE_INFO_KEY      = "MSSQL_TABLE_INFO"
+	MSSQL_INDEX_INFO_KEY      = "MSSQL_INDEX_INFO"
+	MSSQL_COLUMN_MA_KEY       = "MSSQL_COLUMN_MA"
+	MSSQL_TABLE_DETAIL_KEY    = "MSSQL_TABLE_DETAIL"
+	MSSQL_TABLE_INDEX_DDL_KEY = "MSSQL_TABLE_INDEX_DDL"
+)
+
+type MssqlDialect struct {
+	dc *dbi.DbConn
+}
+
+func (md *MssqlDialect) GetDbServer() (*dbi.DbServer, error) {
+	_, res, err := md.dc.Query("SELECT @@VERSION as version")
+	if err != nil {
+		return nil, err
+	}
+	ds := &dbi.DbServer{
+		Version: anyx.ConvString(res[0]["version"]),
+	}
+	return ds, nil
+}
+
+func (md *MssqlDialect) GetDbNames() ([]string, error) {
+	_, res, err := md.dc.Query(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_DBS_KEY))
+	if err != nil {
+		return nil, err
+	}
+
+	databases := make([]string, 0)
+	for _, re := range res {
+		databases = append(databases, anyx.ConvString(re["dbname"]))
+	}
+
+	return databases, nil
+}
+
+// 从连接信息中获取数据库和schema信息
+func (md *MssqlDialect) currentSchema() string {
+	dbName := md.dc.Info.Database
+	schema := ""
+	arr := strings.Split(dbName, "/")
+	if len(arr) == 2 {
+		schema = arr[1]
+	}
+	return schema
+}
+
+// 获取表基础元信息, 如表名等
+func (md *MssqlDialect) GetTables() ([]dbi.Table, error) {
+	_, res, err := md.dc.Query(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_TABLE_INFO_KEY), md.currentSchema())
+
+	if err != nil {
+		return nil, err
+	}
+
+	tables := make([]dbi.Table, 0)
+	for _, re := range res {
+		tables = append(tables, dbi.Table{
+			TableName:    anyx.ConvString(re["tableName"]),
+			TableComment: anyx.ConvString(re["tableComment"]),
+			CreateTime:   anyx.ConvString(re["createTime"]),
+			TableRows:    anyx.ConvInt(re["tableRows"]),
+			DataLength:   anyx.ConvInt64(re["dataLength"]),
+			IndexLength:  anyx.ConvInt64(re["indexLength"]),
+		})
+	}
+	return tables, nil
+}
+
+// 获取列元信息, 如列名等
+func (md *MssqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
+	dbType := md.dc.Info.Type
+	tableName := strings.Join(collx.ArrayMap[string, string](tableNames, func(val string) string {
+		return fmt.Sprintf("'%s'", dbType.RemoveQuote(val))
+	}), ",")
+
+	_, res, err := md.dc.Query(fmt.Sprintf(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_COLUMN_MA_KEY), tableName), md.currentSchema())
+	if err != nil {
+		return nil, err
+	}
+
+	columns := make([]dbi.Column, 0)
+	for _, re := range res {
+		columns = append(columns, dbi.Column{
+			TableName:     anyx.ToString(re["TABLE_NAME"]),
+			ColumnName:    anyx.ToString(re["COLUMN_NAME"]),
+			ColumnType:    anyx.ToString(re["COLUMN_TYPE"]),
+			ColumnComment: anyx.ToString(re["COLUMN_COMMENT"]),
+			Nullable:      anyx.ToString(re["NULLABLE"]),
+			IsPrimaryKey:  anyx.ConvInt(re["IS_PRIMARY_KEY"]) == 1,
+			IsIdentity:    anyx.ConvInt(re["IS_IDENTITY"]) == 1,
+			ColumnDefault: anyx.ToString(re["COLUMN_DEFAULT"]),
+			NumScale:      anyx.ToString(re["NUM_SCALE"]),
+		})
+	}
+	return columns, nil
+}
+
+// 获取表主键字段名，不存在主键标识则默认第一个字段
+func (md *MssqlDialect) GetPrimaryKey(tablename string) (string, error) {
+	columns, err := md.GetColumns(tablename)
+	if err != nil {
+		return "", err
+	}
+	if len(columns) == 0 {
+		return "", errorx.NewBiz("[%s] 表不存在", tablename)
+	}
+
+	for _, v := range columns {
+		if v.IsPrimaryKey {
+			return v.ColumnName, nil
+		}
+	}
+
+	return columns[0].ColumnName, nil
+}
+
+// 获取表索引信息
+func (md *MssqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
+	_, res, err := md.dc.Query(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_INDEX_INFO_KEY), md.currentSchema(), tableName)
+	if err != nil {
+		return nil, err
+	}
+
+	indexs := make([]dbi.Index, 0)
+	for _, re := range res {
+		indexs = append(indexs, dbi.Index{
+			IndexName:    anyx.ConvString(re["indexName"]),
+			ColumnName:   anyx.ConvString(re["columnName"]),
+			IndexType:    anyx.ConvString(re["indexType"]),
+			IndexComment: anyx.ConvString(re["indexComment"]),
+			IsUnique:     anyx.ConvInt(re["isUnique"]) == 1,
+			SeqInIndex:   anyx.ConvInt(re["seqInIndex"]),
+		})
+	}
+	// 把查询结果以索引名分组，索引字段以逗号连接
+	result := make([]dbi.Index, 0)
+	key := ""
+	for _, v := range indexs {
+		// 当前的索引名
+		in := v.IndexName
+		// 过滤掉主键索引,主键索引名为PK__开头的
+		if strings.HasPrefix(in, "PK__") {
+			continue
+		}
+		if key == in {
+			// 索引字段已根据名称和顺序排序，故取最后一个即可
+			i := len(result) - 1
+			// 同索引字段以逗号连接
+			result[i].ColumnName = result[i].ColumnName + "," + v.ColumnName
+		} else {
+			key = in
+			result = append(result, v)
+		}
+	}
+	return result, nil
+}
+
+func (md MssqlDialect) CopyTableDDL(tableName string, newTableName string) (string, error) {
+	if newTableName == "" {
+		newTableName = tableName
+	}
+
+	// 根据列信息生成建表语句
+	var builder strings.Builder
+	var commentBuilder strings.Builder
+
+	// 查询表名和表注释, 设置表注释
+	_, res, err := md.dc.Query(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_TABLE_DETAIL_KEY), md.currentSchema(), tableName)
+	if err != nil {
+		return "", err
+	}
+	tableComment := ""
+	if len(res) > 0 {
+		tableComment = anyx.ToString(res[0]["tableComment"])
+		if tableComment != "" {
+			// 注释转义单引号
+			tableComment = strings.ReplaceAll(tableComment, "'", "\\'")
+			commentBuilder.WriteString(fmt.Sprintf("\nEXEC sp_addextendedproperty N'MS_Description', N'%s', N'SCHEMA', N'%s', N'TABLE',N'%s';\n", tableComment, md.currentSchema(), newTableName))
+		}
+	}
+
+	baseTable := fmt.Sprintf("%s.%s", md.dc.Info.Type.QuoteIdentifier(md.currentSchema()), md.dc.Info.Type.QuoteIdentifier(newTableName))
+
+	// 查询列信息
+	columns, err := md.GetColumns(tableName)
+	if err != nil {
+		return "", err
+	}
+
+	builder.WriteString(fmt.Sprintf("CREATE TABLE %s (\n", baseTable))
+	pks := make([]string, 0)
+	for i, v := range columns {
+		nullAble := "NULL"
+		if v.Nullable == "NO" {
+			nullAble = "NOT NULL"
+		}
+		builder.WriteString(fmt.Sprintf("\t[%s] %s %s", v.ColumnName, v.ColumnType, nullAble))
+		if v.IsIdentity {
+			builder.WriteString(" IDENTITY(1,11)")
+		}
+		if v.ColumnDefault != "" {
+			builder.WriteString(fmt.Sprintf(" DEFAULT %s", v.ColumnDefault))
+		}
+		if v.IsPrimaryKey {
+			pks = append(pks, fmt.Sprintf("[%s]", v.ColumnName))
+		}
+		if i < len(columns)-1 {
+			builder.WriteString(",")
+		}
+		builder.WriteString("\n")
+	}
+	// 设置主键
+	if len(pks) > 0 {
+		builder.WriteString(fmt.Sprintf("\tCONSTRAINT PK_%s PRIMARY KEY ( %s )", newTableName, strings.Join(pks, ",")))
+	}
+	builder.WriteString("\n);\n")
+
+	// 设置字段注释
+	for _, v := range columns {
+		if v.ColumnComment != "" {
+			// 注释转义单引号
+			v.ColumnComment = strings.ReplaceAll(v.ColumnComment, "'", "\\'")
+			commentBuilder.WriteString(fmt.Sprintf("\nEXEC sp_addextendedproperty N'MS_Description', N'%s', N'SCHEMA', N'%s', N'TABLE',N'%s', N'COLUMN', N'%s';\n", v.ColumnComment, md.currentSchema(), newTableName, v.ColumnName))
+		}
+	}
+
+	// 设置索引
+	indexs, err := md.GetTableIndex(tableName)
+	if err != nil {
+		return "", err
+	}
+	for _, v := range indexs {
+		builder.WriteString(fmt.Sprintf("\nCREATE NONCLUSTERED INDEX [%s] ON %s (%s);\n", v.IndexName, baseTable, v.ColumnName))
+		// 设置索引注释
+		if v.IndexComment != "" {
+			// 注释转义单引号
+			v.IndexComment = strings.ReplaceAll(v.IndexComment, "'", "\\'")
+			commentBuilder.WriteString(fmt.Sprintf("\nEXEC sp_addextendedproperty N'MS_Description', N'%s', N'SCHEMA', N'%s', N'TABLE',N'%s', N'INDEX', N'%s';\n", v.IndexComment, md.currentSchema(), newTableName, v.IndexName))
+		}
+	}
+	return builder.String() + commentBuilder.String(), nil
+}
+
+// 获取建表ddl
+func (md *MssqlDialect) GetTableDDL(tableName string) (string, error) {
+	return md.CopyTableDDL(tableName, "")
+}
+
+func (md *MssqlDialect) WalkTableRecord(tableName string, walkFn dbi.WalkQueryRowsFunc) error {
+	return md.dc.WalkQueryRows(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
+}
+
+func (md *MssqlDialect) GetSchemas() ([]string, error) {
+	_, res, err := md.dc.Query(dbi.GetLocalSql(MSSQL_META_FILE, MSSQL_DB_SCHEMAS_KEY))
+	if err != nil {
+		return nil, err
+	}
+
+	schemas := make([]string, 0)
+	for _, re := range res {
+		schemas = append(schemas, anyx.ConvString(re["SCHEMA_NAME"]))
+	}
+	return schemas, nil
+}
+
+// GetDbProgram 获取数据库程序模块，用于数据库备份与恢复
+func (md *MssqlDialect) GetDbProgram() dbi.DbProgram {
+	panic("implement me")
+}
+
+func (md *MssqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
+	schema := md.currentSchema()
+
+	// 生成占位符字符串：如：(?,?)
+	// 重复字符串并用逗号连接
+	repeated := strings.Repeat("?,", len(columns))
+	// 去除最后一个逗号，占位符由括号包裹
+	placeholder := fmt.Sprintf("(%s)", strings.TrimSuffix(repeated, ","))
+
+	// 重复占位符字符串n遍
+	repeated = strings.Repeat(placeholder+",", len(values))
+	// 去除最后一个逗号
+	placeholder = strings.TrimSuffix(repeated, ",")
+
+	baseTable := fmt.Sprintf("%s.%s", md.dc.Info.Type.QuoteIdentifier(schema), md.dc.Info.Type.QuoteIdentifier(tableName))
+
+	sqlStr := fmt.Sprintf("insert into %s (%s) values %s", baseTable, strings.Join(columns, ","), placeholder)
+	// 执行批量insert sql
+	// 把二维数组转为一维数组
+	var args []any
+	for _, v := range values {
+		args = append(args, v...)
+	}
+	// 设置允许填充自增列之后，显示指定列名可以插入自增列
+	_, _ = md.dc.Exec(fmt.Sprintf("set identity_insert \"%s\" on", tableName))
+
+	exec, err := md.dc.TxExec(tx, sqlStr, args...)
+
+	_, _ = md.dc.Exec(fmt.Sprintf("set identity_insert \"%s\" off", tableName))
+
+	return exec, err
+}
+
+func (md *MssqlDialect) GetDataConverter() dbi.DataConverter {
+	return new(DataConverter)
+}
+
+var (
+	// 数字类型
+	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
+	// 日期时间类型
+	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
+	// 日期类型
+	dateRegexp = regexp.MustCompile(`(?i)date`)
+	// 时间类型
+	timeRegexp = regexp.MustCompile(`(?i)time`)
+)
+
+type DataConverter struct {
+}
+
+func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	// 日期时间类型
+	if datetimeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	// 日期类型
+	if dateRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDate
+	}
+	// 时间类型
+	if timeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	return anyx.ToString(dbColumnValue)
+}
+
+func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
+	return dbColumnValue
+}
+
+func (md *MssqlDialect) CopyTable(copy *dbi.DbCopyTable) error {
+
+	schema := md.currentSchema()
+
+	// 生成新表名,为老表明+_copy_时间戳
+	newTableName := copy.TableName + "_copy_" + time.Now().Format("20060102150405")
+
+	// 复制建表语句
+	ddl, err := md.CopyTableDDL(copy.TableName, newTableName)
+	if err != nil {
+		return err
+	}
+
+	// 执行建表
+	_, err = md.dc.Exec(ddl)
+	if err != nil {
+		return err
+	}
+	// 复制数据
+	if copy.CopyData {
+		go func() {
+			// 查询所有的列
+			columns, err := md.GetColumns(copy.TableName)
+			if err != nil {
+				logx.Warnf("复制表[%s]数据失败: %s", copy.TableName, err.Error())
+				return
+			}
+			// 取出每列名, 需要显示指定列名插入数据
+			columnNames := make([]string, 0)
+			hasIdentity := false
+			for _, v := range columns {
+				columnNames = append(columnNames, fmt.Sprintf("[%s]", v.ColumnName))
+				if v.IsIdentity {
+					hasIdentity = true
+				}
+
+			}
+			columnsSql := strings.Join(columnNames, ",")
+
+			// 复制数据
+			// 设置允许填充自增列之后，显示指定列名可以插入自增列
+			identityInsertOn := ""
+			identityInsertOff := ""
+			if hasIdentity {
+				identityInsertOn = fmt.Sprintf("SET IDENTITY_INSERT [%s].[%s] ON", schema, newTableName)
+				identityInsertOff = fmt.Sprintf("SET IDENTITY_INSERT [%s].[%s] OFF", schema, newTableName)
+			}
+			_, err = md.dc.Exec(fmt.Sprintf(" %s INSERT INTO [%s].[%s] (%s) SELECT * FROM [%s].[%s] %s", identityInsertOn, schema, newTableName, columnsSql, schema, copy.TableName, identityInsertOff))
+			if err != nil {
+				logx.Warnf("复制表[%s]数据失败: %s", copy.TableName, err.Error())
+			}
+		}()
+	}
+
+	return err
+}

server/internal/db/dbm/mssql/meta.go

@@ -0,0 +1,48 @@
+package mssql
+
+import (
+	"database/sql"
+	"fmt"
+	_ "github.com/microsoft/go-mssqldb"
+	"mayfly-go/internal/db/dbm/dbi"
+	"net/url"
+	"strings"
+)
+
+func init() {
+	meta := new(Meta)
+	dbi.Register(dbi.DbTypeMssql, meta)
+}
+
+type Meta struct {
+}
+
+func (md *Meta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
+	err := d.IfUseSshTunnelChangeIpPort()
+	if err != nil {
+		return nil, err
+	}
+	query := url.Values{}
+	// The application name (default is go-mssqldb)
+	query.Add("app name", "mayfly")
+	// 指定与服务器协商加密的最低TLS版本
+	query.Add("tlsmin", "1.0")
+	// 连接超时时间10秒
+	query.Add("connection timeout", "10")
+	if d.Database != "" {
+		ss := strings.Split(d.Database, "/")
+		if len(ss) > 1 {
+			query.Add("database", ss[0])
+			query.Add("schema", ss[1])
+		} else {
+			query.Add("database", d.Database)
+		}
+	}
+	const driverName = "mssql"
+	dsn := fmt.Sprintf("sqlserver://%s:%s@%s:%d?%s", url.PathEscape(d.Username), url.PathEscape(d.Password), d.Host, d.Port, query.Encode())
+	return sql.Open(driverName, dsn)
+}
+
+func (md *Meta) GetDialect(conn *dbi.DbConn) dbi.Dialect {
+	return &MssqlDialect{conn}
+}

server/internal/db/dbm/mysql/dialect.go

@@ -1,7 +1,6 @@
 package mysql
 
 import (
-	"context"
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
@@ -10,6 +9,7 @@ import (
 	"mayfly-go/pkg/utils/collx"
 	"regexp"
 	"strings"
+	"time"
 )
 
 const (
@@ -90,7 +90,8 @@ func (md *MysqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
 			ColumnType:    anyx.ConvString(re["columnType"]),
 			ColumnComment: anyx.ConvString(re["columnComment"]),
 			Nullable:      anyx.ConvString(re["nullable"]),
-			ColumnKey:     anyx.ConvString(re["columnKey"]),
+			IsPrimaryKey:  anyx.ConvInt(re["isPrimaryKey"]) == 1,
+			IsIdentity:    anyx.ConvInt(re["isIdentity"]) == 1,
 			ColumnDefault: anyx.ConvString(re["columnDefault"]),
 			NumScale:      anyx.ConvString(re["numScale"]),
 		})
@@ -109,7 +110,7 @@ func (md *MysqlDialect) GetPrimaryKey(tablename string) (string, error) {
 	}
 
 	for _, v := range columns {
-		if v.ColumnKey == "PRI" {
+		if v.IsPrimaryKey {
 			return v.ColumnName, nil
 		}
 	}
@@ -131,7 +132,7 @@ func (md *MysqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
 			ColumnName:   anyx.ConvString(re["columnName"]),
 			IndexType:    anyx.ConvString(re["indexType"]),
 			IndexComment: anyx.ConvString(re["indexComment"]),
-			NonUnique:    anyx.ConvInt(re["nonUnique"]),
+			IsUnique:     anyx.ConvInt(re["isUnique"]) == 1,
 			SeqInIndex:   anyx.ConvInt(re["seqInIndex"]),
 		})
 	}
@@ -163,10 +164,6 @@ func (md *MysqlDialect) GetTableDDL(tableName string) (string, error) {
 	return anyx.ConvString(res[0]["Create Table"]) + ";", nil
 }
 
-func (md *MysqlDialect) WalkTableRecord(tableName string, walkFn dbi.WalkQueryRowsFunc) error {
-	return md.dc.WalkQueryRows(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
-}
-
 func (md *MysqlDialect) GetSchemas() ([]string, error) {
 	return nil, nil
 }
@@ -176,25 +173,6 @@ func (md *MysqlDialect) GetDbProgram() dbi.DbProgram {
 	return NewDbProgramMysql(md.dc)
 }
 
-func (md *MysqlDialect) GetDataType(dbColumnType string) dbi.DataType {
-	if regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`).MatchString(dbColumnType) {
-		return dbi.DataTypeNumber
-	}
-	// 日期时间类型
-	if regexp.MustCompile(`(?i)datetime|timestamp`).MatchString(dbColumnType) {
-		return dbi.DataTypeDateTime
-	}
-	// 日期类型
-	if regexp.MustCompile(`(?i)date`).MatchString(dbColumnType) {
-		return dbi.DataTypeDate
-	}
-	// 时间类型
-	if regexp.MustCompile(`(?i)time`).MatchString(dbColumnType) {
-		return dbi.DataTypeTime
-	}
-	return dbi.DataTypeString
-}
-
 func (md *MysqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
 	// 生成占位符字符串：如：(?,?)
 	// 重复字符串并用逗号连接
@@ -220,7 +198,69 @@ func (md *MysqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []stri
 	return md.dc.TxExec(tx, sqlStr, args...)
 }
 
-func (md *MysqlDialect) FormatStrData(dbColumnValue string, dataType dbi.DataType) string {
-	// mysql不需要格式化时间日期等
+func (md *MysqlDialect) GetDataConverter() dbi.DataConverter {
+	return new(DataConverter)
+}
+
+var (
+	// 数字类型
+	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
+	// 日期时间类型
+	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
+	// 日期类型
+	dateRegexp = regexp.MustCompile(`(?i)date`)
+	// 时间类型
+	timeRegexp = regexp.MustCompile(`(?i)time`)
+)
+
+type DataConverter struct {
+}
+
+func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	// 日期时间类型
+	if datetimeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	// 日期类型
+	if dateRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDate
+	}
+	// 时间类型
+	if timeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	return anyx.ToString(dbColumnValue)
+}
+
+func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
 	return dbColumnValue
 }
+
+func (md *MysqlDialect) CopyTable(copy *dbi.DbCopyTable) error {
+
+	tableName := copy.TableName
+
+	// 生成新表名,为老表明+_copy_时间戳
+	newTableName := tableName + "_copy_" + time.Now().Format("20060102150405")
+
+	// 复制表结构创建表
+	_, err := md.dc.Exec(fmt.Sprintf("create table %s like %s", newTableName, tableName))
+	if err != nil {
+		return err
+	}
+
+	// 复制数据
+	if copy.CopyData {
+		go func() {
+			_, _ = md.dc.Exec(fmt.Sprintf("insert into %s select * from %s", newTableName, tableName))
+		}()
+	}
+	return err
+}

server/internal/db/dbm/mysql/meta.go

@@ -6,21 +6,14 @@ import (
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
 	"net"
-	"sync"
 
 	"github.com/go-sql-driver/mysql"
 )
 
-var (
-	meta dbi.Meta
-	once sync.Once
-)
-
-func GetMeta() dbi.Meta {
-	once.Do(func() {
-		meta = new(MysqlMeta)
-	})
-	return meta
+func init() {
+	meta := new(MysqlMeta)
+	dbi.Register(dbi.DbTypeMysql, meta)
+	dbi.Register(dbi.DbTypeMariadb, meta)
 }
 
 type MysqlMeta struct {

server/internal/db/dbm/mysql/program.go

@@ -15,11 +15,12 @@ import (
 	"strings"
 	"time"
 
-	"github.com/pkg/errors"
 	"mayfly-go/internal/db/config"
 	"mayfly-go/internal/db/dbm/dbi"
 	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/pkg/logx"
+
+	"github.com/pkg/errors"
 )
 
 var _ dbi.DbProgram = (*DbProgramMysql)(nil)
@@ -77,6 +78,15 @@ func (svc *DbProgramMysql) GetBinlogFilePath(fileName string) string {
 }
 
 func (svc *DbProgramMysql) Backup(ctx context.Context, backupHistory *entity.DbBackupHistory) (*entity.BinlogInfo, error) {
+	binlogEnabled, err := svc.CheckBinlogEnabled(ctx)
+	if err != nil {
+		return nil, err
+	}
+	rowFormatEnabled, err := svc.CheckBinlogRowFormat(ctx)
+	if err != nil {
+		return nil, err
+	}
+
 	dir := svc.getDbBackupDir(backupHistory.DbInstanceId, backupHistory.DbBackupId)
 	if err := os.MkdirAll(dir, os.ModePerm); err != nil {
 		return nil, err
@@ -95,10 +105,11 @@ func (svc *DbProgramMysql) Backup(ctx context.Context, backupHistory *entity.DbB
 		"--add-drop-database",
 		"--result-file", tmpFile,
 		"--single-transaction",
-		"--master-data=2",
 		"--databases", backupHistory.DbName,
 	}
-
+	if binlogEnabled && rowFormatEnabled {
+		args = append(args, "--master-data=2")
+	}
 	cmd := exec.CommandContext(ctx, svc.getMysqlBin().MysqldumpPath, args...)
 	logx.Debugf("backup database using mysqldump binary: %s", cmd.String())
 	if err := runCmd(cmd); err != nil {
@@ -115,7 +126,10 @@ func (svc *DbProgramMysql) Backup(ctx context.Context, backupHistory *entity.DbB
 	if err != nil {
 		return nil, err
 	}
-	binlogInfo, err := readBinlogInfoFromBackup(reader)
+	binlogInfo := &entity.BinlogInfo{}
+	if binlogEnabled && rowFormatEnabled {
+		binlogInfo, err = readBinlogInfoFromBackup(reader)
+	}
 	_ = reader.Close()
 	if err != nil {
 		return nil, errors.Wrapf(err, "从备份文件中读取 binlog 信息失败")
@@ -128,6 +142,12 @@ func (svc *DbProgramMysql) Backup(ctx context.Context, backupHistory *entity.DbB
 	return binlogInfo, nil
 }
 
+func (svc *DbProgramMysql) RemoveBackupHistory(_ context.Context, dbBackupId uint64, dbBackupHistoryUuid string) error {
+	fileName := filepath.Join(svc.getDbBackupDir(svc.dbInfo().InstanceId, dbBackupId),
+		fmt.Sprintf("%v.sql", dbBackupHistoryUuid))
+	return os.Remove(fileName)
+}
+
 func (svc *DbProgramMysql) RestoreBackupHistory(ctx context.Context, dbName string, dbBackupId uint64, dbBackupHistoryUuid string) error {
 	dbInfo := svc.dbInfo()
 	args := []string{
@@ -467,7 +487,7 @@ func (svc *DbProgramMysql) GetBinlogEventPositionAtOrAfterTime(ctx context.Conte
 			return posParsed, nil
 		}
 	}
-	return 0, errors.Errorf("在 %s 之后没有 binlog 事件", targetTime.Format(time.DateTime))
+	return 0, errors.Errorf("在 %s 之后没有 binlog 事件", targetTime.Local().Format(time.DateTime))
 }
 
 // ReplayBinlog replays the binlog for `originDatabase` from `startBinlogInfo.Position` to `targetTs`, read binlog from `binlogDir`.
@@ -568,18 +588,24 @@ func (svc *DbProgramMysql) ReplayBinlog(ctx context.Context, originalDatabase, t
 		return errors.Wrap(err, "启动 mysqlbinlog 程序失败")
 	}
 	defer func() {
+		_ = mysqlbinlogCmd.Cancel()
 		if err := mysqlbinlogCmd.Wait(); err != nil {
-			if replayErr != nil {
-				replayErr = errors.Wrap(replayErr, "运行 mysqlbinlog 程序失败")
-			} else {
-				replayErr = errors.Errorf("运行 mysqlbinlog 程序失败: %s", mysqlbinlogErr.String())
+			if mysqlbinlogErr.Len() > 0 {
+				logx.Errorf("运行 mysqlbinlog 程序失败: %s", mysqlbinlogErr.String())
+				if replayErr != nil {
+					replayErr = errors.Wrap(replayErr, "运行 mysqlbinlog 程序失败: "+mysqlbinlogErr.String())
+				} else {
+					replayErr = errors.Errorf("运行 mysqlbinlog 程序失败: %s", mysqlbinlogErr.String())
+				}
 			}
 		}
 	}()
 	if err := mysqlCmd.Start(); err != nil {
+		logx.Error("启动 mysql 程序失败")
 		return errors.Wrap(err, "启动 mysql 程序失败")
 	}
 	if err := mysqlCmd.Wait(); err != nil {
+		logx.Errorf("运行 mysql 程序失败: %s", mysqlErr.String())
 		return errors.Errorf("运行 mysql 程序失败: %s", mysqlErr.String())
 	}
 
@@ -606,27 +632,30 @@ func (svc *DbProgramMysql) getServerVariable(_ context.Context, varName string)
 }
 
 // CheckBinlogEnabled checks whether binlog is enabled for the current instance.
-func (svc *DbProgramMysql) CheckBinlogEnabled(ctx context.Context) error {
+func (svc *DbProgramMysql) CheckBinlogEnabled(ctx context.Context) (bool, error) {
 	value, err := svc.getServerVariable(ctx, "log_bin")
-	if err != nil {
-		return err
+	switch {
+	case err == nil:
+		return strings.ToUpper(value) == "ON", nil
+	case errors.Is(err, sql.ErrNoRows):
+		return false, nil
+	default:
+		return false, err
 	}
-	if strings.ToUpper(value) != "ON" {
-		return errors.Errorf("数据库未启用 binlog")
-	}
-	return nil
 }
 
-// CheckBinlogRowFormat checks whether the binlog format is ROW.
-func (svc *DbProgramMysql) CheckBinlogRowFormat(ctx context.Context) error {
+// CheckBinlogRowFormat checks whether the binlog format is ROW or MIXED.
+func (svc *DbProgramMysql) CheckBinlogRowFormat(ctx context.Context) (bool, error) {
 	value, err := svc.getServerVariable(ctx, "binlog_format")
-	if err != nil {
-		return err
+	switch {
+	case err == nil:
+		value = strings.ToUpper(value)
+		return value == "ROW" || value == "MIXED", nil
+	case errors.Is(err, sql.ErrNoRows):
+		return false, nil
+	default:
+		return false, err
 	}
-	if strings.ToUpper(value) != "ROW" {
-		return errors.Errorf("binlog 格式 %s 不是行模式", value)
-	}
-	return nil
 }
 
 func runCmd(cmd *exec.Cmd) error {

server/internal/db/dbm/mysql/program_e2e_test.go

@@ -153,8 +153,9 @@ func (s *DbInstanceSuite) TestRestorePontInTime() {
 	s.createTable(dbNameBackupTest, tableNameRestorePITTest, "")
 	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "")
 	time.Sleep(time.Second)
-	targetTime := time.Now()
 
+	// 首次恢复数据库
+	firstTargetTime := time.Now()
 	s.dropTable(dbNameBackupTest, tableNameBackupTest, "")
 	s.selectTable(dbNameBackupTest, tableNameBackupTest, "运行 mysql 程序失败")
 	s.createTable(dbNameBackupTest, tableNameNoBackupTest, "")
@@ -165,7 +166,28 @@ func (s *DbInstanceSuite) TestRestorePontInTime() {
 	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "运行 mysql 程序失败")
 	s.selectTable(dbNameBackupTest, tableNameNoBackupTest, "运行 mysql 程序失败")
 
-	s.testReplayBinlog(backupHistory, targetTime)
+	s.testReplayBinlog(backupHistory, firstTargetTime)
+	s.selectTable(dbNameBackupTest, tableNameBackupTest, "")
+	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "")
+	s.selectTable(dbNameBackupTest, tableNameNoBackupTest, "运行 mysql 程序失败")
+
+	s.testBackup(backupHistory)
+
+	// 再次恢复数据库
+	secondTargetTime := time.Now()
+	s.dropTable(dbNameBackupTest, tableNameBackupTest, "")
+	s.selectTable(dbNameBackupTest, tableNameBackupTest, "运行 mysql 程序失败")
+	s.dropTable(dbNameBackupTest, tableNameRestorePITTest, "")
+	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "运行 mysql 程序失败")
+	s.createTable(dbNameBackupTest, tableNameNoBackupTest, "")
+	s.selectTable(dbNameBackupTest, tableNameNoBackupTest, "")
+
+	s.testRestore(backupHistory)
+	s.selectTable(dbNameBackupTest, tableNameBackupTest, "")
+	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "")
+	s.selectTable(dbNameBackupTest, tableNameNoBackupTest, "运行 mysql 程序失败")
+
+	s.testReplayBinlog(backupHistory, secondTargetTime)
 	s.selectTable(dbNameBackupTest, tableNameBackupTest, "")
 	s.selectTable(dbNameBackupTest, tableNameRestorePITTest, "")
 	s.selectTable(dbNameBackupTest, tableNameNoBackupTest, "运行 mysql 程序失败")

server/internal/db/dbm/oracle/dialect.go

@@ -1,14 +1,12 @@
 package oracle
 
 import (
-	"context"
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
 	"mayfly-go/pkg/errorx"
 	"mayfly-go/pkg/utils/anyx"
 	"mayfly-go/pkg/utils/collx"
-	"reflect"
 	"regexp"
 	"strings"
 	"time"
@@ -105,7 +103,8 @@ func (od *OracleDialect) GetColumns(tableNames ...string) ([]dbi.Column, error)
 			ColumnType:    anyx.ConvString(re["COLUMN_TYPE"]),
 			ColumnComment: anyx.ConvString(re["COLUMN_COMMENT"]),
 			Nullable:      anyx.ConvString(re["NULLABLE"]),
-			ColumnKey:     anyx.ConvString(re["COLUMN_KEY"]),
+			IsPrimaryKey:  anyx.ConvInt(re["IS_PRIMARY_KEY"]) == 1,
+			IsIdentity:    anyx.ConvInt(re["IS_IDENTITY"]) == 1,
 			ColumnDefault: defaultVal,
 			NumScale:      anyx.ConvString(re["NUM_SCALE"]),
 		})
@@ -122,7 +121,7 @@ func (od *OracleDialect) GetPrimaryKey(tablename string) (string, error) {
 		return "", errorx.NewBiz("[%s] 表不存在", tablename)
 	}
 	for _, v := range columns {
-		if v.ColumnKey == "PRI" {
+		if v.IsPrimaryKey {
 			return v.ColumnName, nil
 		}
 	}
@@ -144,7 +143,7 @@ func (od *OracleDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
 			ColumnName:   anyx.ConvString(re["COLUMN_NAME"]),
 			IndexType:    anyx.ConvString(re["INDEX_TYPE"]),
 			IndexComment: anyx.ConvString(re["INDEX_COMMENT"]),
-			NonUnique:    anyx.ConvInt(re["NON_UNIQUE"]),
+			IsUnique:     anyx.ConvInt(re["IS_UNIQUE"]) == 1,
 			SeqInIndex:   anyx.ConvInt(re["SEQ_IN_INDEX"]),
 		})
 	}
@@ -234,10 +233,6 @@ func (od *OracleDialect) GetTableDDL(tableName string) (string, error) {
 	return builder.String(), nil
 }
 
-func (od *OracleDialect) WalkTableRecord(tableName string, walkFn dbi.WalkQueryRowsFunc) error {
-	return od.dc.WalkQueryRows(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
-}
-
 // 获取DM当前连接的库可访问的schemaNames
 func (od *OracleDialect) GetSchemas() ([]string, error) {
 	sql := dbi.GetLocalSql(ORACLE_META_FILE, ORACLE_DB_SCHEMAS)
@@ -257,25 +252,6 @@ func (od *OracleDialect) GetDbProgram() dbi.DbProgram {
 	panic("implement me")
 }
 
-func (od *OracleDialect) GetDataType(dbColumnType string) dbi.DataType {
-	if regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`).MatchString(dbColumnType) {
-		return dbi.DataTypeNumber
-	}
-	// 日期时间类型
-	if regexp.MustCompile(`(?i)datetime|timestamp`).MatchString(dbColumnType) {
-		return dbi.DataTypeDateTime
-	}
-	// 日期类型
-	if regexp.MustCompile(`(?i)date`).MatchString(dbColumnType) {
-		return dbi.DataTypeDate
-	}
-	// 时间类型
-	if regexp.MustCompile(`(?i)time`).MatchString(dbColumnType) {
-		return dbi.DataTypeTime
-	}
-	return dbi.DataTypeString
-}
-
 func (od *OracleDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
 	//INSERT ALL
 	//INTO my_table(field_1,field_2) VALUES (value_1,value_2)
@@ -301,20 +277,6 @@ func (od *OracleDialect) BatchInsert(tx *sql.Tx, tableName string, columns []str
 	for i := 0; i < len(args); i += len(columns) {
 		var placeholder []string
 		for j := 0; j < len(columns); j++ {
-			// 判断字符串数据格式是时间"2023-06-25 10:40:10"  占位符需要变成 to_date(:x, 'fmt')
-			if reflect.TypeOf(args[i+j]) == reflect.TypeOf("") {
-				if regexp.MustCompile(`^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$`).MatchString(args[i+j].(string)) {
-					placeholder = append(placeholder, fmt.Sprintf("to_date(:%d, 'yyyy-mm-dd hh24:mi:ss')", i+j+1))
-				} else if regexp.MustCompile(`^\d{4}-\d{2}-\d{2}$`).MatchString(args[i+j].(string)) {
-					// 只有年月日的数据，oracle会自动补零时分秒，如：2024-01-02: to_date('2024-01-02','yyyy-mm-dd') 输出：2024-01-02 00:00:00
-					placeholder = append(placeholder, fmt.Sprintf("to_date(:%d, 'yyyy-mm-dd')", i+j+1))
-				} else if regexp.MustCompile(`^\d{2}:\d{2}:\d{2}$`).MatchString(args[i+j].(string)) {
-					// 只有时间的数据，oracle会拼接当前月份的年月日，如当前月份是2024-01: to_date('13:23:11','hh24:mi:ss') 输出：2024-01-01 13:23:11
-					placeholder = append(placeholder, fmt.Sprintf("to_date(:%d, 'hh24:mi:ss')", i+j+1))
-				}
-				continue
-			}
-
 			placeholder = append(placeholder, fmt.Sprintf(":%d", i+j+1))
 		}
 		sqlArr = append(sqlArr, fmt.Sprintf("INTO %s (%s) VALUES (%s)", od.dc.Info.Type.QuoteIdentifier(tableName), strings.Join(columns, ","), strings.Join(placeholder, ",")))
@@ -326,17 +288,58 @@ func (od *OracleDialect) BatchInsert(tx *sql.Tx, tableName string, columns []str
 	return res, err
 }
 
-func (od *OracleDialect) FormatStrData(dbColumnValue string, dataType dbi.DataType) string {
+func (od *OracleDialect) GetDataConverter() dbi.DataConverter {
+	return new(DataConverter)
+}
+
+var (
+	// 数字类型
+	numberTypeRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
+	// 日期时间类型
+	datetimeTypeRegexp = regexp.MustCompile(`(?i)date|timestamp`)
+)
+
+type DataConverter struct {
+}
+
+func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberTypeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	// 日期时间类型
+	if datetimeTypeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	str := anyx.ToString(dbColumnValue)
 	switch dataType {
+	// oracle把日期类型数据格式化输出
 	case dbi.DataTypeDateTime: // "2024-01-02T22:08:22.275697+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.DateTime)
-	case dbi.DataTypeDate: // "2024-01-02T00:00:00+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
-		return res.Format(time.DateOnly)
-	case dbi.DataTypeTime: // "0000-01-01T22:08:22.275688+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
-		return res.Format(time.TimeOnly)
+	}
+	return str
+}
+
+func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
+	// oracle把日期类型的数据转化为time类型
+	if dataType == dbi.DataTypeDateTime {
+		res, _ := time.Parse(time.RFC3339, anyx.ConvString(dbColumnValue))
+		return res
 	}
 	return dbColumnValue
 }
+
+func (od *OracleDialect) CopyTable(copy *dbi.DbCopyTable) error {
+	// 生成新表名,为老表明+_copy_时间戳
+	newTableName := strings.ToUpper(copy.TableName + "_copy_" + time.Now().Format("20060102150405"))
+	condition := ""
+	if !copy.CopyData {
+		condition = " where 1 = 2"
+	}
+	_, err := od.dc.Exec(fmt.Sprintf("create table \"%s\" as select * from \"%s\" %s", newTableName, copy.TableName, condition))
+	return err
+}

server/internal/db/dbm/oracle/meta.go

@@ -5,21 +5,12 @@ import (
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
 	"strings"
-	"sync"
 
 	go_ora "github.com/sijms/go-ora/v2"
 )
 
-var (
-	meta dbi.Meta
-	once sync.Once
-)
-
-func GetMeta() dbi.Meta {
-	once.Do(func() {
-		meta = new(OraMeta)
-	})
-	return meta
+func init() {
+	dbi.Register(dbi.DbTypeOracle, new(OraMeta))
 }
 
 type OraMeta struct {

server/internal/db/dbm/postgres/dialect.go

@@ -1,7 +1,6 @@
 package postgres
 
 import (
-	"context"
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
@@ -26,8 +25,8 @@ type PgsqlDialect struct {
 	dc *dbi.DbConn
 }
 
-func (pd *PgsqlDialect) GetDbServer() (*dbi.DbServer, error) {
-	_, res, err := pd.dc.Query("SHOW server_version")
+func (md *PgsqlDialect) GetDbServer() (*dbi.DbServer, error) {
+	_, res, err := md.dc.Query("SHOW server_version")
 	if err != nil {
 		return nil, err
 	}
@@ -37,8 +36,8 @@ func (pd *PgsqlDialect) GetDbServer() (*dbi.DbServer, error) {
 	return ds, nil
 }
 
-func (pd *PgsqlDialect) GetDbNames() ([]string, error) {
-	_, res, err := pd.dc.Query("SELECT datname AS dbname FROM pg_database WHERE datistemplate = false AND has_database_privilege(datname, 'CONNECT')")
+func (md *PgsqlDialect) GetDbNames() ([]string, error) {
+	_, res, err := md.dc.Query("SELECT datname AS dbname FROM pg_database WHERE datistemplate = false AND has_database_privilege(datname, 'CONNECT')")
 	if err != nil {
 		return nil, err
 	}
@@ -52,8 +51,8 @@ func (pd *PgsqlDialect) GetDbNames() ([]string, error) {
 }
 
 // 获取表基础元信息, 如表名等
-func (pd *PgsqlDialect) GetTables() ([]dbi.Table, error) {
-	_, res, err := pd.dc.Query(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_TABLE_INFO_KEY))
+func (md *PgsqlDialect) GetTables() ([]dbi.Table, error) {
+	_, res, err := md.dc.Query(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_TABLE_INFO_KEY))
 	if err != nil {
 		return nil, err
 	}
@@ -73,13 +72,13 @@ func (pd *PgsqlDialect) GetTables() ([]dbi.Table, error) {
 }
 
 // 获取列元信息, 如列名等
-func (pd *PgsqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
-	dbType := pd.dc.Info.Type
+func (md *PgsqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
+	dbType := md.dc.Info.Type
 	tableName := strings.Join(collx.ArrayMap[string, string](tableNames, func(val string) string {
 		return fmt.Sprintf("'%s'", dbType.RemoveQuote(val))
 	}), ",")
 
-	_, res, err := pd.dc.Query(fmt.Sprintf(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_COLUMN_MA_KEY), tableName))
+	_, res, err := md.dc.Query(fmt.Sprintf(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_COLUMN_MA_KEY), tableName))
 	if err != nil {
 		return nil, err
 	}
@@ -92,7 +91,8 @@ func (pd *PgsqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
 			ColumnType:    anyx.ConvString(re["columnType"]),
 			ColumnComment: anyx.ConvString(re["columnComment"]),
 			Nullable:      anyx.ConvString(re["nullable"]),
-			ColumnKey:     anyx.ConvString(re["columnKey"]),
+			IsPrimaryKey:  anyx.ConvInt(re["isPrimaryKey"]) == 1,
+			IsIdentity:    anyx.ConvInt(re["isIdentity"]) == 1,
 			ColumnDefault: anyx.ConvString(re["columnDefault"]),
 			NumScale:      anyx.ConvString(re["numScale"]),
 		})
@@ -100,8 +100,8 @@ func (pd *PgsqlDialect) GetColumns(tableNames ...string) ([]dbi.Column, error) {
 	return columns, nil
 }
 
-func (pd *PgsqlDialect) GetPrimaryKey(tablename string) (string, error) {
-	columns, err := pd.GetColumns(tablename)
+func (md *PgsqlDialect) GetPrimaryKey(tablename string) (string, error) {
+	columns, err := md.GetColumns(tablename)
 	if err != nil {
 		return "", err
 	}
@@ -109,7 +109,7 @@ func (pd *PgsqlDialect) GetPrimaryKey(tablename string) (string, error) {
 		return "", errorx.NewBiz("[%s] 表不存在", tablename)
 	}
 	for _, v := range columns {
-		if v.ColumnKey == "PRI" {
+		if v.IsPrimaryKey {
 			return v.ColumnName, nil
 		}
 	}
@@ -118,8 +118,8 @@ func (pd *PgsqlDialect) GetPrimaryKey(tablename string) (string, error) {
 }
 
 // 获取表索引信息
-func (pd *PgsqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
-	_, res, err := pd.dc.Query(fmt.Sprintf(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_INDEX_INFO_KEY), tableName))
+func (md *PgsqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
+	_, res, err := md.dc.Query(fmt.Sprintf(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_INDEX_INFO_KEY), tableName))
 	if err != nil {
 		return nil, err
 	}
@@ -131,7 +131,7 @@ func (pd *PgsqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
 			ColumnName:   anyx.ConvString(re["columnName"]),
 			IndexType:    anyx.ConvString(re["IndexType"]),
 			IndexComment: anyx.ConvString(re["indexComment"]),
-			NonUnique:    anyx.ConvInt(re["nonUnique"]),
+			IsUnique:     anyx.ConvInt(re["isUnique"]) == 1,
 			SeqInIndex:   anyx.ConvInt(re["seqInIndex"]),
 		})
 	}
@@ -155,17 +155,17 @@ func (pd *PgsqlDialect) GetTableIndex(tableName string) ([]dbi.Index, error) {
 }
 
 // 获取建表ddl
-func (pd *PgsqlDialect) GetTableDDL(tableName string) (string, error) {
-	_, err := pd.dc.Exec(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_TABLE_DDL_KEY))
+func (md *PgsqlDialect) GetTableDDL(tableName string) (string, error) {
+	_, err := md.dc.Exec(dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_TABLE_DDL_KEY))
 	if err != nil {
 		return "", err
 	}
 
-	_, schemaRes, _ := pd.dc.Query("select current_schema() as schema")
+	_, schemaRes, _ := md.dc.Query("select current_schema() as schema")
 	schemaName := schemaRes[0]["schema"].(string)
 
 	ddlSql := fmt.Sprintf("select showcreatetable('%s','%s') as sql", schemaName, tableName)
-	_, res, err := pd.dc.Query(ddlSql)
+	_, res, err := md.dc.Query(ddlSql)
 	if err != nil {
 		return "", err
 	}
@@ -173,14 +173,10 @@ func (pd *PgsqlDialect) GetTableDDL(tableName string) (string, error) {
 	return res[0]["sql"].(string), nil
 }
 
-func (pd *PgsqlDialect) WalkTableRecord(tableName string, walkFn dbi.WalkQueryRowsFunc) error {
-	return pd.dc.WalkQueryRows(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName), walkFn)
-}
-
 // 获取pgsql当前连接的库可访问的schemaNames
-func (pd *PgsqlDialect) GetSchemas() ([]string, error) {
+func (md *PgsqlDialect) GetSchemas() ([]string, error) {
 	sql := dbi.GetLocalSql(PGSQL_META_FILE, PGSQL_DB_SCHEMAS)
-	_, res, err := pd.dc.Query(sql)
+	_, res, err := md.dc.Query(sql)
 	if err != nil {
 		return nil, err
 	}
@@ -192,30 +188,11 @@ func (pd *PgsqlDialect) GetSchemas() ([]string, error) {
 }
 
 // GetDbProgram 获取数据库程序模块，用于数据库备份与恢复
-func (pd *PgsqlDialect) GetDbProgram() dbi.DbProgram {
+func (md *PgsqlDialect) GetDbProgram() dbi.DbProgram {
 	panic("implement me")
 }
 
-func (pd *PgsqlDialect) GetDataType(dbColumnType string) dbi.DataType {
-	if regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`).MatchString(dbColumnType) {
-		return dbi.DataTypeNumber
-	}
-	// 日期时间类型
-	if regexp.MustCompile(`(?i)datetime|timestamp`).MatchString(dbColumnType) {
-		return dbi.DataTypeDateTime
-	}
-	// 日期类型
-	if regexp.MustCompile(`(?i)date`).MatchString(dbColumnType) {
-		return dbi.DataTypeDate
-	}
-	// 时间类型
-	if regexp.MustCompile(`(?i)time`).MatchString(dbColumnType) {
-		return dbi.DataTypeTime
-	}
-	return dbi.DataTypeString
-}
-
-func (pd *PgsqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
+func (md *PgsqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []string, values [][]any) (int64, error) {
 	// 执行批量insert sql，跟mysql一样  pg或高斯支持批量insert语法
 	// insert into table_name (column1, column2, ...) values (value1, value2, ...), (value1, value2, ...), ...
 
@@ -235,23 +212,127 @@ func (pd *PgsqlDialect) BatchInsert(tx *sql.Tx, tableName string, columns []stri
 		placeholders = append(placeholders, "("+strings.Join(placeholder, ", ")+")")
 	}
 
-	sqlStr := fmt.Sprintf("insert into %s (%s) values %s", pd.dc.Info.Type.QuoteIdentifier(tableName), strings.Join(columns, ","), strings.Join(placeholders, ", "))
+	sqlStr := fmt.Sprintf("insert into %s (%s) values %s", md.dc.Info.Type.QuoteIdentifier(tableName), strings.Join(columns, ","), strings.Join(placeholders, ", "))
 	// 执行批量insert sql
 
-	return pd.dc.TxExec(tx, sqlStr, args...)
+	return md.dc.TxExec(tx, sqlStr, args...)
 }
 
-func (pd *PgsqlDialect) FormatStrData(dbColumnValue string, dataType dbi.DataType) string {
+func (md *PgsqlDialect) GetDataConverter() dbi.DataConverter {
+	return new(DataConverter)
+}
+
+var (
+	// 数字类型
+	numberRegexp = regexp.MustCompile(`(?i)int|double|float|number|decimal|byte|bit`)
+	// 日期时间类型
+	datetimeRegexp = regexp.MustCompile(`(?i)datetime|timestamp`)
+	// 日期类型
+	dateRegexp = regexp.MustCompile(`(?i)date`)
+	// 时间类型
+	timeRegexp = regexp.MustCompile(`(?i)time`)
+)
+
+type DataConverter struct {
+}
+
+func (dc *DataConverter) GetDataType(dbColumnType string) dbi.DataType {
+	if numberRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeNumber
+	}
+	// 日期时间类型
+	if datetimeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDateTime
+	}
+	// 日期类型
+	if dateRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeDate
+	}
+	// 时间类型
+	if timeRegexp.MatchString(dbColumnType) {
+		return dbi.DataTypeTime
+	}
+	return dbi.DataTypeString
+}
+
+func (dc *DataConverter) FormatData(dbColumnValue any, dataType dbi.DataType) string {
+	str := fmt.Sprintf("%v", dbColumnValue)
 	switch dataType {
 	case dbi.DataTypeDateTime: // "2024-01-02T22:16:28.545377+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.DateTime)
 	case dbi.DataTypeDate: //  "2024-01-02T00:00:00Z"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.DateOnly)
 	case dbi.DataTypeTime: // "0000-01-01T22:16:28.545075+08:00"
-		res, _ := time.Parse(time.RFC3339, dbColumnValue)
+		res, _ := time.Parse(time.RFC3339, str)
 		return res.Format(time.TimeOnly)
 	}
+	return anyx.ConvString(dbColumnValue)
+}
+
+func (dc *DataConverter) ParseData(dbColumnValue any, dataType dbi.DataType) any {
 	return dbColumnValue
 }
+
+func (md *PgsqlDialect) IsGauss() bool {
+	return strings.Contains(md.dc.Info.Params, "gauss")
+}
+
+func (md *PgsqlDialect) CopyTable(copy *dbi.DbCopyTable) error {
+	tableName := copy.TableName
+	// 生成新表名,为老表明+_copy_时间戳
+	newTableName := tableName + "_copy_" + time.Now().Format("20060102150405")
+	// 执行根据旧表创建新表
+	_, err := md.dc.Exec(fmt.Sprintf("create table %s (like %s)", newTableName, tableName))
+	if err != nil {
+		return err
+	}
+
+	// 复制数据
+	if copy.CopyData {
+		go func() {
+			_, _ = md.dc.Exec(fmt.Sprintf("insert into %s select * from %s", newTableName, tableName))
+		}()
+	}
+
+	// 查询旧表的自增字段名 重新设置新表的序列序列器
+	_, res, err := md.dc.Query(fmt.Sprintf("select column_name from information_schema.columns where table_name = '%s' and column_default like 'nextval%%'", tableName))
+	if err != nil {
+		return err
+	}
+
+	for _, re := range res {
+		colName := anyx.ConvString(re["column_name"])
+		if colName != "" {
+
+			// 查询自增列当前最大值
+			_, maxRes, err := md.dc.Query(fmt.Sprintf("select max(%s) max_val from %s", colName, tableName))
+			if err != nil {
+				return err
+			}
+			maxVal := anyx.ConvInt(maxRes[0]["max_val"])
+			// 序列起始值为1或当前最大值+1
+			if maxVal <= 0 {
+				maxVal = 1
+			} else {
+				maxVal += 1
+			}
+
+			// 之所以不用tableName_colName_seq是因为gauss会自动创建同名的序列，且无法修改序列起始值，所以直接使用新序列值
+			newSeqName := fmt.Sprintf("%s_%s_copy_seq", newTableName, colName)
+
+			// 创建自增序列，当前最大值为旧表最大值
+			_, err = md.dc.Exec(fmt.Sprintf("CREATE SEQUENCE %s START %d INCREMENT 1", newSeqName, maxVal))
+			if err != nil {
+				return err
+			}
+			// 将新表的自增主键序列与主键列相关联
+			_, err = md.dc.Exec(fmt.Sprintf("alter table %s alter column %s set default nextval('%s')", newTableName, colName, newSeqName))
+			if err != nil {
+				return err
+			}
+		}
+	}
+	return err
+}

server/internal/db/dbm/postgres/meta.go

@@ -9,29 +9,25 @@ import (
 	"mayfly-go/pkg/utils/netx"
 	"net"
 	"strings"
-	"sync"
 	"time"
 
 	pq "gitee.com/liuzongyang/libpq"
 )
 
-var (
-	meta dbi.Meta
-	once sync.Once
-)
+func init() {
+	dbi.Register(dbi.DbTypePostgres, new(PostgresMeta))
 
-func GetMeta() dbi.Meta {
-	once.Do(func() {
-		meta = new(PostgresMeta)
-	})
-	return meta
+	gauss := new(PostgresMeta)
+	gauss.Param = "dbtype=gauss"
+	dbi.Register(dbi.DbTypeGauss, gauss)
 }
 
 type PostgresMeta struct {
+	Param string
 }
 
 func (md *PostgresMeta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
-	driverName := string(d.Type)
+	driverName := "postgres"
 	// SSH Conect
 	if d.SshTunnelMachineId > 0 {
 		// 如果使用了隧道，则使用`postgres:ssh:隧道机器id`注册名
@@ -76,6 +72,10 @@ func (md *PostgresMeta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
 		dsn = fmt.Sprintf("%s %s", dsn, strings.Join(strings.Split(d.Params, "&"), " "))
 	}
 
+	if md.Param != "" && !strings.Contains(dsn, "dbtype") {
+		dsn = fmt.Sprintf("%s %s", dsn, md.Param)
+	}
+
 	return sql.Open(driverName, dsn)
 }