mirror of
				https://gitee.com/dromara/mayfly-go
				synced 2025-11-04 08:20:25 +08:00 
			
		
		
		
	Compare commits
	
		
			6 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					76d6fc3ba5 | ||
| 
						 | 
					f0540559bb | ||
| 
						 | 
					802e379f60 | ||
| 
						 | 
					8c9253da80 | ||
| 
						 | 
					5271bd21e8 | ||
| 
						 | 
					db554ebdc9 | 
							
								
								
									
										2
									
								
								LICENSE
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								LICENSE
									
									
									
									
									
								
							@@ -1,6 +1,6 @@
 | 
			
		||||
                                 Apache License
 | 
			
		||||
                           Version 2.0, January 2004
 | 
			
		||||
                        http://www.apache.org/licenses/
 | 
			
		||||
                        http://www.apache.org/licenses/ 
 | 
			
		||||
 | 
			
		||||
   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										36
									
								
								README.en.md
									
									
									
									
									
								
							
							
						
						
									
										36
									
								
								README.en.md
									
									
									
									
									
								
							@@ -1,36 +0,0 @@
 | 
			
		||||
# mayfly-go
 | 
			
		||||
 | 
			
		||||
#### Description
 | 
			
		||||
golang实现linux运维等
 | 
			
		||||
 | 
			
		||||
#### Software Architecture
 | 
			
		||||
Software architecture description
 | 
			
		||||
 | 
			
		||||
#### Installation
 | 
			
		||||
 | 
			
		||||
1.  xxxx
 | 
			
		||||
2.  xxxx
 | 
			
		||||
3.  xxxx
 | 
			
		||||
 | 
			
		||||
#### Instructions
 | 
			
		||||
 | 
			
		||||
1.  xxxx
 | 
			
		||||
2.  xxxx
 | 
			
		||||
3.  xxxx
 | 
			
		||||
 | 
			
		||||
#### Contribution
 | 
			
		||||
 | 
			
		||||
1.  Fork the repository
 | 
			
		||||
2.  Create Feat_xxx branch
 | 
			
		||||
3.  Commit your code
 | 
			
		||||
4.  Create Pull Request
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#### Gitee Feature
 | 
			
		||||
 | 
			
		||||
1.  You can use Readme\_XXX.md to support different languages, such as Readme\_en.md, Readme\_zh.md
 | 
			
		||||
2.  Gitee blog [blog.gitee.com](https://blog.gitee.com)
 | 
			
		||||
3.  Explore open source project [https://gitee.com/explore](https://gitee.com/explore)
 | 
			
		||||
4.  The most valuable open source project [GVP](https://gitee.com/gvp)
 | 
			
		||||
5.  The manual of Gitee [https://gitee.com/help](https://gitee.com/help)
 | 
			
		||||
6.  The most popular members  [https://gitee.com/gitee-stars/](https://gitee.com/gitee-stars/)
 | 
			
		||||
							
								
								
									
										5123
									
								
								mayfly_go_web/package-lock.json
									
									
									
										generated
									
									
									
										Normal file
									
								
							
							
						
						
									
										5123
									
								
								mayfly_go_web/package-lock.json
									
									
									
										generated
									
									
									
										Normal file
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							@@ -13,7 +13,8 @@
 | 
			
		||||
    "countup.js": "^2.0.7",
 | 
			
		||||
    "cropperjs": "^1.5.11",
 | 
			
		||||
    "echarts": "^5.3.3",
 | 
			
		||||
    "element-plus": "^2.2.9",
 | 
			
		||||
    "element-plus": "^2.2.10",
 | 
			
		||||
    "jsencrypt": "^3.2.1",
 | 
			
		||||
    "jsoneditor": "^9.9.0",
 | 
			
		||||
    "lodash": "^4.17.21",
 | 
			
		||||
    "mitt": "^3.0.0",
 | 
			
		||||
@@ -23,7 +24,7 @@
 | 
			
		||||
    "sql-formatter": "^7.0.3",
 | 
			
		||||
    "vue": "^3.2.37",
 | 
			
		||||
    "vue-clipboard3": "^1.0.1",
 | 
			
		||||
    "vue-router": "^4.0.16",
 | 
			
		||||
    "vue-router": "^4.1.2",
 | 
			
		||||
    "vuex": "^4.0.2",
 | 
			
		||||
    "xterm": "^4.19.0",
 | 
			
		||||
    "xterm-addon-fit": "^0.5.0"
 | 
			
		||||
 
 | 
			
		||||
@@ -1,4 +1,5 @@
 | 
			
		||||
window.globalConfig = {
 | 
			
		||||
    "BaseApiUrl": "http://localhost:8888",
 | 
			
		||||
    "BaseWsUrl": "ws://localhost:8888"
 | 
			
		||||
    // 默认为空,以访问根目录为api请求地址。若前后端分离部署可单独配置该后端api请求地址
 | 
			
		||||
    "BaseApiUrl": "",
 | 
			
		||||
    "BaseWsUrl": ""
 | 
			
		||||
}
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
const config = {
 | 
			
		||||
    baseApiUrl:  `${(window as any).globalConfig.BaseApiUrl}/api`,
 | 
			
		||||
    baseWsUrl: `${(window as any).globalConfig.BaseWsUrl}/api`
 | 
			
		||||
    baseApiUrl: `${(window as any).globalConfig.BaseApiUrl}/api`,
 | 
			
		||||
    baseWsUrl: `${(window as any).globalConfig.BaseWsUrl || `${location.protocol == 'https:' ? 'wss:' : 'ws:'}//${location.host}`}/api`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export default config
 | 
			
		||||
@@ -2,6 +2,8 @@ import request from './request'
 | 
			
		||||
 | 
			
		||||
export default {
 | 
			
		||||
    login: (param: any) => request.request('POST', '/sys/accounts/login', param, null),
 | 
			
		||||
    changePwd: (param: any) => request.request('POST', '/sys/accounts/change-pwd', param, null),
 | 
			
		||||
    getPublicKey: () => request.request('GET', '/common/public-key', null, null),
 | 
			
		||||
    captcha: () => request.request('GET', '/sys/captcha', null, null),
 | 
			
		||||
    logout: (param: any) => request.request('POST', '/sys/accounts/logout/{token}', param, null),
 | 
			
		||||
    getMenuRoute: (param: any) => request.request('Get', '/sys/resources/account', param, null)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										37
									
								
								mayfly_go_web/src/common/rsa.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										37
									
								
								mayfly_go_web/src/common/rsa.ts
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,37 @@
 | 
			
		||||
import openApi from './openApi';
 | 
			
		||||
import JSEncrypt from 'jsencrypt'
 | 
			
		||||
import { notBlank } from './assert';
 | 
			
		||||
 | 
			
		||||
var encryptor: any = null
 | 
			
		||||
 | 
			
		||||
export async function getRsaPublicKey() {
 | 
			
		||||
    let publicKey = sessionStorage.getItem('RsaPublicKey')
 | 
			
		||||
    if (publicKey) {
 | 
			
		||||
        return publicKey
 | 
			
		||||
    }
 | 
			
		||||
    publicKey = await openApi.getPublicKey() as string
 | 
			
		||||
    sessionStorage.setItem('RsaPublicKey', publicKey)
 | 
			
		||||
    return publicKey
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * 公钥加密指定值
 | 
			
		||||
 * 
 | 
			
		||||
 * @param value value
 | 
			
		||||
 * @returns 加密后的值
 | 
			
		||||
 */
 | 
			
		||||
export async function RsaEncrypt(value: any) {
 | 
			
		||||
    // 不存在则返回空值
 | 
			
		||||
    if (!value) {
 | 
			
		||||
        return ""
 | 
			
		||||
    }
 | 
			
		||||
    if (encryptor != null) {
 | 
			
		||||
        return encryptor.encrypt(value)
 | 
			
		||||
    }
 | 
			
		||||
    console.log(value)
 | 
			
		||||
    encryptor = new JSEncrypt()
 | 
			
		||||
    const publicKey = await getRsaPublicKey() as string;
 | 
			
		||||
    notBlank(publicKey, "获取公钥失败")
 | 
			
		||||
    encryptor.setPublicKey(publicKey)//设置公钥
 | 
			
		||||
    return encryptor.encrypt(value)
 | 
			
		||||
}
 | 
			
		||||
@@ -10,6 +10,7 @@ export const imports = {
 | 
			
		||||
    "ResourceList": () => import('@/views/system/resource'),
 | 
			
		||||
    "RoleList": () => import('@/views/system/role'),
 | 
			
		||||
    "AccountList": () => import('@/views/system/account'),
 | 
			
		||||
    "SyslogList": () => import('@/views/system/syslog/SyslogList.vue'),
 | 
			
		||||
    // project
 | 
			
		||||
    "ProjectList": () => import('@/views/ops/project/ProjectList.vue'),
 | 
			
		||||
    // db
 | 
			
		||||
 
 | 
			
		||||
@@ -947,12 +947,6 @@
 | 
			
		||||
.el-select-dropdown .el-scrollbar__wrap {
 | 
			
		||||
	overflow-x: scroll !important;
 | 
			
		||||
}
 | 
			
		||||
.el-select-dropdown__wrap {
 | 
			
		||||
	max-height: 274px !important; /*修复Select 选择器高度问题*/
 | 
			
		||||
}
 | 
			
		||||
.el-cascader-menu__wrap.el-scrollbar__wrap {
 | 
			
		||||
	height: 204px !important; /*修复Cascader 级联选择器高度问题*/
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/* Drawer 抽屉
 | 
			
		||||
------------------------------- */
 | 
			
		||||
 
 | 
			
		||||
@@ -1,54 +1,75 @@
 | 
			
		||||
<template>
 | 
			
		||||
    <el-form ref="loginFormRef" :model="loginForm" :rules="rules" class="login-content-form" size="large">
 | 
			
		||||
        <el-form-item prop="username">
 | 
			
		||||
            <el-input type="text" placeholder="请输入用户名" prefix-icon="user" v-model="loginForm.username" clearable autocomplete="off">
 | 
			
		||||
            </el-input>
 | 
			
		||||
        </el-form-item>
 | 
			
		||||
        <el-form-item prop="password">
 | 
			
		||||
            <el-input
 | 
			
		||||
                type="password"
 | 
			
		||||
                placeholder="请输入密码"
 | 
			
		||||
                prefix-icon="lock"
 | 
			
		||||
                v-model="loginForm.password"
 | 
			
		||||
                autocomplete="off"
 | 
			
		||||
                show-password
 | 
			
		||||
            >
 | 
			
		||||
            </el-input>
 | 
			
		||||
        </el-form-item>
 | 
			
		||||
        <el-form-item prop="captcha">
 | 
			
		||||
            <el-row :gutter="15">
 | 
			
		||||
                <el-col :span="16">
 | 
			
		||||
    <div>
 | 
			
		||||
        <el-form ref="loginFormRef" :model="loginForm" :rules="rules" class="login-content-form" size="large">
 | 
			
		||||
            <el-form-item prop="username">
 | 
			
		||||
                <el-input type="text" placeholder="请输入用户名" prefix-icon="user" v-model="loginForm.username" clearable autocomplete="off">
 | 
			
		||||
                </el-input>
 | 
			
		||||
            </el-form-item>
 | 
			
		||||
            <el-form-item prop="password">
 | 
			
		||||
                <el-input type="password" placeholder="请输入密码" prefix-icon="lock" v-model="loginForm.password" autocomplete="off" show-password>
 | 
			
		||||
                </el-input>
 | 
			
		||||
            </el-form-item>
 | 
			
		||||
            <el-form-item prop="captcha">
 | 
			
		||||
                <el-row :gutter="15">
 | 
			
		||||
                    <el-col :span="16">
 | 
			
		||||
                        <el-input
 | 
			
		||||
                            type="text"
 | 
			
		||||
                            maxlength="6"
 | 
			
		||||
                            placeholder="请输入验证码"
 | 
			
		||||
                            prefix-icon="position"
 | 
			
		||||
                            v-model="loginForm.captcha"
 | 
			
		||||
                            clearable
 | 
			
		||||
                            autocomplete="off"
 | 
			
		||||
                            @keyup.enter="login"
 | 
			
		||||
                        ></el-input>
 | 
			
		||||
                    </el-col>
 | 
			
		||||
                    <el-col :span="8">
 | 
			
		||||
                        <div class="login-content-code">
 | 
			
		||||
                            <img
 | 
			
		||||
                                class="login-content-code-img"
 | 
			
		||||
                                @click="getCaptcha"
 | 
			
		||||
                                width="130px"
 | 
			
		||||
                                height="40px"
 | 
			
		||||
                                :src="captchaImage"
 | 
			
		||||
                                style="cursor: pointer"
 | 
			
		||||
                            />
 | 
			
		||||
                        </div>
 | 
			
		||||
                    </el-col>
 | 
			
		||||
                </el-row>
 | 
			
		||||
            </el-form-item>
 | 
			
		||||
            <el-form-item>
 | 
			
		||||
                <el-button type="primary" class="login-content-submit" round @click="login" :loading="loading.signIn">
 | 
			
		||||
                    <span>登 录</span>
 | 
			
		||||
                </el-button>
 | 
			
		||||
            </el-form-item>
 | 
			
		||||
        </el-form>
 | 
			
		||||
 | 
			
		||||
        <el-dialog title="修改密码" v-model="changePwdDialog.visible" :close-on-click-modal="false" width="450px" :destroy-on-close="true">
 | 
			
		||||
            <el-form :model="changePwdDialog.form" :rules="changePwdDialog.rules" ref="changePwdFormRef" label-width="65px">
 | 
			
		||||
                <el-form-item prop="username" label="用户名" required>
 | 
			
		||||
                    <el-input v-model.trim="changePwdDialog.form.username" disabled></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="oldPassword" label="旧密码" required>
 | 
			
		||||
                    <el-input v-model.trim="changePwdDialog.form.oldPassword" autocomplete="new-password" type="password"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="newPassword" label="新密码" required>
 | 
			
		||||
                    <el-input
 | 
			
		||||
                        type="text"
 | 
			
		||||
                        maxlength="6"
 | 
			
		||||
                        placeholder="请输入验证码"
 | 
			
		||||
                        prefix-icon="position"
 | 
			
		||||
                        v-model="loginForm.captcha"
 | 
			
		||||
                        clearable
 | 
			
		||||
                        autocomplete="off"
 | 
			
		||||
                        @keyup.enter="login"
 | 
			
		||||
                        v-model.trim="changePwdDialog.form.newPassword"
 | 
			
		||||
                        placeholder="须为8位以上且包含字⺟⼤⼩写+数字+特殊符号"
 | 
			
		||||
                        type="password"
 | 
			
		||||
                        autocomplete="new-password"
 | 
			
		||||
                    ></el-input>
 | 
			
		||||
                </el-col>
 | 
			
		||||
                <el-col :span="8">
 | 
			
		||||
                    <div class="login-content-code">
 | 
			
		||||
                        <img
 | 
			
		||||
                            class="login-content-code-img"
 | 
			
		||||
                            @click="getCaptcha"
 | 
			
		||||
                            width="130px"
 | 
			
		||||
                            height="40px"
 | 
			
		||||
                            :src="captchaImage"
 | 
			
		||||
                            style="cursor: pointer"
 | 
			
		||||
                        />
 | 
			
		||||
                    </div>
 | 
			
		||||
                </el-col>
 | 
			
		||||
            </el-row>
 | 
			
		||||
        </el-form-item>
 | 
			
		||||
        <el-form-item>
 | 
			
		||||
            <el-button type="primary" class="login-content-submit" round @click="login" :loading="loading.signIn">
 | 
			
		||||
                <span>登 录</span>
 | 
			
		||||
            </el-button>
 | 
			
		||||
        </el-form-item>
 | 
			
		||||
    </el-form>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
            </el-form>
 | 
			
		||||
 | 
			
		||||
            <template #footer>
 | 
			
		||||
                <div class="dialog-footer">
 | 
			
		||||
                    <el-button @click="cancelChangePwd">取 消</el-button>
 | 
			
		||||
                    <el-button @click="changePwd" type="primary" :loading="loading.changePwd">确 定</el-button>
 | 
			
		||||
                </div>
 | 
			
		||||
            </template>
 | 
			
		||||
        </el-dialog>
 | 
			
		||||
    </div>
 | 
			
		||||
</template>
 | 
			
		||||
 | 
			
		||||
<script lang="ts">
 | 
			
		||||
@@ -60,7 +81,9 @@ import { useStore } from '@/store/index.ts';
 | 
			
		||||
import { setSession } from '@/common/utils/storage.ts';
 | 
			
		||||
import { formatAxis } from '@/common/utils/formatTime.ts';
 | 
			
		||||
import openApi from '@/common/openApi';
 | 
			
		||||
import { RsaEncrypt } from '@/common/rsa';
 | 
			
		||||
import { letterAvatar } from '@/common/utils/string';
 | 
			
		||||
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'AccountLogin',
 | 
			
		||||
    setup() {
 | 
			
		||||
@@ -68,6 +91,8 @@ export default defineComponent({
 | 
			
		||||
        const route = useRoute();
 | 
			
		||||
        const router = useRouter();
 | 
			
		||||
        const loginFormRef: any = ref(null);
 | 
			
		||||
        const changePwdFormRef: any = ref(null);
 | 
			
		||||
 | 
			
		||||
        const state = reactive({
 | 
			
		||||
            captchaImage: '',
 | 
			
		||||
            loginForm: {
 | 
			
		||||
@@ -76,6 +101,24 @@ export default defineComponent({
 | 
			
		||||
                captcha: '',
 | 
			
		||||
                cid: '',
 | 
			
		||||
            },
 | 
			
		||||
            changePwdDialog: {
 | 
			
		||||
                visible: false,
 | 
			
		||||
                form: {
 | 
			
		||||
                    username: '',
 | 
			
		||||
                    oldPassword: '',
 | 
			
		||||
                    newPassword: '',
 | 
			
		||||
                },
 | 
			
		||||
                rules: {
 | 
			
		||||
                    newPassword: [
 | 
			
		||||
                        { required: true, message: '请输入新密码', trigger: 'blur' },
 | 
			
		||||
                        {
 | 
			
		||||
                            pattern: /^(?=.*[A-Za-z])(?=.*\d)(?=.*[`~!@#$%^&*()_+<>?:"{},.\/\\;'[\]])[A-Za-z\d`~!@#$%^&*()_+<>?:"{},.\/\\;'[\]]{8,}$/,
 | 
			
		||||
                            message: '须为8位以上且包含字⺟⼤⼩写+数字+特殊符号',
 | 
			
		||||
                            trigger: 'blur',
 | 
			
		||||
                        },
 | 
			
		||||
                    ],
 | 
			
		||||
                },
 | 
			
		||||
            },
 | 
			
		||||
            rules: {
 | 
			
		||||
                username: [{ required: true, message: '请输入用户名', trigger: 'blur' }],
 | 
			
		||||
                password: [{ required: true, message: '请输入密码', trigger: 'blur' }],
 | 
			
		||||
@@ -83,6 +126,7 @@ export default defineComponent({
 | 
			
		||||
            },
 | 
			
		||||
            loading: {
 | 
			
		||||
                signIn: false,
 | 
			
		||||
                changePwd: false,
 | 
			
		||||
            },
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
@@ -116,15 +160,26 @@ export default defineComponent({
 | 
			
		||||
        const onSignIn = async () => {
 | 
			
		||||
            state.loading.signIn = true;
 | 
			
		||||
            let loginRes;
 | 
			
		||||
            const originPwd = state.loginForm.password;
 | 
			
		||||
            try {
 | 
			
		||||
                loginRes = await openApi.login(state.loginForm);
 | 
			
		||||
                // // 存储 token 到浏览器缓存
 | 
			
		||||
                const loginReq = { ...state.loginForm };
 | 
			
		||||
                loginReq.password = await RsaEncrypt(originPwd);
 | 
			
		||||
                loginRes = await openApi.login(loginReq);
 | 
			
		||||
                // 存储 token 到浏览器缓存
 | 
			
		||||
                setSession('token', loginRes.token);
 | 
			
		||||
                setSession('menus', loginRes.menus);
 | 
			
		||||
            } catch (e) {
 | 
			
		||||
            } catch (e: any) {
 | 
			
		||||
                state.loading.signIn = false;
 | 
			
		||||
                state.loginForm.captcha = '';
 | 
			
		||||
                getCaptcha();
 | 
			
		||||
                // 密码强度不足
 | 
			
		||||
                if (e.code && e.code == 401) {
 | 
			
		||||
                    state.changePwdDialog.form.username = state.loginForm.username;
 | 
			
		||||
                    state.changePwdDialog.form.oldPassword = originPwd;
 | 
			
		||||
                    state.changePwdDialog.form.newPassword = '';
 | 
			
		||||
                    state.changePwdDialog.visible = true;
 | 
			
		||||
                } else {
 | 
			
		||||
                    getCaptcha();
 | 
			
		||||
                }
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            // 用户信息
 | 
			
		||||
@@ -174,11 +229,44 @@ export default defineComponent({
 | 
			
		||||
            }, 300);
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const changePwd = () => {
 | 
			
		||||
            changePwdFormRef.value.validate(async (valid: boolean) => {
 | 
			
		||||
                if (!valid) {
 | 
			
		||||
                    return false;
 | 
			
		||||
                }
 | 
			
		||||
                try {
 | 
			
		||||
                    state.loading.changePwd = true;
 | 
			
		||||
                    const form = state.changePwdDialog.form;
 | 
			
		||||
                    const changePwdReq: any = { ...form };
 | 
			
		||||
                    changePwdReq.oldPassword = await RsaEncrypt(form.oldPassword);
 | 
			
		||||
                    changePwdReq.newPassword = await RsaEncrypt(form.newPassword);
 | 
			
		||||
                    await openApi.changePwd(changePwdReq);
 | 
			
		||||
                    ElMessage.success('密码修改成功, 新密码已填充至登录密码框');
 | 
			
		||||
                    state.loginForm.password = state.changePwdDialog.form.newPassword;
 | 
			
		||||
                    state.changePwdDialog.visible = false;
 | 
			
		||||
                    getCaptcha();
 | 
			
		||||
                } finally {
 | 
			
		||||
                    state.loading.changePwd = false;
 | 
			
		||||
                }
 | 
			
		||||
            });
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const cancelChangePwd = () => {
 | 
			
		||||
            state.changePwdDialog.visible = false;
 | 
			
		||||
            state.changePwdDialog.form.newPassword = '';
 | 
			
		||||
            state.changePwdDialog.form.oldPassword = '';
 | 
			
		||||
            state.changePwdDialog.form.username = '';
 | 
			
		||||
            getCaptcha();
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        return {
 | 
			
		||||
            getCaptcha,
 | 
			
		||||
            currentTime,
 | 
			
		||||
            loginFormRef,
 | 
			
		||||
            changePwdFormRef,
 | 
			
		||||
            login,
 | 
			
		||||
            changePwd,
 | 
			
		||||
            cancelChangePwd,
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
        };
 | 
			
		||||
    },
 | 
			
		||||
 
 | 
			
		||||
@@ -23,10 +23,13 @@
 | 
			
		||||
                    </el-select>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="host" label="host:" required>
 | 
			
		||||
                    <el-input v-model.trim="form.host" placeholder="请输入主机ip" auto-complete="off"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="port" label="port:" required>
 | 
			
		||||
                    <el-input type="number" v-model.trim="form.port" placeholder="请输入端口"></el-input>
 | 
			
		||||
                    <el-col :span="18">
 | 
			
		||||
                        <el-input 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>
 | 
			
		||||
@@ -44,28 +47,38 @@
 | 
			
		||||
                    <el-input v-model="form.params" placeholder="其他连接参数,形如: key1=value1&key2=value2"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="database" label="数据库名:" required>
 | 
			
		||||
                    <el-tag
 | 
			
		||||
                        v-for="db in databaseList"
 | 
			
		||||
                        :key="db"
 | 
			
		||||
                        class="ml5 mt5"
 | 
			
		||||
                        type="success"
 | 
			
		||||
                        effect="plain"
 | 
			
		||||
                        closable
 | 
			
		||||
                        :disable-transitions="false"
 | 
			
		||||
                        @close="handleClose(db)"
 | 
			
		||||
                    <el-select
 | 
			
		||||
                        @change="changeDatabase"
 | 
			
		||||
                        @focus="getAllDatabase"
 | 
			
		||||
                        v-model="databaseList"
 | 
			
		||||
                        multiple
 | 
			
		||||
                        collapse-tags
 | 
			
		||||
                        collapse-tags-tooltip
 | 
			
		||||
                        filterable
 | 
			
		||||
                        allow-create
 | 
			
		||||
                        placeholder="请确保数据库实例信息填写完整后选择数据库"
 | 
			
		||||
                        style="width: 100%"
 | 
			
		||||
                    >
 | 
			
		||||
                        {{ db }}
 | 
			
		||||
                    </el-tag>
 | 
			
		||||
                    <el-input
 | 
			
		||||
                        v-if="inputDbVisible"
 | 
			
		||||
                        ref="InputDbRef"
 | 
			
		||||
                        v-model="inputDbValue"
 | 
			
		||||
                        style="width: 120px; margin-left: 5px; margin-top: 5px"
 | 
			
		||||
                        size="small"
 | 
			
		||||
                        @keyup.enter="handleInputDbConfirm"
 | 
			
		||||
                        @blur="handleInputDbConfirm"
 | 
			
		||||
                    />
 | 
			
		||||
                    <el-button v-else class="ml5 mt5" size="small" @click="showInputDb"> + 添加数据库 </el-button>
 | 
			
		||||
                        <el-option v-for="db in allDatabases" :key="db" :label="db" :value="db" />
 | 
			
		||||
                    </el-select>
 | 
			
		||||
                </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="2" v-if="form.enableSshTunnel == 1"> 机器: </el-col>
 | 
			
		||||
                    <el-col :span="19" 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>
 | 
			
		||||
 | 
			
		||||
@@ -80,12 +93,13 @@
 | 
			
		||||
</template>
 | 
			
		||||
 | 
			
		||||
<script lang="ts">
 | 
			
		||||
import { toRefs, reactive, nextTick, watch, defineComponent, ref } from 'vue';
 | 
			
		||||
import { toRefs, reactive, watch, defineComponent, ref } from 'vue';
 | 
			
		||||
import { dbApi } from './api';
 | 
			
		||||
import { projectApi } from '../project/api.ts';
 | 
			
		||||
import { machineApi } from '../machine/api.ts';
 | 
			
		||||
import { ElMessage } from 'element-plus';
 | 
			
		||||
import type { ElInput } from 'element-plus';
 | 
			
		||||
import { notBlank } from '@/common/assert';
 | 
			
		||||
import { RsaEncrypt } from '@/common/rsa';
 | 
			
		||||
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'DbEdit',
 | 
			
		||||
@@ -105,15 +119,14 @@ export default defineComponent({
 | 
			
		||||
    },
 | 
			
		||||
    setup(props: any, { emit }) {
 | 
			
		||||
        const dbForm: any = ref(null);
 | 
			
		||||
        const InputDbRef = ref<InstanceType<typeof ElInput>>();
 | 
			
		||||
 | 
			
		||||
        const state = reactive({
 | 
			
		||||
            dialogVisible: false,
 | 
			
		||||
            projects: [],
 | 
			
		||||
            envs: [],
 | 
			
		||||
            allDatabases: [] as any,
 | 
			
		||||
            databaseList: [] as any,
 | 
			
		||||
            inputDbVisible: false,
 | 
			
		||||
            inputDbValue: '',
 | 
			
		||||
            sshTunnelMachineList: [],
 | 
			
		||||
            form: {
 | 
			
		||||
                id: null,
 | 
			
		||||
                name: null,
 | 
			
		||||
@@ -126,6 +139,8 @@ export default defineComponent({
 | 
			
		||||
                projectId: null,
 | 
			
		||||
                envId: null,
 | 
			
		||||
                env: null,
 | 
			
		||||
                enableSshTunnel: null,
 | 
			
		||||
                sshTunnelMachineId: null,
 | 
			
		||||
            },
 | 
			
		||||
            btnLoading: false,
 | 
			
		||||
            rules: {
 | 
			
		||||
@@ -160,14 +175,7 @@ export default defineComponent({
 | 
			
		||||
                host: [
 | 
			
		||||
                    {
 | 
			
		||||
                        required: true,
 | 
			
		||||
                        message: '请输入主机ip',
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
                port: [
 | 
			
		||||
                    {
 | 
			
		||||
                        required: true,
 | 
			
		||||
                        message: '请输入端口',
 | 
			
		||||
                        message: '请输入主机ip和port',
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
@@ -189,6 +197,10 @@ export default defineComponent({
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        watch(props, (newValue) => {
 | 
			
		||||
            state.dialogVisible = newValue.visible;
 | 
			
		||||
            if (!state.dialogVisible) {
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            state.projects = newValue.projects;
 | 
			
		||||
            if (newValue.db) {
 | 
			
		||||
                getEnvs(newValue.db.projectId);
 | 
			
		||||
@@ -197,33 +209,12 @@ export default defineComponent({
 | 
			
		||||
                state.databaseList = newValue.db.database.split(' ');
 | 
			
		||||
            } else {
 | 
			
		||||
                state.envs = [];
 | 
			
		||||
                state.form = { port: 3306 } as any;
 | 
			
		||||
                state.form = { port: 3306, enableSshTunnel: -1 } as any;
 | 
			
		||||
                state.databaseList = [];
 | 
			
		||||
            }
 | 
			
		||||
            state.dialogVisible = newValue.visible;
 | 
			
		||||
            getSshTunnelMachines();
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        const handleClose = (db: string) => {
 | 
			
		||||
            state.databaseList.splice(state.databaseList.indexOf(db), 1);
 | 
			
		||||
            changeDatabase();
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const showInputDb = () => {
 | 
			
		||||
            state.inputDbVisible = true;
 | 
			
		||||
            nextTick(() => {
 | 
			
		||||
                InputDbRef.value!.input!.focus();
 | 
			
		||||
            });
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const handleInputDbConfirm = () => {
 | 
			
		||||
            if (state.inputDbValue) {
 | 
			
		||||
                state.databaseList.push(state.inputDbValue);
 | 
			
		||||
                changeDatabase();
 | 
			
		||||
            }
 | 
			
		||||
            state.inputDbVisible = false;
 | 
			
		||||
            state.inputDbValue = '';
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        /**
 | 
			
		||||
         * 改变表单中的数据库字段,方便表单错误提示。如全部删光,可提示请添加数据库
 | 
			
		||||
         */
 | 
			
		||||
@@ -231,6 +222,13 @@ export default defineComponent({
 | 
			
		||||
            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 getEnvs = async (projectId: any) => {
 | 
			
		||||
            state.envs = await projectApi.projectEnvs.request({ projectId });
 | 
			
		||||
        };
 | 
			
		||||
@@ -255,14 +253,24 @@ export default defineComponent({
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const getAllDatabase = async () => {
 | 
			
		||||
            if (state.allDatabases.length != 0) {
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            const reqForm = { ...state.form };
 | 
			
		||||
            reqForm.password = await RsaEncrypt(reqForm.password);
 | 
			
		||||
            state.allDatabases = await dbApi.getAllDatabase.request(reqForm);
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const btnOk = async () => {
 | 
			
		||||
            if (!state.form.id) {
 | 
			
		||||
                notBlank(state.form.password, '新增操作,密码不可为空');
 | 
			
		||||
            }
 | 
			
		||||
            dbForm.value.validate((valid: boolean) => {
 | 
			
		||||
            dbForm.value.validate(async (valid: boolean) => {
 | 
			
		||||
                if (valid) {
 | 
			
		||||
                    state.form.port = Number.parseInt(state.form.port as any);
 | 
			
		||||
                    dbApi.saveDb.request(state.form).then(() => {
 | 
			
		||||
                    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;
 | 
			
		||||
@@ -280,9 +288,8 @@ export default defineComponent({
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const resetInputDb = () => {
 | 
			
		||||
            state.inputDbVisible = false;
 | 
			
		||||
            state.databaseList = [];
 | 
			
		||||
            state.inputDbValue = '';
 | 
			
		||||
            state.allDatabases = [];
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const cancel = () => {
 | 
			
		||||
@@ -296,10 +303,9 @@ export default defineComponent({
 | 
			
		||||
        return {
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
            dbForm,
 | 
			
		||||
            InputDbRef,
 | 
			
		||||
            handleClose,
 | 
			
		||||
            showInputDb,
 | 
			
		||||
            handleInputDbConfirm,
 | 
			
		||||
            getAllDatabase,
 | 
			
		||||
            changeDatabase,
 | 
			
		||||
            getSshTunnelMachines,
 | 
			
		||||
            changeProject,
 | 
			
		||||
            changeEnv,
 | 
			
		||||
            btnOk,
 | 
			
		||||
 
 | 
			
		||||
@@ -502,6 +502,8 @@ export default defineComponent({
 | 
			
		||||
                state.tableInfoDialog.infos = await dbApi.tableInfos.request({ id: row.id, db });
 | 
			
		||||
                state.dbId = row.id;
 | 
			
		||||
                state.db = db;
 | 
			
		||||
            } catch (e) {
 | 
			
		||||
                state.tableInfoDialog.visible = false;
 | 
			
		||||
            } finally {
 | 
			
		||||
                state.tableInfoDialog.loading = false;
 | 
			
		||||
            }
 | 
			
		||||
 
 | 
			
		||||
@@ -4,6 +4,7 @@ export const dbApi = {
 | 
			
		||||
    // 获取权限列表
 | 
			
		||||
    dbs: Api.create("/dbs", 'get'),
 | 
			
		||||
    saveDb: Api.create("/dbs", 'post'),
 | 
			
		||||
    getAllDatabase: Api.create("/dbs/databases", 'post'),
 | 
			
		||||
    deleteDb: Api.create("/dbs/{id}", 'delete'),
 | 
			
		||||
    dumpDb: Api.create("/dbs/{id}/dump", 'post'),
 | 
			
		||||
    tableInfos: Api.create("/dbs/{id}/t-infos", 'get'),
 | 
			
		||||
@@ -22,5 +23,5 @@ export const dbApi = {
 | 
			
		||||
    getSqlNames: Api.create("/dbs/{id}/sql-names", 'get'),
 | 
			
		||||
    deleteDbSql: Api.create("/dbs/{id}/sql", 'delete'),
 | 
			
		||||
    // 获取数据库sql执行记录
 | 
			
		||||
    getSqlExecs: Api.create("/dbs/{id}/sql-execs", 'get'),
 | 
			
		||||
    getSqlExecs: Api.create("/dbs/{dbId}/sql-execs", 'get'),
 | 
			
		||||
}
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
<template>
 | 
			
		||||
    <div>
 | 
			
		||||
        <el-dialog :title="title" v-model="dialogVisible" :close-on-click-modal="false" :destroy-on-close="true" :before-close="cancel" width="35%">
 | 
			
		||||
        <el-dialog :title="title" v-model="dialogVisible" :close-on-click-modal="false" :destroy-on-close="true" :before-close="cancel" width="38%">
 | 
			
		||||
            <el-form :model="form" ref="machineForm" :rules="rules" label-width="85px">
 | 
			
		||||
                <el-form-item prop="projectId" label="项目:" required>
 | 
			
		||||
                    <el-select style="width: 100%" v-model="form.projectId" placeholder="请选择项目" @change="changeProject" filterable>
 | 
			
		||||
@@ -11,15 +11,24 @@
 | 
			
		||||
                    <el-input v-model.trim="form.name" placeholder="请输入机器别名" auto-complete="off"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="ip" label="ip:" required>
 | 
			
		||||
                    <el-input v-model.trim="form.ip" placeholder="请输入主机ip" auto-complete="off"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="port" label="port:" required>
 | 
			
		||||
                    <el-input type="number" v-model.number="form.port" placeholder="请输入端口"></el-input>
 | 
			
		||||
                    <el-col :span="18">
 | 
			
		||||
                        <el-input v-model.trim="form.ip" 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-form-item prop="authMethod" label="认证方式:" required>
 | 
			
		||||
                    <el-select style="width: 100%" v-model="form.authMethod" placeholder="请选择认证方式">
 | 
			
		||||
                        <el-option key="1" label="Password" :value="1"> </el-option>
 | 
			
		||||
                        <el-option key="2" label="PublicKey" :value="2"> </el-option>
 | 
			
		||||
                    </el-select>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item v-if="form.authMethod == 1" prop="password" label="密码:">
 | 
			
		||||
                    <el-input
 | 
			
		||||
                        type="password"
 | 
			
		||||
                        show-password
 | 
			
		||||
@@ -28,9 +37,30 @@
 | 
			
		||||
                        autocomplete="new-password"
 | 
			
		||||
                    ></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item v-if="form.authMethod == 2" prop="password" label="秘钥:">
 | 
			
		||||
                    <el-input type="textarea" :rows="3" v-model="form.password" placeholder="请将私钥文件内容拷贝至此,修改操作可不填"></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="remark" label="备注:">
 | 
			
		||||
                    <el-input type="textarea" v-model="form.remark"></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="2" v-if="form.enableSshTunnel == 1"> 机器: </el-col>
 | 
			
		||||
                    <el-col :span="19" 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>
 | 
			
		||||
@@ -48,6 +78,7 @@ import { toRefs, reactive, watch, defineComponent, ref } from 'vue';
 | 
			
		||||
import { machineApi } from './api';
 | 
			
		||||
import { ElMessage } from 'element-plus';
 | 
			
		||||
import { notBlank } from '@/common/assert';
 | 
			
		||||
import { RsaEncrypt } from '@/common/rsa';
 | 
			
		||||
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'MachineEdit',
 | 
			
		||||
@@ -70,15 +101,19 @@ export default defineComponent({
 | 
			
		||||
        const state = reactive({
 | 
			
		||||
            dialogVisible: false,
 | 
			
		||||
            projects: [],
 | 
			
		||||
            sshTunnelMachineList: [],
 | 
			
		||||
            form: {
 | 
			
		||||
                id: null,
 | 
			
		||||
                projectId: null,
 | 
			
		||||
                projectName: null,
 | 
			
		||||
                name: null,
 | 
			
		||||
                authMethod: 1,
 | 
			
		||||
                port: 22,
 | 
			
		||||
                username: null,
 | 
			
		||||
                password: null,
 | 
			
		||||
                username: '',
 | 
			
		||||
                password: '',
 | 
			
		||||
                remark: '',
 | 
			
		||||
                enableSshTunnel: null,
 | 
			
		||||
                sshTunnelMachineId: null,
 | 
			
		||||
            },
 | 
			
		||||
            btnLoading: false,
 | 
			
		||||
            rules: {
 | 
			
		||||
@@ -106,14 +141,7 @@ export default defineComponent({
 | 
			
		||||
                ip: [
 | 
			
		||||
                    {
 | 
			
		||||
                        required: true,
 | 
			
		||||
                        message: '请输入主机ip',
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
                port: [
 | 
			
		||||
                    {
 | 
			
		||||
                        required: true,
 | 
			
		||||
                        message: '请输入端口',
 | 
			
		||||
                        message: '请输入主机ip和端口',
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
@@ -124,19 +152,41 @@ export default defineComponent({
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
                authMethod: [
 | 
			
		||||
                    {
 | 
			
		||||
                        required: true,
 | 
			
		||||
                        message: '请选择认证方式',
 | 
			
		||||
                        trigger: ['change', 'blur'],
 | 
			
		||||
                    },
 | 
			
		||||
                ],
 | 
			
		||||
            },
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        watch(props, async (newValue) => {
 | 
			
		||||
            state.dialogVisible = newValue.visible;
 | 
			
		||||
            if (!state.dialogVisible) {
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            state.projects = newValue.projects;
 | 
			
		||||
            if (newValue.machine) {
 | 
			
		||||
                state.form = { ...newValue.machine };
 | 
			
		||||
            } else {
 | 
			
		||||
                state.form = { port: 22 } as any;
 | 
			
		||||
                state.form = { port: 22, authMethod: 1 } as any;
 | 
			
		||||
            }
 | 
			
		||||
            getSshTunnelMachines();
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        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 getSshTunnelMachine = (machineId: any) => {
 | 
			
		||||
            return state.sshTunnelMachineList.find((x: any) => x.id == machineId);
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const changeProject = (projectId: number) => {
 | 
			
		||||
            for (let p of state.projects as any) {
 | 
			
		||||
                if (p.id == projectId) {
 | 
			
		||||
@@ -149,18 +199,29 @@ export default defineComponent({
 | 
			
		||||
            if (!state.form.id) {
 | 
			
		||||
                notBlank(state.form.password, '新增操作,密码不可为空');
 | 
			
		||||
            }
 | 
			
		||||
            machineForm.value.validate((valid: boolean) => {
 | 
			
		||||
            machineForm.value.validate(async (valid: boolean) => {
 | 
			
		||||
                if (valid) {
 | 
			
		||||
                    machineApi.saveMachine.request(state.form).then(() => {
 | 
			
		||||
                    const form: any = state.form;
 | 
			
		||||
                    if (form.enableSshTunnel == 1) {
 | 
			
		||||
                        const tunnelMachine: any = getSshTunnelMachine(form.sshTunnelMachineId);
 | 
			
		||||
                        if (tunnelMachine.ip == form.ip && tunnelMachine.port == form.port) {
 | 
			
		||||
                            ElMessage.error('隧道机器不能与本机器一致');
 | 
			
		||||
                            return;
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                    const reqForm: any = { ...form };
 | 
			
		||||
                    if (reqForm.authMethod == 1) {
 | 
			
		||||
                        reqForm.password = await RsaEncrypt(state.form.password);
 | 
			
		||||
                    }
 | 
			
		||||
                    state.btnLoading = true;
 | 
			
		||||
                    try {
 | 
			
		||||
                        await machineApi.saveMachine.request(reqForm);
 | 
			
		||||
                        ElMessage.success('保存成功');
 | 
			
		||||
                        emit('val-change', state.form);
 | 
			
		||||
                        state.btnLoading = true;
 | 
			
		||||
                        setTimeout(() => {
 | 
			
		||||
                            state.btnLoading = false;
 | 
			
		||||
                        }, 1000);
 | 
			
		||||
 | 
			
		||||
                        cancel();
 | 
			
		||||
                    });
 | 
			
		||||
                    } finally {
 | 
			
		||||
                        state.btnLoading = false;
 | 
			
		||||
                    }
 | 
			
		||||
                } else {
 | 
			
		||||
                    ElMessage.error('请正确填写信息');
 | 
			
		||||
                    return false;
 | 
			
		||||
@@ -176,6 +237,7 @@ export default defineComponent({
 | 
			
		||||
        return {
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
            machineForm,
 | 
			
		||||
            getSshTunnelMachines,
 | 
			
		||||
            changeProject,
 | 
			
		||||
            btnOk,
 | 
			
		||||
            cancel,
 | 
			
		||||
 
 | 
			
		||||
@@ -260,13 +260,13 @@ export default defineComponent({
 | 
			
		||||
            search();
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const openFormDialog = (redis: any) => {
 | 
			
		||||
        const openFormDialog = (machine: any) => {
 | 
			
		||||
            let dialogTitle;
 | 
			
		||||
            if (redis) {
 | 
			
		||||
            if (machine) {
 | 
			
		||||
                state.machineEditDialog.data = state.currentData as any;
 | 
			
		||||
                dialogTitle = '编辑机器';
 | 
			
		||||
            } else {
 | 
			
		||||
                state.machineEditDialog.data = { port: 22 } as any;
 | 
			
		||||
                state.machineEditDialog.data = null;
 | 
			
		||||
                dialogTitle = '添加机器';
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -15,7 +15,7 @@
 | 
			
		||||
                            </el-form-item>
 | 
			
		||||
 | 
			
		||||
                            <el-form-item label="库" label-width="20px">
 | 
			
		||||
                                <el-select v-model="database" placeholder="请选择库" @change="changeDatabase">
 | 
			
		||||
                                <el-select v-model="database" placeholder="请选择库" @change="changeDatabase" filterable>
 | 
			
		||||
                                    <el-option v-for="item in databases" :key="item.Name" :label="item.Name" :value="item.Name">
 | 
			
		||||
                                        <span style="float: left">{{ item.Name }}</span>
 | 
			
		||||
                                        <span style="float: right; color: #8492a6; margin-left: 4px; font-size: 13px">{{
 | 
			
		||||
@@ -26,12 +26,8 @@
 | 
			
		||||
                            </el-form-item>
 | 
			
		||||
 | 
			
		||||
                            <el-form-item label="集合" label-width="40px">
 | 
			
		||||
                                <el-select v-model="collection" placeholder="请选择集合" @change="changeCollection">
 | 
			
		||||
                                <el-select v-model="collection" placeholder="请选择集合" @change="changeCollection" filterable>
 | 
			
		||||
                                    <el-option v-for="item in collections" :key="item" :label="item" :value="item">
 | 
			
		||||
                                        <!-- <span style="float: left">{{ item.uri }}</span>
 | 
			
		||||
                                            <span style="float: right; color: #8492a6; margin-left: 6px; font-size: 13px">{{
 | 
			
		||||
                                                ` [${item.name}]`
 | 
			
		||||
                                            }}</span> -->
 | 
			
		||||
                                    </el-option>
 | 
			
		||||
                                </el-select>
 | 
			
		||||
                            </el-form-item>
 | 
			
		||||
@@ -124,7 +120,7 @@
 | 
			
		||||
            </template>
 | 
			
		||||
        </el-dialog>
 | 
			
		||||
 | 
			
		||||
        <el-dialog width="800px" title="json编辑器" v-model="jsoneditorDialog.visible" @close="onCloseJsonEditDialog" :close-on-click-modal="false">
 | 
			
		||||
        <el-dialog width="70%" title="json编辑器" v-model="jsoneditorDialog.visible" @close="onCloseJsonEditDialog" :close-on-click-modal="false">
 | 
			
		||||
            <json-edit v-model="jsoneditorDialog.doc" />
 | 
			
		||||
        </el-dialog>
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -1,7 +1,7 @@
 | 
			
		||||
<template>
 | 
			
		||||
    <div>
 | 
			
		||||
        <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" :close-on-click-modal="false" width="35%" :destroy-on-close="true">
 | 
			
		||||
            <el-form :model="form" ref="mongoForm" :rules="rules" label-width="65px">
 | 
			
		||||
        <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" :close-on-click-modal="false" width="38%" :destroy-on-close="true">
 | 
			
		||||
            <el-form :model="form" ref="mongoForm" :rules="rules" label-width="85px">
 | 
			
		||||
                <el-form-item prop="projectId" label="项目" required>
 | 
			
		||||
                    <el-select style="width: 100%" v-model="form.projectId" placeholder="请选择项目" @change="changeProject" filterable>
 | 
			
		||||
                        <el-option v-for="item in projects" :key="item.id" :label="`${item.name} [${item.remark}]`" :value="item.id"> </el-option>
 | 
			
		||||
@@ -25,6 +25,24 @@
 | 
			
		||||
                        auto-complete="off"
 | 
			
		||||
                    ></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="2" v-if="form.enableSshTunnel == 1"> 机器: </el-col>
 | 
			
		||||
                    <el-col :span="19" 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>
 | 
			
		||||
@@ -41,7 +59,9 @@
 | 
			
		||||
import { toRefs, reactive, watch, defineComponent, ref } from 'vue';
 | 
			
		||||
import { mongoApi } from './api';
 | 
			
		||||
import { projectApi } from '../project/api.ts';
 | 
			
		||||
import { machineApi } from '../machine/api.ts';
 | 
			
		||||
import { ElMessage } from 'element-plus';
 | 
			
		||||
import { RsaEncrypt } from '@/common/rsa';
 | 
			
		||||
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'MongoEdit',
 | 
			
		||||
@@ -65,10 +85,13 @@ export default defineComponent({
 | 
			
		||||
            dialogVisible: false,
 | 
			
		||||
            projects: [],
 | 
			
		||||
            envs: [],
 | 
			
		||||
            sshTunnelMachineList: [],
 | 
			
		||||
            form: {
 | 
			
		||||
                id: null,
 | 
			
		||||
                name: null,
 | 
			
		||||
                uri: null,
 | 
			
		||||
                enableSshTunnel: -1,
 | 
			
		||||
                sshTunnelMachineId: null,
 | 
			
		||||
                project: null,
 | 
			
		||||
                projectId: null,
 | 
			
		||||
                envId: null,
 | 
			
		||||
@@ -109,6 +132,9 @@ export default defineComponent({
 | 
			
		||||
 | 
			
		||||
        watch(props, async (newValue) => {
 | 
			
		||||
            state.dialogVisible = newValue.visible;
 | 
			
		||||
            if (!state.dialogVisible) {
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            state.projects = newValue.projects;
 | 
			
		||||
            if (newValue.mongo) {
 | 
			
		||||
                getEnvs(newValue.mongo.projectId);
 | 
			
		||||
@@ -117,8 +143,16 @@ export default defineComponent({
 | 
			
		||||
                state.envs = [];
 | 
			
		||||
                state.form = { db: 0 } as any;
 | 
			
		||||
            }
 | 
			
		||||
            getSshTunnelMachines();
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        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 getEnvs = async (projectId: any) => {
 | 
			
		||||
            state.envs = await projectApi.projectEnvs.request({ projectId });
 | 
			
		||||
        };
 | 
			
		||||
@@ -144,9 +178,11 @@ export default defineComponent({
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const btnOk = async () => {
 | 
			
		||||
            mongoForm.value.validate((valid: boolean) => {
 | 
			
		||||
            mongoForm.value.validate(async (valid: boolean) => {
 | 
			
		||||
                if (valid) {
 | 
			
		||||
                    mongoApi.saveMongo.request(state.form).then(() => {
 | 
			
		||||
                    const reqForm = { ...state.form };
 | 
			
		||||
                    reqForm.uri = await RsaEncrypt(reqForm.uri);
 | 
			
		||||
                    mongoApi.saveMongo.request(reqForm).then(() => {
 | 
			
		||||
                        ElMessage.success('保存成功');
 | 
			
		||||
                        emit('val-change', state.form);
 | 
			
		||||
                        state.btnLoading = true;
 | 
			
		||||
@@ -172,6 +208,7 @@ export default defineComponent({
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
            mongoForm,
 | 
			
		||||
            changeProject,
 | 
			
		||||
            getSshTunnelMachines,
 | 
			
		||||
            changeEnv,
 | 
			
		||||
            btnOk,
 | 
			
		||||
            cancel,
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,6 @@
 | 
			
		||||
<template>
 | 
			
		||||
    <div>
 | 
			
		||||
        <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" :close-on-click-modal="false" :destroy-on-close="true" width="35%">
 | 
			
		||||
        <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="redisForm" :rules="rules" label-width="85px">
 | 
			
		||||
                <el-form-item prop="projectId" label="项目:" required>
 | 
			
		||||
                    <el-select style="width: 100%" v-model="form.projectId" placeholder="请选择项目" @change="changeProject" filterable>
 | 
			
		||||
@@ -20,14 +20,19 @@
 | 
			
		||||
                    </el-select>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="host" label="host:" required>
 | 
			
		||||
                    <el-input v-model.trim="form.host" placeholder="请输入host:port,集群模式用','分割" auto-complete="off" type="textarea"></el-input>
 | 
			
		||||
                    <el-input
 | 
			
		||||
                        v-model.trim="form.host"
 | 
			
		||||
                        placeholder="请输入host:port,集群模式用','分割"
 | 
			
		||||
                        auto-complete="off"
 | 
			
		||||
                        type="textarea"
 | 
			
		||||
                    ></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
                <el-form-item prop="password" label="密码:">
 | 
			
		||||
                    <el-input
 | 
			
		||||
                        type="password"
 | 
			
		||||
                        show-password
 | 
			
		||||
                        v-model.trim="form.password"
 | 
			
		||||
                        placeholder="请输入密码"
 | 
			
		||||
                        placeholder="请输入密码, 修改操作可不填"
 | 
			
		||||
                        autocomplete="new-password"
 | 
			
		||||
                    ></el-input>
 | 
			
		||||
                </el-form-item>
 | 
			
		||||
@@ -37,12 +42,29 @@
 | 
			
		||||
                <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="2" v-if="form.enableSshTunnel == 1"> 机器: </el-col>
 | 
			
		||||
                    <el-col :span="19" 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>
 | 
			
		||||
                    <el-button type="primary" :loading="btnLoading" @click="btnOk">确 定</el-button>
 | 
			
		||||
                </div>
 | 
			
		||||
            </template>
 | 
			
		||||
        </el-dialog>
 | 
			
		||||
@@ -53,7 +75,9 @@
 | 
			
		||||
import { toRefs, reactive, watch, defineComponent, ref } from 'vue';
 | 
			
		||||
import { redisApi } from './api';
 | 
			
		||||
import { projectApi } from '../project/api.ts';
 | 
			
		||||
import { machineApi } from '../machine/api.ts';
 | 
			
		||||
import { ElMessage } from 'element-plus';
 | 
			
		||||
import { RsaEncrypt } from '@/common/rsa';
 | 
			
		||||
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'RedisEdit',
 | 
			
		||||
@@ -77,17 +101,20 @@ export default defineComponent({
 | 
			
		||||
            dialogVisible: false,
 | 
			
		||||
            projects: [],
 | 
			
		||||
            envs: [],
 | 
			
		||||
            sshTunnelMachineList: [],
 | 
			
		||||
            form: {
 | 
			
		||||
                id: null,
 | 
			
		||||
                name: null,
 | 
			
		||||
                mode: "standalone",
 | 
			
		||||
                mode: 'standalone',
 | 
			
		||||
                host: null,
 | 
			
		||||
                password: null,
 | 
			
		||||
                project: null,
 | 
			
		||||
                projectId: null,
 | 
			
		||||
                envId: null,
 | 
			
		||||
                env: null,
 | 
			
		||||
                remark: "",
 | 
			
		||||
                remark: '',
 | 
			
		||||
                enableSshTunnel: null,
 | 
			
		||||
                sshTunnelMachineId: null,
 | 
			
		||||
            },
 | 
			
		||||
            btnLoading: false,
 | 
			
		||||
            rules: {
 | 
			
		||||
@@ -131,16 +158,27 @@ export default defineComponent({
 | 
			
		||||
 | 
			
		||||
        watch(props, async (newValue) => {
 | 
			
		||||
            state.dialogVisible = newValue.visible;
 | 
			
		||||
            if (!state.dialogVisible) {
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
            state.projects = newValue.projects;
 | 
			
		||||
            if (newValue.redis) {
 | 
			
		||||
                getEnvs(newValue.redis.projectId);
 | 
			
		||||
                state.form = { ...newValue.redis };
 | 
			
		||||
            } else {
 | 
			
		||||
                state.envs = [];
 | 
			
		||||
                state.form = { db: 0 } as any;
 | 
			
		||||
                state.form = { db: 0, enableSshTunnel: -1 } as any;
 | 
			
		||||
            }
 | 
			
		||||
            getSshTunnelMachines();
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        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 getEnvs = async (projectId: any) => {
 | 
			
		||||
            state.envs = await projectApi.projectEnvs.request({ projectId });
 | 
			
		||||
        };
 | 
			
		||||
@@ -166,9 +204,11 @@ export default defineComponent({
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const btnOk = async () => {
 | 
			
		||||
            redisForm.value.validate((valid: boolean) => {
 | 
			
		||||
            redisForm.value.validate(async (valid: boolean) => {
 | 
			
		||||
                if (valid) {
 | 
			
		||||
                    redisApi.saveRedis.request(state.form).then(() => {
 | 
			
		||||
                    const reqForm = { ...state.form };
 | 
			
		||||
                    reqForm.password = await RsaEncrypt(reqForm.password);
 | 
			
		||||
                    redisApi.saveRedis.request(reqForm).then(() => {
 | 
			
		||||
                        ElMessage.success('保存成功');
 | 
			
		||||
                        emit('val-change', state.form);
 | 
			
		||||
                        state.btnLoading = true;
 | 
			
		||||
@@ -193,6 +233,7 @@ export default defineComponent({
 | 
			
		||||
        return {
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
            redisForm,
 | 
			
		||||
            getSshTunnelMachines,
 | 
			
		||||
            changeProject,
 | 
			
		||||
            changeEnv,
 | 
			
		||||
            btnOk,
 | 
			
		||||
 
 | 
			
		||||
@@ -22,7 +22,7 @@
 | 
			
		||||
                <el-table-column prop="env" label="环境" min-width="100"></el-table-column>
 | 
			
		||||
                <el-table-column prop="host" label="host:port" min-width="150" show-overflow-tooltip> </el-table-column>
 | 
			
		||||
                <el-table-column prop="mode" label="mode" min-width="100"></el-table-column>
 | 
			
		||||
                <el-table-column prop="remark" label="备注" min-width="100"></el-table-column>
 | 
			
		||||
                <el-table-column prop="remark" label="备注" min-width="120" show-overflow-tooltip></el-table-column>
 | 
			
		||||
                <el-table-column prop="createTime" label="创建时间" min-width="160">
 | 
			
		||||
                    <template #default="scope">
 | 
			
		||||
                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
			
		||||
 
 | 
			
		||||
@@ -33,5 +33,5 @@ export const accountApi = {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export const logApi = {
 | 
			
		||||
    list: Api.create("/sys/logs", "get")
 | 
			
		||||
    list: Api.create("/syslogs", "get")
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -10,15 +10,14 @@
 | 
			
		||||
 | 
			
		||||
            <div style="float: right">
 | 
			
		||||
                <el-input
 | 
			
		||||
                    placeholder="请输入角色名称!"
 | 
			
		||||
                    placeholder="请输入角色名称"
 | 
			
		||||
                    class="mr2"
 | 
			
		||||
                    size="small"
 | 
			
		||||
                    style="width: 300px"
 | 
			
		||||
                    style="width: 200px"
 | 
			
		||||
                    v-model="query.name"
 | 
			
		||||
                    @clear="search"
 | 
			
		||||
                    clearable
 | 
			
		||||
                ></el-input>
 | 
			
		||||
                <el-button @click="search" type="success" icon="search" size="small"></el-button>
 | 
			
		||||
                <el-button @click="search" type="success" icon="search"></el-button>
 | 
			
		||||
            </div>
 | 
			
		||||
            <el-table :data="roles" @current-change="choose" ref="table" style="width: 100%">
 | 
			
		||||
                <el-table-column label="选择" width="50px">
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										103
									
								
								mayfly_go_web/src/views/system/syslog/SyslogList.vue
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										103
									
								
								mayfly_go_web/src/views/system/syslog/SyslogList.vue
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,103 @@
 | 
			
		||||
<template>
 | 
			
		||||
    <div class="role-list">
 | 
			
		||||
        <el-card>
 | 
			
		||||
            <div style="float: right">
 | 
			
		||||
                <el-select
 | 
			
		||||
                    remote
 | 
			
		||||
                    :remote-method="getAccount"
 | 
			
		||||
                    v-model="query.creatorId"
 | 
			
		||||
                    filterable
 | 
			
		||||
                    placeholder="请输入并选择账号"
 | 
			
		||||
                    clearable
 | 
			
		||||
                    class="mr5"
 | 
			
		||||
                >
 | 
			
		||||
                    <el-option v-for="item in accounts" :key="item.id" :label="item.username" :value="item.id"> </el-option>
 | 
			
		||||
                </el-select>
 | 
			
		||||
                <el-select v-model="query.type" filterable placeholder="请选择操作结果" clearable class="mr5">
 | 
			
		||||
                    <el-option label="成功" :value="1"> </el-option>
 | 
			
		||||
                    <el-option label="失败" :value="2"> </el-option>
 | 
			
		||||
                </el-select>
 | 
			
		||||
                <el-button @click="search" type="success" icon="search"></el-button>
 | 
			
		||||
            </div>
 | 
			
		||||
            <el-table :data="logs" style="width: 100%">
 | 
			
		||||
                <el-table-column prop="creator" label="操作人" min-width="100" show-overflow-tooltip></el-table-column>
 | 
			
		||||
                <el-table-column prop="createTime" label="操作时间" min-width="160">
 | 
			
		||||
                    <template #default="scope">
 | 
			
		||||
                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
			
		||||
                    </template>
 | 
			
		||||
                </el-table-column>
 | 
			
		||||
                <el-table-column prop="type" label="结果" min-width="65">
 | 
			
		||||
                    <template #default="scope">
 | 
			
		||||
                        <el-tag v-if="scope.row.type == 1" type="success" size="small">成功</el-tag>
 | 
			
		||||
                        <el-tag v-if="scope.row.type == 2" type="danger" size="small">失败</el-tag>
 | 
			
		||||
                    </template>
 | 
			
		||||
                </el-table-column>
 | 
			
		||||
                <el-table-column prop="description" label="描述" min-width="160" show-overflow-tooltip></el-table-column>
 | 
			
		||||
 | 
			
		||||
                <el-table-column prop="reqParam" label="请求信息" min-width="300" show-overflow-tooltip></el-table-column>
 | 
			
		||||
                <el-table-column prop="resp" label="响应信息" min-width="200" show-overflow-tooltip></el-table-column>
 | 
			
		||||
            </el-table>
 | 
			
		||||
            <el-row style="margin-top: 20px" type="flex" justify="end">
 | 
			
		||||
                <el-pagination
 | 
			
		||||
                    style="text-align: right"
 | 
			
		||||
                    @current-change="handlePageChange"
 | 
			
		||||
                    :total="total"
 | 
			
		||||
                    layout="prev, pager, next, total, jumper"
 | 
			
		||||
                    v-model:current-page="query.pageNum"
 | 
			
		||||
                    :page-size="query.pageSize"
 | 
			
		||||
                ></el-pagination>
 | 
			
		||||
            </el-row>
 | 
			
		||||
        </el-card>
 | 
			
		||||
    </div>
 | 
			
		||||
</template>
 | 
			
		||||
 | 
			
		||||
<script lang="ts">
 | 
			
		||||
import { toRefs, reactive, onMounted, defineComponent } from 'vue';
 | 
			
		||||
import { logApi, accountApi } from '../api';
 | 
			
		||||
export default defineComponent({
 | 
			
		||||
    name: 'SyslogList',
 | 
			
		||||
    components: {},
 | 
			
		||||
    setup() {
 | 
			
		||||
        const state = reactive({
 | 
			
		||||
            query: {
 | 
			
		||||
                pageNum: 1,
 | 
			
		||||
                pageSize: 10,
 | 
			
		||||
                name: null,
 | 
			
		||||
            },
 | 
			
		||||
            total: 0,
 | 
			
		||||
            logs: [],
 | 
			
		||||
            accounts: [],
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        onMounted(() => {
 | 
			
		||||
            search();
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        const search = async () => {
 | 
			
		||||
            let res = await logApi.list.request(state.query);
 | 
			
		||||
            state.logs = res.list;
 | 
			
		||||
            state.total = res.total;
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const handlePageChange = (curPage: number) => {
 | 
			
		||||
            state.query.pageNum = curPage;
 | 
			
		||||
            search();
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        const getAccount = (username: any) => {
 | 
			
		||||
            accountApi.list.request({ username }).then((res) => {
 | 
			
		||||
                state.accounts = res.list;
 | 
			
		||||
            });
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        return {
 | 
			
		||||
            ...toRefs(state),
 | 
			
		||||
            search,
 | 
			
		||||
            handlePageChange,
 | 
			
		||||
            getAccount,
 | 
			
		||||
        };
 | 
			
		||||
    },
 | 
			
		||||
});
 | 
			
		||||
</script>
 | 
			
		||||
<style lang="scss">
 | 
			
		||||
</style>
 | 
			
		||||
@@ -30,7 +30,6 @@ const viteConfig: UserConfig = {
 | 
			
		||||
				target: 'http://localhost:8888',
 | 
			
		||||
				ws: true,
 | 
			
		||||
				changeOrigin: true,
 | 
			
		||||
				rewrite: (path) => path.replace(/^\/api/, '/'),
 | 
			
		||||
			},
 | 
			
		||||
		},
 | 
			
		||||
	},
 | 
			
		||||
 
 | 
			
		||||
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							@@ -1,6 +1,6 @@
 | 
			
		||||
app:
 | 
			
		||||
  name: mayfly-go
 | 
			
		||||
  version: 1.2.0
 | 
			
		||||
  version: 1.2.3
 | 
			
		||||
  
 | 
			
		||||
server:
 | 
			
		||||
  # debug release test
 | 
			
		||||
 
 | 
			
		||||
@@ -6,15 +6,16 @@ require (
 | 
			
		||||
	github.com/dgrijalva/jwt-go v3.2.0+incompatible // jwt
 | 
			
		||||
	github.com/gin-gonic/gin v1.8.1
 | 
			
		||||
	github.com/go-redis/redis/v8 v8.11.5
 | 
			
		||||
	github.com/go-sql-driver/mysql v1.6.0
 | 
			
		||||
	github.com/gorilla/websocket v1.5.0
 | 
			
		||||
	github.com/lib/pq v1.10.6
 | 
			
		||||
	github.com/mojocn/base64Captcha v1.3.5 // 验证码
 | 
			
		||||
	github.com/pkg/sftp v1.13.4
 | 
			
		||||
	github.com/pkg/sftp v1.13.5
 | 
			
		||||
	github.com/robfig/cron/v3 v3.0.1 // 定时任务
 | 
			
		||||
	github.com/sirupsen/logrus v1.8.1
 | 
			
		||||
	github.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2
 | 
			
		||||
	go.mongodb.org/mongo-driver v1.9.1 // mongo
 | 
			
		||||
	golang.org/x/crypto v0.0.0-20220525230936-793ad666bf5e // ssh
 | 
			
		||||
	golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d // ssh
 | 
			
		||||
	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b
 | 
			
		||||
	// gorm
 | 
			
		||||
	gorm.io/driver/mysql v1.3.4
 | 
			
		||||
@@ -28,7 +29,6 @@ require (
 | 
			
		||||
	github.com/go-playground/locales v0.14.0 // indirect
 | 
			
		||||
	github.com/go-playground/universal-translator v0.18.0 // indirect
 | 
			
		||||
	github.com/go-playground/validator/v10 v10.10.1 // indirect
 | 
			
		||||
	github.com/go-sql-driver/mysql v1.6.0 // indirect
 | 
			
		||||
	github.com/go-stack/stack v1.8.0 // indirect
 | 
			
		||||
	github.com/goccy/go-json v0.9.7 // indirect
 | 
			
		||||
	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
 | 
			
		||||
 
 | 
			
		||||
@@ -2,7 +2,7 @@ package initialize
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	common_index_router "mayfly-go/internal/common/router"
 | 
			
		||||
	common_router "mayfly-go/internal/common/router"
 | 
			
		||||
	devops_router "mayfly-go/internal/devops/router"
 | 
			
		||||
	sys_router "mayfly-go/internal/sys/router"
 | 
			
		||||
	"mayfly-go/pkg/config"
 | 
			
		||||
@@ -46,13 +46,15 @@ func InitRouter() *gin.Engine {
 | 
			
		||||
	// 设置路由组
 | 
			
		||||
	api := router.Group("/api")
 | 
			
		||||
	{
 | 
			
		||||
		common_index_router.InitIndexRouter(api)
 | 
			
		||||
		common_router.InitIndexRouter(api)
 | 
			
		||||
		common_router.InitCommonRouter(api)
 | 
			
		||||
 | 
			
		||||
		sys_router.InitCaptchaRouter(api)
 | 
			
		||||
		sys_router.InitAccountRouter(api) // 注册account路由
 | 
			
		||||
		sys_router.InitResourceRouter(api)
 | 
			
		||||
		sys_router.InitRoleRouter(api)
 | 
			
		||||
		sys_router.InitSystemRouter(api)
 | 
			
		||||
		sys_router.InitSyslogRouter(api)
 | 
			
		||||
 | 
			
		||||
		devops_router.InitProjectRouter(api)
 | 
			
		||||
		devops_router.InitDbRouter(api)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										10
									
								
								server/initialize/savelog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								server/initialize/savelog.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,10 @@
 | 
			
		||||
package initialize
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	sys_application "mayfly-go/internal/sys/application"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func InitSaveLogFunc() ctx.SaveLogFunc {
 | 
			
		||||
	return sys_application.SyslogApp.SaveFromReq
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										16
									
								
								server/internal/common/api/common.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										16
									
								
								server/internal/common/api/common.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,16 @@
 | 
			
		||||
package api
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Common struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (i *Common) RasPublicKey(rc *ctx.ReqCtx) {
 | 
			
		||||
	publicKeyStr, err := utils.GetRsaPublicKey()
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "rsa生成公私钥失败")
 | 
			
		||||
	rc.ResData = publicKeyStr
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										21
									
								
								server/internal/common/router/common.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								server/internal/common/router/common.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
package router
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/internal/common/api"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func InitCommonRouter(router *gin.RouterGroup) {
 | 
			
		||||
	common := router.Group("common")
 | 
			
		||||
	c := &api.Common{}
 | 
			
		||||
	{
 | 
			
		||||
		// 获取公钥
 | 
			
		||||
		common.GET("public-key", func(g *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(g).
 | 
			
		||||
				WithNeedToken(false).
 | 
			
		||||
				Handle(c.RasPublicKey)
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										16
									
								
								server/internal/constant/constant.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										16
									
								
								server/internal/constant/constant.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,16 @@
 | 
			
		||||
package constant
 | 
			
		||||
 | 
			
		||||
import "time"
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	MachineConnExpireTime = 60 * time.Minute
 | 
			
		||||
	DbConnExpireTime      = 45 * time.Minute
 | 
			
		||||
	RedisConnExpireTime   = 30 * time.Minute
 | 
			
		||||
	MongoConnExpireTime   = 30 * time.Minute
 | 
			
		||||
 | 
			
		||||
/****  开发测试使用   ****/
 | 
			
		||||
// MachineConnExpireTime = 20 * time.Second
 | 
			
		||||
// DbConnExpireTime      = 20 * time.Second
 | 
			
		||||
// RedisConnExpireTime   = 20 * time.Second
 | 
			
		||||
// MongoConnExpireTime   = 20 * time.Second
 | 
			
		||||
)
 | 
			
		||||
@@ -45,12 +45,40 @@ func (d *Db) Save(rc *ctx.ReqCtx) {
 | 
			
		||||
	form := &form.DbForm{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
			
		||||
 | 
			
		||||
	db := new(entity.Db)
 | 
			
		||||
	utils.Copy(db, form)
 | 
			
		||||
 | 
			
		||||
	// 密码解密,并使用解密后的赋值
 | 
			
		||||
	originPwd, err := utils.DefaultRsaDecrypt(form.Password, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
			
		||||
	db.Password = originPwd
 | 
			
		||||
 | 
			
		||||
	// 密码脱敏记录日志
 | 
			
		||||
	form.Password = "****"
 | 
			
		||||
	rc.ReqParam = form
 | 
			
		||||
 | 
			
		||||
	db.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	d.DbApp.Save(db)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取数据库实例的所有数据库名
 | 
			
		||||
func (d *Db) GetDatabaseNames(rc *ctx.ReqCtx) {
 | 
			
		||||
	form := &form.DbForm{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
			
		||||
 | 
			
		||||
	db := new(entity.Db)
 | 
			
		||||
	utils.Copy(db, form)
 | 
			
		||||
	db.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	d.DbApp.Save(db)
 | 
			
		||||
 | 
			
		||||
	// 密码解密,并使用解密后的赋值
 | 
			
		||||
	originPwd, err := utils.DefaultRsaDecrypt(form.Password, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
			
		||||
	db.Password = originPwd
 | 
			
		||||
 | 
			
		||||
	// 如果id不为空,并且密码为空则从数据库查询
 | 
			
		||||
	if form.Id != 0 && db.Password == "" {
 | 
			
		||||
		db = d.DbApp.GetById(form.Id)
 | 
			
		||||
	}
 | 
			
		||||
	rc.ResData = d.DbApp.GetDatabases(db)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (d *Db) DeleteDb(rc *ctx.ReqCtx) {
 | 
			
		||||
@@ -136,6 +164,8 @@ func (d *Db) ExecSqlFile(rc *ctx.ReqCtx) {
 | 
			
		||||
	filename := fileheader.Filename
 | 
			
		||||
	dbId, db := GetIdAndDb(g)
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintf("dbId: %d, db: %s, filename: %s", dbId, db, filename)
 | 
			
		||||
 | 
			
		||||
	go func() {
 | 
			
		||||
		db := d.DbApp.GetDbInstance(dbId, db)
 | 
			
		||||
 | 
			
		||||
@@ -192,7 +222,6 @@ func (d *Db) DumpSql(rc *ctx.ReqCtx) {
 | 
			
		||||
	g.Header("Content-Type", "application/octet-stream")
 | 
			
		||||
	g.Header("Content-Disposition", "attachment; filename="+filename)
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintf("数据库id: %d -- %s", dbId, db)
 | 
			
		||||
	writer := g.Writer
 | 
			
		||||
	writer.WriteString("-- ----------------------------")
 | 
			
		||||
	writer.WriteString("\n-- 导出平台: mayfly-go")
 | 
			
		||||
@@ -258,6 +287,8 @@ func (d *Db) DumpSql(rc *ctx.ReqCtx) {
 | 
			
		||||
		writer.WriteString("COMMIT;\n")
 | 
			
		||||
	}
 | 
			
		||||
	rc.NoRes = true
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintf("dbId: %d, db: %s, tables: %s, dumpType: %s", dbId, db, tablesStr, dumpType)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// @router /api/db/:dbId/t-metadata [get]
 | 
			
		||||
 
 | 
			
		||||
@@ -9,11 +9,14 @@ type DbForm struct {
 | 
			
		||||
	Username  string `binding:"required" json:"username"`
 | 
			
		||||
	Password  string `json:"password"`
 | 
			
		||||
	Params    string `json:"params"`
 | 
			
		||||
	Database  string `binding:"required" json:"database"`
 | 
			
		||||
	Database  string `json:"database"`
 | 
			
		||||
	ProjectId uint64 `binding:"required" json:"projectId"`
 | 
			
		||||
	Project   string `json:"project"`
 | 
			
		||||
	Env       string `json:"env"`
 | 
			
		||||
	EnvId     uint64 `binding:"required" json:"envId"`
 | 
			
		||||
 | 
			
		||||
	EnableSshTunnel    int8   `json:"enableSshTunnel"`
 | 
			
		||||
	SshTunnelMachineId uint64 `json:"sshTunnelMachineId"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type DbSqlSaveForm struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,18 +1,18 @@
 | 
			
		||||
package form
 | 
			
		||||
 | 
			
		||||
type MachineForm struct {
 | 
			
		||||
	Id          uint64 `json:"id"`
 | 
			
		||||
	ProjectId   uint64 `json:"projectId"`
 | 
			
		||||
	ProjectName string `json:"projectName"`
 | 
			
		||||
	Name        string `json:"name" binding:"required"`
 | 
			
		||||
	// IP地址
 | 
			
		||||
	Ip string `json:"ip" binding:"required"`
 | 
			
		||||
	// 用户名
 | 
			
		||||
	Username string `json:"username" binding:"required"`
 | 
			
		||||
	Password string `json:"password"`
 | 
			
		||||
	// 端口号
 | 
			
		||||
	Port   int    `json:"port" binding:"required"`
 | 
			
		||||
	Remark string `json:"remark"`
 | 
			
		||||
	Id                 uint64 `json:"id"`
 | 
			
		||||
	ProjectId          uint64 `json:"projectId"`
 | 
			
		||||
	ProjectName        string `json:"projectName"`
 | 
			
		||||
	Name               string `json:"name" binding:"required"`
 | 
			
		||||
	Ip                 string `json:"ip" binding:"required"`       // IP地址
 | 
			
		||||
	Username           string `json:"username" binding:"required"` // 用户名
 | 
			
		||||
	AuthMethod         int8   `json:"authMethod" binding:"required"`
 | 
			
		||||
	Password           string `json:"password"`
 | 
			
		||||
	Port               int    `json:"port" binding:"required"` // 端口号
 | 
			
		||||
	Remark             string `json:"remark"`
 | 
			
		||||
	EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MachineRunForm struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +1,15 @@
 | 
			
		||||
package form
 | 
			
		||||
 | 
			
		||||
type Mongo struct {
 | 
			
		||||
	Id        uint64
 | 
			
		||||
	Uri       string `binding:"required" json:"uri"`
 | 
			
		||||
	Name      string `binding:"required" json:"name"`
 | 
			
		||||
	ProjectId uint64 `binding:"required" json:"projectId"`
 | 
			
		||||
	Project   string `json:"project"`
 | 
			
		||||
	Env       string `json:"env"`
 | 
			
		||||
	EnvId     uint64 `binding:"required" json:"envId"`
 | 
			
		||||
	Id                 uint64
 | 
			
		||||
	Uri                string `binding:"required" json:"uri"`
 | 
			
		||||
	EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	Name               string `binding:"required" json:"name"`
 | 
			
		||||
	ProjectId          uint64 `binding:"required" json:"projectId"`
 | 
			
		||||
	Project            string `json:"project"`
 | 
			
		||||
	Env                string `json:"env"`
 | 
			
		||||
	EnvId              uint64 `binding:"required" json:"envId"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MongoCommand struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,15 +1,18 @@
 | 
			
		||||
package form
 | 
			
		||||
 | 
			
		||||
type Redis struct {
 | 
			
		||||
	Id        uint64
 | 
			
		||||
	Host      string `binding:"required" json:"host"`
 | 
			
		||||
	Password  string `json:"password"`
 | 
			
		||||
	Mode      string `json:"mode"`
 | 
			
		||||
	Db        int    `json:"db"`
 | 
			
		||||
	ProjectId uint64 `binding:"required" json:"projectId"`
 | 
			
		||||
	Project   string `json:"project"`
 | 
			
		||||
	Env       string `json:"env"`
 | 
			
		||||
	EnvId     uint64 `binding:"required" json:"envId"`
 | 
			
		||||
	Id                 uint64
 | 
			
		||||
	Host               string `binding:"required" json:"host"`
 | 
			
		||||
	Password           string `json:"password"`
 | 
			
		||||
	Mode               string `json:"mode"`
 | 
			
		||||
	Db                 int    `json:"db"`
 | 
			
		||||
	EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	ProjectId          uint64 `binding:"required" json:"projectId"`
 | 
			
		||||
	Project            string `json:"project"`
 | 
			
		||||
	Env                string `json:"env"`
 | 
			
		||||
	EnvId              uint64 `binding:"required" json:"envId"`
 | 
			
		||||
	Remark             string `json:"remark"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type KeyInfo struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -54,11 +54,22 @@ func (m *Machine) SaveMachine(rc *ctx.ReqCtx) {
 | 
			
		||||
	machineForm := new(form.MachineForm)
 | 
			
		||||
	ginx.BindJsonAndValid(g, machineForm)
 | 
			
		||||
 | 
			
		||||
	entity := new(entity.Machine)
 | 
			
		||||
	utils.Copy(entity, machineForm)
 | 
			
		||||
	me := new(entity.Machine)
 | 
			
		||||
	utils.Copy(me, machineForm)
 | 
			
		||||
 | 
			
		||||
	entity.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	m.MachineApp.Save(entity)
 | 
			
		||||
	if me.AuthMethod == entity.MachineAuthMethodPassword {
 | 
			
		||||
		// 密码解密,并使用解密后的赋值
 | 
			
		||||
		originPwd, err := utils.DefaultRsaDecrypt(machineForm.Password, true)
 | 
			
		||||
		biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
			
		||||
		me.Password = originPwd
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 密码脱敏记录日志
 | 
			
		||||
	machineForm.Password = "****"
 | 
			
		||||
	rc.ReqParam = machineForm
 | 
			
		||||
 | 
			
		||||
	me.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	m.MachineApp.Save(me)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *Machine) ChangeStatus(rc *ctx.ReqCtx) {
 | 
			
		||||
 
 | 
			
		||||
@@ -38,6 +38,11 @@ func (m *Mongo) Save(rc *ctx.ReqCtx) {
 | 
			
		||||
 | 
			
		||||
	mongo := new(entity.Mongo)
 | 
			
		||||
	utils.Copy(mongo, form)
 | 
			
		||||
	// 解密uri,并使用解密后的赋值
 | 
			
		||||
	originUri, err := utils.DefaultRsaDecrypt(form.Uri, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密uri错误: %s")
 | 
			
		||||
	mongo.Uri = originUri
 | 
			
		||||
 | 
			
		||||
	mongo.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	m.MongoApp.Save(mongo)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -43,7 +43,8 @@ func (p *Project) GetProjects(rc *ctx.ReqCtx) {
 | 
			
		||||
func (p *Project) SaveProject(rc *ctx.ReqCtx) {
 | 
			
		||||
	project := &entity.Project{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, project)
 | 
			
		||||
	rc.ReqParam = project
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintf("projectId: %d, projectName: %s, remark: %s", project.Id, project.Name, project.Remark)
 | 
			
		||||
 | 
			
		||||
	project.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	p.ProjectApp.SaveProject(project)
 | 
			
		||||
@@ -81,7 +82,8 @@ func (p *Project) GetProjectMembers(rc *ctx.ReqCtx) {
 | 
			
		||||
func (p *Project) SaveProjectMember(rc *ctx.ReqCtx) {
 | 
			
		||||
	projectMem := &entity.ProjectMember{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, projectMem)
 | 
			
		||||
	rc.ReqParam = projectMem
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintf("projectId: %d, username: %s", projectMem.ProjectId, projectMem.Username)
 | 
			
		||||
 | 
			
		||||
	// 校验账号,并赋值username
 | 
			
		||||
	account := &sys_entity.Account{}
 | 
			
		||||
 
 | 
			
		||||
@@ -36,10 +36,18 @@ func (r *Redis) Save(rc *ctx.ReqCtx) {
 | 
			
		||||
	form := &form.Redis{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = form
 | 
			
		||||
 | 
			
		||||
	redis := new(entity.Redis)
 | 
			
		||||
	utils.Copy(redis, form)
 | 
			
		||||
 | 
			
		||||
	// 密码解密,并使用解密后的赋值
 | 
			
		||||
	originPwd, err := utils.DefaultRsaDecrypt(redis.Password, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
			
		||||
	redis.Password = originPwd
 | 
			
		||||
 | 
			
		||||
	// 密码脱敏记录日志
 | 
			
		||||
	form.Password = "****"
 | 
			
		||||
	rc.ReqParam = form
 | 
			
		||||
 | 
			
		||||
	redis.SetBaseInfo(rc.LoginAccount)
 | 
			
		||||
	r.RedisApp.Save(redis)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -19,4 +19,7 @@ type SelectDataDbVO struct {
 | 
			
		||||
	CreateTime *time.Time `json:"createTime"`
 | 
			
		||||
	Creator    *string    `json:"creator"`
 | 
			
		||||
	CreatorId  *int64     `json:"creatorId"`
 | 
			
		||||
 | 
			
		||||
	EnableSshTunnel    *int8   `json:"enableSshTunnel"`
 | 
			
		||||
	SshTunnelMachineId *uint64 `json:"sshTunnelMachineId"`
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -5,17 +5,19 @@ import "time"
 | 
			
		||||
type Redis struct {
 | 
			
		||||
	Id *int64 `json:"id"`
 | 
			
		||||
	// Name       *string    `json:"name"`
 | 
			
		||||
	Host       *string    `json:"host"`
 | 
			
		||||
	Db         int        `json:"db"`
 | 
			
		||||
	ProjectId  *int64     `json:"projectId"`
 | 
			
		||||
	Project    *string    `json:"project"`
 | 
			
		||||
	Mode       *string    `json:"mode"`
 | 
			
		||||
	Remark     *string    `json:"remark"`
 | 
			
		||||
	Env        *string    `json:"env"`
 | 
			
		||||
	EnvId      *int64     `json:"envId"`
 | 
			
		||||
	CreateTime *time.Time `json:"createTime"`
 | 
			
		||||
	Creator    *string    `json:"creator"`
 | 
			
		||||
	CreatorId  *int64     `json:"creatorId"`
 | 
			
		||||
	Host               *string    `json:"host"`
 | 
			
		||||
	Db                 int        `json:"db"`
 | 
			
		||||
	ProjectId          *int64     `json:"projectId"`
 | 
			
		||||
	Project            *string    `json:"project"`
 | 
			
		||||
	Mode               *string    `json:"mode"`
 | 
			
		||||
	EnableSshTunnel    *int8      `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId *uint64    `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	Remark             *string    `json:"remark"`
 | 
			
		||||
	Env                *string    `json:"env"`
 | 
			
		||||
	EnvId              *int64     `json:"envId"`
 | 
			
		||||
	CreateTime         *time.Time `json:"createTime"`
 | 
			
		||||
	Creator            *string    `json:"creator"`
 | 
			
		||||
	CreatorId          *int64     `json:"creatorId"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Keys struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -15,22 +15,25 @@ type AccountVO struct {
 | 
			
		||||
 | 
			
		||||
type MachineVO struct {
 | 
			
		||||
	//models.BaseModel
 | 
			
		||||
	Id          *uint64    `json:"id"`
 | 
			
		||||
	ProjectId   uint64     `json:"projectId"`
 | 
			
		||||
	ProjectName string     `json:"projectName"`
 | 
			
		||||
	Name        *string    `json:"name"`
 | 
			
		||||
	Username    *string    `json:"username"`
 | 
			
		||||
	Ip          *string    `json:"ip"`
 | 
			
		||||
	Port        *int       `json:"port"`
 | 
			
		||||
	Status      *int8      `json:"status"`
 | 
			
		||||
	CreateTime  *time.Time `json:"createTime"`
 | 
			
		||||
	Creator     *string    `json:"creator"`
 | 
			
		||||
	CreatorId   *int64     `json:"creatorId"`
 | 
			
		||||
	UpdateTime  *time.Time `json:"updateTime"`
 | 
			
		||||
	Modifier    *string    `json:"modifier"`
 | 
			
		||||
	ModifierId  *int64     `json:"modifierId"`
 | 
			
		||||
	HasCli      bool       `json:"hasCli" gorm:"-"`
 | 
			
		||||
	Remark      *string    `json:"remark"`
 | 
			
		||||
	Id                 *uint64    `json:"id"`
 | 
			
		||||
	ProjectId          uint64     `json:"projectId"`
 | 
			
		||||
	ProjectName        string     `json:"projectName"`
 | 
			
		||||
	Name               *string    `json:"name"`
 | 
			
		||||
	Username           *string    `json:"username"`
 | 
			
		||||
	Ip                 *string    `json:"ip"`
 | 
			
		||||
	Port               *int       `json:"port"`
 | 
			
		||||
	AuthMethod         *int8      `json:"authMethod"`
 | 
			
		||||
	Status             *int8      `json:"status"`
 | 
			
		||||
	EnableSshTunnel    *int8      `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId *uint64    `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	CreateTime         *time.Time `json:"createTime"`
 | 
			
		||||
	Creator            *string    `json:"creator"`
 | 
			
		||||
	CreatorId          *int64     `json:"creatorId"`
 | 
			
		||||
	UpdateTime         *time.Time `json:"updateTime"`
 | 
			
		||||
	Modifier           *string    `json:"modifier"`
 | 
			
		||||
	ModifierId         *int64     `json:"modifierId"`
 | 
			
		||||
	HasCli             bool       `json:"hasCli" gorm:"-"`
 | 
			
		||||
	Remark             *string    `json:"remark"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MachineScriptVO struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,24 +1,29 @@
 | 
			
		||||
package application
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"database/sql"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/internal/constant"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/repository"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/persistence"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/cache"
 | 
			
		||||
	"mayfly-go/pkg/global"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"net"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	_ "github.com/lib/pq"
 | 
			
		||||
	"github.com/go-sql-driver/mysql"
 | 
			
		||||
	"github.com/lib/pq"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Db interface {
 | 
			
		||||
@@ -42,6 +47,9 @@ type Db interface {
 | 
			
		||||
	// @param id 数据库实例id
 | 
			
		||||
	// @param db 数据库
 | 
			
		||||
	GetDbInstance(id uint64, db string) *DbInstance
 | 
			
		||||
 | 
			
		||||
	// 获取数据库实例的所有数据库列表
 | 
			
		||||
	GetDatabases(entity *entity.Db) []string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type dbAppImpl struct {
 | 
			
		||||
@@ -75,10 +83,11 @@ func (d *dbAppImpl) GetById(id uint64, cols ...string) *entity.Db {
 | 
			
		||||
 | 
			
		||||
func (d *dbAppImpl) Save(dbEntity *entity.Db) {
 | 
			
		||||
	// 默认tcp连接
 | 
			
		||||
	dbEntity.Network = "tcp"
 | 
			
		||||
	dbEntity.Network = dbEntity.GetNetwork()
 | 
			
		||||
 | 
			
		||||
	// 测试连接
 | 
			
		||||
	if dbEntity.Password != "" {
 | 
			
		||||
		TestConnection(*dbEntity)
 | 
			
		||||
		TestConnection(dbEntity)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 查找是否存在该库
 | 
			
		||||
@@ -102,6 +111,8 @@ func (d *dbAppImpl) Save(dbEntity *entity.Db) {
 | 
			
		||||
 | 
			
		||||
	var oldDbs []interface{}
 | 
			
		||||
	for _, v := range strings.Split(old.Database, " ") {
 | 
			
		||||
		// 关闭数据库连接
 | 
			
		||||
		CloseDb(dbEntity.Id, v)
 | 
			
		||||
		oldDbs = append(oldDbs, v)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
@@ -114,14 +125,11 @@ func (d *dbAppImpl) Save(dbEntity *entity.Db) {
 | 
			
		||||
		return i1.(string) == i2.(string)
 | 
			
		||||
	})
 | 
			
		||||
	for _, v := range delDb {
 | 
			
		||||
		// 先关闭数据库连接
 | 
			
		||||
		CloseDb(dbEntity.Id, v.(string))
 | 
			
		||||
		// 删除该库关联的所有sql记录
 | 
			
		||||
		d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: dbId, Db: v.(string)})
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	d.dbRepo.Update(dbEntity)
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (d *dbAppImpl) Delete(id uint64) {
 | 
			
		||||
@@ -136,11 +144,34 @@ func (d *dbAppImpl) Delete(id uint64) {
 | 
			
		||||
	d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: id})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (d *dbAppImpl) GetDatabases(ed *entity.Db) []string {
 | 
			
		||||
	databases := make([]string, 0)
 | 
			
		||||
	var dbConn *sql.DB
 | 
			
		||||
	var metaDb string
 | 
			
		||||
	var getDatabasesSql string
 | 
			
		||||
	if ed.Type == entity.DbTypeMysql {
 | 
			
		||||
		metaDb = "information_schema"
 | 
			
		||||
		getDatabasesSql = "SELECT SCHEMA_NAME AS dbname FROM SCHEMATA"
 | 
			
		||||
	} else {
 | 
			
		||||
		metaDb = "postgres"
 | 
			
		||||
		getDatabasesSql = "SELECT datname AS dbname FROM pg_database"
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	dbConn, err := GetDbConn(ed, metaDb)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "数据库连接失败: %s")
 | 
			
		||||
	defer dbConn.Close()
 | 
			
		||||
 | 
			
		||||
	_, res, err := SelectDataByDb(dbConn, getDatabasesSql)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "获取数据库列表失败")
 | 
			
		||||
	for _, re := range res {
 | 
			
		||||
		databases = append(databases, re["dbname"].(string))
 | 
			
		||||
	}
 | 
			
		||||
	return databases
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var mutex sync.Mutex
 | 
			
		||||
 | 
			
		||||
func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
			
		||||
	mutex.Lock()
 | 
			
		||||
	defer mutex.Unlock()
 | 
			
		||||
	// Id不为0,则为需要缓存
 | 
			
		||||
	needCache := id != 0
 | 
			
		||||
	if needCache {
 | 
			
		||||
@@ -149,20 +180,21 @@ func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
			
		||||
			return load.(*DbInstance)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	biz.IsTrue(mutex.TryLock(), "有数据库实例在连接中...请稍后重试")
 | 
			
		||||
	defer mutex.Unlock()
 | 
			
		||||
 | 
			
		||||
	d := da.GetById(id)
 | 
			
		||||
	biz.NotNil(d, "数据库信息不存在")
 | 
			
		||||
	biz.IsTrue(strings.Contains(d.Database, db), "未配置该库的操作权限")
 | 
			
		||||
	global.Log.Infof("连接db: %s:%d/%s", d.Host, d.Port, db)
 | 
			
		||||
 | 
			
		||||
	// 将数据库替换为要访问的数据库,原本数据库为空格拼接的所有库
 | 
			
		||||
	d.Database = db
 | 
			
		||||
	DB, err := sql.Open(d.Type, getDsn(d))
 | 
			
		||||
	biz.ErrIsNil(err, fmt.Sprintf("Open %s failed, err:%v\n", d.Type, err))
 | 
			
		||||
	perr := DB.Ping()
 | 
			
		||||
	if perr != nil {
 | 
			
		||||
	cacheKey := GetDbCacheKey(id, db)
 | 
			
		||||
	dbi := &DbInstance{Id: cacheKey, Type: d.Type, ProjectId: d.ProjectId, sshTunnelMachineId: d.SshTunnelMachineId}
 | 
			
		||||
 | 
			
		||||
	DB, err := GetDbConn(d, db)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		dbi.Close()
 | 
			
		||||
		global.Log.Errorf("连接db失败: %s:%d/%s", d.Host, d.Port, db)
 | 
			
		||||
		panic(biz.NewBizErr(fmt.Sprintf("数据库连接失败: %s", perr.Error())))
 | 
			
		||||
		panic(biz.NewBizErr(fmt.Sprintf("数据库连接失败: %s", err.Error())))
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 最大连接周期,超过时间的连接就close
 | 
			
		||||
@@ -172,8 +204,8 @@ func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
			
		||||
	// 设置闲置连接数
 | 
			
		||||
	DB.SetMaxIdleConns(1)
 | 
			
		||||
 | 
			
		||||
	cacheKey := GetDbCacheKey(id, db)
 | 
			
		||||
	dbi := &DbInstance{Id: cacheKey, Type: d.Type, ProjectId: d.ProjectId, db: DB}
 | 
			
		||||
	dbi.db = DB
 | 
			
		||||
	global.Log.Infof("连接db: %s:%d/%s", d.Host, d.Port, db)
 | 
			
		||||
	if needCache {
 | 
			
		||||
		dbCache.Put(cacheKey, dbi)
 | 
			
		||||
	}
 | 
			
		||||
@@ -182,14 +214,27 @@ func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
			
		||||
 | 
			
		||||
//------------------------------------------------------------------------------
 | 
			
		||||
 | 
			
		||||
// 客户端连接缓存,30分钟内没有访问则会被关闭, key为数据库实例id:数据库
 | 
			
		||||
var dbCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
			
		||||
// 客户端连接缓存,指定时间内没有访问则会被关闭, key为数据库实例id:数据库
 | 
			
		||||
var dbCache = cache.NewTimedCache(constant.DbConnExpireTime, 5*time.Second).
 | 
			
		||||
	WithUpdateAccessTime(true).
 | 
			
		||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
			
		||||
		global.Log.Info(fmt.Sprintf("删除db连接缓存 id = %s", key))
 | 
			
		||||
		value.(*DbInstance).Close()
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	machine.AddCheckSshTunnelMachineUseFunc(func(machineId uint64) bool {
 | 
			
		||||
		// 遍历所有db连接实例,若存在redis实例使用该ssh隧道机器,则返回true,表示还在使用中...
 | 
			
		||||
		items := dbCache.Items()
 | 
			
		||||
		for _, v := range items {
 | 
			
		||||
			if v.Value.(*DbInstance).sshTunnelMachineId == machineId {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return false
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetDbCacheKey(dbId uint64, db string) string {
 | 
			
		||||
	return fmt.Sprintf("%d:%s", dbId, db)
 | 
			
		||||
}
 | 
			
		||||
@@ -201,37 +246,46 @@ func GetDbInstanceByCache(id string) *DbInstance {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestConnection(d entity.Db) {
 | 
			
		||||
func TestConnection(d *entity.Db) {
 | 
			
		||||
	// 验证第一个库是否可以连接即可
 | 
			
		||||
	d.Database = strings.Split(d.Database, " ")[0]
 | 
			
		||||
	DB, err := sql.Open(d.Type, getDsn(&d))
 | 
			
		||||
	biz.ErrIsNil(err, "Open %s failed, err:%v\n", d.Type, err)
 | 
			
		||||
	DB, err := GetDbConn(d, strings.Split(d.Database, " ")[0])
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "数据库连接失败: %s")
 | 
			
		||||
	defer DB.Close()
 | 
			
		||||
	perr := DB.Ping()
 | 
			
		||||
	biz.ErrIsNilAppendErr(perr, "数据库连接失败: %s")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// db实例
 | 
			
		||||
type DbInstance struct {
 | 
			
		||||
	Id        string
 | 
			
		||||
	Type      string
 | 
			
		||||
	ProjectId uint64
 | 
			
		||||
	db        *sql.DB
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 执行查询语句
 | 
			
		||||
// 依次返回 列名数组,结果map,错误
 | 
			
		||||
func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interface{}, error) {
 | 
			
		||||
	execSql = strings.Trim(execSql, " ")
 | 
			
		||||
	isSelect := strings.HasPrefix(execSql, "SELECT") || strings.HasPrefix(execSql, "select")
 | 
			
		||||
	isShow := strings.HasPrefix(execSql, "show")
 | 
			
		||||
	isExplain := strings.HasPrefix(execSql, "explain")
 | 
			
		||||
 | 
			
		||||
	if !isSelect && !isShow && !isExplain {
 | 
			
		||||
		return nil, nil, errors.New("该sql非查询语句")
 | 
			
		||||
// 获取数据库连接
 | 
			
		||||
func GetDbConn(d *entity.Db, db string) (*sql.DB, error) {
 | 
			
		||||
	// SSH Conect
 | 
			
		||||
	if d.EnableSshTunnel == 1 && d.SshTunnelMachineId != 0 {
 | 
			
		||||
		sshTunnelMachine := MachineApp.GetSshTunnelMachine(d.SshTunnelMachineId)
 | 
			
		||||
		defer machine.CloseSshTunnelMachine(d.SshTunnelMachineId, 0)
 | 
			
		||||
		if d.Type == entity.DbTypeMysql {
 | 
			
		||||
			mysql.RegisterDialContext(d.Network, func(ctx context.Context, addr string) (net.Conn, error) {
 | 
			
		||||
				return MachineApp.GetSshTunnelMachine(d.SshTunnelMachineId).GetDialConn("tcp", addr)
 | 
			
		||||
			})
 | 
			
		||||
		} else if d.Type == entity.DbTypePostgres {
 | 
			
		||||
			_, err := pq.DialOpen(&PqSqlDialer{sshTunnelMachine: sshTunnelMachine}, getDsn(d, db))
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				panic(biz.NewBizErr(fmt.Sprintf("postgres隧道连接失败: %s", err.Error())))
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	rows, err := d.db.Query(execSql)
 | 
			
		||||
	DB, err := sql.Open(d.Type, getDsn(d, db))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	err = DB.Ping()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		DB.Close()
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return DB, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func SelectDataByDb(db *sql.DB, selectSql string) ([]string, []map[string]interface{}, error) {
 | 
			
		||||
	rows, err := db.Query(selectSql)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, nil, err
 | 
			
		||||
	}
 | 
			
		||||
@@ -316,6 +370,45 @@ func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interfac
 | 
			
		||||
	return colNames, result, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type PqSqlDialer struct {
 | 
			
		||||
	sshTunnelMachine *machine.SshTunnelMachine
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (pd *PqSqlDialer) Dial(network, address string) (net.Conn, error) {
 | 
			
		||||
	if sshConn, err := pd.sshTunnelMachine.GetDialConn("tcp", address); err == nil {
 | 
			
		||||
		// 将ssh conn包装,否则redis内部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
			
		||||
		return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
			
		||||
	} else {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
func (pd *PqSqlDialer) DialTimeout(network, address string, timeout time.Duration) (net.Conn, error) {
 | 
			
		||||
	return pd.Dial(network, address)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// db实例
 | 
			
		||||
type DbInstance struct {
 | 
			
		||||
	Id                 string
 | 
			
		||||
	Type               string
 | 
			
		||||
	ProjectId          uint64
 | 
			
		||||
	db                 *sql.DB
 | 
			
		||||
	sshTunnelMachineId uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 执行查询语句
 | 
			
		||||
// 依次返回 列名数组,结果map,错误
 | 
			
		||||
func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interface{}, error) {
 | 
			
		||||
	execSql = strings.Trim(execSql, " ")
 | 
			
		||||
	isSelect := strings.HasPrefix(execSql, "SELECT") || strings.HasPrefix(execSql, "select")
 | 
			
		||||
	isShow := strings.HasPrefix(execSql, "show")
 | 
			
		||||
	isExplain := strings.HasPrefix(execSql, "explain")
 | 
			
		||||
 | 
			
		||||
	if !isSelect && !isShow && !isExplain {
 | 
			
		||||
		return nil, nil, errors.New("该sql非查询语句")
 | 
			
		||||
	}
 | 
			
		||||
	return SelectDataByDb(d.db, execSql)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 执行 update, insert, delete,建表等sql
 | 
			
		||||
// 返回影响条数和错误
 | 
			
		||||
func (d *DbInstance) Exec(sql string) (int64, error) {
 | 
			
		||||
@@ -328,22 +421,27 @@ func (d *DbInstance) Exec(sql string) (int64, error) {
 | 
			
		||||
 | 
			
		||||
// 关闭连接
 | 
			
		||||
func (d *DbInstance) Close() {
 | 
			
		||||
	d.db.Close()
 | 
			
		||||
	if d.db != nil {
 | 
			
		||||
		if err := d.db.Close(); err != nil {
 | 
			
		||||
			global.Log.Errorf("关闭数据库实例[%s]连接失败: %s", d.Id, err.Error())
 | 
			
		||||
		}
 | 
			
		||||
		d.db = nil
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取dataSourceName
 | 
			
		||||
func getDsn(d *entity.Db) string {
 | 
			
		||||
func getDsn(d *entity.Db, db string) string {
 | 
			
		||||
	var dsn string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
		dsn = fmt.Sprintf("%s:%s@%s(%s:%d)/%s?timeout=8s", d.Username, d.Password, d.Network, d.Host, d.Port, d.Database)
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		dsn = fmt.Sprintf("%s:%s@%s(%s:%d)/%s?timeout=8s", d.Username, d.Password, d.Network, d.Host, d.Port, db)
 | 
			
		||||
		if d.Params != "" {
 | 
			
		||||
			dsn = fmt.Sprintf("%s&%s", dsn, d.Params)
 | 
			
		||||
		}
 | 
			
		||||
		return dsn
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if d.Type == "postgres" {
 | 
			
		||||
		dsn = fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", d.Host, d.Port, d.Username, d.Password, d.Database)
 | 
			
		||||
	if d.Type == entity.DbTypePostgres {
 | 
			
		||||
		dsn = fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", d.Host, d.Port, d.Username, d.Password, db)
 | 
			
		||||
		if d.Params != "" {
 | 
			
		||||
			dsn = fmt.Sprintf("%s %s", dsn, strings.Join(strings.Split(d.Params, "&"), " "))
 | 
			
		||||
		}
 | 
			
		||||
@@ -438,7 +536,7 @@ const (
 | 
			
		||||
 | 
			
		||||
func (d *DbInstance) GetTableMetedatas() []map[string]interface{} {
 | 
			
		||||
	var sql string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		sql = MYSQL_TABLE_MA
 | 
			
		||||
	} else if d.Type == "postgres" {
 | 
			
		||||
		sql = PGSQL_TABLE_MA
 | 
			
		||||
@@ -458,10 +556,10 @@ func (d *DbInstance) GetColumnMetadatas(tableNames ...string) []map[string]inter
 | 
			
		||||
 | 
			
		||||
	var countSqlTmp string
 | 
			
		||||
	var sqlTmp string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		countSqlTmp = MYSQL_COLOUMN_MA_COUNT
 | 
			
		||||
		sqlTmp = MYSQL_COLUMN_MA
 | 
			
		||||
	} else if d.Type == "postgres" {
 | 
			
		||||
	} else if d.Type == entity.DbTypePostgres {
 | 
			
		||||
		countSqlTmp = PGSQL_COLUMN_MA_COUNT
 | 
			
		||||
		sqlTmp = PGSQL_COLUMN_MA
 | 
			
		||||
	}
 | 
			
		||||
@@ -493,9 +591,9 @@ func (d *DbInstance) GetPrimaryKey(tablename string) string {
 | 
			
		||||
 | 
			
		||||
func (d *DbInstance) GetTableInfos() []map[string]interface{} {
 | 
			
		||||
	var sql string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		sql = MYSQL_TABLE_INFO
 | 
			
		||||
	} else if d.Type == "postgres" {
 | 
			
		||||
	} else if d.Type == entity.DbTypePostgres {
 | 
			
		||||
		sql = PGSQL_TABLE_INFO
 | 
			
		||||
	}
 | 
			
		||||
	_, res, _ := d.SelectData(sql)
 | 
			
		||||
@@ -504,9 +602,9 @@ func (d *DbInstance) GetTableInfos() []map[string]interface{} {
 | 
			
		||||
 | 
			
		||||
func (d *DbInstance) GetTableIndex(tableName string) []map[string]interface{} {
 | 
			
		||||
	var sql string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		sql = fmt.Sprintf(MYSQL_INDEX_INFO, tableName)
 | 
			
		||||
	} else if d.Type == "postgres" {
 | 
			
		||||
	} else if d.Type == entity.DbTypePostgres {
 | 
			
		||||
		sql = fmt.Sprintf(PGSQL_INDEX_INFO, tableName)
 | 
			
		||||
	}
 | 
			
		||||
	_, res, _ := d.SelectData(sql)
 | 
			
		||||
@@ -515,7 +613,7 @@ func (d *DbInstance) GetTableIndex(tableName string) []map[string]interface{} {
 | 
			
		||||
 | 
			
		||||
func (d *DbInstance) GetCreateTableDdl(tableName string) []map[string]interface{} {
 | 
			
		||||
	var sql string
 | 
			
		||||
	if d.Type == "mysql" {
 | 
			
		||||
	if d.Type == entity.DbTypeMysql {
 | 
			
		||||
		sql = fmt.Sprintf("show create table %s ", tableName)
 | 
			
		||||
	}
 | 
			
		||||
	_, res, _ := d.SelectData(sql)
 | 
			
		||||
 
 | 
			
		||||
@@ -32,6 +32,9 @@ type Machine interface {
 | 
			
		||||
 | 
			
		||||
	// 获取机器连接
 | 
			
		||||
	GetCli(id uint64) *machine.Cli
 | 
			
		||||
 | 
			
		||||
	// 获取ssh隧道机器连接
 | 
			
		||||
	GetSshTunnelMachine(id uint64) *machine.SshTunnelMachine
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type machineAppImpl struct {
 | 
			
		||||
@@ -53,7 +56,7 @@ func (m *machineAppImpl) Count(condition *entity.Machine) int64 {
 | 
			
		||||
func (m *machineAppImpl) Save(me *entity.Machine) {
 | 
			
		||||
	// ’修改机器信息且密码不为空‘ or ‘新增’需要测试是否可连接
 | 
			
		||||
	if (me.Id != 0 && me.Password != "") || me.Id == 0 {
 | 
			
		||||
		biz.ErrIsNilAppendErr(machine.TestConn(me), "该机器无法连接: %s")
 | 
			
		||||
		biz.ErrIsNilAppendErr(machine.TestConn(*me, func(u uint64) *entity.Machine { return m.GetById(u) }), "该机器无法连接: %s")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	oldMachine := &entity.Machine{Ip: me.Ip, Port: me.Port, Username: me.Username}
 | 
			
		||||
@@ -126,3 +129,13 @@ func (m *machineAppImpl) GetCli(id uint64) *machine.Cli {
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "获取客户端错误: %s")
 | 
			
		||||
	return cli
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *machineAppImpl) GetSshTunnelMachine(id uint64) *machine.SshTunnelMachine {
 | 
			
		||||
	sshTunnel, err := machine.GetSshTunnelMachine(id, func(machineId uint64) *entity.Machine {
 | 
			
		||||
		machine := m.GetById(machineId)
 | 
			
		||||
		biz.IsTrue(machine.Status == entity.MachineStatusEnable, "该机器已被停用")
 | 
			
		||||
		return machine
 | 
			
		||||
	})
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "获取ssh隧道连接失败: %s")
 | 
			
		||||
	return sshTunnel
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -2,13 +2,17 @@ package application
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"mayfly-go/internal/constant"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/repository"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/persistence"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/cache"
 | 
			
		||||
	"mayfly-go/pkg/global"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"net"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"go.mongodb.org/mongo-driver/mongo"
 | 
			
		||||
@@ -80,36 +84,50 @@ func (d *mongoAppImpl) Save(m *entity.Mongo) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (d *mongoAppImpl) GetMongoCli(id uint64) *mongo.Client {
 | 
			
		||||
	cli, err := GetMongoCli(id, func(u uint64) string {
 | 
			
		||||
		mongo := d.GetById(id)
 | 
			
		||||
	mongoInstance, err := GetMongoInstance(id, func(u uint64) *entity.Mongo {
 | 
			
		||||
		mongo := d.GetById(u)
 | 
			
		||||
		biz.NotNil(mongo, "mongo信息不存在")
 | 
			
		||||
		return mongo.Uri
 | 
			
		||||
		return mongo
 | 
			
		||||
	})
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "连接mongo失败: %s")
 | 
			
		||||
	return cli
 | 
			
		||||
	return mongoInstance.Cli
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// -----------------------------------------------------------
 | 
			
		||||
 | 
			
		||||
//mongo客户端连接缓存,30分钟内没有访问则会被关闭
 | 
			
		||||
var mongoCliCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
			
		||||
//mongo客户端连接缓存,指定时间内没有访问则会被关闭
 | 
			
		||||
var mongoCliCache = cache.NewTimedCache(constant.MongoConnExpireTime, 5*time.Second).
 | 
			
		||||
	WithUpdateAccessTime(true).
 | 
			
		||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
			
		||||
		global.Log.Info("关闭mongo连接: id = ", key)
 | 
			
		||||
		value.(*mongo.Client).Disconnect(context.TODO())
 | 
			
		||||
		global.Log.Info("删除mongo连接缓存: id = ", key)
 | 
			
		||||
		value.(*MongoInstance).Close()
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
func GetMongoCli(mongoId uint64, getMongoUri func(uint64) string) (*mongo.Client, error) {
 | 
			
		||||
	cli, err := mongoCliCache.ComputeIfAbsent(mongoId, func(key interface{}) (interface{}, error) {
 | 
			
		||||
		c, err := connect(getMongoUri(mongoId))
 | 
			
		||||
func init() {
 | 
			
		||||
	machine.AddCheckSshTunnelMachineUseFunc(func(machineId uint64) bool {
 | 
			
		||||
		// 遍历所有mongo连接实例,若存在redis实例使用该ssh隧道机器,则返回true,表示还在使用中...
 | 
			
		||||
		items := mongoCliCache.Items()
 | 
			
		||||
		for _, v := range items {
 | 
			
		||||
			if v.Value.(*MongoInstance).sshTunnelMachineId == machineId {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return false
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取mongo的连接实例
 | 
			
		||||
func GetMongoInstance(mongoId uint64, getMongoEntity func(uint64) *entity.Mongo) (*MongoInstance, error) {
 | 
			
		||||
	mi, err := mongoCliCache.ComputeIfAbsent(mongoId, func(_ interface{}) (interface{}, error) {
 | 
			
		||||
		c, err := connect(getMongoEntity(mongoId))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return c, nil
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	if cli != nil {
 | 
			
		||||
		return cli.(*mongo.Client), err
 | 
			
		||||
	if mi != nil {
 | 
			
		||||
		return mi.(*MongoInstance), err
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
@@ -118,16 +136,61 @@ func DeleteMongoCache(mongoId uint64) {
 | 
			
		||||
	mongoCliCache.Delete(mongoId)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MongoInstance struct {
 | 
			
		||||
	Id                 uint64
 | 
			
		||||
	ProjectId          uint64
 | 
			
		||||
	Cli                *mongo.Client
 | 
			
		||||
	sshTunnelMachineId uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (mi *MongoInstance) Close() {
 | 
			
		||||
	if mi.Cli != nil {
 | 
			
		||||
		if err := mi.Cli.Disconnect(context.Background()); err != nil {
 | 
			
		||||
			global.Log.Errorf("关闭mongo实例[%d]连接失败: %s", mi.Id, err)
 | 
			
		||||
		}
 | 
			
		||||
		mi.Cli = nil
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 连接mongo,并返回client
 | 
			
		||||
func connect(uri string) (*mongo.Client, error) {
 | 
			
		||||
	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
 | 
			
		||||
func connect(me *entity.Mongo) (*MongoInstance, error) {
 | 
			
		||||
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 | 
			
		||||
	defer cancel()
 | 
			
		||||
	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri).SetMaxPoolSize(2))
 | 
			
		||||
 | 
			
		||||
	mongoInstance := &MongoInstance{Id: me.Id, ProjectId: me.ProjectId}
 | 
			
		||||
 | 
			
		||||
	mongoOptions := options.Client().ApplyURI(me.Uri).
 | 
			
		||||
		SetMaxPoolSize(1)
 | 
			
		||||
	// 启用ssh隧道则连接隧道机器
 | 
			
		||||
	if me.EnableSshTunnel == 1 {
 | 
			
		||||
		mongoInstance.sshTunnelMachineId = me.SshTunnelMachineId
 | 
			
		||||
		mongoOptions.SetDialer(&MongoSshDialer{machineId: me.SshTunnelMachineId})
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	client, err := mongo.Connect(ctx, mongoOptions)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		mongoInstance.Close()
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	if err = client.Ping(context.TODO(), nil); err != nil {
 | 
			
		||||
		mongoInstance.Close()
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	global.Log.Infof("连接mongo: %s", me.Uri)
 | 
			
		||||
	mongoInstance.Cli = client
 | 
			
		||||
	return mongoInstance, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type MongoSshDialer struct {
 | 
			
		||||
	machineId uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (sd *MongoSshDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
 | 
			
		||||
	if sshConn, err := MachineApp.GetSshTunnelMachine(sd.machineId).GetDialConn(network, address); err == nil {
 | 
			
		||||
		// 将ssh conn包装,否则内部部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
			
		||||
		return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
			
		||||
	} else {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return client, err
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -3,13 +3,17 @@ package application
 | 
			
		||||
import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/internal/constant"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/repository"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
			
		||||
	"mayfly-go/internal/devops/infrastructure/persistence"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/cache"
 | 
			
		||||
	"mayfly-go/pkg/global"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"net"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
@@ -65,7 +69,10 @@ func (r *redisAppImpl) GetRedisBy(condition *entity.Redis, cols ...string) error
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *redisAppImpl) Save(re *entity.Redis) {
 | 
			
		||||
	TestRedisConnection(re)
 | 
			
		||||
	// ’修改信息且密码不为空‘ or ‘新增’需要测试是否可连接
 | 
			
		||||
	if (re.Id != 0 && re.Password != "") || re.Id == 0 {
 | 
			
		||||
		TestRedisConnection(re)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 查找是否存在该库
 | 
			
		||||
	oldRedis := &entity.Redis{Host: re.Host, Db: re.Db}
 | 
			
		||||
@@ -106,25 +113,23 @@ func (r *redisAppImpl) GetRedisInstance(id uint64) *RedisInstance {
 | 
			
		||||
	biz.NotNil(re, "redis信息不存在")
 | 
			
		||||
 | 
			
		||||
	redisMode := re.Mode
 | 
			
		||||
	ri := &RedisInstance{Id: id, ProjectId: re.ProjectId, Mode: redisMode}
 | 
			
		||||
	var ri *RedisInstance
 | 
			
		||||
	if redisMode == "" || redisMode == entity.RedisModeStandalone {
 | 
			
		||||
		rcli := getRedisCient(re)
 | 
			
		||||
		ri = getRedisCient(re)
 | 
			
		||||
		// 测试连接
 | 
			
		||||
		_, e := rcli.Ping(context.Background()).Result()
 | 
			
		||||
		_, e := ri.Cli.Ping(context.Background()).Result()
 | 
			
		||||
		if e != nil {
 | 
			
		||||
			rcli.Close()
 | 
			
		||||
			ri.Close()
 | 
			
		||||
			panic(biz.NewBizErr(fmt.Sprintf("redis连接失败: %s", e.Error())))
 | 
			
		||||
		}
 | 
			
		||||
		ri.Cli = rcli
 | 
			
		||||
	} else if redisMode == entity.RedisModeCluster {
 | 
			
		||||
		ccli := getRedisClusterClient(re)
 | 
			
		||||
		ri = getRedisClusterClient(re)
 | 
			
		||||
		// 测试连接
 | 
			
		||||
		_, e := ccli.Ping(context.Background()).Result()
 | 
			
		||||
		_, e := ri.ClusterCli.Ping(context.Background()).Result()
 | 
			
		||||
		if e != nil {
 | 
			
		||||
			ccli.Close()
 | 
			
		||||
			ri.Close()
 | 
			
		||||
			panic(biz.NewBizErr(fmt.Sprintf("redis集群连接失败: %s", e.Error())))
 | 
			
		||||
		}
 | 
			
		||||
		ri.ClusterCli = ccli
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	global.Log.Infof("连接redis: %s", re.Host)
 | 
			
		||||
@@ -134,27 +139,57 @@ func (r *redisAppImpl) GetRedisInstance(id uint64) *RedisInstance {
 | 
			
		||||
	return ri
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getRedisCient(re *entity.Redis) *redis.Client {
 | 
			
		||||
	return redis.NewClient(&redis.Options{
 | 
			
		||||
		Addr:        re.Host,
 | 
			
		||||
		Password:    re.Password, // no password set
 | 
			
		||||
		DB:          re.Db,       // use default DB
 | 
			
		||||
		DialTimeout: 8 * time.Second,
 | 
			
		||||
	})
 | 
			
		||||
func getRedisCient(re *entity.Redis) *RedisInstance {
 | 
			
		||||
	ri := &RedisInstance{Id: re.Id, ProjectId: re.ProjectId, Mode: re.Mode}
 | 
			
		||||
 | 
			
		||||
	redisOptions := &redis.Options{
 | 
			
		||||
		Addr:         re.Host,
 | 
			
		||||
		Password:     re.Password, // no password set
 | 
			
		||||
		DB:           re.Db,       // use default DB
 | 
			
		||||
		DialTimeout:  8 * time.Second,
 | 
			
		||||
		ReadTimeout:  -1, // Disable timeouts, because SSH does not support deadlines.
 | 
			
		||||
		WriteTimeout: -1,
 | 
			
		||||
	}
 | 
			
		||||
	if re.EnableSshTunnel == 1 {
 | 
			
		||||
		ri.sshTunnelMachineId = re.SshTunnelMachineId
 | 
			
		||||
		redisOptions.Dialer = getRedisDialer(re.SshTunnelMachineId)
 | 
			
		||||
	}
 | 
			
		||||
	ri.Cli = redis.NewClient(redisOptions)
 | 
			
		||||
	return ri
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getRedisClusterClient(re *entity.Redis) *redis.ClusterClient {
 | 
			
		||||
	return redis.NewClusterClient(&redis.ClusterOptions{
 | 
			
		||||
func getRedisClusterClient(re *entity.Redis) *RedisInstance {
 | 
			
		||||
	ri := &RedisInstance{Id: re.Id, ProjectId: re.ProjectId, Mode: re.Mode}
 | 
			
		||||
 | 
			
		||||
	redisClusterOptions := &redis.ClusterOptions{
 | 
			
		||||
		Addrs:       strings.Split(re.Host, ","),
 | 
			
		||||
		Password:    re.Password,
 | 
			
		||||
		DialTimeout: 8 * time.Second,
 | 
			
		||||
	})
 | 
			
		||||
	}
 | 
			
		||||
	if re.EnableSshTunnel == 1 {
 | 
			
		||||
		ri.sshTunnelMachineId = re.SshTunnelMachineId
 | 
			
		||||
		redisClusterOptions.Dialer = getRedisDialer(re.SshTunnelMachineId)
 | 
			
		||||
	}
 | 
			
		||||
	ri.ClusterCli = redis.NewClusterClient(redisClusterOptions)
 | 
			
		||||
	return ri
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getRedisDialer(machineId uint64) func(ctx context.Context, network, addr string) (net.Conn, error) {
 | 
			
		||||
	sshTunnel := MachineApp.GetSshTunnelMachine(machineId)
 | 
			
		||||
	return func(_ context.Context, network, addr string) (net.Conn, error) {
 | 
			
		||||
		if sshConn, err := sshTunnel.GetDialConn(network, addr); err == nil {
 | 
			
		||||
			// 将ssh conn包装,否则redis内部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
			
		||||
			return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
			
		||||
		} else {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//------------------------------------------------------------------------------
 | 
			
		||||
 | 
			
		||||
// redis客户端连接缓存,30分钟内没有访问则会被关闭
 | 
			
		||||
var redisCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
			
		||||
// redis客户端连接缓存,指定时间内没有访问则会被关闭
 | 
			
		||||
var redisCache = cache.NewTimedCache(constant.RedisConnExpireTime, 5*time.Second).
 | 
			
		||||
	WithUpdateAccessTime(true).
 | 
			
		||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
			
		||||
		global.Log.Info(fmt.Sprintf("删除redis连接缓存 id = %d", key))
 | 
			
		||||
@@ -166,16 +201,29 @@ func CloseRedis(id uint64) {
 | 
			
		||||
	redisCache.Delete(id)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	machine.AddCheckSshTunnelMachineUseFunc(func(machineId uint64) bool {
 | 
			
		||||
		// 遍历所有redis连接实例,若存在redis实例使用该ssh隧道机器,则返回true,表示还在使用中...
 | 
			
		||||
		items := redisCache.Items()
 | 
			
		||||
		for _, v := range items {
 | 
			
		||||
			if v.Value.(*RedisInstance).sshTunnelMachineId == machineId {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return false
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestRedisConnection(re *entity.Redis) {
 | 
			
		||||
	var cmd redis.Cmdable
 | 
			
		||||
	if re.Mode == "" || re.Mode == entity.RedisModeStandalone {
 | 
			
		||||
		rcli := getRedisCient(re)
 | 
			
		||||
		defer rcli.Close()
 | 
			
		||||
		cmd = rcli
 | 
			
		||||
		cmd = rcli.Cli
 | 
			
		||||
	} else if re.Mode == entity.RedisModeCluster {
 | 
			
		||||
		ccli := getRedisClusterClient(re)
 | 
			
		||||
		defer ccli.Close()
 | 
			
		||||
		cmd = ccli
 | 
			
		||||
		cmd = ccli.ClusterCli
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// 测试连接
 | 
			
		||||
@@ -185,11 +233,12 @@ func TestRedisConnection(re *entity.Redis) {
 | 
			
		||||
 | 
			
		||||
// redis实例
 | 
			
		||||
type RedisInstance struct {
 | 
			
		||||
	Id         uint64
 | 
			
		||||
	ProjectId  uint64
 | 
			
		||||
	Mode       string
 | 
			
		||||
	Cli        *redis.Client
 | 
			
		||||
	ClusterCli *redis.ClusterClient
 | 
			
		||||
	Id                 uint64
 | 
			
		||||
	ProjectId          uint64
 | 
			
		||||
	Mode               string
 | 
			
		||||
	Cli                *redis.Client
 | 
			
		||||
	ClusterCli         *redis.ClusterClient
 | 
			
		||||
	sshTunnelMachineId uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取命令执行接口的具体实现
 | 
			
		||||
@@ -212,10 +261,15 @@ func (r *RedisInstance) Scan(cursor uint64, match string, count int64) ([]string
 | 
			
		||||
 | 
			
		||||
func (r *RedisInstance) Close() {
 | 
			
		||||
	if r.Mode == entity.RedisModeStandalone {
 | 
			
		||||
		r.Cli.Close()
 | 
			
		||||
		return
 | 
			
		||||
		if err := r.Cli.Close(); err != nil {
 | 
			
		||||
			global.Log.Errorf("关闭redis单机实例[%d]连接失败: %s", r.Id, err.Error())
 | 
			
		||||
		}
 | 
			
		||||
		r.Cli = nil
 | 
			
		||||
	}
 | 
			
		||||
	if r.Mode == entity.RedisModeCluster {
 | 
			
		||||
		r.ClusterCli.Close()
 | 
			
		||||
		if err := r.ClusterCli.Close(); err != nil {
 | 
			
		||||
			global.Log.Errorf("关闭redis集群实例[%d]连接失败: %s", r.Id, err.Error())
 | 
			
		||||
		}
 | 
			
		||||
		r.ClusterCli = nil
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,7 @@
 | 
			
		||||
package entity
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@@ -20,4 +21,25 @@ type Db struct {
 | 
			
		||||
	Project   string
 | 
			
		||||
	EnvId     uint64
 | 
			
		||||
	Env       string
 | 
			
		||||
 | 
			
		||||
	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取数据库连接网络, 若没有使用ssh隧道,则直接返回。否则返回拼接的网络需要注册至指定dial
 | 
			
		||||
func (d Db) GetNetwork() string {
 | 
			
		||||
	network := d.Network
 | 
			
		||||
	if d.EnableSshTunnel == -1 {
 | 
			
		||||
		if network == "" {
 | 
			
		||||
			return "tcp"
 | 
			
		||||
		} else {
 | 
			
		||||
			return network
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	return fmt.Sprintf("%s+ssh:%d", d.Type, d.SshTunnelMachineId)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	DbTypeMysql    = "mysql"
 | 
			
		||||
	DbTypePostgres = "postgres"
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -6,18 +6,23 @@ import (
 | 
			
		||||
 | 
			
		||||
type Machine struct {
 | 
			
		||||
	model.Model
 | 
			
		||||
	ProjectId   uint64 `json:"projectId"`
 | 
			
		||||
	ProjectName string `json:"projectName"`
 | 
			
		||||
	Name        string `json:"name"`
 | 
			
		||||
	Ip          string `json:"ip"`       // IP地址
 | 
			
		||||
	Username    string `json:"username"` // 用户名
 | 
			
		||||
	Password    string `json:"-"`
 | 
			
		||||
	Port        int    `json:"port"`   // 端口号
 | 
			
		||||
	Status      int8   `json:"status"` // 状态 1:启用;2:停用
 | 
			
		||||
	Remark      string `json:"remark"` // 备注
 | 
			
		||||
	ProjectId          uint64 `json:"projectId"`
 | 
			
		||||
	ProjectName        string `json:"projectName"`
 | 
			
		||||
	Name               string `json:"name"`
 | 
			
		||||
	Ip                 string `json:"ip"`         // IP地址
 | 
			
		||||
	Username           string `json:"username"`   // 用户名
 | 
			
		||||
	AuthMethod         int8   `json:"authMethod"` // 授权认证方式
 | 
			
		||||
	Password           string `json:"-"`
 | 
			
		||||
	Port               int    `json:"port"`               // 端口号
 | 
			
		||||
	Status             int8   `json:"status"`             // 状态 1:启用;2:停用
 | 
			
		||||
	Remark             string `json:"remark"`             // 备注
 | 
			
		||||
	EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	MachineStatusEnable  int8 = 1  // 启用状态
 | 
			
		||||
	MachineStatusDisable int8 = -1 // 禁用状态
 | 
			
		||||
	MachineStatusEnable        int8 = 1  // 启用状态
 | 
			
		||||
	MachineStatusDisable       int8 = -1 // 禁用状态
 | 
			
		||||
	MachineAuthMethodPassword  int8 = 1  // 密码登录
 | 
			
		||||
	MachineAuthMethodPublicKey int8 = 2  // 公钥免密登录
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -5,10 +5,12 @@ import "mayfly-go/pkg/model"
 | 
			
		||||
type Mongo struct {
 | 
			
		||||
	model.Model
 | 
			
		||||
 | 
			
		||||
	Name      string `orm:"column(name)" json:"name"`
 | 
			
		||||
	Uri       string `orm:"column(uri)" json:"uri"`
 | 
			
		||||
	ProjectId uint64 `json:"projectId"`
 | 
			
		||||
	Project   string `json:"project"`
 | 
			
		||||
	EnvId     uint64 `json:"envId"`
 | 
			
		||||
	Env       string `json:"env"`
 | 
			
		||||
	Name               string `orm:"column(name)" json:"name"`
 | 
			
		||||
	Uri                string `orm:"column(uri)" json:"uri"`
 | 
			
		||||
	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	ProjectId          uint64 `json:"projectId"`
 | 
			
		||||
	Project            string `json:"project"`
 | 
			
		||||
	EnvId              uint64 `json:"envId"`
 | 
			
		||||
	Env                string `json:"env"`
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -7,15 +7,17 @@ import (
 | 
			
		||||
type Redis struct {
 | 
			
		||||
	model.Model
 | 
			
		||||
 | 
			
		||||
	Host      string `orm:"column(host)" json:"host"`
 | 
			
		||||
	Mode      string `json:"mode"`
 | 
			
		||||
	Password  string `orm:"column(password)" json:"-"`
 | 
			
		||||
	Db        int    `orm:"column(database)" json:"db"`
 | 
			
		||||
	Remark    string
 | 
			
		||||
	ProjectId uint64
 | 
			
		||||
	Project   string
 | 
			
		||||
	EnvId     uint64
 | 
			
		||||
	Env       string
 | 
			
		||||
	Host               string `orm:"column(host)" json:"host"`
 | 
			
		||||
	Mode               string `json:"mode"`
 | 
			
		||||
	Password           string `orm:"column(password)" json:"-"`
 | 
			
		||||
	Db                 int    `orm:"column(database)" json:"db"`
 | 
			
		||||
	EnableSshTunnel    int8   `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
 | 
			
		||||
	SshTunnelMachineId uint64 `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
			
		||||
	Remark             string
 | 
			
		||||
	ProjectId          uint64
 | 
			
		||||
	Project            string
 | 
			
		||||
	EnvId              uint64
 | 
			
		||||
	Env                string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
 
 | 
			
		||||
@@ -3,6 +3,7 @@ package machine
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/internal/constant"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/cache"
 | 
			
		||||
@@ -18,62 +19,12 @@ import (
 | 
			
		||||
// 客户端信息
 | 
			
		||||
type Cli struct {
 | 
			
		||||
	machine *entity.Machine
 | 
			
		||||
	// ssh客户端
 | 
			
		||||
	client *ssh.Client
 | 
			
		||||
 | 
			
		||||
	sftpClient *sftp.Client
 | 
			
		||||
}
 | 
			
		||||
	client     *ssh.Client  // ssh客户端
 | 
			
		||||
	sftpClient *sftp.Client // sftp客户端
 | 
			
		||||
 | 
			
		||||
// 机器客户端连接缓存,45分钟内没有访问则会被关闭
 | 
			
		||||
var cliCache = cache.NewTimedCache(45*time.Minute, 5*time.Second).
 | 
			
		||||
	WithUpdateAccessTime(true).
 | 
			
		||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
			
		||||
		value.(*Cli).Close()
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
// 是否存在指定id的客户端连接
 | 
			
		||||
func HasCli(machineId uint64) bool {
 | 
			
		||||
	if _, ok := cliCache.Get(machineId); ok {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 删除指定机器客户端,并关闭客户端连接
 | 
			
		||||
func DeleteCli(id uint64) {
 | 
			
		||||
	cliCache.Delete(id)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 从缓存中获取客户端信息,不存在则回调获取机器信息函数,并新建
 | 
			
		||||
func GetCli(machineId uint64, getMachine func(uint64) *entity.Machine) (*Cli, error) {
 | 
			
		||||
	cli, err := cliCache.ComputeIfAbsent(machineId, func(key interface{}) (interface{}, error) {
 | 
			
		||||
		c, err := newClient(getMachine(machineId))
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		return c, nil
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	if cli != nil {
 | 
			
		||||
		return cli.(*Cli), err
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//根据机器信息创建客户端对象
 | 
			
		||||
func newClient(machine *entity.Machine) (*Cli, error) {
 | 
			
		||||
	if machine == nil {
 | 
			
		||||
		return nil, errors.New("机器不存在")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	global.Log.Infof("[%s]机器连接:%s:%d", machine.Name, machine.Ip, machine.Port)
 | 
			
		||||
	cli := new(Cli)
 | 
			
		||||
	cli.machine = machine
 | 
			
		||||
	err := cli.connect()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return cli, nil
 | 
			
		||||
	enableSshTunnel    int8
 | 
			
		||||
	sshTunnelMachineId uint64
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//连接
 | 
			
		||||
@@ -83,16 +34,7 @@ func (c *Cli) connect() error {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	m := c.machine
 | 
			
		||||
	config := ssh.ClientConfig{
 | 
			
		||||
		User: m.Username,
 | 
			
		||||
		Auth: []ssh.AuthMethod{ssh.Password(m.Password)},
 | 
			
		||||
		HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
 | 
			
		||||
			return nil
 | 
			
		||||
		},
 | 
			
		||||
		Timeout: 5 * time.Second,
 | 
			
		||||
	}
 | 
			
		||||
	addr := fmt.Sprintf("%s:%d", m.Ip, m.Port)
 | 
			
		||||
	sshClient, err := ssh.Dial("tcp", addr, &config)
 | 
			
		||||
	sshClient, err := GetSshClient(m)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
@@ -100,26 +42,7 @@ func (c *Cli) connect() error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 测试连接
 | 
			
		||||
func TestConn(m *entity.Machine) error {
 | 
			
		||||
	config := ssh.ClientConfig{
 | 
			
		||||
		User: m.Username,
 | 
			
		||||
		Auth: []ssh.AuthMethod{ssh.Password(m.Password)},
 | 
			
		||||
		HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
 | 
			
		||||
			return nil
 | 
			
		||||
		},
 | 
			
		||||
		Timeout: 5 * time.Second,
 | 
			
		||||
	}
 | 
			
		||||
	addr := fmt.Sprintf("%s:%d", m.Ip, m.Port)
 | 
			
		||||
	sshClient, err := ssh.Dial("tcp", addr, &config)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	defer sshClient.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 关闭client和并从缓存中移除
 | 
			
		||||
// 关闭client并从缓存中移除,如果使用隧道则也关闭
 | 
			
		||||
func (c *Cli) Close() {
 | 
			
		||||
	m := c.machine
 | 
			
		||||
	global.Log.Info(fmt.Sprintf("关闭机器客户端连接-> id: %d, name: %s, ip: %s", m.Id, m.Name, m.Ip))
 | 
			
		||||
@@ -131,6 +54,9 @@ func (c *Cli) Close() {
 | 
			
		||||
		c.sftpClient.Close()
 | 
			
		||||
		c.sftpClient = nil
 | 
			
		||||
	}
 | 
			
		||||
	if c.enableSshTunnel == 1 {
 | 
			
		||||
		CloseSshTunnelMachine(c.sshTunnelMachineId, c.machine.Id)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取sftp client
 | 
			
		||||
@@ -184,3 +110,144 @@ func (c *Cli) Run(shell string) (*string, error) {
 | 
			
		||||
func (c *Cli) GetMachine() *entity.Machine {
 | 
			
		||||
	return c.machine
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 机器客户端连接缓存,指定时间内没有访问则会被关闭
 | 
			
		||||
var cliCache = cache.NewTimedCache(constant.MachineConnExpireTime, 5*time.Second).
 | 
			
		||||
	WithUpdateAccessTime(true).
 | 
			
		||||
	OnEvicted(func(_, value interface{}) {
 | 
			
		||||
		value.(*Cli).Close()
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	AddCheckSshTunnelMachineUseFunc(func(machineId uint64) bool {
 | 
			
		||||
		// 遍历所有机器连接实例,若存在机器连接实例使用该ssh隧道机器,则返回true,表示还在使用中...
 | 
			
		||||
		items := cliCache.Items()
 | 
			
		||||
		for _, v := range items {
 | 
			
		||||
			if v.Value.(*Cli).sshTunnelMachineId == machineId {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return false
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 是否存在指定id的客户端连接
 | 
			
		||||
func HasCli(machineId uint64) bool {
 | 
			
		||||
	if _, ok := cliCache.Get(machineId); ok {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 删除指定机器客户端,并关闭客户端连接
 | 
			
		||||
func DeleteCli(id uint64) {
 | 
			
		||||
	cliCache.Delete(id)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 从缓存中获取客户端信息,不存在则回调获取机器信息函数,并新建
 | 
			
		||||
func GetCli(machineId uint64, getMachine func(uint64) *entity.Machine) (*Cli, error) {
 | 
			
		||||
	cli, err := cliCache.ComputeIfAbsent(machineId, func(_ interface{}) (interface{}, error) {
 | 
			
		||||
		me := getMachine(machineId)
 | 
			
		||||
		err := IfUseSshTunnelChangeIpPort(me, getMachine)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return nil, fmt.Errorf("ssh隧道连接失败: %s", err.Error())
 | 
			
		||||
		}
 | 
			
		||||
		c, err := newClient(me)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			CloseSshTunnelMachine(me.SshTunnelMachineId, me.Id)
 | 
			
		||||
			return nil, err
 | 
			
		||||
		}
 | 
			
		||||
		c.enableSshTunnel = me.EnableSshTunnel
 | 
			
		||||
		c.sshTunnelMachineId = me.SshTunnelMachineId
 | 
			
		||||
		return c, nil
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	if cli != nil {
 | 
			
		||||
		return cli.(*Cli), err
 | 
			
		||||
	}
 | 
			
		||||
	return nil, err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 测试连接,使用传值的方式,而非引用。因为如果使用了ssh隧道,则ip和端口会变为本地映射地址与端口
 | 
			
		||||
func TestConn(me entity.Machine, getSshTunnelMachine func(uint64) *entity.Machine) error {
 | 
			
		||||
	originId := me.Id
 | 
			
		||||
	if originId == 0 {
 | 
			
		||||
		// 随机设置一个ip,如果使用了隧道则用于临时保存隧道
 | 
			
		||||
		me.Id = uint64(time.Now().Nanosecond())
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	err := IfUseSshTunnelChangeIpPort(&me, getSshTunnelMachine)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "ssh隧道连接失败: %s")
 | 
			
		||||
	if me.EnableSshTunnel == 1 {
 | 
			
		||||
		defer CloseSshTunnelMachine(me.SshTunnelMachineId, me.Id)
 | 
			
		||||
	}
 | 
			
		||||
	sshClient, err := GetSshClient(&me)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	defer sshClient.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 如果使用了ssh隧道,则修改机器ip port为暴露的ip port
 | 
			
		||||
func IfUseSshTunnelChangeIpPort(me *entity.Machine, getMachine func(uint64) *entity.Machine) error {
 | 
			
		||||
	if me.EnableSshTunnel != 1 {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
	sshTunnelMachine, err := GetSshTunnelMachine(me.SshTunnelMachineId, func(u uint64) *entity.Machine {
 | 
			
		||||
		return getMachine(u)
 | 
			
		||||
	})
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	exposeIp, exposePort, err := sshTunnelMachine.OpenSshTunnel(me.Id, me.Ip, me.Port)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return err
 | 
			
		||||
	}
 | 
			
		||||
	// 修改机器ip地址
 | 
			
		||||
	me.Ip = exposeIp
 | 
			
		||||
	me.Port = exposePort
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetSshClient(m *entity.Machine) (*ssh.Client, error) {
 | 
			
		||||
	config := ssh.ClientConfig{
 | 
			
		||||
		User: m.Username,
 | 
			
		||||
		HostKeyCallback: func(hostname string, remote net.Addr, key ssh.PublicKey) error {
 | 
			
		||||
			return nil
 | 
			
		||||
		},
 | 
			
		||||
		Timeout: 5 * time.Second,
 | 
			
		||||
	}
 | 
			
		||||
	if m.AuthMethod == entity.MachineAuthMethodPassword {
 | 
			
		||||
		config.Auth = []ssh.AuthMethod{ssh.Password(m.Password)}
 | 
			
		||||
	} else if m.AuthMethod == entity.MachineAuthMethodPublicKey {
 | 
			
		||||
		if signer, err := ssh.ParsePrivateKey([]byte(m.Password)); err != nil {
 | 
			
		||||
			return nil, err
 | 
			
		||||
		} else {
 | 
			
		||||
			config.Auth = []ssh.AuthMethod{ssh.PublicKeys(signer)}
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	addr := fmt.Sprintf("%s:%d", m.Ip, m.Port)
 | 
			
		||||
	sshClient, err := ssh.Dial("tcp", addr, &config)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return sshClient, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
//根据机器信息创建客户端对象
 | 
			
		||||
func newClient(machine *entity.Machine) (*Cli, error) {
 | 
			
		||||
	if machine == nil {
 | 
			
		||||
		return nil, errors.New("机器不存在")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	global.Log.Infof("[%s]机器连接:%s:%d", machine.Name, machine.Ip, machine.Port)
 | 
			
		||||
	cli := new(Cli)
 | 
			
		||||
	cli.machine = machine
 | 
			
		||||
	err := cli.connect()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return cli, nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										240
									
								
								server/internal/devops/infrastructure/machine/sshtunnel.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										240
									
								
								server/internal/devops/infrastructure/machine/sshtunnel.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,240 @@
 | 
			
		||||
package machine
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/pkg/global"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"net"
 | 
			
		||||
	"os"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"golang.org/x/crypto/ssh"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	sshTunnelMachines map[uint64]*SshTunnelMachine = make(map[uint64]*SshTunnelMachine)
 | 
			
		||||
 | 
			
		||||
	mutex sync.Mutex
 | 
			
		||||
 | 
			
		||||
	// 所有检测ssh隧道机器是否被使用的函数
 | 
			
		||||
	checkSshTunnelMachineHasUseFuncs []CheckSshTunnelMachineHasUseFunc
 | 
			
		||||
 | 
			
		||||
	// 是否开启检查ssh隧道机器是否被使用,只有使用到了隧道机器才启用
 | 
			
		||||
	startCheckSshTunnelHasUse bool = false
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// 检查ssh隧道机器是否有被使用
 | 
			
		||||
type CheckSshTunnelMachineHasUseFunc func(uint64) bool
 | 
			
		||||
 | 
			
		||||
func startCheckUse() {
 | 
			
		||||
	global.Log.Info("开启定时检测ssh隧道机器是否还有被使用")
 | 
			
		||||
	heartbeat := time.Duration(10) * time.Minute
 | 
			
		||||
	tick := time.NewTicker(heartbeat)
 | 
			
		||||
	go func() {
 | 
			
		||||
		for range tick.C {
 | 
			
		||||
			func() {
 | 
			
		||||
				if !mutex.TryLock() {
 | 
			
		||||
					return
 | 
			
		||||
				}
 | 
			
		||||
				defer mutex.Unlock()
 | 
			
		||||
				// 遍历隧道机器,都未被使用将会被关闭
 | 
			
		||||
				for mid, sshTunnelMachine := range sshTunnelMachines {
 | 
			
		||||
					global.Log.Debugf("开始定时检查ssh隧道机器[%d]是否还有被使用...", mid)
 | 
			
		||||
					for _, checkUseFunc := range checkSshTunnelMachineHasUseFuncs {
 | 
			
		||||
						// 如果一个在使用则返回不关闭,不继续后续检查
 | 
			
		||||
						if checkUseFunc(mid) {
 | 
			
		||||
							return
 | 
			
		||||
						}
 | 
			
		||||
					}
 | 
			
		||||
					// 都未被使用,则关闭
 | 
			
		||||
					sshTunnelMachine.Close()
 | 
			
		||||
				}
 | 
			
		||||
			}()
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 添加ssh隧道机器检测是否使用函数
 | 
			
		||||
func AddCheckSshTunnelMachineUseFunc(checkFunc CheckSshTunnelMachineHasUseFunc) {
 | 
			
		||||
	if checkSshTunnelMachineHasUseFuncs == nil {
 | 
			
		||||
		checkSshTunnelMachineHasUseFuncs = make([]CheckSshTunnelMachineHasUseFunc, 0)
 | 
			
		||||
	}
 | 
			
		||||
	checkSshTunnelMachineHasUseFuncs = append(checkSshTunnelMachineHasUseFuncs, checkFunc)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ssh隧道机器
 | 
			
		||||
type SshTunnelMachine struct {
 | 
			
		||||
	machineId uint64 // 隧道机器id
 | 
			
		||||
	SshClient *ssh.Client
 | 
			
		||||
	mutex     sync.Mutex
 | 
			
		||||
	tunnels   map[uint64]*Tunnel // 机器id -> 隧道
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (stm *SshTunnelMachine) OpenSshTunnel(id uint64, ip string, port int) (exposedIp string, exposedPort int, err error) {
 | 
			
		||||
	stm.mutex.Lock()
 | 
			
		||||
	defer stm.mutex.Unlock()
 | 
			
		||||
 | 
			
		||||
	localPort, err := utils.GetAvailablePort()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", 0, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	hostname, err := os.Hostname()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", 0, err
 | 
			
		||||
	}
 | 
			
		||||
	// debug
 | 
			
		||||
	//hostname = "0.0.0.0"
 | 
			
		||||
 | 
			
		||||
	localAddr := fmt.Sprintf("%s:%d", hostname, localPort)
 | 
			
		||||
	listener, err := net.Listen("tcp", localAddr)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", 0, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	tunnel := &Tunnel{
 | 
			
		||||
		id:         id,
 | 
			
		||||
		machineId:  stm.machineId,
 | 
			
		||||
		localHost:  hostname,
 | 
			
		||||
		localPort:  localPort,
 | 
			
		||||
		remoteHost: ip,
 | 
			
		||||
		remotePort: port,
 | 
			
		||||
		listener:   listener,
 | 
			
		||||
	}
 | 
			
		||||
	go tunnel.Open(stm.SshClient)
 | 
			
		||||
	stm.tunnels[tunnel.id] = tunnel
 | 
			
		||||
 | 
			
		||||
	return tunnel.localHost, tunnel.localPort, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (st *SshTunnelMachine) GetDialConn(network string, addr string) (net.Conn, error) {
 | 
			
		||||
	st.mutex.Lock()
 | 
			
		||||
	defer st.mutex.Unlock()
 | 
			
		||||
	return st.SshClient.Dial(network, addr)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (stm *SshTunnelMachine) Close() {
 | 
			
		||||
	stm.mutex.Lock()
 | 
			
		||||
	defer stm.mutex.Unlock()
 | 
			
		||||
 | 
			
		||||
	for id, tunnel := range stm.tunnels {
 | 
			
		||||
		if tunnel != nil {
 | 
			
		||||
			tunnel.Close()
 | 
			
		||||
			delete(stm.tunnels, id)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if stm.SshClient != nil {
 | 
			
		||||
		global.Log.Infof("ssh隧道机器[%d]未被使用, 关闭隧道...", stm.machineId)
 | 
			
		||||
		stm.SshClient.Close()
 | 
			
		||||
	}
 | 
			
		||||
	delete(sshTunnelMachines, stm.machineId)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取ssh隧道机器,方便统一管理充当ssh隧道的机器,避免创建多个ssh client
 | 
			
		||||
func GetSshTunnelMachine(machineId uint64, getMachine func(uint64) *entity.Machine) (*SshTunnelMachine, error) {
 | 
			
		||||
	sshTunnelMachine := sshTunnelMachines[machineId]
 | 
			
		||||
	if sshTunnelMachine != nil {
 | 
			
		||||
		return sshTunnelMachine, nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	mutex.Lock()
 | 
			
		||||
	defer mutex.Unlock()
 | 
			
		||||
 | 
			
		||||
	me := getMachine(machineId)
 | 
			
		||||
	sshClient, err := GetSshClient(me)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	sshTunnelMachine = &SshTunnelMachine{SshClient: sshClient, machineId: machineId, tunnels: map[uint64]*Tunnel{}}
 | 
			
		||||
 | 
			
		||||
	global.Log.Infof("初次连接ssh隧道机器[%d][%s:%d]", machineId, me.Ip, me.Port)
 | 
			
		||||
	sshTunnelMachines[machineId] = sshTunnelMachine
 | 
			
		||||
 | 
			
		||||
	// 如果实用了隧道机器且还没开始定时检查是否还被实用,则执行定时任务检测隧道是否还被使用
 | 
			
		||||
	if !startCheckSshTunnelHasUse {
 | 
			
		||||
		startCheckUse()
 | 
			
		||||
		startCheckSshTunnelHasUse = true
 | 
			
		||||
	}
 | 
			
		||||
	return sshTunnelMachine, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 关闭ssh隧道机器的指定隧道
 | 
			
		||||
func CloseSshTunnelMachine(machineId uint64, tunnelId uint64) {
 | 
			
		||||
	sshTunnelMachine := sshTunnelMachines[machineId]
 | 
			
		||||
	if sshTunnelMachine == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	sshTunnelMachine.mutex.Lock()
 | 
			
		||||
	defer sshTunnelMachine.mutex.Unlock()
 | 
			
		||||
	t := sshTunnelMachine.tunnels[tunnelId]
 | 
			
		||||
	if t != nil {
 | 
			
		||||
		t.Close()
 | 
			
		||||
		delete(sshTunnelMachine.tunnels, tunnelId)
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Tunnel struct {
 | 
			
		||||
	id                uint64 // 唯一标识
 | 
			
		||||
	machineId         uint64 // 隧道机器id
 | 
			
		||||
	localHost         string // 本地监听地址
 | 
			
		||||
	localPort         int    // 本地端口
 | 
			
		||||
	remoteHost        string // 远程连接地址
 | 
			
		||||
	remotePort        int    // 远程端口
 | 
			
		||||
	listener          net.Listener
 | 
			
		||||
	localConnections  []net.Conn
 | 
			
		||||
	remoteConnections []net.Conn
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *Tunnel) Open(sshClient *ssh.Client) {
 | 
			
		||||
	localAddr := fmt.Sprintf("%s:%d", r.localHost, r.localPort)
 | 
			
		||||
 | 
			
		||||
	for {
 | 
			
		||||
		global.Log.Debugf("隧道 %v 等待客户端访问 %v", r.id, localAddr)
 | 
			
		||||
		localConn, err := r.listener.Accept()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			global.Log.Debugf("隧道 %v 接受连接失败 %v, 退出循环", r.id, err.Error())
 | 
			
		||||
			global.Log.Debug("-------------------------------------------------")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		r.localConnections = append(r.localConnections, localConn)
 | 
			
		||||
 | 
			
		||||
		global.Log.Debugf("隧道 %v 新增本地连接 %v", r.id, localConn.RemoteAddr().String())
 | 
			
		||||
		remoteAddr := fmt.Sprintf("%s:%d", r.remoteHost, r.remotePort)
 | 
			
		||||
		global.Log.Debugf("隧道 %v 连接远程地址 %v ...", r.id, remoteAddr)
 | 
			
		||||
		remoteConn, err := sshClient.Dial("tcp", remoteAddr)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			global.Log.Debugf("隧道 %v 连接远程地址 %v, 退出循环", r.id, err.Error())
 | 
			
		||||
			global.Log.Debug("-------------------------------------------------")
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		r.remoteConnections = append(r.remoteConnections, remoteConn)
 | 
			
		||||
 | 
			
		||||
		global.Log.Debugf("隧道 %v 连接远程主机成功", r.id)
 | 
			
		||||
		go copyConn(localConn, remoteConn)
 | 
			
		||||
		go copyConn(remoteConn, localConn)
 | 
			
		||||
		global.Log.Debugf("隧道 %v 开始转发数据 [%v]->[%v]", r.id, localAddr, remoteAddr)
 | 
			
		||||
		global.Log.Debug("~~~~~~~~~~~~~~~~~~~~分割线~~~~~~~~~~~~~~~~~~~~~~~~")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *Tunnel) Close() {
 | 
			
		||||
	for i := range r.localConnections {
 | 
			
		||||
		_ = r.localConnections[i].Close()
 | 
			
		||||
	}
 | 
			
		||||
	r.localConnections = nil
 | 
			
		||||
	for i := range r.remoteConnections {
 | 
			
		||||
		_ = r.remoteConnections[i].Close()
 | 
			
		||||
	}
 | 
			
		||||
	r.remoteConnections = nil
 | 
			
		||||
	_ = r.listener.Close()
 | 
			
		||||
	global.Log.Debugf("隧道 %d 监听器关闭", r.id)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func copyConn(writer, reader net.Conn) {
 | 
			
		||||
	_, _ = io.Copy(writer, reader)
 | 
			
		||||
}
 | 
			
		||||
@@ -4,6 +4,7 @@ import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/devops/domain/repository"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@@ -51,9 +52,9 @@ func (m *machineRepo) GetById(id uint64, cols ...string) *entity.Machine {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *machineRepo) Create(entity *entity.Machine) {
 | 
			
		||||
	model.Insert(entity)
 | 
			
		||||
	biz.ErrIsNilAppendErr(model.Insert(entity), "创建机器信息失败: %s")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *machineRepo) UpdateById(entity *entity.Machine) {
 | 
			
		||||
	model.UpdateById(entity)
 | 
			
		||||
	biz.ErrIsNilAppendErr(model.UpdateById(entity), "更新机器信息失败: %s")
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -24,14 +24,19 @@ func InitDbRouter(router *gin.RouterGroup) {
 | 
			
		||||
			rc.Handle(d.Dbs)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveDb := ctx.NewLogInfo("保存数据库信息")
 | 
			
		||||
		saveDb := ctx.NewLogInfo("保存数据库信息").WithSave(true)
 | 
			
		||||
		db.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(saveDb).
 | 
			
		||||
				Handle(d.Save)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		deleteDb := ctx.NewLogInfo("删除数据库信息")
 | 
			
		||||
		db.POST("databases", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				Handle(d.GetDatabaseNames)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		deleteDb := ctx.NewLogInfo("删除数据库信息").WithSave(true)
 | 
			
		||||
		db.DELETE(":dbId", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(deleteDb).
 | 
			
		||||
@@ -50,20 +55,23 @@ func InitDbRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(d.GetCreateTableDdl)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// db.GET(":dbId/exec-sql", controllers.SelectData)
 | 
			
		||||
		execSqlLog := ctx.NewLogInfo("执行Sql语句")
 | 
			
		||||
		db.POST(":dbId/exec-sql", func(g *gin.Context) {
 | 
			
		||||
			rc := ctx.NewReqCtxWithGin(g).WithLog(ctx.NewLogInfo("执行Sql语句"))
 | 
			
		||||
			rc := ctx.NewReqCtxWithGin(g).WithLog(execSqlLog)
 | 
			
		||||
			rc.Handle(d.ExecSql)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		execSqlFileLog := ctx.NewLogInfo("执行Sql文件").WithSave(true)
 | 
			
		||||
		db.POST(":dbId/exec-sql-file", func(g *gin.Context) {
 | 
			
		||||
			rc := ctx.NewReqCtxWithGin(g).WithLog(ctx.NewLogInfo("执行Sql文件"))
 | 
			
		||||
			rc.Handle(d.ExecSqlFile)
 | 
			
		||||
			ctx.NewReqCtxWithGin(g).
 | 
			
		||||
				WithLog(execSqlFileLog).
 | 
			
		||||
				Handle(d.ExecSqlFile)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		dumpLog := ctx.NewLogInfo("导出sql文件").WithSave(true)
 | 
			
		||||
		db.GET(":dbId/dump", func(g *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(g).
 | 
			
		||||
				WithLog(ctx.NewLogInfo("Sql文件dump")).
 | 
			
		||||
				WithLog(dumpLog).
 | 
			
		||||
				Handle(d.DumpSql)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -29,7 +29,7 @@ func InitMachineRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 终止进程
 | 
			
		||||
		killProcessL := ctx.NewLogInfo("终止进程")
 | 
			
		||||
		killProcessL := ctx.NewLogInfo("终止进程").WithSave(true)
 | 
			
		||||
		killProcessP := ctx.NewPermission("machine:killprocess")
 | 
			
		||||
		machines.DELETE(":machineId/process", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
@@ -38,28 +38,28 @@ func InitMachineRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(m.KillProcess)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveMachine := ctx.NewLogInfo("保存机器信息")
 | 
			
		||||
		saveMachine := ctx.NewLogInfo("保存机器信息").WithSave(true)
 | 
			
		||||
		machines.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(saveMachine).
 | 
			
		||||
				Handle(m.SaveMachine)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("调整机器状态")
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("调整机器状态").WithSave(true)
 | 
			
		||||
		machines.PUT(":machineId/:status", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(changeStatus).
 | 
			
		||||
				Handle(m.ChangeStatus)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delMachine := ctx.NewLogInfo("删除机器")
 | 
			
		||||
		delMachine := ctx.NewLogInfo("删除机器").WithSave(true)
 | 
			
		||||
		machines.DELETE(":machineId", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(delMachine).
 | 
			
		||||
				Handle(m.DeleteMachine)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		closeCli := ctx.NewLogInfo("关闭机器客户端")
 | 
			
		||||
		closeCli := ctx.NewLogInfo("关闭机器客户端").WithSave(true)
 | 
			
		||||
		machines.DELETE(":machineId/close-cli", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(closeCli).Handle(m.CloseCli)
 | 
			
		||||
		})
 | 
			
		||||
 
 | 
			
		||||
@@ -24,7 +24,7 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 新增修改机器文件
 | 
			
		||||
		addFileConf := ctx.NewLogInfo("新增机器文件配置")
 | 
			
		||||
		addFileConf := ctx.NewLogInfo("新增机器文件配置").WithSave(true)
 | 
			
		||||
		afcP := ctx.NewPermission("machine:file:add")
 | 
			
		||||
		machineFile.POST(":machineId/files", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(addFileConf).
 | 
			
		||||
@@ -33,7 +33,7 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 删除机器文件
 | 
			
		||||
		delFileConf := ctx.NewLogInfo("删除机器文件配置")
 | 
			
		||||
		delFileConf := ctx.NewLogInfo("删除机器文件配置").WithSave(true)
 | 
			
		||||
		dfcP := ctx.NewPermission("machine:file:del")
 | 
			
		||||
		machineFile.DELETE(":machineId/files/:fileId", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(delFileConf).
 | 
			
		||||
@@ -41,7 +41,7 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(mf.DeleteFile)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		getContent := ctx.NewLogInfo("读取机器文件内容")
 | 
			
		||||
		getContent := ctx.NewLogInfo("读取机器文件内容").WithSave(true)
 | 
			
		||||
		machineFile.GET(":machineId/files/:fileId/read", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(getContent).
 | 
			
		||||
				Handle(mf.ReadFileContent)
 | 
			
		||||
@@ -53,7 +53,7 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(mf.GetDirEntry)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		writeFile := ctx.NewLogInfo("写入or下载文件内容")
 | 
			
		||||
		writeFile := ctx.NewLogInfo("写入or下载文件内容").WithSave(true)
 | 
			
		||||
		wfP := ctx.NewPermission("machine:file:write")
 | 
			
		||||
		machineFile.POST(":machineId/files/:fileId/write", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(writeFile).
 | 
			
		||||
@@ -61,14 +61,14 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(mf.WriteFileContent)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		createFile := ctx.NewLogInfo("创建机器文件or目录")
 | 
			
		||||
		createFile := ctx.NewLogInfo("创建机器文件or目录").WithSave(true)
 | 
			
		||||
		machineFile.POST(":machineId/files/:fileId/create-file", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(createFile).
 | 
			
		||||
				WithRequiredPermission(wfP).
 | 
			
		||||
				Handle(mf.CreateFile)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		uploadFile := ctx.NewLogInfo("文件上传")
 | 
			
		||||
		uploadFile := ctx.NewLogInfo("文件上传").WithSave(true)
 | 
			
		||||
		ufP := ctx.NewPermission("machine:file:upload")
 | 
			
		||||
		machineFile.POST(":machineId/files/:fileId/upload", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(uploadFile).
 | 
			
		||||
@@ -76,7 +76,7 @@ func InitMachineFileRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(mf.UploadFile)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		removeFile := ctx.NewLogInfo("删除文件or文件夹")
 | 
			
		||||
		removeFile := ctx.NewLogInfo("删除文件or文件夹").WithSave(true)
 | 
			
		||||
		rfP := ctx.NewPermission("machine:file:rm")
 | 
			
		||||
		machineFile.DELETE(":machineId/files/:fileId/remove", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(removeFile).
 | 
			
		||||
 
 | 
			
		||||
@@ -22,7 +22,7 @@ func InitMachineScriptRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(ms.MachineScripts)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveMachienScriptLog := ctx.NewLogInfo("保存脚本")
 | 
			
		||||
		saveMachienScriptLog := ctx.NewLogInfo("保存脚本").WithSave(true)
 | 
			
		||||
		smsP := ctx.NewPermission("machine:script:save")
 | 
			
		||||
		// 保存脚本
 | 
			
		||||
		machines.POST(":machineId/scripts", func(c *gin.Context) {
 | 
			
		||||
@@ -31,7 +31,7 @@ func InitMachineScriptRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(ms.SaveMachineScript)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		deleteLog := ctx.NewLogInfo("删除脚本")
 | 
			
		||||
		deleteLog := ctx.NewLogInfo("删除脚本").WithSave(true)
 | 
			
		||||
		dP := ctx.NewPermission("machine:script:del")
 | 
			
		||||
		// 保存脚本
 | 
			
		||||
		machines.DELETE(":machineId/scripts/:scriptId", func(c *gin.Context) {
 | 
			
		||||
@@ -40,7 +40,7 @@ func InitMachineScriptRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(ms.DeleteMachineScript)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		runLog := ctx.NewLogInfo("执行机器脚本")
 | 
			
		||||
		runLog := ctx.NewLogInfo("执行机器脚本").WithSave(true)
 | 
			
		||||
		rP := ctx.NewPermission("machine:script:run")
 | 
			
		||||
		// 运行脚本
 | 
			
		||||
		machines.GET(":machineId/scripts/:scriptId/run", func(c *gin.Context) {
 | 
			
		||||
 
 | 
			
		||||
@@ -25,7 +25,7 @@ func InitProjectRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(m.GetProjects)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveProjectLog := ctx.NewLogInfo("保存项目信息")
 | 
			
		||||
		saveProjectLog := ctx.NewLogInfo("保存项目信息").WithSave(true)
 | 
			
		||||
		savePP := ctx.NewPermission("project:save")
 | 
			
		||||
		// 保存项目下的环境信息
 | 
			
		||||
		project.POST("", func(c *gin.Context) {
 | 
			
		||||
@@ -34,7 +34,7 @@ func InitProjectRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(m.SaveProject)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delProjectLog := ctx.NewLogInfo("删除项目信息")
 | 
			
		||||
		delProjectLog := ctx.NewLogInfo("删除项目信息").WithSave(true)
 | 
			
		||||
		delPP := ctx.NewPermission("project:del")
 | 
			
		||||
		// 删除项目
 | 
			
		||||
		project.DELETE("", func(c *gin.Context) {
 | 
			
		||||
@@ -48,7 +48,7 @@ func InitProjectRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(m.GetProjectEnvs)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveProjectEnvLog := ctx.NewLogInfo("新增项目环境信息")
 | 
			
		||||
		saveProjectEnvLog := ctx.NewLogInfo("新增项目环境信息").WithSave(true)
 | 
			
		||||
		savePeP := ctx.NewPermission("project:env:add")
 | 
			
		||||
		// 保存项目下的环境信息
 | 
			
		||||
		project.POST("/:projectId/envs", func(c *gin.Context) {
 | 
			
		||||
@@ -63,7 +63,7 @@ func InitProjectRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 保存项目成员
 | 
			
		||||
		saveProjectMemLog := ctx.NewLogInfo("新增项目成员")
 | 
			
		||||
		saveProjectMemLog := ctx.NewLogInfo("新增项目成员").WithSave(true)
 | 
			
		||||
		savePmP := ctx.NewPermission("project:member:add")
 | 
			
		||||
		project.POST("/:projectId/members", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(saveProjectMemLog).
 | 
			
		||||
@@ -72,7 +72,7 @@ func InitProjectRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 删除项目成员
 | 
			
		||||
		delProjectMemLog := ctx.NewLogInfo("删除项目成员")
 | 
			
		||||
		delProjectMemLog := ctx.NewLogInfo("删除项目成员").WithSave(true)
 | 
			
		||||
		savePmdP := ctx.NewPermission("project:member:del")
 | 
			
		||||
		project.DELETE("/:projectId/members/:accountId", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(delProjectMemLog).
 | 
			
		||||
 
 | 
			
		||||
@@ -21,12 +21,12 @@ func InitRedisRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(rs.RedisList)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		save := ctx.NewLogInfo("保存redis信息")
 | 
			
		||||
		save := ctx.NewLogInfo("保存redis信息").WithSave(true)
 | 
			
		||||
		redis.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(save).Handle(rs.Save)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delRedis := ctx.NewLogInfo("删除redis信息")
 | 
			
		||||
		delRedis := ctx.NewLogInfo("删除redis信息").WithSave(true)
 | 
			
		||||
		redis.DELETE(":id", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(delRedis).Handle(rs.DeleteRedis)
 | 
			
		||||
		})
 | 
			
		||||
@@ -45,8 +45,7 @@ func InitRedisRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 删除key
 | 
			
		||||
		deleteKeyL := ctx.NewLogInfo("redis删除key")
 | 
			
		||||
		// deleteKey := ctx.NewPermission("project:save")
 | 
			
		||||
		deleteKeyL := ctx.NewLogInfo("redis删除key").WithSave(true)
 | 
			
		||||
		redis.DELETE(":id/key", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(deleteKeyL).Handle(rs.DeleteKey)
 | 
			
		||||
		})
 | 
			
		||||
 
 | 
			
		||||
@@ -10,7 +10,9 @@ import (
 | 
			
		||||
	"mayfly-go/pkg/captcha"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
	"mayfly-go/pkg/ginx"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"regexp"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"time"
 | 
			
		||||
@@ -29,15 +31,20 @@ type Account struct {
 | 
			
		||||
func (a *Account) Login(rc *ctx.ReqCtx) {
 | 
			
		||||
	loginForm := &form.LoginForm{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, loginForm)
 | 
			
		||||
	rc.ReqParam = loginForm.Username
 | 
			
		||||
 | 
			
		||||
	// 校验验证码
 | 
			
		||||
	biz.IsTrue(captcha.Verify(loginForm.Cid, loginForm.Captcha), "验证码错误")
 | 
			
		||||
 | 
			
		||||
	account := &entity.Account{Username: loginForm.Username, Password: utils.Md5(loginForm.Password)}
 | 
			
		||||
	originPwd, err := utils.DefaultRsaDecrypt(loginForm.Password, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
			
		||||
 | 
			
		||||
	account := &entity.Account{Username: loginForm.Username, Password: utils.Md5(originPwd)}
 | 
			
		||||
	biz.ErrIsNil(a.AccountApp.GetAccount(account, "Id", "Username", "Status", "LastLoginTime", "LastLoginIp"), "用户名或密码错误")
 | 
			
		||||
	biz.IsTrue(account.IsEnable(), "该账号不可用")
 | 
			
		||||
 | 
			
		||||
	// 校验密码强度是否符合
 | 
			
		||||
	biz.IsTrueBy(CheckPasswordLever(originPwd), biz.NewBizErrCode(401, "您的密码安全等级较低,请修改后重新登录"))
 | 
			
		||||
 | 
			
		||||
	var resources vo.AccountResourceVOList
 | 
			
		||||
	// 获取账号菜单资源
 | 
			
		||||
	a.ResourceApp.GetAccountResources(account.Id, &resources)
 | 
			
		||||
@@ -54,8 +61,13 @@ func (a *Account) Login(rc *ctx.ReqCtx) {
 | 
			
		||||
	// 保存该账号的权限codes
 | 
			
		||||
	ctx.SavePermissionCodes(account.Id, permissions)
 | 
			
		||||
 | 
			
		||||
	clientIp := rc.GinCtx.ClientIP()
 | 
			
		||||
	// 保存登录消息
 | 
			
		||||
	go a.saveLogin(account, rc.GinCtx.ClientIP())
 | 
			
		||||
	go a.saveLogin(account, clientIp)
 | 
			
		||||
 | 
			
		||||
	rc.ReqParam = fmt.Sprintln("登录ip: ", clientIp)
 | 
			
		||||
	// 赋值loginAccount 主要用于记录操作日志,因为操作日志保存请求上下文没有该信息不保存日志
 | 
			
		||||
	rc.LoginAccount = &model.LoginAccount{Id: account.Id, Username: account.Username}
 | 
			
		||||
 | 
			
		||||
	rc.ResData = map[string]interface{}{
 | 
			
		||||
		"token":         ctx.CreateToken(account.Id, account.Username),
 | 
			
		||||
@@ -67,6 +79,48 @@ func (a *Account) Login(rc *ctx.ReqCtx) {
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Account) ChangePassword(rc *ctx.ReqCtx) {
 | 
			
		||||
	form := new(form.AccountChangePasswordForm)
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
			
		||||
 | 
			
		||||
	originOldPwd, err := utils.DefaultRsaDecrypt(form.OldPassword, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密旧密码错误: %s")
 | 
			
		||||
 | 
			
		||||
	account := &entity.Account{Username: form.Username, Password: utils.Md5(originOldPwd)}
 | 
			
		||||
	biz.ErrIsNil(a.AccountApp.GetAccount(account, "Id", "Username", "Status"), "旧密码不正确")
 | 
			
		||||
 | 
			
		||||
	originNewPwd, err := utils.DefaultRsaDecrypt(form.NewPassword, true)
 | 
			
		||||
	biz.ErrIsNilAppendErr(err, "解密新密码错误: %s")
 | 
			
		||||
	biz.IsTrue(CheckPasswordLever(originNewPwd), "密码强度必须8位以上且包含字⺟⼤⼩写+数字+特殊符号")
 | 
			
		||||
 | 
			
		||||
	updateAccount := new(entity.Account)
 | 
			
		||||
	updateAccount.Id = account.Id
 | 
			
		||||
	updateAccount.Password = utils.Md5(originNewPwd)
 | 
			
		||||
	a.AccountApp.Update(updateAccount)
 | 
			
		||||
 | 
			
		||||
	// 赋值loginAccount 主要用于记录操作日志,因为操作日志保存请求上下文没有该信息不保存日志
 | 
			
		||||
	rc.LoginAccount = &model.LoginAccount{Id: account.Id, Username: account.Username}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func CheckPasswordLever(ps string) bool {
 | 
			
		||||
	if len(ps) < 8 {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	num := `[0-9]{1}`
 | 
			
		||||
	a_z := `[a-zA-Z]{1}`
 | 
			
		||||
	symbol := `[!@#~$%^&*()+|_.,]{1}`
 | 
			
		||||
	if b, err := regexp.MatchString(num, ps); !b || err != nil {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	if b, err := regexp.MatchString(a_z, ps); !b || err != nil {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	if b, err := regexp.MatchString(symbol, ps); !b || err != nil {
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 保存更新账号登录信息
 | 
			
		||||
func (a *Account) saveLogin(account *entity.Account, ip string) {
 | 
			
		||||
	// 更新账号最后登录时间
 | 
			
		||||
@@ -100,7 +154,7 @@ func (a *Account) saveLogin(account *entity.Account, ip string) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取个人账号信息
 | 
			
		||||
func (a Account) AccountInfo(rc *ctx.ReqCtx) {
 | 
			
		||||
func (a *Account) AccountInfo(rc *ctx.ReqCtx) {
 | 
			
		||||
	ap := new(vo.AccountPersonVO)
 | 
			
		||||
	// 角色信息
 | 
			
		||||
	roles := new([]vo.AccountRoleVO)
 | 
			
		||||
@@ -111,7 +165,7 @@ func (a Account) AccountInfo(rc *ctx.ReqCtx) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 更新个人账号信息
 | 
			
		||||
func (a Account) UpdateAccount(rc *ctx.ReqCtx) {
 | 
			
		||||
func (a *Account) UpdateAccount(rc *ctx.ReqCtx) {
 | 
			
		||||
	updateForm := &form.AccountUpdateForm{}
 | 
			
		||||
	ginx.BindJsonAndValid(rc.GinCtx, updateForm)
 | 
			
		||||
 | 
			
		||||
@@ -121,13 +175,14 @@ func (a Account) UpdateAccount(rc *ctx.ReqCtx) {
 | 
			
		||||
	updateAccount.Id = rc.LoginAccount.Id
 | 
			
		||||
 | 
			
		||||
	if updateAccount.Password != "" {
 | 
			
		||||
		biz.IsTrue(CheckPasswordLever(updateAccount.Password), "密码强度必须8位以上且包含字⺟⼤⼩写+数字+特殊符号")
 | 
			
		||||
		updateAccount.Password = utils.Md5(updateAccount.Password)
 | 
			
		||||
	}
 | 
			
		||||
	a.AccountApp.Update(updateAccount)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取账号接收的消息列表
 | 
			
		||||
func (a Account) GetMsgs(rc *ctx.ReqCtx) {
 | 
			
		||||
func (a *Account) GetMsgs(rc *ctx.ReqCtx) {
 | 
			
		||||
	condition := &entity.Msg{
 | 
			
		||||
		RecipientId: int64(rc.LoginAccount.Id),
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -7,3 +7,9 @@ type AccountCreateForm struct {
 | 
			
		||||
type AccountUpdateForm struct {
 | 
			
		||||
	Password *string `json:"password" binding:"min=6,max=16"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type AccountChangePasswordForm struct {
 | 
			
		||||
	Username    string `json:"username"`
 | 
			
		||||
	OldPassword string `json:"oldPassword"`
 | 
			
		||||
	NewPassword string `json:"newPassword"`
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -29,6 +29,7 @@ func (r *Role) SaveRole(rc *ctx.ReqCtx) {
 | 
			
		||||
	g := rc.GinCtx
 | 
			
		||||
	form := &form.RoleForm{}
 | 
			
		||||
	ginx.BindJsonAndValid(g, form)
 | 
			
		||||
	rc.ReqParam = form
 | 
			
		||||
 | 
			
		||||
	role := new(entity.Role)
 | 
			
		||||
	utils.Copy(role, form)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										21
									
								
								server/internal/sys/api/syslog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								server/internal/sys/api/syslog.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
package api
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/internal/sys/application"
 | 
			
		||||
	"mayfly-go/internal/sys/domain/entity"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
	"mayfly-go/pkg/ginx"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Syslog struct {
 | 
			
		||||
	SyslogApp application.Syslog
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (r *Syslog) Syslogs(rc *ctx.ReqCtx) {
 | 
			
		||||
	g := rc.GinCtx
 | 
			
		||||
	condition := &entity.Syslog{
 | 
			
		||||
		Type:      int8(ginx.QueryInt(g, "type", 0)),
 | 
			
		||||
		CreatorId: uint64(ginx.QueryInt(g, "creatorId", 0)),
 | 
			
		||||
	}
 | 
			
		||||
	rc.ResData = r.SyslogApp.GetPageList(condition, ginx.GetPageParam(g), new([]entity.Syslog), "create_time DESC")
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										79
									
								
								server/internal/sys/application/syslog_app.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										79
									
								
								server/internal/sys/application/syslog_app.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,79 @@
 | 
			
		||||
package application
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"mayfly-go/internal/sys/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/sys/domain/repository"
 | 
			
		||||
	"mayfly-go/internal/sys/infrastructure/persistence"
 | 
			
		||||
	"mayfly-go/pkg/biz"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
	"mayfly-go/pkg/utils"
 | 
			
		||||
	"reflect"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Syslog interface {
 | 
			
		||||
	GetPageList(condition *entity.Syslog, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult
 | 
			
		||||
 | 
			
		||||
	// 从请求上下文的参数保存系统日志
 | 
			
		||||
	SaveFromReq(req *ctx.ReqCtx)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type syslogAppImpl struct {
 | 
			
		||||
	syslogRepo repository.Syslog
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 实现类单例
 | 
			
		||||
var SyslogApp Syslog = &syslogAppImpl{
 | 
			
		||||
	syslogRepo: persistence.SyslogDao,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *syslogAppImpl) GetPageList(condition *entity.Syslog, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult {
 | 
			
		||||
	return m.syslogRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *syslogAppImpl) SaveFromReq(req *ctx.ReqCtx) {
 | 
			
		||||
	lg := req.LoginAccount
 | 
			
		||||
	if lg == nil {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	syslog := new(entity.Syslog)
 | 
			
		||||
	syslog.CreateTime = time.Now()
 | 
			
		||||
	syslog.Creator = lg.Username
 | 
			
		||||
	syslog.CreatorId = lg.Id
 | 
			
		||||
	syslog.Description = req.LogInfo.Description
 | 
			
		||||
 | 
			
		||||
	if req.LogInfo.LogResp {
 | 
			
		||||
		respB, _ := json.Marshal(req.ResData)
 | 
			
		||||
		syslog.Resp = string(respB)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	reqParam := req.ReqParam
 | 
			
		||||
	if !utils.IsBlank(reflect.ValueOf(reqParam)) {
 | 
			
		||||
		// 如果是字符串类型,则不使用json序列化
 | 
			
		||||
		if reqStr, ok := reqParam.(string); ok {
 | 
			
		||||
			syslog.ReqParam = reqStr
 | 
			
		||||
		} else {
 | 
			
		||||
			reqB, _ := json.Marshal(reqParam)
 | 
			
		||||
			syslog.ReqParam = string(reqB)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if err := req.Err; err != nil {
 | 
			
		||||
		syslog.Type = entity.SyslogTypeError
 | 
			
		||||
		var errMsg string
 | 
			
		||||
		switch t := err.(type) {
 | 
			
		||||
		case *biz.BizError:
 | 
			
		||||
			errMsg = fmt.Sprintf("errCode: %d, errMsg: %s", t.Code(), t.Error())
 | 
			
		||||
		case error:
 | 
			
		||||
			errMsg = t.Error()
 | 
			
		||||
		}
 | 
			
		||||
		syslog.Resp = errMsg
 | 
			
		||||
	} else {
 | 
			
		||||
		syslog.Type = entity.SyslogTypeNorman
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	m.syslogRepo.Insert(syslog)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										25
									
								
								server/internal/sys/domain/entity/syslog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								server/internal/sys/domain/entity/syslog.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
			
		||||
package entity
 | 
			
		||||
 | 
			
		||||
import "time"
 | 
			
		||||
 | 
			
		||||
// 系统操作日志
 | 
			
		||||
type Syslog struct {
 | 
			
		||||
	Id         uint64    `json:"id"`
 | 
			
		||||
	CreateTime time.Time `json:"createTime"`
 | 
			
		||||
	CreatorId  uint64    `json:"creatorId"`
 | 
			
		||||
	Creator    string    `json:"creator"`
 | 
			
		||||
 | 
			
		||||
	Type        int8   `json:"type"`
 | 
			
		||||
	Description string `json:"description"`
 | 
			
		||||
	ReqParam    string `json:"reqParam"` // 请求参数
 | 
			
		||||
	Resp        string `json:"resp"`     // 响应结构
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Syslog) TableName() string {
 | 
			
		||||
	return "t_sys_log"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	SyslogTypeNorman int8 = 1 // 正常状态
 | 
			
		||||
	SyslogTypeError  int8 = 2 // 错误状态
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										12
									
								
								server/internal/sys/domain/repository/syslog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								server/internal/sys/domain/repository/syslog.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
package repository
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/internal/sys/domain/entity"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Syslog interface {
 | 
			
		||||
	GetPageList(condition *entity.Syslog, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult
 | 
			
		||||
 | 
			
		||||
	Insert(log *entity.Syslog)
 | 
			
		||||
}
 | 
			
		||||
@@ -0,0 +1,19 @@
 | 
			
		||||
package persistence
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/internal/sys/domain/entity"
 | 
			
		||||
	"mayfly-go/internal/sys/domain/repository"
 | 
			
		||||
	"mayfly-go/pkg/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type syslogRepo struct{}
 | 
			
		||||
 | 
			
		||||
var SyslogDao repository.Syslog = &syslogRepo{}
 | 
			
		||||
 | 
			
		||||
func (m *syslogRepo) GetPageList(condition *entity.Syslog, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult {
 | 
			
		||||
	return model.GetPage(pageParam, condition, toEntity, orderBy...)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *syslogRepo) Insert(syslog *entity.Syslog) {
 | 
			
		||||
	model.Insert(syslog)
 | 
			
		||||
}
 | 
			
		||||
@@ -18,9 +18,20 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
			
		||||
	}
 | 
			
		||||
	{
 | 
			
		||||
		// 用户登录
 | 
			
		||||
		loginLog := ctx.NewLogInfo("用户登录").WithSave(true)
 | 
			
		||||
		account.POST("login", func(g *gin.Context) {
 | 
			
		||||
			rc := ctx.NewReqCtxWithGin(g).WithNeedToken(false).WithLog(ctx.NewLogInfo("用户登录"))
 | 
			
		||||
			rc.Handle(a.Login)
 | 
			
		||||
			ctx.NewReqCtxWithGin(g).
 | 
			
		||||
				WithNeedToken(false).
 | 
			
		||||
				WithLog(loginLog).
 | 
			
		||||
				Handle(a.Login)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		changePwdLog := ctx.NewLogInfo("用户修改密码").WithSave(true)
 | 
			
		||||
		account.POST("change-pwd", func(g *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(g).
 | 
			
		||||
				WithNeedToken(false).
 | 
			
		||||
				WithLog(changePwdLog).
 | 
			
		||||
				Handle(a.ChangePassword)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 获取个人账号信息
 | 
			
		||||
@@ -44,7 +55,7 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(a.Accounts)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		createAccount := ctx.NewLogInfo("创建账号")
 | 
			
		||||
		createAccount := ctx.NewLogInfo("创建账号").WithSave(true)
 | 
			
		||||
		addAccountPermission := ctx.NewPermission("account:add")
 | 
			
		||||
		account.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
@@ -53,14 +64,14 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(a.CreateAccount)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("修改账号状态")
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("修改账号状态").WithSave(true)
 | 
			
		||||
		account.PUT("change-status/:id/:status", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
				WithLog(changeStatus).
 | 
			
		||||
				Handle(a.ChangeStatus)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delAccount := ctx.NewLogInfo("删除账号")
 | 
			
		||||
		delAccount := ctx.NewLogInfo("删除账号").WithSave(true)
 | 
			
		||||
		delAccountPermission := ctx.NewPermission("account:del")
 | 
			
		||||
		account.DELETE(":id", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
@@ -75,7 +86,7 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		// 保存用户角色
 | 
			
		||||
		saveAccountRole := ctx.NewLogInfo("保存用户角色")
 | 
			
		||||
		saveAccountRole := ctx.NewLogInfo("保存用户角色").WithSave(true)
 | 
			
		||||
		sarPermission := ctx.NewPermission("account:saveRoles")
 | 
			
		||||
		account.POST("/roles", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(saveAccountRole).
 | 
			
		||||
 
 | 
			
		||||
@@ -12,10 +12,6 @@ func InitResourceRouter(router *gin.RouterGroup) {
 | 
			
		||||
	r := &api.Resource{ResourceApp: application.ResourceApp}
 | 
			
		||||
	db := router.Group("sys/resources")
 | 
			
		||||
	{
 | 
			
		||||
		// db.GET("/account", func(c *gin.Context) {
 | 
			
		||||
		// 	ctx.NewReqCtxWithGin(c).Handle(r.ResourceTree)
 | 
			
		||||
		// })
 | 
			
		||||
 | 
			
		||||
		db.GET("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(r.GetAllResourceTree)
 | 
			
		||||
		})
 | 
			
		||||
@@ -24,7 +20,7 @@ func InitResourceRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(r.GetById)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveResource := ctx.NewLogInfo("保存资源")
 | 
			
		||||
		saveResource := ctx.NewLogInfo("保存资源").WithSave(true)
 | 
			
		||||
		srPermission := ctx.NewPermission("resource:add")
 | 
			
		||||
		db.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
@@ -33,7 +29,7 @@ func InitResourceRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(r.SaveResource)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("修改资源状态")
 | 
			
		||||
		changeStatus := ctx.NewLogInfo("修改资源状态").WithSave(true)
 | 
			
		||||
		csPermission := ctx.NewPermission("resource:changeStatus")
 | 
			
		||||
		db.PUT(":id/:status", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
@@ -42,7 +38,7 @@ func InitResourceRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(r.ChangeStatus)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delResource := ctx.NewLogInfo("删除资源")
 | 
			
		||||
		delResource := ctx.NewLogInfo("删除资源").WithSave(true)
 | 
			
		||||
		dePermission := ctx.NewPermission("resource:delete")
 | 
			
		||||
		db.DELETE(":id", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).
 | 
			
		||||
 
 | 
			
		||||
@@ -20,7 +20,7 @@ func InitRoleRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(r.Roles)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveRole := ctx.NewLogInfo("保存角色")
 | 
			
		||||
		saveRole := ctx.NewLogInfo("保存角色").WithSave(true)
 | 
			
		||||
		sPermission := ctx.NewPermission("role:add")
 | 
			
		||||
		db.POST("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(saveRole).
 | 
			
		||||
@@ -28,7 +28,7 @@ func InitRoleRouter(router *gin.RouterGroup) {
 | 
			
		||||
				Handle(r.SaveRole)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		delRole := ctx.NewLogInfo("删除角色")
 | 
			
		||||
		delRole := ctx.NewLogInfo("删除角色").WithSave(true)
 | 
			
		||||
		drPermission := ctx.NewPermission("role:del")
 | 
			
		||||
		db.DELETE(":id", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(delRole).
 | 
			
		||||
@@ -44,7 +44,7 @@ func InitRoleRouter(router *gin.RouterGroup) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(r.RoleResource)
 | 
			
		||||
		})
 | 
			
		||||
 | 
			
		||||
		saveResource := ctx.NewLogInfo("保存角色资源")
 | 
			
		||||
		saveResource := ctx.NewLogInfo("保存角色资源").WithSave(true)
 | 
			
		||||
		srPermission := ctx.NewPermission("role:saveResources")
 | 
			
		||||
		db.POST(":id/resources", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).WithLog(saveResource).
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										21
									
								
								server/internal/sys/router/syslog.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								server/internal/sys/router/syslog.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
package router
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"mayfly-go/internal/sys/api"
 | 
			
		||||
	"mayfly-go/internal/sys/application"
 | 
			
		||||
	"mayfly-go/pkg/ctx"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func InitSyslogRouter(router *gin.RouterGroup) {
 | 
			
		||||
	s := &api.Syslog{
 | 
			
		||||
		SyslogApp: application.SyslogApp,
 | 
			
		||||
	}
 | 
			
		||||
	sys := router.Group("syslogs")
 | 
			
		||||
	{
 | 
			
		||||
		sys.GET("", func(c *gin.Context) {
 | 
			
		||||
			ctx.NewReqCtxWithGin(c).Handle(s.Syslogs)
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
@@ -29,7 +29,9 @@ CREATE TABLE `t_db` (
 | 
			
		||||
  `type` varchar(20) COLLATE utf8mb4_bin NOT NULL COMMENT '数据库实例类型(mysql...)',
 | 
			
		||||
  `database` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '数据库,空格分割多个数据库',
 | 
			
		||||
  `params` varchar(125) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '其他连接参数',
 | 
			
		||||
  `network` varchar(8) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `network` varchar(20) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
			
		||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
			
		||||
  `project_id` bigint(20) DEFAULT NULL,
 | 
			
		||||
  `project` varchar(64) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `env_id` bigint(20) DEFAULT NULL COMMENT '环境id',
 | 
			
		||||
@@ -107,7 +109,10 @@ CREATE TABLE `t_machine` (
 | 
			
		||||
  `ip` varchar(36) COLLATE utf8mb4_bin NOT NULL,
 | 
			
		||||
  `port` int(12) NOT NULL,
 | 
			
		||||
  `username` varchar(12) COLLATE utf8mb4_bin NOT NULL,
 | 
			
		||||
  `password` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `auth_method` tinyint(2) NULL DEFAULT NULL COMMENT '1.密码登录2.publickey登录',
 | 
			
		||||
  `password` varchar(3200) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
			
		||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
			
		||||
  `status` tinyint(2) NOT NULL COMMENT '状态: 1:启用; -1:禁用',
 | 
			
		||||
  `remark` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `need_monitor` tinyint(2) DEFAULT NULL,
 | 
			
		||||
@@ -258,6 +263,8 @@ CREATE TABLE `t_redis` (
 | 
			
		||||
  `password` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `db` int(32) DEFAULT NULL,
 | 
			
		||||
  `mode` varchar(32) DEFAULT NULL,
 | 
			
		||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
			
		||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
			
		||||
  `remark` varchar(125) DEFAULT NULL,
 | 
			
		||||
  `project_id` bigint(20) DEFAULT NULL,
 | 
			
		||||
  `project` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
@@ -363,12 +370,11 @@ CREATE TABLE `t_sys_resource` (
 | 
			
		||||
-- Records of t_sys_resource
 | 
			
		||||
-- ----------------------------
 | 
			
		||||
BEGIN;
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (1, 0, 1, 1, '首页', '/home', 1, '{\"component\":\"Home\",\"icon\":\"Home\",\"isAffix\":true,\"isKeepAlive\":true,\"routeName\":\"Home\"}', 1, 'admin', 1, 'admin', '2021-05-25 16:44:41', '2021-05-27 09:12:56');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (1, 0, 1, 1, '首页', '/home', 1, '{\"component\":\"Home\",\"icon\":\"HomeFilled\",\"isAffix\":true,\"isKeepAlive\":true,\"routeName\":\"Home\"}', 1, 'admin', 1, 'admin', '2021-05-25 16:44:41', '2021-05-27 09:12:56');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (2, 0, 1, 1, '运维', '/ops', 3, '{\"icon\":\"Monitor\",\"isKeepAlive\":true,\"redirect\":\"machine/list\",\"routeName\":\"Ops\"}', 1, 'admin', 1, 'admin', '2021-05-25 16:48:16', '2021-06-08 14:20:24');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (3, 2, 1, 1, '机器列表', 'machines', 2, '{\"component\":\"MachineList\",\"icon\":\"Menu\",\"isKeepAlive\":true,\"routeName\":\"MachineList\"}', 2, 'admin', 1, 'admin', '2021-05-25 16:50:04', '2021-06-30 16:20:08');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (4, 0, 1, 1, '系统管理', '/sys', 5, '{\"icon\":\"Setting\",\"isKeepAlive\":true,\"redirect\":\"/sys/resources\",\"routeName\":\"sys\"}', 1, 'admin', 1, 'admin', '2021-05-26 15:20:20', '2021-07-29 18:03:06');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (5, 4, 1, 1, '资源管理', 'resources', 3, '{\"component\":\"ResourceList\",\"icon\":\"Menu\",\"isKeepAlive\":true,\"routeName\":\"ResourceList\"}', 1, 'admin', 1, 'admin', '2021-05-26 15:23:07', '2021-06-08 11:27:55');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (9, 0, 1, 1, 'iframes', '/iframes', 6, '{\"component\":\"RouterParent\",\"icon\":\"Menu\",\"isIframe\":true,\"isKeepAlive\":true,\"link\":\"https://www.baidu.com\",\"routeName\":\"Iframe\"}', 1, 'admin', 1, 'admin', '2021-05-27 09:58:37', '2021-07-29 18:03:13');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (11, 4, 1, 1, '角色管理', 'roles', 2, '{\"component\":\"RoleList\",\"icon\":\"Menu\",\"isKeepAlive\":true,\"routeName\":\"RoleList\"}', 1, 'admin', 1, 'admin', '2021-05-27 11:15:35', '2021-06-03 09:59:41');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (12, 3, 2, 1, '机器终端按钮', 'machine:terminal', 4, '', 1, 'admin', 1, 'admin', '2021-05-28 14:06:02', '2021-05-31 17:47:59');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (14, 4, 1, 1, '账号管理', 'accounts', 1, '{\"component\":\"AccountList\",\"icon\":\"Menu\",\"isKeepAlive\":true,\"routeName\":\"AccountList\"}', 1, 'admin', 1, 'admin', '2021-05-28 14:56:25', '2021-06-03 09:39:22');
 | 
			
		||||
@@ -383,7 +389,7 @@ INSERT INTO `t_sys_resource` VALUES (22, 11, 2, 1, '角色删除按钮', 'role:d
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (23, 11, 2, 1, '角色新增按钮', 'role:add', 3, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:02:44', '2021-05-31 19:33:39');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (24, 11, 2, 1, '角色编辑按钮', 'role:update', 4, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:02:57', '2021-05-31 19:33:40');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (25, 5, 2, 1, '资源新增按钮', 'resource:add', 1, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:03:33', '2021-05-31 19:31:47');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (26, 5, 2, 1, '资源删除按钮', 'resource:del', 2, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:03:47', '2021-05-31 19:29:40');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (26, 5, 2, 1, '资源删除按钮', 'resource:delete', 2, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:03:47', '2021-05-31 19:29:40');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (27, 5, 2, 1, '资源编辑按钮', 'resource:update', 3, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:04:03', '2021-05-31 19:29:40');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (28, 5, 2, 1, '资源禁用启用按钮', 'resource:changeStatus', 4, NULL, 1, 'admin', 1, 'admin', '2021-05-31 18:04:33', '2021-05-31 18:04:33');
 | 
			
		||||
INSERT INTO `t_sys_resource` VALUES (29, 14, 2, 1, '账号添加按钮', 'account:add', 3, NULL, 1, 'admin', 1, 'admin', '2021-05-31 19:23:42', '2021-05-31 19:23:42');
 | 
			
		||||
@@ -429,6 +435,8 @@ INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `wei
 | 
			
		||||
INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (81, 80, 2, 1, '基本权限', 'mongo:base', 1, 'null', 1, 'admin', 1, 'admin', '2022-05-13 14:04:16', '2022-05-13 14:04:16');
 | 
			
		||||
INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (82, 79, 1, 1, 'Mongo管理', 'mongo-manage', 2, '{\"component\":\"MongoList\",\"icon\":\"Menu\",\"isKeepAlive\":true,\"routeName\":\"MongoList\"}', 1, 'admin', 1, 'admin', '2022-05-16 18:13:06', '2022-05-16 18:13:06');
 | 
			
		||||
INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (83, 82, 2, 1, '基本权限', 'mongo:manage:base', 1, 'null', 1, 'admin', 1, 'admin', '2022-05-16 18:13:25', '2022-05-16 18:13:25');
 | 
			
		||||
INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (84, 4, 1, 1, '操作日志', 'syslogs', 4, '{\"component\":\"SyslogList\",\"icon\":\"Tickets\",\"routeName\":\"SyslogList\"}', 1, 'admin', 1, 'admin', '2022-07-13 19:57:07', '2022-07-13 22:58:19');
 | 
			
		||||
INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (85, 84, 2, 1, '操作日志基本权限', 'syslog', 1, 'null', 1, 'admin', 1, 'admin', '2022-07-13 19:57:55', '2022-07-13 19:57:55');
 | 
			
		||||
COMMIT;
 | 
			
		||||
 | 
			
		||||
-- ----------------------------
 | 
			
		||||
@@ -625,8 +633,34 @@ INSERT INTO `t_sys_role_resource` VALUES (496, 8, 61, 1, 'admin', '2021-11-05 15
 | 
			
		||||
INSERT INTO `t_sys_role_resource` VALUES (497, 8, 62, 1, 'admin', '2021-11-05 15:59:16');
 | 
			
		||||
INSERT INTO `t_sys_role_resource` VALUES (498, 8, 63, 1, 'admin', '2021-11-05 15:59:16');
 | 
			
		||||
INSERT INTO `t_sys_role_resource` VALUES (499, 8, 64, 1, 'admin', '2021-11-05 15:59:16');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (500, 1, 72, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (501, 1, 71, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (502, 1, 79, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (503, 1, 80, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (504, 1, 81, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (505, 1, 82, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (506, 1, 83, 1, 'admin', '2022-07-14 11:03:09');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (507, 1, 84, 1, 'admin', '2022-07-14 11:10:11');
 | 
			
		||||
INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (508, 1, 85, 1, 'admin', '2022-07-14 11:10:11');
 | 
			
		||||
COMMIT;
 | 
			
		||||
 | 
			
		||||
-- ----------------------------
 | 
			
		||||
-- 表结构: t_sys_log 
 | 
			
		||||
-- ----------------------------
 | 
			
		||||
DROP TABLE IF EXISTS `t_sys_log`;
 | 
			
		||||
CREATE TABLE `t_sys_log` (
 | 
			
		||||
  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
			
		||||
  `type` tinyint(4) NOT NULL COMMENT '类型',
 | 
			
		||||
  `description` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin DEFAULT NULL COMMENT '描述',
 | 
			
		||||
  `req_param` varchar(1000) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin DEFAULT NULL COMMENT '请求信息',
 | 
			
		||||
  `resp` varchar(1000) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin DEFAULT NULL COMMENT '响应信息',
 | 
			
		||||
  `creator` varchar(36) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin NOT NULL COMMENT '调用者',
 | 
			
		||||
  `creator_id` bigint(20) NOT NULL COMMENT '调用者id',
 | 
			
		||||
  `create_time` datetime NOT NULL COMMENT '操作时间',
 | 
			
		||||
  PRIMARY KEY (`id`),
 | 
			
		||||
  KEY `idx_creator_id` (`creator_id`) USING BTREE
 | 
			
		||||
) ENGINE=InnoDB AUTO_INCREMENT=41 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin COMMENT='系统操作日志';
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
-- ----------------------------
 | 
			
		||||
-- Table structure for t_mongo
 | 
			
		||||
@@ -636,6 +670,8 @@ CREATE TABLE `t_mongo` (
 | 
			
		||||
  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
			
		||||
  `name` varchar(36) COLLATE utf8mb4_bin NOT NULL COMMENT '名称',
 | 
			
		||||
  `uri` varchar(255) COLLATE utf8mb4_bin NOT NULL COMMENT '连接uri',
 | 
			
		||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
			
		||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
			
		||||
  `project_id` bigint(20) NOT NULL,
 | 
			
		||||
  `project` varchar(36) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
			
		||||
  `env_id` bigint(20) DEFAULT NULL,
 | 
			
		||||
 
 | 
			
		||||
@@ -7,28 +7,28 @@ type BizError struct {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	Success       *BizError = NewBizErrCode(200, "success")
 | 
			
		||||
	BizErr        *BizError = NewBizErrCode(400, "biz error")
 | 
			
		||||
	ServerError   *BizError = NewBizErrCode(500, "server error")
 | 
			
		||||
	PermissionErr *BizError = NewBizErrCode(501, "token error")
 | 
			
		||||
	Success       BizError = NewBizErrCode(200, "success")
 | 
			
		||||
	BizErr        BizError = NewBizErrCode(400, "biz error")
 | 
			
		||||
	ServerError   BizError = NewBizErrCode(500, "server error")
 | 
			
		||||
	PermissionErr BizError = NewBizErrCode(501, "token error")
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// 错误消息
 | 
			
		||||
func (e *BizError) Error() string {
 | 
			
		||||
func (e BizError) Error() string {
 | 
			
		||||
	return e.err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 错误码
 | 
			
		||||
func (e *BizError) Code() int16 {
 | 
			
		||||
func (e BizError) Code() int16 {
 | 
			
		||||
	return e.code
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 创建业务逻辑错误结构体,默认为业务逻辑错误
 | 
			
		||||
func NewBizErr(msg string) *BizError {
 | 
			
		||||
	return &BizError{code: BizErr.code, err: msg}
 | 
			
		||||
func NewBizErr(msg string) BizError {
 | 
			
		||||
	return BizError{code: BizErr.code, err: msg}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 创建业务逻辑错误结构体,可设置指定错误code
 | 
			
		||||
func NewBizErrCode(code int16, msg string) *BizError {
 | 
			
		||||
	return &BizError{code: code, err: msg}
 | 
			
		||||
func NewBizErrCode(code int16, msg string) BizError {
 | 
			
		||||
	return BizError{code: code, err: msg}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -12,20 +12,38 @@ import (
 | 
			
		||||
	"github.com/sirupsen/logrus"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type SaveLogFunc func(*ReqCtx)
 | 
			
		||||
 | 
			
		||||
var saveLog SaveLogFunc
 | 
			
		||||
 | 
			
		||||
// 设置保存日志处理函数
 | 
			
		||||
func SetSaveLogFunc(sl SaveLogFunc) {
 | 
			
		||||
	saveLog = sl
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type LogInfo struct {
 | 
			
		||||
	LogResp     bool   // 是否记录返回结果
 | 
			
		||||
	Description string // 请求描述
 | 
			
		||||
	Save        bool   // 是否保存日志
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 新建日志信息
 | 
			
		||||
func NewLogInfo(description string) *LogInfo {
 | 
			
		||||
	return &LogInfo{Description: description, LogResp: false}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 是否记录返回结果
 | 
			
		||||
func (i *LogInfo) WithLogResp(logResp bool) *LogInfo {
 | 
			
		||||
	i.LogResp = logResp
 | 
			
		||||
	return i
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 是否保存日志
 | 
			
		||||
func (i *LogInfo) WithSave(saveLog bool) *LogInfo {
 | 
			
		||||
	i.Save = saveLog
 | 
			
		||||
	return i
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func LogHandler(rc *ReqCtx) error {
 | 
			
		||||
	li := rc.LogInfo
 | 
			
		||||
	if li == nil {
 | 
			
		||||
@@ -41,6 +59,10 @@ func LogHandler(rc *ReqCtx) error {
 | 
			
		||||
	req := rc.GinCtx.Request
 | 
			
		||||
	lfs[req.Method] = req.URL.Path
 | 
			
		||||
 | 
			
		||||
	// 如果需要保存日志,并且保存日志处理函数存在则执行保存日志函数
 | 
			
		||||
	if li.Save && saveLog != nil {
 | 
			
		||||
		go saveLog(rc)
 | 
			
		||||
	}
 | 
			
		||||
	if err := rc.Err; err != nil {
 | 
			
		||||
		logger.Log.WithFields(lfs).Error(getErrMsg(rc, err))
 | 
			
		||||
		return nil
 | 
			
		||||
@@ -73,7 +95,7 @@ func getErrMsg(rc *ReqCtx, err interface{}) string {
 | 
			
		||||
 | 
			
		||||
	var errMsg string
 | 
			
		||||
	switch t := err.(type) {
 | 
			
		||||
	case *biz.BizError:
 | 
			
		||||
	case biz.BizError:
 | 
			
		||||
		errMsg = fmt.Sprintf("\n<-e errCode: %d, errMsg: %s", t.Code(), t.Error())
 | 
			
		||||
	case error:
 | 
			
		||||
		errMsg = fmt.Sprintf("\n<-e errMsg: %s\n%s", t.Error(), string(debug.Stack()))
 | 
			
		||||
 
 | 
			
		||||
@@ -53,7 +53,7 @@ func (rc *ReqCtx) Handle(handler HandlerFunc) {
 | 
			
		||||
 | 
			
		||||
	begin := time.Now()
 | 
			
		||||
	handler(rc)
 | 
			
		||||
	rc.timed = time.Now().Sub(begin).Milliseconds()
 | 
			
		||||
	rc.timed = time.Since(begin).Milliseconds()
 | 
			
		||||
	if !rc.NoRes {
 | 
			
		||||
		ginx.SuccessRes(ginCtx, rc.ResData)
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -62,7 +62,7 @@ func SuccessRes(g *gin.Context, data interface{}) {
 | 
			
		||||
// 返回失败结果集
 | 
			
		||||
func ErrorRes(g *gin.Context, err interface{}) {
 | 
			
		||||
	switch t := err.(type) {
 | 
			
		||||
	case *biz.BizError:
 | 
			
		||||
	case biz.BizError:
 | 
			
		||||
		g.JSON(http.StatusOK, model.Error(t))
 | 
			
		||||
	case error:
 | 
			
		||||
		g.JSON(http.StatusOK, model.ServerError())
 | 
			
		||||
 
 | 
			
		||||
@@ -44,7 +44,7 @@ func SuccessNoData() *Result {
 | 
			
		||||
	return &Result{Code: SuccessCode, Msg: SuccessMsg}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Error(bizerr *biz.BizError) *Result {
 | 
			
		||||
func Error(bizerr biz.BizError) *Result {
 | 
			
		||||
	return &Result{Code: bizerr.Code(), Msg: bizerr.Error()}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -13,6 +13,9 @@ func RunWebServer() {
 | 
			
		||||
	ctx.UseBeforeHandlerInterceptor(ctx.PermissionHandler)
 | 
			
		||||
	// 日志处理器
 | 
			
		||||
	ctx.UseAfterHandlerInterceptor(ctx.LogHandler)
 | 
			
		||||
	// 设置日志保存函数
 | 
			
		||||
	ctx.SetSaveLogFunc(initialize.InitSaveLogFunc())
 | 
			
		||||
 | 
			
		||||
	// 注册路由
 | 
			
		||||
	web := initialize.InitRouter()
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -1,8 +1,15 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"crypto/md5"
 | 
			
		||||
	"crypto/rand"
 | 
			
		||||
	"crypto/rsa"
 | 
			
		||||
	"crypto/x509"
 | 
			
		||||
	"encoding/base64"
 | 
			
		||||
	"encoding/hex"
 | 
			
		||||
	"encoding/pem"
 | 
			
		||||
	"errors"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// md5
 | 
			
		||||
@@ -11,3 +18,115 @@ func Md5(str string) string {
 | 
			
		||||
	h.Write([]byte(str))
 | 
			
		||||
	return hex.EncodeToString(h.Sum(nil))
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 系统统一RSA秘钥对
 | 
			
		||||
var RsaPair []string
 | 
			
		||||
 | 
			
		||||
// 生成RSA私钥和公钥字符串
 | 
			
		||||
// bits 证书大小
 | 
			
		||||
// @return privateKeyStr publicKeyStr error
 | 
			
		||||
func GenerateRSAKey(bits int) (string, string, error) {
 | 
			
		||||
	var privateKeyStr, publicKeyStr string
 | 
			
		||||
 | 
			
		||||
	//GenerateKey函数使用随机数据生成器random生成一对具有指定字位数的RSA密钥
 | 
			
		||||
	//Reader是一个全局、共享的密码用强随机数生成器
 | 
			
		||||
	privateKey, err := rsa.GenerateKey(rand.Reader, bits)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return privateKeyStr, publicKeyStr, err
 | 
			
		||||
	}
 | 
			
		||||
	//保存私钥
 | 
			
		||||
	//通过x509标准将得到的ras私钥序列化为ASN.1 的 DER编码字符串
 | 
			
		||||
	X509PrivateKey := x509.MarshalPKCS1PrivateKey(privateKey)
 | 
			
		||||
	//构建一个pem.Block结构体对象
 | 
			
		||||
	privateBlock := pem.Block{Type: "RSA Private Key", Bytes: X509PrivateKey}
 | 
			
		||||
 | 
			
		||||
	privateBuf := new(bytes.Buffer)
 | 
			
		||||
	pem.Encode(privateBuf, &privateBlock)
 | 
			
		||||
	privateKeyStr = privateBuf.String()
 | 
			
		||||
 | 
			
		||||
	//保存公钥
 | 
			
		||||
	//获取公钥的数据
 | 
			
		||||
	publicKey := privateKey.PublicKey
 | 
			
		||||
	//X509对公钥编码
 | 
			
		||||
	X509PublicKey, err := x509.MarshalPKIXPublicKey(&publicKey)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return publicKeyStr, privateKeyStr, err
 | 
			
		||||
	}
 | 
			
		||||
	//创建一个pem.Block结构体对象
 | 
			
		||||
	publicBlock := pem.Block{Type: "RSA Public Key", Bytes: X509PublicKey}
 | 
			
		||||
 | 
			
		||||
	publicBuf := new(bytes.Buffer)
 | 
			
		||||
	pem.Encode(publicBuf, &publicBlock)
 | 
			
		||||
	publicKeyStr = publicBuf.String()
 | 
			
		||||
 | 
			
		||||
	return privateKeyStr, publicKeyStr, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// rsa解密
 | 
			
		||||
func RsaDecrypt(privateKeyStr string, data []byte) ([]byte, error) {
 | 
			
		||||
	block, _ := pem.Decode([]byte(privateKeyStr))
 | 
			
		||||
	if block == nil {
 | 
			
		||||
		return nil, errors.New("private key error")
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	priv, err := x509.ParsePKCS1PrivateKey(block.Bytes)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
	}
 | 
			
		||||
	return rsa.DecryptPKCS1v15(rand.Reader, priv, data)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 使用系统默认的私钥解密
 | 
			
		||||
// @param base64 字符串是否使用base64编码
 | 
			
		||||
func DefaultRsaDecrypt(data string, useBase64 bool) (string, error) {
 | 
			
		||||
	// 空字符串不解密
 | 
			
		||||
	if data == "" {
 | 
			
		||||
		return "", nil
 | 
			
		||||
	}
 | 
			
		||||
	if useBase64 {
 | 
			
		||||
		if decodeBase64, err := base64.StdEncoding.DecodeString(data); err != nil {
 | 
			
		||||
			return "", err
 | 
			
		||||
		} else {
 | 
			
		||||
			data = string(decodeBase64)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	priKey, err := GetRsaPrivateKey()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	val, err := RsaDecrypt(priKey, []byte(data))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	return string(val), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取系统的RSA公钥
 | 
			
		||||
func GetRsaPublicKey() (string, error) {
 | 
			
		||||
	if len(RsaPair) == 2 {
 | 
			
		||||
		return RsaPair[1], nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	privateKey, publicKey, err := GenerateRSAKey(1024)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	RsaPair = append(RsaPair, privateKey)
 | 
			
		||||
	RsaPair = append(RsaPair, publicKey)
 | 
			
		||||
	return publicKey, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// 获取系统私钥
 | 
			
		||||
func GetRsaPrivateKey() (string, error) {
 | 
			
		||||
	if len(RsaPair) == 2 {
 | 
			
		||||
		return RsaPair[0], nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	privateKey, publicKey, err := GenerateRSAKey(1024)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return "", err
 | 
			
		||||
	}
 | 
			
		||||
	RsaPair = append(RsaPair, privateKey)
 | 
			
		||||
	RsaPair = append(RsaPair, publicKey)
 | 
			
		||||
	return privateKey, nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										21
									
								
								server/pkg/utils/net.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								server/pkg/utils/net.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import "net"
 | 
			
		||||
 | 
			
		||||
// GetAvailablePort 获取可用端口
 | 
			
		||||
func GetAvailablePort() (int, error) {
 | 
			
		||||
	addr, err := net.ResolveTCPAddr("tcp", "localhost:0")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return 0, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	l, err := net.ListenTCP("tcp", addr)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return 0, err
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	defer func(l *net.TCPListener) {
 | 
			
		||||
		_ = l.Close()
 | 
			
		||||
	}(l)
 | 
			
		||||
	return l.Addr().(*net.TCPAddr).Port, nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										36
									
								
								server/pkg/utils/ssh_conn_wrap.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								server/pkg/utils/ssh_conn_wrap.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,36 @@
 | 
			
		||||
package utils
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"net"
 | 
			
		||||
	"time"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type WrapSshConn struct {
 | 
			
		||||
	Conn net.Conn
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *WrapSshConn) Read(b []byte) (n int, err error) {
 | 
			
		||||
	return c.Conn.Read(b)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (c *WrapSshConn) Write(b []byte) (n int, err error) {
 | 
			
		||||
	return c.Conn.Write(b)
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) Close() error {
 | 
			
		||||
	return c.Conn.Close()
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) LocalAddr() net.Addr {
 | 
			
		||||
	return c.Conn.LocalAddr()
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) RemoteAddr() net.Addr {
 | 
			
		||||
	return c.Conn.RemoteAddr()
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) SetDeadline(t time.Time) error {
 | 
			
		||||
	return c.Conn.SetDeadline(t)
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) SetReadDeadline(t time.Time) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
func (c *WrapSshConn) SetWriteDeadline(t time.Time) error {
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -1,7 +1,8 @@
 | 
			
		||||
后端配置:
 | 
			
		||||
  服务端口,mysql等信息在config.yml里配置即可。
 | 
			
		||||
前端配置:
 | 
			
		||||
  static/config.js中的api地址配成启动后的后端服务的真实地址即可
 | 
			
		||||
相关配置文件: 
 | 
			
		||||
  后端:
 | 
			
		||||
    config.yml: 服务端口,mysql等信息在此配置即可。
 | 
			
		||||
  前端:
 | 
			
		||||
    static/config.js: 若前后端分开部署则将该文件中的api地址配成后端服务的真实地址即可,否则无需修改。
 | 
			
		||||
 | 
			
		||||
服务启动:./startup.sh
 | 
			
		||||
服务关闭:./shutdown.sh
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user