mayfly-go/mayfly_go_web/src/views/ops/db/DbEdit.vue

<template>
    <div>
        <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" :close-on-click-modal="false"
            :destroy-on-close="true" width="38%">
            <el-form :model="form" ref="dbForm" :rules="rules" label-width="95px">
                <el-form-item prop="tagId" label="标签:" required>
                    <tag-select v-model:tag-id="form.tagId" v-model:tag-path="form.tagPath" style="width: 100%" />
                </el-form-item>

                <el-form-item prop="name" label="别名:" required>
                    <el-input v-model.trim="form.name" placeholder="请输入数据库别名" auto-complete="off"></el-input>
                </el-form-item>
                <el-form-item prop="type" label="类型:" required>
                    <el-select style="width: 100%" v-model="form.type" placeholder="请选择数据库类型">
                        <el-option key="item.id" label="mysql" value="mysql"> </el-option>
                        <el-option key="item.id" label="postgres" value="postgres"> </el-option>
                    </el-select>
                </el-form-item>
                <el-form-item prop="host" label="host:" required>
                    <el-col :span="18">
                        <el-input :disabled="form.id !== undefined" v-model.trim="form.host" placeholder="请输入主机ip"
                            auto-complete="off"></el-input>
                    </el-col>
                    <el-col style="text-align: center" :span="1">:</el-col>
                    <el-col :span="5">
                        <el-input type="number" v-model.number="form.port" placeholder="请输入端口"></el-input>
                    </el-col>
                </el-form-item>
                <el-form-item prop="username" label="用户名:" required>
                    <el-input v-model.trim="form.username" placeholder="请输入用户名"></el-input>
                </el-form-item>
                <el-form-item prop="password" label="密码:">
                    <el-input type="password" show-password v-model.trim="form.password" placeholder="请输入密码，修改操作可不填"
                        autocomplete="new-password">
                        <template v-if="form.id && form.id != 0" #suffix>
                            <el-popover @hide="pwd = ''" placement="right" title="原密码" :width="200" trigger="click"
                                :content="pwd">
                                <template #reference>
                                    <el-link @click="getDbPwd" :underline="false" type="primary" class="mr5">原密码
                                    </el-link>
                                </template>
                            </el-popover>
                        </template>
                    </el-input>
                </el-form-item>
                <el-form-item prop="params" label="连接参数:">
                    <el-input v-model.trim="form.params" placeholder="其他连接参数，形如: key1=value1&key2=value2">
                        <template #suffix>
                            <el-link target="_blank" href="https://github.com/go-sql-driver/mysql#parameters"
                                :underline="false" type="primary" class="mr5">参数参考</el-link>
                        </template>
                    </el-input>
                </el-form-item>
                <el-form-item prop="database" label="数据库名:" required>
                    <el-col :span="19">
                        <el-select @change="changeDatabase" v-model="databaseList" multiple clearable collapse-tags
                            collapse-tags-tooltip filterable allow-create placeholder="请确保数据库实例信息填写完整后获取库名"
                            style="width: 100%">
                            <el-option v-for="db in allDatabases" :key="db" :label="db" :value="db" />
                        </el-select>
                    </el-col>
                    <el-col style="text-align: center" :span="1">
                        <el-divider direction="vertical" border-style="dashed" />
                    </el-col>
                    <el-col :span="4">
                        <el-link @click="getAllDatabase" :underline="false" type="success">获取库名</el-link>
                    </el-col>
                </el-form-item>

                <el-form-item prop="remark" label="备注:">
                    <el-input v-model.trim="form.remark" auto-complete="off" type="textarea"></el-input>
                </el-form-item>

                <el-form-item prop="enableSshTunnel" label="SSH隧道:">
                    <el-col :span="3">
                        <el-checkbox @change="getSshTunnelMachines" v-model="form.enableSshTunnel" :true-label="1"
                            :false-label="-1"></el-checkbox>
                    </el-col>
                    <el-col :span="5" v-if="form.enableSshTunnel == 1"> 机器: </el-col>
                    <el-col :span="16" v-if="form.enableSshTunnel == 1">
                        <el-select style="width: 100%" v-model="form.sshTunnelMachineId" placeholder="请选择SSH隧道机器">
                            <el-option v-for="item in sshTunnelMachineList" :key="item.id"
                                :label="`${item.ip}:${item.port} [${item.name}]`" :value="item.id">
                            </el-option>
                        </el-select>
                    </el-col>
                </el-form-item>
            </el-form>

            <template #footer>
                <div class="dialog-footer">
                    <el-button @click="cancel()">取 消</el-button>
                    <el-button type="primary" :loading="btnLoading" @click="btnOk">确 定</el-button>
                </div>
            </template>
        </el-dialog>
    </div>
</template>

<script lang="ts" setup>
import { toRefs, reactive, watch, ref } from 'vue';
import { dbApi } from './api';
import { machineApi } from '../machine/api.ts';
import { ElMessage } from 'element-plus';
import { notBlank } from '@/common/assert';
import { RsaEncrypt } from '@/common/rsa';
import TagSelect from '../component/TagSelect.vue';

const props = defineProps({
    visible: {
        type: Boolean,
    },
    db: {
        type: [Boolean, Object],
    },
    title: {
        type: String,
    },
})

//定义事件
const emit = defineEmits(['update:visible', 'cancel', 'val-change'])

const rules = {
    tagId: [
        {
            required: true,
            message: '请选择标签',
            trigger: ['change', 'blur'],
        },
    ],
    name: [
        {
            required: true,
            message: '请输入别名',
            trigger: ['change', 'blur'],
        },
    ],
    type: [
        {
            required: true,
            message: '请选择数据库类型',
            trigger: ['change', 'blur'],
        },
    ],
    host: [
        {
            required: true,
            message: '请输入主机ip和port',
            trigger: ['change', 'blur'],
        },
    ],
    username: [
        {
            required: true,
            message: '请输入用户名',
            trigger: ['change', 'blur'],
        },
    ],
    database: [
        {
            required: true,
            message: '请添加数据库',
            trigger: ['change', 'blur'],
        },
    ],
}

const dbForm: any = ref(null);

const state = reactive({
    dialogVisible: false,
    allDatabases: [] as any,
    databaseList: [] as any,
    sshTunnelMachineList: [] as any,
    form: {
        id: null,
        tagId: null as any,
        tagPath: null as any,
        type: null,
        name: null,
        host: '',
        port: 3306,
        username: null,
        password: null,
        params: null,
        database: '',
        project: null,
        projectId: null,
        envId: null,
        env: null,
        remark: '',
        enableSshTunnel: null,
        sshTunnelMachineId: null,
    },
    // 原密码
    pwd: '',
    btnLoading: false,
});

const {
    dialogVisible,
    allDatabases,
    databaseList,
    sshTunnelMachineList,
    form,
    pwd,
    btnLoading,
} = toRefs(state)

watch(props, (newValue: any) => {
    state.dialogVisible = newValue.visible;
    if (!state.dialogVisible) {
        return;
    }
    if (newValue.db) {
        state.form = { ...newValue.db };
        // 将数据库名使用空格切割，获取所有数据库列表
        state.databaseList = newValue.db.database.split(' ');
    } else {
        state.form = { port: 3306, enableSshTunnel: -1 } as any;
        state.databaseList = [];
    }
    getSshTunnelMachines();
});

/**
 * 改变表单中的数据库字段，方便表单错误提示。如全部删光，可提示请添加数据库
 */
const changeDatabase = () => {
    state.form.database = state.databaseList.length == 0 ? '' : state.databaseList.join(' ');
};

const getSshTunnelMachines = async () => {
    if (state.form.enableSshTunnel == 1 && state.sshTunnelMachineList.length == 0) {
        const res = await machineApi.list.request({ pageNum: 1, pageSize: 100 });
        state.sshTunnelMachineList = res.list;
    }
};

const getAllDatabase = async () => {
    const reqForm = { ...state.form };
    reqForm.password = await RsaEncrypt(reqForm.password);
    state.allDatabases = await dbApi.getAllDatabase.request(reqForm);
    ElMessage.success('获取成功, 请选择需要管理操作的数据库');
};

const getDbPwd = async () => {
    state.pwd = await dbApi.getDbPwd.request({ id: state.form.id });
};

const btnOk = async () => {
    if (!state.form.id) {
        notBlank(state.form.password, '新增操作，密码不可为空');
    }
    dbForm.value.validate(async (valid: boolean) => {
        if (valid) {
            const reqForm = { ...state.form };
            reqForm.password = await RsaEncrypt(reqForm.password);
            dbApi.saveDb.request(reqForm).then(() => {
                ElMessage.success('保存成功');
                emit('val-change', state.form);
                state.btnLoading = true;
                setTimeout(() => {
                    state.btnLoading = false;
                }, 1000);

                cancel();
            });
        } else {
            ElMessage.error('请正确填写信息');
            return false;
        }
    });
};

const resetInputDb = () => {
    state.databaseList = [];
    state.allDatabases = [];
};

const cancel = () => {
    emit('update:visible', false);
    emit('cancel');
    setTimeout(() => {
        resetInputDb();
    }, 500);
};
</script>
<style lang="scss">

</style>