mirror of
				https://gitee.com/dromara/mayfly-go
				synced 2025-11-04 08:20:25 +08:00 
			
		
		
		
	Compare commits
	
		
			25 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					7761fe0288 | ||
| 
						 | 
					09e6bdcf7e | ||
| 
						 | 
					61a4d87f59 | ||
| 
						 | 
					c219ec33b0 | ||
| 
						 | 
					fd86f36218 | ||
| 
						 | 
					efac41f392 | ||
| 
						 | 
					52df61ae0d | ||
| 
						 | 
					cf2bc6785c | ||
| 
						 | 
					98a4c92576 | ||
| 
						 | 
					b1ee9b65ff | ||
| 
						 | 
					99cc4c5e5e | ||
| 
						 | 
					226bb8f089 | ||
| 
						 | 
					37ed5134e8 | ||
| 
						 | 
					0f54d4a472 | ||
| 
						 | 
					64805360d6 | ||
| 
						 | 
					7f69fe2ad9 | ||
| 
						 | 
					f913510d3c | ||
| 
						 | 
					f2d9e7786d | ||
| 
						 | 
					e1afb1ed54 | ||
| 
						 | 
					12f8cf0111 | ||
| 
						 | 
					daa2ef5203 | ||
| 
						 | 
					1e3e183930 | ||
| 
						 | 
					366563a0fe | ||
| 
						 | 
					577802e5ad | ||
| 
						 | 
					76d6fc3ba5 | 
							
								
								
									
										22
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										22
									
								
								README.md
									
									
									
									
									
								
							@@ -1,7 +1,25 @@
 | 
				
			|||||||
# 🌈mayfly-go
 | 
					# 🌈mayfly-go
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					<p align="center">
 | 
				
			||||||
 | 
					  <a href="https://gitee.com/objs/mayfly-go" target="_blank">
 | 
				
			||||||
 | 
					    <img src="https://gitee.com/objs/mayfly-go/badge/star.svg?theme=white" alt="star"/>
 | 
				
			||||||
 | 
					    <img src="https://gitee.com/objs/mayfly-go/badge/fork.svg" alt="fork"/>
 | 
				
			||||||
 | 
					  </a>
 | 
				
			||||||
 | 
					  <a href="https://github.com/may-fly/mayfly-go" target="_blank">
 | 
				
			||||||
 | 
					    <img src="https://img.shields.io/github/stars/may-fly/mayfly-go.svg?style=social" alt="github star"/>
 | 
				
			||||||
 | 
					    <img src="https://img.shields.io/github/forks/may-fly/mayfly-go.svg?style=social" alt="github fork"/>
 | 
				
			||||||
 | 
					  </a>
 | 
				
			||||||
 | 
					  <a href="https://github.com/golang/go" target="_blank">
 | 
				
			||||||
 | 
					    <img src="https://img.shields.io/badge/Golang-1.18%2B-yellow.svg" alt="golang"/>
 | 
				
			||||||
 | 
					  </a>
 | 
				
			||||||
 | 
					  <a href="https://cn.vuejs.org" target="_blank">
 | 
				
			||||||
 | 
					    <img src="https://img.shields.io/badge/Vue-3.x-green.svg" alt="vue">
 | 
				
			||||||
 | 
					  </a>
 | 
				
			||||||
 | 
					</p>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 介绍
 | 
					### 介绍
 | 
				
			||||||
简单基于DDD(领域驱动设计)分层架构实现的web版 **linux、数据库(mysql postgres)、redis(单机 集群)、mongo统一管理操作平台**
 | 
					基于DDD分层实现的web版 **linux(终端 文件 脚本 进程)、数据库(mysql postgres)、redis(单机 哨兵 集群)、mongo统一管理操作平台**
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 开发语言与主要框架
 | 
					### 开发语言与主要框架
 | 
				
			||||||
@@ -10,7 +28,7 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 交流及问题反馈加 QQ 群
 | 
					### 交流及问题反馈加 QQ 群
 | 
				
			||||||
<a target="_blank" href="https://qm.qq.com/cgi-bin/qm/qr?jump_from=webapi">119699946</a>
 | 
					<a target="_blank" href="https://qm.qq.com/cgi-bin/qm/qr?k=IdJSHW0jTMhmWFHBUS9a83wxtrxDDhFj&jump_from=webapi">119699946</a>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
### 系统相关资料
 | 
					### 系统相关资料
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -30,9 +30,9 @@ function buildWeb() {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
    echo_yellow "-------------------打包前端开始-------------------"
 | 
					    echo_yellow "-------------------打包前端开始-------------------"
 | 
				
			||||||
    yarn run build
 | 
					    yarn run build
 | 
				
			||||||
    if [ "${copy2Server}" == "1" ] ; then
 | 
					    if [ "${copy2Server}" == "2" ] ; then
 | 
				
			||||||
        echo_green '将打包后的静态文件拷贝至server/static'
 | 
					        echo_green '将打包后的静态文件拷贝至server/static/static'
 | 
				
			||||||
        rm -rf ${server_folder}/static && mkdir -p ${server_folder}/static && cp -r ${web_folder}/dist/* ${server_folder}/static
 | 
					        rm -rf ${server_folder}/static/static && mkdir -p ${server_folder}/static/static && cp -r ${web_folder}/dist/* ${server_folder}/static/static
 | 
				
			||||||
    fi
 | 
					    fi
 | 
				
			||||||
    echo_yellow ">>>>>>>>>>>>>>>>>>>打包前端结束<<<<<<<<<<<<<<<<<<<<\n"
 | 
					    echo_yellow ">>>>>>>>>>>>>>>>>>>打包前端结束<<<<<<<<<<<<<<<<<<<<\n"
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -44,6 +44,7 @@ function build() {
 | 
				
			|||||||
    toFolder=$1
 | 
					    toFolder=$1
 | 
				
			||||||
    os=$2
 | 
					    os=$2
 | 
				
			||||||
    arch=$3
 | 
					    arch=$3
 | 
				
			||||||
 | 
					    copyStatic=$4
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    echo_yellow "-------------------${os}-${arch}打包构建开始-------------------"
 | 
					    echo_yellow "-------------------${os}-${arch}打包构建开始-------------------"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -67,8 +68,10 @@ function build() {
 | 
				
			|||||||
    echo_green "移动二进制文件至'${toFolder}'"
 | 
					    echo_green "移动二进制文件至'${toFolder}'"
 | 
				
			||||||
    mv ${server_folder}/${execFileName} ${toFolder}
 | 
					    mv ${server_folder}/${execFileName} ${toFolder}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    echo_green "拷贝前端静态页面至'${toFolder}/static'"
 | 
					    if [ "${copy2Server}" == "1" ] ; then
 | 
				
			||||||
    mkdir -p ${toFolder}/static && cp -r ${web_folder}/dist/* ${toFolder}/static
 | 
					        echo_green "拷贝前端静态页面至'${toFolder}/static'"
 | 
				
			||||||
 | 
					        mkdir -p ${toFolder}/static && cp -r ${web_folder}/dist/* ${toFolder}/static
 | 
				
			||||||
 | 
					    fi
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    echo_green "拷贝脚本等资源文件[config.yml、mayfly-go.sql、readme.txt、startup.sh、shutdown.sh]"
 | 
					    echo_green "拷贝脚本等资源文件[config.yml、mayfly-go.sql、readme.txt、startup.sh、shutdown.sh]"
 | 
				
			||||||
    cp ${server_folder}/config.yml ${toFolder}
 | 
					    cp ${server_folder}/config.yml ${toFolder}
 | 
				
			||||||
@@ -81,15 +84,15 @@ function build() {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
function buildLinuxAmd64() {
 | 
					function buildLinuxAmd64() {
 | 
				
			||||||
    build "$1/mayfly-go-linux-amd64" "linux" "amd64"
 | 
					    build "$1/mayfly-go-linux-amd64" "linux" "amd64" $2
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
function buildLinuxArm64() {
 | 
					function buildLinuxArm64() {
 | 
				
			||||||
    build "$1/mayfly-go-linux-arm64" "linux" "arm64"
 | 
					    build "$1/mayfly-go-linux-arm64" "linux" "arm64" $2
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
function buildWindows() {
 | 
					function buildWindows() {
 | 
				
			||||||
    build "$1/mayfly-go-windows" "windows" "amd64"
 | 
					    build "$1/mayfly-go-windows" "windows" "amd64" $2
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
function runBuild() {
 | 
					function runBuild() {
 | 
				
			||||||
@@ -103,34 +106,30 @@ function runBuild() {
 | 
				
			|||||||
    cd ${toPath}
 | 
					    cd ${toPath}
 | 
				
			||||||
    toPath=`pwd`
 | 
					    toPath=`pwd`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    read -p "是否构建前端[0|其他->否 1->是 2->构建并拷贝至server/static]: " runBuildWeb
 | 
					    read -p "是否构建前端[0|其他->否 1->是 2->构建并拷贝至server/static/static]: " runBuildWeb
 | 
				
			||||||
    read -p "请选择构建版本[0|其他->全部 1->linux-amd64 2->linux-arm64 3->windows]: " buildType
 | 
					    read -p "请选择构建版本[0|其他->全部 1->linux-amd64 2->linux-arm64 3->windows]: " buildType
 | 
				
			||||||
    
 | 
					    
 | 
				
			||||||
    if [ "${runBuildWeb}" == "1" ];then
 | 
					    
 | 
				
			||||||
        buildWeb
 | 
					    if [ "${runBuildWeb}" == "1" ] || [ "${runBuildWeb}" == "2" ] ; then
 | 
				
			||||||
    fi
 | 
					        buildWeb ${runBuildWeb}
 | 
				
			||||||
    if [ "${runBuildWeb}" == "2" ];then
 | 
					 | 
				
			||||||
        buildWeb 1
 | 
					 | 
				
			||||||
    fi
 | 
					    fi
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    if [ "${buildType}" == "1" ];then
 | 
					    case ${buildType} in
 | 
				
			||||||
        buildLinuxAmd64 ${toPath}
 | 
					         "1")
 | 
				
			||||||
        exit;
 | 
					            buildLinuxAmd64 ${toPath} ${runBuildWeb}
 | 
				
			||||||
    fi
 | 
					        ;;
 | 
				
			||||||
 | 
					         "2")
 | 
				
			||||||
    if [ "${buildType}" == "2" ];then
 | 
					            buildLinuxArm64 ${toPath} ${runBuildWeb}
 | 
				
			||||||
        buildLinuxArm64 ${toPath}
 | 
					        ;;
 | 
				
			||||||
        exit;
 | 
					        "3")
 | 
				
			||||||
    fi
 | 
					            buildWindows ${toPath} ${runBuildWeb}
 | 
				
			||||||
 | 
					        ;;
 | 
				
			||||||
    if [ "${buildType}" == "3" ];then
 | 
					        *)
 | 
				
			||||||
        buildWindows ${toPath}
 | 
					            buildLinuxAmd64 ${toPath} ${runBuildWeb}
 | 
				
			||||||
        exit;
 | 
					            buildLinuxArm64 ${toPath} ${runBuildWeb}
 | 
				
			||||||
    fi
 | 
					            buildWindows ${toPath} ${runBuildWeb}
 | 
				
			||||||
 | 
					        ;;
 | 
				
			||||||
    buildLinuxAmd64 ${toPath}
 | 
					    esac
 | 
				
			||||||
    buildLinuxArm64 ${toPath}
 | 
					 | 
				
			||||||
    buildWindows ${toPath}
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
runBuild
 | 
					runBuild
 | 
				
			||||||
@@ -18,8 +18,7 @@
 | 
				
			|||||||
	</head>
 | 
						</head>
 | 
				
			||||||
	<body>
 | 
						<body>
 | 
				
			||||||
		<div id="app"></div>
 | 
							<div id="app"></div>
 | 
				
			||||||
        <script type="text/javascript" src="./config.js"></script>
 | 
					        <script type="application/javascript" src="./config.js"></script>
 | 
				
			||||||
		<script type="module" src="/src/main.ts"></script>
 | 
							<script type="module" src="/src/main.ts"></script>
 | 
				
			||||||
		<!-- <script type="text/javascript" src="https://api.map.baidu.com/api?v=3.0&ak=wsijQt8sLXrCW71YesmispvYHitfG9gv&s=1"></script> -->
 | 
					 | 
				
			||||||
	</body>
 | 
						</body>
 | 
				
			||||||
</html>
 | 
					</html>
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -7,21 +7,21 @@
 | 
				
			|||||||
    "lint-fix": "eslint --fix --ext .js --ext .jsx --ext .vue src/"
 | 
					    "lint-fix": "eslint --fix --ext .js --ext .jsx --ext .vue src/"
 | 
				
			||||||
  },
 | 
					  },
 | 
				
			||||||
  "dependencies": {
 | 
					  "dependencies": {
 | 
				
			||||||
    "@element-plus/icons-vue": "^2.0.6",
 | 
					    "@element-plus/icons-vue": "^2.0.9",
 | 
				
			||||||
    "axios": "^0.27.2",
 | 
					    "axios": "^0.27.2",
 | 
				
			||||||
    "codemirror": "^5.65.5",
 | 
					    "codemirror": "^5.65.5",
 | 
				
			||||||
    "countup.js": "^2.0.7",
 | 
					    "countup.js": "^2.0.7",
 | 
				
			||||||
    "cropperjs": "^1.5.11",
 | 
					    "cropperjs": "^1.5.11",
 | 
				
			||||||
    "echarts": "^5.3.3",
 | 
					    "echarts": "^5.3.3",
 | 
				
			||||||
    "element-plus": "^2.2.9",
 | 
					    "element-plus": "^2.2.14",
 | 
				
			||||||
    "jsencrypt": "^3.2.1",
 | 
					    "jsencrypt": "^3.2.1",
 | 
				
			||||||
    "jsoneditor": "^9.9.0",
 | 
					    "jsoneditor": "^9.9.0",
 | 
				
			||||||
    "lodash": "^4.17.21",
 | 
					    "lodash": "^4.17.21",
 | 
				
			||||||
    "mitt": "^3.0.0",
 | 
					    "mitt": "^3.0.0",
 | 
				
			||||||
    "nprogress": "^0.2.0",
 | 
					    "nprogress": "^0.2.0",
 | 
				
			||||||
    "screenfull": "^5.1.0",
 | 
					    "screenfull": "^6.0.2",
 | 
				
			||||||
    "sortablejs": "^1.13.0",
 | 
					    "sortablejs": "^1.13.0",
 | 
				
			||||||
    "sql-formatter": "^7.0.3",
 | 
					    "sql-formatter": "^9.2.0",
 | 
				
			||||||
    "vue": "^3.2.37",
 | 
					    "vue": "^3.2.37",
 | 
				
			||||||
    "vue-clipboard3": "^1.0.1",
 | 
					    "vue-clipboard3": "^1.0.1",
 | 
				
			||||||
    "vue-router": "^4.1.2",
 | 
					    "vue-router": "^4.1.2",
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -11,6 +11,8 @@ import { useStore } from '@/store/index.ts';
 | 
				
			|||||||
import { getLocal } from '@/common/utils/storage.ts';
 | 
					import { getLocal } from '@/common/utils/storage.ts';
 | 
				
			||||||
import LockScreen from '@/views/layout/lockScreen/index.vue';
 | 
					import LockScreen from '@/views/layout/lockScreen/index.vue';
 | 
				
			||||||
import Setings from '@/views/layout/navBars/breadcrumb/setings.vue';
 | 
					import Setings from '@/views/layout/navBars/breadcrumb/setings.vue';
 | 
				
			||||||
 | 
					import Watermark from '@/common/utils/wartermark.ts';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
    name: 'app',
 | 
					    name: 'app',
 | 
				
			||||||
    components: { LockScreen, Setings },
 | 
					    components: { LockScreen, Setings },
 | 
				
			||||||
@@ -57,6 +59,8 @@ export default defineComponent({
 | 
				
			|||||||
            () => route.path,
 | 
					            () => route.path,
 | 
				
			||||||
            () => {
 | 
					            () => {
 | 
				
			||||||
                nextTick(() => {
 | 
					                nextTick(() => {
 | 
				
			||||||
 | 
					                    // 路由变化更新水印
 | 
				
			||||||
 | 
					                    Watermark.use();
 | 
				
			||||||
                    document.title = `${route.meta.title} - ${getThemeConfig.value.globalTitle}` || getThemeConfig.value.globalTitle;
 | 
					                    document.title = `${route.meta.title} - ${getThemeConfig.value.globalTitle}` || getThemeConfig.value.globalTitle;
 | 
				
			||||||
                });
 | 
					                });
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,10 +1,11 @@
 | 
				
			|||||||
import request from './request'
 | 
					import request from './request'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default {
 | 
					export default {
 | 
				
			||||||
    login: (param: any) => request.request('POST', '/sys/accounts/login', param, null),
 | 
					    login: (param: any) => request.request('POST', '/sys/accounts/login', param),
 | 
				
			||||||
    changePwd: (param: any) => request.request('POST', '/sys/accounts/change-pwd', param, null),
 | 
					    changePwd: (param: any) => request.request('POST', '/sys/accounts/change-pwd', param),
 | 
				
			||||||
    getPublicKey: () => request.request('GET', '/common/public-key', null, null),
 | 
					    getPublicKey: () => request.request('GET', '/common/public-key'),
 | 
				
			||||||
    captcha: () => request.request('GET', '/sys/captcha', null, null),
 | 
					    getConfigValue: (param: any) => request.request('GET', '/sys/configs/value', param),
 | 
				
			||||||
    logout: (param: any) => request.request('POST', '/sys/accounts/logout/{token}', param, null),
 | 
					    captcha: () => request.request('GET', '/sys/captcha'),
 | 
				
			||||||
    getMenuRoute: (param: any) => request.request('Get', '/sys/resources/account', param, null)
 | 
					    logout: (param: any) => request.request('POST', '/sys/accounts/logout/{token}', param),
 | 
				
			||||||
 | 
					    getMenuRoute: (param: any) => request.request('Get', '/sys/resources/account', param)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -28,7 +28,6 @@ export async function RsaEncrypt(value: any) {
 | 
				
			|||||||
    if (encryptor != null) {
 | 
					    if (encryptor != null) {
 | 
				
			||||||
        return encryptor.encrypt(value)
 | 
					        return encryptor.encrypt(value)
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
    console.log(value)
 | 
					 | 
				
			||||||
    encryptor = new JSEncrypt()
 | 
					    encryptor = new JSEncrypt()
 | 
				
			||||||
    const publicKey = await getRsaPublicKey() as string;
 | 
					    const publicKey = await getRsaPublicKey() as string;
 | 
				
			||||||
    notBlank(publicKey, "获取公钥失败")
 | 
					    notBlank(publicKey, "获取公钥失败")
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										48
									
								
								mayfly_go_web/src/common/sysconfig.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										48
									
								
								mayfly_go_web/src/common/sysconfig.ts
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,48 @@
 | 
				
			|||||||
 | 
					import openApi from './openApi';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 登录是否使用验证码配置key
 | 
				
			||||||
 | 
					const UseLoginCaptchaConfigKey = "UseLoginCaptcha"
 | 
				
			||||||
 | 
					const UseWartermarkConfigKey = "UseWartermark"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * 获取系统配置值
 | 
				
			||||||
 | 
					 * 
 | 
				
			||||||
 | 
					 * @param key 配置key
 | 
				
			||||||
 | 
					 * @returns 配置值
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					export async function getConfigValue(key: string) : Promise<string> {
 | 
				
			||||||
 | 
					    return await openApi.getConfigValue({key}) as string
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * 获取bool类型系统配置值
 | 
				
			||||||
 | 
					 * 
 | 
				
			||||||
 | 
					 * @param key 配置key
 | 
				
			||||||
 | 
					 * @param defaultValue 默认值
 | 
				
			||||||
 | 
					 * @returns 是否为ture,1: true;其他: false
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					export async function getBoolConfigValue(key :string, defaultValue :boolean) : Promise<boolean> {
 | 
				
			||||||
 | 
					    const value = await getConfigValue(key)
 | 
				
			||||||
 | 
					    if (!value) {
 | 
				
			||||||
 | 
					        return defaultValue;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    return value == "1";
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * 是否使用登录验证码
 | 
				
			||||||
 | 
					 * 
 | 
				
			||||||
 | 
					 * @returns 
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					export async function useLoginCaptcha() : Promise<boolean> {
 | 
				
			||||||
 | 
					    return await getBoolConfigValue(UseLoginCaptchaConfigKey, true)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/**
 | 
				
			||||||
 | 
					 * 是否启用水印
 | 
				
			||||||
 | 
					 * 
 | 
				
			||||||
 | 
					 * @returns 
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					 export async function useWartermark() : Promise<boolean> {
 | 
				
			||||||
 | 
					    return await getBoolConfigValue(UseWartermarkConfigKey, true)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -35,3 +35,22 @@ export function removeSession(key: string) {
 | 
				
			|||||||
export function clearSession() {
 | 
					export function clearSession() {
 | 
				
			||||||
    window.sessionStorage.clear();
 | 
					    window.sessionStorage.clear();
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export function getUserInfo4Session() {
 | 
				
			||||||
 | 
					    return getSession("userInfo")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export function setUserInfo2Session(userinfo: any) {
 | 
				
			||||||
 | 
					    setSession("userInfo", userinfo)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取是否开启水印
 | 
				
			||||||
 | 
					export function getUseWatermark4Session() {
 | 
				
			||||||
 | 
					    return getSession("useWatermark")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export function setUseWatermark2Session(useWatermark: boolean) {
 | 
				
			||||||
 | 
					    setSession("useWatermark", useWatermark)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -1,21 +1,26 @@
 | 
				
			|||||||
 | 
					import { getUseWatermark4Session, getUserInfo4Session } from '@/common/utils/storage.ts';
 | 
				
			||||||
 | 
					import { dateFormat } from '@/common/utils/date.ts'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 页面添加水印效果
 | 
					// 页面添加水印效果
 | 
				
			||||||
const setWatermark = (str: any) => {
 | 
					const setWatermark = (str: any) => {
 | 
				
			||||||
    const id = '1.23452384164.123412416';
 | 
					    const id = '1.23452384164.123412416';
 | 
				
			||||||
    if (document.getElementById(id) !== null) document.body.removeChild(document.getElementById(id) as any);
 | 
					    if (document.getElementById(id) !== null) document.body.removeChild(document.getElementById(id) as any);
 | 
				
			||||||
    const can = document.createElement('canvas');
 | 
					    const can = document.createElement('canvas');
 | 
				
			||||||
    can.width = 250;
 | 
					    can.width = 400;
 | 
				
			||||||
    can.height = 180;
 | 
					    can.height = 250;
 | 
				
			||||||
    const cans: any = can.getContext('2d');
 | 
					    const cans: any = can.getContext('2d');
 | 
				
			||||||
    cans.rotate((-20 * Math.PI) / 180);
 | 
					    cans.rotate((-20 * Math.PI) / 180);
 | 
				
			||||||
    cans.font = '12px Vedana';
 | 
					    cans.font = '14px Vedana';
 | 
				
			||||||
    cans.fillStyle = 'rgba(200, 200, 200, 0.30)';
 | 
					    cans.fillStyle = 'rgba(200, 200, 200, 0.35)';
 | 
				
			||||||
    cans.textAlign = 'center';
 | 
					    cans.textAlign = 'left';
 | 
				
			||||||
    cans.textBaseline = 'Middle';
 | 
					    cans.textBaseline = 'Middle';
 | 
				
			||||||
    cans.fillText(str, can.width / 10, can.height / 2);
 | 
					    // cans.fillText('mayfly go', can.width / 4, can.height )
 | 
				
			||||||
 | 
					    cans.fillText(str, can.width / 8, can.height / 2);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    const div = document.createElement('div');
 | 
					    const div = document.createElement('div');
 | 
				
			||||||
    div.id = id;
 | 
					    div.id = id;
 | 
				
			||||||
    div.style.pointerEvents = 'none';
 | 
					    div.style.pointerEvents = 'none';
 | 
				
			||||||
    div.style.top = '35px';
 | 
					    div.style.top = '30px';
 | 
				
			||||||
    div.style.left = '0px';
 | 
					    div.style.left = '0px';
 | 
				
			||||||
    div.style.position = 'fixed';
 | 
					    div.style.position = 'fixed';
 | 
				
			||||||
    div.style.zIndex = '10000000';
 | 
					    div.style.zIndex = '10000000';
 | 
				
			||||||
@@ -26,16 +31,34 @@ const setWatermark = (str: any) => {
 | 
				
			|||||||
    return id;
 | 
					    return id;
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					function set(str: any) {
 | 
				
			||||||
 | 
					    let id = setWatermark(str);
 | 
				
			||||||
 | 
					    if (document.getElementById(id) === null) id = setWatermark(str);
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					function del() {
 | 
				
			||||||
 | 
					    let id = '1.23452384164.123412416';
 | 
				
			||||||
 | 
					    if (document.getElementById(id) !== null) document.body.removeChild(document.getElementById(id) as any);
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const watermark = {
 | 
					const watermark = {
 | 
				
			||||||
 | 
					    use: () => {
 | 
				
			||||||
 | 
					        setTimeout(() => {
 | 
				
			||||||
 | 
					            const userinfo = getUserInfo4Session()
 | 
				
			||||||
 | 
					            if (userinfo && getUseWatermark4Session()) {
 | 
				
			||||||
 | 
					                set(`${userinfo.username} ${dateFormat('yyyy-MM-dd HH:mm:ss', new Date())}`)
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                del();
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        }, 1500)
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
    // 设置水印
 | 
					    // 设置水印
 | 
				
			||||||
    set: (str: any) => {
 | 
					    set: (str: any) => {
 | 
				
			||||||
        let id = setWatermark(str);
 | 
					        set(str)
 | 
				
			||||||
        if (document.getElementById(id) === null) id = setWatermark(str);
 | 
					 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
    // 删除水印
 | 
					    // 删除水印
 | 
				
			||||||
    del: () => {
 | 
					    del: () => {
 | 
				
			||||||
        let id = '1.23452384164.123412416';
 | 
					        del();
 | 
				
			||||||
        if (document.getElementById(id) !== null) document.body.removeChild(document.getElementById(id) as any);
 | 
					 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
};
 | 
					};
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,6 +2,7 @@ import RouterParent from '@/views/layout/routerView/parent.vue';
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
export const imports = {
 | 
					export const imports = {
 | 
				
			||||||
    'RouterParent': RouterParent,
 | 
					    'RouterParent': RouterParent,
 | 
				
			||||||
 | 
					    
 | 
				
			||||||
    "Home": () => import('@/views/home/index.vue'),
 | 
					    "Home": () => import('@/views/home/index.vue'),
 | 
				
			||||||
    'Personal': () => import('@/views/personal/index.vue'),
 | 
					    'Personal': () => import('@/views/personal/index.vue'),
 | 
				
			||||||
    // machine
 | 
					    // machine
 | 
				
			||||||
@@ -11,6 +12,8 @@ export const imports = {
 | 
				
			|||||||
    "RoleList": () => import('@/views/system/role'),
 | 
					    "RoleList": () => import('@/views/system/role'),
 | 
				
			||||||
    "AccountList": () => import('@/views/system/account'),
 | 
					    "AccountList": () => import('@/views/system/account'),
 | 
				
			||||||
    "SyslogList": () => import('@/views/system/syslog/SyslogList.vue'),
 | 
					    "SyslogList": () => import('@/views/system/syslog/SyslogList.vue'),
 | 
				
			||||||
 | 
					    "ConfigList": () => import('@/views/system/config/ConfigList.vue'),
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // project
 | 
					    // project
 | 
				
			||||||
    "ProjectList": () => import('@/views/ops/project/ProjectList.vue'),
 | 
					    "ProjectList": () => import('@/views/ops/project/ProjectList.vue'),
 | 
				
			||||||
    // db
 | 
					    // db
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -52,6 +52,7 @@ export interface ThemeConfigState {
 | 
				
			|||||||
        terminalBackground: string;
 | 
					        terminalBackground: string;
 | 
				
			||||||
        terminalCursor: string;
 | 
					        terminalCursor: string;
 | 
				
			||||||
        terminalFontSize: number;
 | 
					        terminalFontSize: number;
 | 
				
			||||||
 | 
					        terminalFontWeight: string;
 | 
				
			||||||
    };
 | 
					    };
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -113,6 +113,7 @@ const themeConfigModule: Module<ThemeConfigState, RootStateTypes> = {
 | 
				
			|||||||
            // ssh终端cursor色
 | 
					            // ssh终端cursor色
 | 
				
			||||||
            terminalCursor: '#268F81',
 | 
					            terminalCursor: '#268F81',
 | 
				
			||||||
            terminalFontSize: 15,
 | 
					            terminalFontSize: 15,
 | 
				
			||||||
 | 
					            terminalFontWeight: 'normal',
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            /* 后端控制路由
 | 
					            /* 后端控制路由
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -947,12 +947,6 @@
 | 
				
			|||||||
.el-select-dropdown .el-scrollbar__wrap {
 | 
					.el-select-dropdown .el-scrollbar__wrap {
 | 
				
			||||||
	overflow-x: scroll !important;
 | 
						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 抽屉
 | 
					/* Drawer 抽屉
 | 
				
			||||||
------------------------------- */
 | 
					------------------------------- */
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -40,7 +40,7 @@
 | 
				
			|||||||
                        </el-input-number>
 | 
					                        </el-input-number>
 | 
				
			||||||
                    </div>
 | 
					                    </div>
 | 
				
			||||||
                </div>
 | 
					                </div>
 | 
				
			||||||
                <!-- <div class="layout-breadcrumb-seting-bar-flex mt15">
 | 
					                <div class="layout-breadcrumb-seting-bar-flex mt15">
 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-label">字体粗细</div>
 | 
					                    <div class="layout-breadcrumb-seting-bar-flex-label">字体粗细</div>
 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-value">
 | 
					                    <div class="layout-breadcrumb-seting-bar-flex-value">
 | 
				
			||||||
                        <el-select @change="setLocalThemeConfig" v-model="getThemeConfig.terminalFontWeight" size="small" style="width: 90px">
 | 
					                        <el-select @change="setLocalThemeConfig" v-model="getThemeConfig.terminalFontWeight" size="small" style="width: 90px">
 | 
				
			||||||
@@ -48,7 +48,7 @@
 | 
				
			|||||||
                            <el-option label="bold" value="bold"> </el-option>
 | 
					                            <el-option label="bold" value="bold"> </el-option>
 | 
				
			||||||
                        </el-select>
 | 
					                        </el-select>
 | 
				
			||||||
                    </div>
 | 
					                    </div>
 | 
				
			||||||
                </div> -->
 | 
					                </div>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <!-- 全局主题 -->
 | 
					                <!-- 全局主题 -->
 | 
				
			||||||
                <el-divider content-position="left">全局主题</el-divider>
 | 
					                <el-divider content-position="left">全局主题</el-divider>
 | 
				
			||||||
@@ -273,23 +273,6 @@
 | 
				
			|||||||
                        <el-switch v-model="getThemeConfig.isInvert" @change="onAddFilterChange('invert')"></el-switch>
 | 
					                        <el-switch v-model="getThemeConfig.isInvert" @change="onAddFilterChange('invert')"></el-switch>
 | 
				
			||||||
                    </div>
 | 
					                    </div>
 | 
				
			||||||
                </div>
 | 
					                </div>
 | 
				
			||||||
                <div class="layout-breadcrumb-seting-bar-flex mt15">
 | 
					 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-label">开启水印</div>
 | 
					 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-value">
 | 
					 | 
				
			||||||
                        <el-switch v-model="getThemeConfig.isWartermark" @change="onWartermarkChange"></el-switch>
 | 
					 | 
				
			||||||
                    </div>
 | 
					 | 
				
			||||||
                </div>
 | 
					 | 
				
			||||||
                <div class="layout-breadcrumb-seting-bar-flex mt14">
 | 
					 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-label">水印文案</div>
 | 
					 | 
				
			||||||
                    <div class="layout-breadcrumb-seting-bar-flex-value">
 | 
					 | 
				
			||||||
                        <el-input
 | 
					 | 
				
			||||||
                            v-model="getThemeConfig.wartermarkText"
 | 
					 | 
				
			||||||
                            size="small"
 | 
					 | 
				
			||||||
                            style="width: 90px"
 | 
					 | 
				
			||||||
                            @input="onWartermarkTextInput($event)"
 | 
					 | 
				
			||||||
                        ></el-input>
 | 
					 | 
				
			||||||
                    </div>
 | 
					 | 
				
			||||||
                </div>
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <!-- 其它设置 -->
 | 
					                <!-- 其它设置 -->
 | 
				
			||||||
                <el-divider content-position="left">其他设置</el-divider>
 | 
					                <el-divider content-position="left">其他设置</el-divider>
 | 
				
			||||||
@@ -440,8 +423,6 @@ import { ElMessage } from 'element-plus';
 | 
				
			|||||||
import ClipboardJS from 'clipboard';
 | 
					import ClipboardJS from 'clipboard';
 | 
				
			||||||
import { useStore } from '@/store/index.ts';
 | 
					import { useStore } from '@/store/index.ts';
 | 
				
			||||||
import { getLightColor } from '@/common/utils/theme.ts';
 | 
					import { getLightColor } from '@/common/utils/theme.ts';
 | 
				
			||||||
import Watermark from '@/common/utils/wartermark.ts';
 | 
					 | 
				
			||||||
import { verifyAndSpace } from '@/common/utils/toolsValidate.ts';
 | 
					 | 
				
			||||||
import { setLocal, getLocal, removeLocal } from '@/common/utils/storage.ts';
 | 
					import { setLocal, getLocal, removeLocal } from '@/common/utils/storage.ts';
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
    name: 'layoutBreadcrumbSeting',
 | 
					    name: 'layoutBreadcrumbSeting',
 | 
				
			||||||
@@ -572,18 +553,6 @@ export default defineComponent({
 | 
				
			|||||||
            setLocalThemeConfig();
 | 
					            setLocalThemeConfig();
 | 
				
			||||||
            setLocal('appFilterStyle', appEle.style.cssText);
 | 
					            setLocal('appFilterStyle', appEle.style.cssText);
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
        // 4、界面显示 --> 开启水印
 | 
					 | 
				
			||||||
        const onWartermarkChange = () => {
 | 
					 | 
				
			||||||
            getThemeConfig.value.isWartermark ? Watermark.set(getThemeConfig.value.wartermarkText) : Watermark.del();
 | 
					 | 
				
			||||||
            setLocalThemeConfig();
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
        // 4、界面显示 --> 水印文案
 | 
					 | 
				
			||||||
        const onWartermarkTextInput = (val: string) => {
 | 
					 | 
				
			||||||
            getThemeConfig.value.wartermarkText = verifyAndSpace(val);
 | 
					 | 
				
			||||||
            if (getThemeConfig.value.wartermarkText === '') return false;
 | 
					 | 
				
			||||||
            if (getThemeConfig.value.isWartermark) Watermark.set(getThemeConfig.value.wartermarkText);
 | 
					 | 
				
			||||||
            setLocalThemeConfig();
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
        // 5、布局切换
 | 
					        // 5、布局切换
 | 
				
			||||||
        const onSetLayout = (layout: string) => {
 | 
					        const onSetLayout = (layout: string) => {
 | 
				
			||||||
            setLocal('oldLayout', layout);
 | 
					            setLocal('oldLayout', layout);
 | 
				
			||||||
@@ -735,8 +704,6 @@ export default defineComponent({
 | 
				
			|||||||
                            const appEl: any = document.querySelector('#app');
 | 
					                            const appEl: any = document.querySelector('#app');
 | 
				
			||||||
                            appEl.style.cssText = getLocal('appFilterStyle');
 | 
					                            appEl.style.cssText = getLocal('appFilterStyle');
 | 
				
			||||||
                        }
 | 
					                        }
 | 
				
			||||||
                        // 开启水印
 | 
					 | 
				
			||||||
                        onWartermarkChange();
 | 
					 | 
				
			||||||
                        // // 语言国际化
 | 
					                        // // 语言国际化
 | 
				
			||||||
                        // if (getLocal('themeConfig')) proxy.$i18n.locale = getLocal('themeConfig').globalI18n;
 | 
					                        // if (getLocal('themeConfig')) proxy.$i18n.locale = getLocal('themeConfig').globalI18n;
 | 
				
			||||||
                    }, 1100);
 | 
					                    }, 1100);
 | 
				
			||||||
@@ -762,8 +729,6 @@ export default defineComponent({
 | 
				
			|||||||
            getThemeConfig,
 | 
					            getThemeConfig,
 | 
				
			||||||
            onDrawerClose,
 | 
					            onDrawerClose,
 | 
				
			||||||
            onAddFilterChange,
 | 
					            onAddFilterChange,
 | 
				
			||||||
            onWartermarkChange,
 | 
					 | 
				
			||||||
            onWartermarkTextInput,
 | 
					 | 
				
			||||||
            onSetLayout,
 | 
					            onSetLayout,
 | 
				
			||||||
            setLocalThemeConfig,
 | 
					            setLocalThemeConfig,
 | 
				
			||||||
            onClassicSplitMenuChange,
 | 
					            onClassicSplitMenuChange,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -14,11 +14,11 @@
 | 
				
			|||||||
                </el-dropdown-menu>
 | 
					                </el-dropdown-menu>
 | 
				
			||||||
            </template>
 | 
					            </template>
 | 
				
			||||||
        </el-dropdown>
 | 
					        </el-dropdown>
 | 
				
			||||||
        <div class="layout-navbars-breadcrumb-user-icon" @click="onSearchClick">
 | 
					        <!-- <div class="layout-navbars-breadcrumb-user-icon" @click="onSearchClick">
 | 
				
			||||||
            <el-icon title="菜单搜索">
 | 
					            <el-icon title="菜单搜索">
 | 
				
			||||||
                <search />
 | 
					                <search />
 | 
				
			||||||
            </el-icon>
 | 
					            </el-icon>
 | 
				
			||||||
        </div>
 | 
					        </div> -->
 | 
				
			||||||
        <div class="layout-navbars-breadcrumb-user-icon" @click="onLayoutSetingClick">
 | 
					        <div class="layout-navbars-breadcrumb-user-icon" @click="onLayoutSetingClick">
 | 
				
			||||||
            <el-icon title="布局设置">
 | 
					            <el-icon title="布局设置">
 | 
				
			||||||
                <setting />
 | 
					                <setting />
 | 
				
			||||||
@@ -28,7 +28,7 @@
 | 
				
			|||||||
            <el-popover
 | 
					            <el-popover
 | 
				
			||||||
                placement="bottom"
 | 
					                placement="bottom"
 | 
				
			||||||
                trigger="click"
 | 
					                trigger="click"
 | 
				
			||||||
                v-model:visible="isShowUserNewsPopover"
 | 
					                :visible="isShowUserNewsPopover"
 | 
				
			||||||
                :width="300"
 | 
					                :width="300"
 | 
				
			||||||
                popper-class="el-popover-pupop-user-news"
 | 
					                popper-class="el-popover-pupop-user-news"
 | 
				
			||||||
            >
 | 
					            >
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@
 | 
				
			|||||||
                <el-input type="password" placeholder="请输入密码" prefix-icon="lock" v-model="loginForm.password" autocomplete="off" show-password>
 | 
					                <el-input type="password" placeholder="请输入密码" prefix-icon="lock" v-model="loginForm.password" autocomplete="off" show-password>
 | 
				
			||||||
                </el-input>
 | 
					                </el-input>
 | 
				
			||||||
            </el-form-item>
 | 
					            </el-form-item>
 | 
				
			||||||
            <el-form-item prop="captcha">
 | 
					            <el-form-item v-if="useLoginCaptcha" prop="captcha">
 | 
				
			||||||
                <el-row :gutter="15">
 | 
					                <el-row :gutter="15">
 | 
				
			||||||
                    <el-col :span="16">
 | 
					                    <el-col :span="16">
 | 
				
			||||||
                        <el-input
 | 
					                        <el-input
 | 
				
			||||||
@@ -78,10 +78,11 @@ import { useRoute, useRouter } from 'vue-router';
 | 
				
			|||||||
import { ElMessage } from 'element-plus';
 | 
					import { ElMessage } from 'element-plus';
 | 
				
			||||||
import { initBackEndControlRoutesFun } from '@/router/index.ts';
 | 
					import { initBackEndControlRoutesFun } from '@/router/index.ts';
 | 
				
			||||||
import { useStore } from '@/store/index.ts';
 | 
					import { useStore } from '@/store/index.ts';
 | 
				
			||||||
import { setSession } from '@/common/utils/storage.ts';
 | 
					import { setSession, setUserInfo2Session, setUseWatermark2Session } from '@/common/utils/storage.ts';
 | 
				
			||||||
import { formatAxis } from '@/common/utils/formatTime.ts';
 | 
					import { formatAxis } from '@/common/utils/formatTime.ts';
 | 
				
			||||||
import openApi from '@/common/openApi';
 | 
					import openApi from '@/common/openApi';
 | 
				
			||||||
import { RsaEncrypt } from '@/common/rsa';
 | 
					import { RsaEncrypt } from '@/common/rsa';
 | 
				
			||||||
 | 
					import { useLoginCaptcha, useWartermark } from '@/common/sysconfig';
 | 
				
			||||||
import { letterAvatar } from '@/common/utils/string';
 | 
					import { letterAvatar } from '@/common/utils/string';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
@@ -94,6 +95,7 @@ export default defineComponent({
 | 
				
			|||||||
        const changePwdFormRef: any = ref(null);
 | 
					        const changePwdFormRef: any = ref(null);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const state = reactive({
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            useLoginCaptcha: true,
 | 
				
			||||||
            captchaImage: '',
 | 
					            captchaImage: '',
 | 
				
			||||||
            loginForm: {
 | 
					            loginForm: {
 | 
				
			||||||
                username: '',
 | 
					                username: '',
 | 
				
			||||||
@@ -130,11 +132,17 @@ export default defineComponent({
 | 
				
			|||||||
            },
 | 
					            },
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        onMounted(() => {
 | 
					        onMounted(async () => {
 | 
				
			||||||
 | 
					            // 移除公钥, 方便后续重新获取
 | 
				
			||||||
 | 
					            sessionStorage.removeItem('RsaPublicKey');
 | 
				
			||||||
 | 
					            state.useLoginCaptcha = await useLoginCaptcha();
 | 
				
			||||||
            getCaptcha();
 | 
					            getCaptcha();
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const getCaptcha = async () => {
 | 
					        const getCaptcha = async () => {
 | 
				
			||||||
 | 
					            if (!state.useLoginCaptcha) {
 | 
				
			||||||
 | 
					                return;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
            let res: any = await openApi.captcha();
 | 
					            let res: any = await openApi.captcha();
 | 
				
			||||||
            state.captchaImage = res.base64Captcha;
 | 
					            state.captchaImage = res.base64Captcha;
 | 
				
			||||||
            state.loginForm.cid = res.cid;
 | 
					            state.loginForm.cid = res.cid;
 | 
				
			||||||
@@ -196,7 +204,7 @@ export default defineComponent({
 | 
				
			|||||||
            };
 | 
					            };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // 存储用户信息到浏览器缓存
 | 
					            // 存储用户信息到浏览器缓存
 | 
				
			||||||
            setSession('userInfo', userInfos);
 | 
					            setUserInfo2Session(userInfos);
 | 
				
			||||||
            // 1、请注意执行顺序(存储用户信息到vuex)
 | 
					            // 1、请注意执行顺序(存储用户信息到vuex)
 | 
				
			||||||
            store.dispatch('userInfos/setUserInfos', userInfos);
 | 
					            store.dispatch('userInfos/setUserInfos', userInfos);
 | 
				
			||||||
            if (!store.state.themeConfig.themeConfig.isRequestRoutes) {
 | 
					            if (!store.state.themeConfig.themeConfig.isRequestRoutes) {
 | 
				
			||||||
@@ -222,10 +230,13 @@ export default defineComponent({
 | 
				
			|||||||
            // 如果是复制粘贴的路径,非首页/登录页,那么登录成功后重定向到对应的路径中
 | 
					            // 如果是复制粘贴的路径,非首页/登录页,那么登录成功后重定向到对应的路径中
 | 
				
			||||||
            route.query?.redirect ? router.push(route.query.redirect as string) : router.push('/');
 | 
					            route.query?.redirect ? router.push(route.query.redirect as string) : router.push('/');
 | 
				
			||||||
            // 登录成功提示
 | 
					            // 登录成功提示
 | 
				
			||||||
            setTimeout(() => {
 | 
					            setTimeout(async () => {
 | 
				
			||||||
                // 关闭 loading
 | 
					                // 关闭 loading
 | 
				
			||||||
                state.loading.signIn = true;
 | 
					                state.loading.signIn = true;
 | 
				
			||||||
                ElMessage.success(`${currentTimeInfo},欢迎回来!`);
 | 
					                ElMessage.success(`${currentTimeInfo},欢迎回来!`);
 | 
				
			||||||
 | 
					                if (await useWartermark()) {
 | 
				
			||||||
 | 
					                    setUseWatermark2Session(true);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
            }, 300);
 | 
					            }, 300);
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -41,34 +41,39 @@
 | 
				
			|||||||
                        v-model.trim="form.password"
 | 
					                        v-model.trim="form.password"
 | 
				
			||||||
                        placeholder="请输入密码,修改操作可不填"
 | 
					                        placeholder="请输入密码,修改操作可不填"
 | 
				
			||||||
                        autocomplete="new-password"
 | 
					                        autocomplete="new-password"
 | 
				
			||||||
                    ></el-input>
 | 
					                    >
 | 
				
			||||||
 | 
					                        <template v-if="form.id && form.id != 0" #suffix>
 | 
				
			||||||
 | 
					                            <el-popover @hide="pwd = ''" placement="right" title="原密码" :width="200" trigger="click" :content="pwd">
 | 
				
			||||||
 | 
					                                <template #reference>
 | 
				
			||||||
 | 
					                                    <el-link @click="getDbPwd" :underline="false" type="primary" class="mr5">原密码</el-link>
 | 
				
			||||||
 | 
					                                </template>
 | 
				
			||||||
 | 
					                            </el-popover>
 | 
				
			||||||
 | 
					                        </template>
 | 
				
			||||||
 | 
					                    </el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item prop="params" label="连接参数:">
 | 
					                <el-form-item prop="params" label="连接参数:">
 | 
				
			||||||
                    <el-input v-model="form.params" placeholder="其他连接参数,形如: key1=value1&key2=value2"></el-input>
 | 
					                    <el-input v-model="form.params" placeholder="其他连接参数,形如: key1=value1&key2=value2"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item prop="database" label="数据库名:" required>
 | 
					                <el-form-item prop="database" label="数据库名:" required>
 | 
				
			||||||
                    <el-tag
 | 
					                    <el-col :span="19">
 | 
				
			||||||
                        v-for="db in databaseList"
 | 
					                        <el-select
 | 
				
			||||||
                        :key="db"
 | 
					                            @change="changeDatabase"
 | 
				
			||||||
                        class="ml5 mt5"
 | 
					                            v-model="databaseList"
 | 
				
			||||||
                        type="success"
 | 
					                            multiple
 | 
				
			||||||
                        effect="plain"
 | 
					                            collapse-tags
 | 
				
			||||||
                        closable
 | 
					                            collapse-tags-tooltip
 | 
				
			||||||
                        :disable-transitions="false"
 | 
					                            filterable
 | 
				
			||||||
                        @close="handleClose(db)"
 | 
					                            allow-create
 | 
				
			||||||
                    >
 | 
					                            placeholder="请确保数据库实例信息填写完整后获取库名"
 | 
				
			||||||
                        {{ db }}
 | 
					                            style="width: 100%"
 | 
				
			||||||
                    </el-tag>
 | 
					                        >
 | 
				
			||||||
                    <el-input
 | 
					                            <el-option v-for="db in allDatabases" :key="db" :label="db" :value="db" />
 | 
				
			||||||
                        v-if="inputDbVisible"
 | 
					                        </el-select>
 | 
				
			||||||
                        ref="InputDbRef"
 | 
					                    </el-col>
 | 
				
			||||||
                        v-model="inputDbValue"
 | 
					                    <el-col style="text-align: center" :span="1"><el-divider direction="vertical" border-style="dashed" /></el-col>
 | 
				
			||||||
                        style="width: 120px; margin-left: 5px; margin-top: 5px"
 | 
					                    <el-col :span="4">
 | 
				
			||||||
                        size="small"
 | 
					                        <el-link @click="getAllDatabase" :underline="false" type="success">获取库名</el-link>
 | 
				
			||||||
                        @keyup.enter="handleInputDbConfirm"
 | 
					                    </el-col>
 | 
				
			||||||
                        @blur="handleInputDbConfirm"
 | 
					 | 
				
			||||||
                    />
 | 
					 | 
				
			||||||
                    <el-button v-else class="ml5 mt5" size="small" @click="showInputDb"> + 添加数据库 </el-button>
 | 
					 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-form-item prop="enableSshTunnel" label="SSH隧道:">
 | 
					                <el-form-item prop="enableSshTunnel" label="SSH隧道:">
 | 
				
			||||||
@@ -101,12 +106,11 @@
 | 
				
			|||||||
</template>
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
<script lang="ts">
 | 
					<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 { dbApi } from './api';
 | 
				
			||||||
import { projectApi } from '../project/api.ts';
 | 
					import { projectApi } from '../project/api.ts';
 | 
				
			||||||
import { machineApi } from '../machine/api.ts';
 | 
					import { machineApi } from '../machine/api.ts';
 | 
				
			||||||
import { ElMessage } from 'element-plus';
 | 
					import { ElMessage } from 'element-plus';
 | 
				
			||||||
import type { ElInput } from 'element-plus';
 | 
					 | 
				
			||||||
import { notBlank } from '@/common/assert';
 | 
					import { notBlank } from '@/common/assert';
 | 
				
			||||||
import { RsaEncrypt } from '@/common/rsa';
 | 
					import { RsaEncrypt } from '@/common/rsa';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -128,16 +132,14 @@ export default defineComponent({
 | 
				
			|||||||
    },
 | 
					    },
 | 
				
			||||||
    setup(props: any, { emit }) {
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
        const dbForm: any = ref(null);
 | 
					        const dbForm: any = ref(null);
 | 
				
			||||||
        const InputDbRef = ref<InstanceType<typeof ElInput>>();
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const state = reactive({
 | 
					        const state = reactive({
 | 
				
			||||||
            dialogVisible: false,
 | 
					            dialogVisible: false,
 | 
				
			||||||
            projects: [],
 | 
					            projects: [],
 | 
				
			||||||
            envs: [],
 | 
					            envs: [],
 | 
				
			||||||
 | 
					            allDatabases: [] as any,
 | 
				
			||||||
            databaseList: [] as any,
 | 
					            databaseList: [] as any,
 | 
				
			||||||
            sshTunnelMachineList: [],
 | 
					            sshTunnelMachineList: [],
 | 
				
			||||||
            inputDbVisible: false,
 | 
					 | 
				
			||||||
            inputDbValue: '',
 | 
					 | 
				
			||||||
            form: {
 | 
					            form: {
 | 
				
			||||||
                id: null,
 | 
					                id: null,
 | 
				
			||||||
                name: null,
 | 
					                name: null,
 | 
				
			||||||
@@ -153,6 +155,8 @@ export default defineComponent({
 | 
				
			|||||||
                enableSshTunnel: null,
 | 
					                enableSshTunnel: null,
 | 
				
			||||||
                sshTunnelMachineId: null,
 | 
					                sshTunnelMachineId: null,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
 | 
					            // 原密码
 | 
				
			||||||
 | 
					            pwd: '',
 | 
				
			||||||
            btnLoading: false,
 | 
					            btnLoading: false,
 | 
				
			||||||
            rules: {
 | 
					            rules: {
 | 
				
			||||||
                projectId: [
 | 
					                projectId: [
 | 
				
			||||||
@@ -226,27 +230,6 @@ export default defineComponent({
 | 
				
			|||||||
            getSshTunnelMachines();
 | 
					            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 = '';
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        /**
 | 
					        /**
 | 
				
			||||||
         * 改变表单中的数据库字段,方便表单错误提示。如全部删光,可提示请添加数据库
 | 
					         * 改变表单中的数据库字段,方便表单错误提示。如全部删光,可提示请添加数据库
 | 
				
			||||||
         */
 | 
					         */
 | 
				
			||||||
@@ -285,6 +268,17 @@ export default defineComponent({
 | 
				
			|||||||
            }
 | 
					            }
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const getAllDatabase = async () => {
 | 
				
			||||||
 | 
					            const reqForm = { ...state.form };
 | 
				
			||||||
 | 
					            reqForm.password = await RsaEncrypt(reqForm.password);
 | 
				
			||||||
 | 
					            state.allDatabases = await dbApi.getAllDatabase.request(reqForm);
 | 
				
			||||||
 | 
					            ElMessage.success('获取成功, 请选择需要管理操作的数据库')
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const getDbPwd = async () => {
 | 
				
			||||||
 | 
					            state.pwd = await dbApi.getDbPwd.request({ id: state.form.id });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const btnOk = async () => {
 | 
					        const btnOk = async () => {
 | 
				
			||||||
            if (!state.form.id) {
 | 
					            if (!state.form.id) {
 | 
				
			||||||
                notBlank(state.form.password, '新增操作,密码不可为空');
 | 
					                notBlank(state.form.password, '新增操作,密码不可为空');
 | 
				
			||||||
@@ -293,7 +287,6 @@ export default defineComponent({
 | 
				
			|||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    const reqForm = { ...state.form };
 | 
					                    const reqForm = { ...state.form };
 | 
				
			||||||
                    reqForm.password = await RsaEncrypt(reqForm.password);
 | 
					                    reqForm.password = await RsaEncrypt(reqForm.password);
 | 
				
			||||||
                    // reqForm.ssh_pass = await RsaEncrypt(reqForm.ssh_pass);
 | 
					 | 
				
			||||||
                    dbApi.saveDb.request(reqForm).then(() => {
 | 
					                    dbApi.saveDb.request(reqForm).then(() => {
 | 
				
			||||||
                        ElMessage.success('保存成功');
 | 
					                        ElMessage.success('保存成功');
 | 
				
			||||||
                        emit('val-change', state.form);
 | 
					                        emit('val-change', state.form);
 | 
				
			||||||
@@ -312,9 +305,8 @@ export default defineComponent({
 | 
				
			|||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const resetInputDb = () => {
 | 
					        const resetInputDb = () => {
 | 
				
			||||||
            state.inputDbVisible = false;
 | 
					 | 
				
			||||||
            state.databaseList = [];
 | 
					            state.databaseList = [];
 | 
				
			||||||
            state.inputDbValue = '';
 | 
					            state.allDatabases = [];
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const cancel = () => {
 | 
					        const cancel = () => {
 | 
				
			||||||
@@ -328,10 +320,9 @@ export default defineComponent({
 | 
				
			|||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            dbForm,
 | 
					            dbForm,
 | 
				
			||||||
            InputDbRef,
 | 
					            getAllDatabase,
 | 
				
			||||||
            handleClose,
 | 
					            getDbPwd,
 | 
				
			||||||
            showInputDb,
 | 
					            changeDatabase,
 | 
				
			||||||
            handleInputDbConfirm,
 | 
					 | 
				
			||||||
            getSshTunnelMachines,
 | 
					            getSshTunnelMachines,
 | 
				
			||||||
            changeProject,
 | 
					            changeProject,
 | 
				
			||||||
            changeEnv,
 | 
					            changeEnv,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -46,7 +46,7 @@
 | 
				
			|||||||
                <el-table-column prop="username" label="用户名" min-width="100"></el-table-column>
 | 
					                <el-table-column prop="username" label="用户名" min-width="100"></el-table-column>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-table-column min-width="115" prop="creator" label="创建账号"></el-table-column>
 | 
					                <el-table-column min-width="115" prop="creator" label="创建账号"></el-table-column>
 | 
				
			||||||
                <el-table-column min-width="160" prop="createTime" label="创建时间">
 | 
					                <el-table-column min-width="160" prop="createTime" label="创建时间" show-overflow-tooltip>
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
					                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
				
			||||||
                    </template>
 | 
					                    </template>
 | 
				
			||||||
@@ -100,9 +100,17 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
                <el-button type="primary" size="small" @click="tableCreateDialog.visible = true">创建表</el-button>
 | 
					                <el-button type="primary" size="small" @click="tableCreateDialog.visible = true">创建表</el-button>
 | 
				
			||||||
            </el-row>
 | 
					            </el-row>
 | 
				
			||||||
            <el-table v-loading="tableInfoDialog.loading" border stripe :data="tableInfoDialog.infos" size="small">
 | 
					            <el-table v-loading="tableInfoDialog.loading" border stripe :data="filterTableInfos" size="small">
 | 
				
			||||||
                <el-table-column property="tableName" label="表名" min-width="150" show-overflow-tooltip></el-table-column>
 | 
					                <el-table-column property="tableName" label="表名" min-width="150" show-overflow-tooltip>
 | 
				
			||||||
                <el-table-column property="tableComment" label="备注" min-width="150" show-overflow-tooltip></el-table-column>
 | 
					                    <template #header>
 | 
				
			||||||
 | 
					                        <el-input v-model="tableInfoDialog.tableNameSearch" size="small" placeholder="表名: 输入可过滤" clearable />
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column property="tableComment" label="备注" min-width="150" show-overflow-tooltip>
 | 
				
			||||||
 | 
					                    <template #header>
 | 
				
			||||||
 | 
					                        <el-input v-model="tableInfoDialog.tableCommentSearch" size="small" placeholder="备注: 输入可过滤" clearable />
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
                <el-table-column
 | 
					                <el-table-column
 | 
				
			||||||
                    prop="tableRows"
 | 
					                    prop="tableRows"
 | 
				
			||||||
                    label="Rows"
 | 
					                    label="Rows"
 | 
				
			||||||
@@ -244,7 +252,7 @@
 | 
				
			|||||||
</template>
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
<script lang='ts'>
 | 
					<script lang='ts'>
 | 
				
			||||||
import { toRefs, reactive, onMounted, defineComponent } from 'vue';
 | 
					import { toRefs, reactive, computed, onMounted, defineComponent } from 'vue';
 | 
				
			||||||
import { ElMessage, ElMessageBox } from 'element-plus';
 | 
					import { ElMessage, ElMessageBox } from 'element-plus';
 | 
				
			||||||
import { formatByteSize } from '@/common/utils/format';
 | 
					import { formatByteSize } from '@/common/utils/format';
 | 
				
			||||||
import DbEdit from './DbEdit.vue';
 | 
					import DbEdit from './DbEdit.vue';
 | 
				
			||||||
@@ -317,6 +325,8 @@ export default defineComponent({
 | 
				
			|||||||
                loading: false,
 | 
					                loading: false,
 | 
				
			||||||
                visible: false,
 | 
					                visible: false,
 | 
				
			||||||
                infos: [],
 | 
					                infos: [],
 | 
				
			||||||
 | 
					                tableNameSearch: '',
 | 
				
			||||||
 | 
					                tableCommentSearch: '',
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            columnDialog: {
 | 
					            columnDialog: {
 | 
				
			||||||
                visible: false,
 | 
					                visible: false,
 | 
				
			||||||
@@ -342,7 +352,26 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        onMounted(async () => {
 | 
					        onMounted(async () => {
 | 
				
			||||||
            search();
 | 
					            search();
 | 
				
			||||||
            state.projects = await projectApi.accountProjects.request(null);
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const filterTableInfos = computed(() => {
 | 
				
			||||||
 | 
					            const infos = state.tableInfoDialog.infos;
 | 
				
			||||||
 | 
					            const tableNameSearch = state.tableInfoDialog.tableNameSearch;
 | 
				
			||||||
 | 
					            const tableCommentSearch = state.tableInfoDialog.tableCommentSearch;
 | 
				
			||||||
 | 
					            if (!tableNameSearch && !tableCommentSearch) {
 | 
				
			||||||
 | 
					                return infos;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            return infos.filter((data: any) => {
 | 
				
			||||||
 | 
					                let tnMatch = true;
 | 
				
			||||||
 | 
					                let tcMatch = true;
 | 
				
			||||||
 | 
					                if (tableNameSearch) {
 | 
				
			||||||
 | 
					                    tnMatch = data.tableName.toLowerCase().includes(tableNameSearch.toLowerCase());
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					                if (tableCommentSearch) {
 | 
				
			||||||
 | 
					                    tcMatch = data.tableComment.includes(tableCommentSearch);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					                return tnMatch && tcMatch;
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const choose = (item: any) => {
 | 
					        const choose = (item: any) => {
 | 
				
			||||||
@@ -369,7 +398,8 @@ export default defineComponent({
 | 
				
			|||||||
            search();
 | 
					            search();
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const editDb = (isAdd = false) => {
 | 
					        const editDb = async (isAdd = false) => {
 | 
				
			||||||
 | 
					            state.projects = await projectApi.accountProjects.request(null);
 | 
				
			||||||
            if (isAdd) {
 | 
					            if (isAdd) {
 | 
				
			||||||
                state.dbEditDialog.data = null;
 | 
					                state.dbEditDialog.data = null;
 | 
				
			||||||
                state.dbEditDialog.title = '新增数据库资源';
 | 
					                state.dbEditDialog.title = '新增数据库资源';
 | 
				
			||||||
@@ -502,6 +532,8 @@ export default defineComponent({
 | 
				
			|||||||
                state.tableInfoDialog.infos = await dbApi.tableInfos.request({ id: row.id, db });
 | 
					                state.tableInfoDialog.infos = await dbApi.tableInfos.request({ id: row.id, db });
 | 
				
			||||||
                state.dbId = row.id;
 | 
					                state.dbId = row.id;
 | 
				
			||||||
                state.db = db;
 | 
					                state.db = db;
 | 
				
			||||||
 | 
					            } catch (e) {
 | 
				
			||||||
 | 
					                state.tableInfoDialog.visible = false;
 | 
				
			||||||
            } finally {
 | 
					            } finally {
 | 
				
			||||||
                state.tableInfoDialog.loading = false;
 | 
					                state.tableInfoDialog.loading = false;
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
@@ -570,6 +602,7 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            filterTableInfos,
 | 
				
			||||||
            enums,
 | 
					            enums,
 | 
				
			||||||
            search,
 | 
					            search,
 | 
				
			||||||
            choose,
 | 
					            choose,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,9 +1,5 @@
 | 
				
			|||||||
<template>
 | 
					<template>
 | 
				
			||||||
    <div>
 | 
					    <div>
 | 
				
			||||||
        <el-button-group :style="btnStyle">
 | 
					 | 
				
			||||||
            <el-button @click="onRunSql" type="success" icon="video-play" size="small" plain>执行</el-button>
 | 
					 | 
				
			||||||
            <el-button @click="formatSql" type="primary" icon="magic-stick" size="small" plain>格式化</el-button>
 | 
					 | 
				
			||||||
        </el-button-group>
 | 
					 | 
				
			||||||
        <div class="toolbar">
 | 
					        <div class="toolbar">
 | 
				
			||||||
            <el-row type="flex" justify="space-between">
 | 
					            <el-row type="flex" justify="space-between">
 | 
				
			||||||
                <el-col :span="24">
 | 
					                <el-col :span="24">
 | 
				
			||||||
@@ -58,6 +54,19 @@
 | 
				
			|||||||
                        <div>
 | 
					                        <div>
 | 
				
			||||||
                            <div class="toolbar">
 | 
					                            <div class="toolbar">
 | 
				
			||||||
                                <div class="fl">
 | 
					                                <div class="fl">
 | 
				
			||||||
 | 
					                                    <el-link @click="onRunSql" :underline="false" class="ml15" icon="VideoPlay"></el-link>
 | 
				
			||||||
 | 
					                                    <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                                    <el-tooltip class="box-item" effect="dark" content="format sql" placement="top">
 | 
				
			||||||
 | 
					                                        <el-link @click="formatSql" type="primary" :underline="false" icon="MagicStick"></el-link>
 | 
				
			||||||
 | 
					                                    </el-tooltip>
 | 
				
			||||||
 | 
					                                    <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                                    <el-tooltip class="box-item" effect="dark" content="commit" placement="top">
 | 
				
			||||||
 | 
					                                        <el-link @click="onCommit" type="success" :underline="false" icon="CircleCheck"></el-link>
 | 
				
			||||||
 | 
					                                    </el-tooltip>
 | 
				
			||||||
 | 
					                                    <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                                    <el-upload
 | 
					                                    <el-upload
 | 
				
			||||||
                                        style="display: inline-block"
 | 
					                                        style="display: inline-block"
 | 
				
			||||||
                                        :before-upload="beforeUpload"
 | 
					                                        :before-upload="beforeUpload"
 | 
				
			||||||
@@ -72,11 +81,10 @@
 | 
				
			|||||||
                                        multiple
 | 
					                                        multiple
 | 
				
			||||||
                                        :limit="100"
 | 
					                                        :limit="100"
 | 
				
			||||||
                                    >
 | 
					                                    >
 | 
				
			||||||
                                        <el-button type="success" icon="video-play" plain size="small">sql脚本执行</el-button>
 | 
					                                        <el-tooltip class="box-item" effect="dark" content="SQL脚本执行" placement="top">
 | 
				
			||||||
 | 
					                                            <el-link type="success" :underline="false" icon="Document"></el-link>
 | 
				
			||||||
 | 
					                                        </el-tooltip>
 | 
				
			||||||
                                    </el-upload>
 | 
					                                    </el-upload>
 | 
				
			||||||
                                    <el-button @click="onCommit" class="ml5 mb5" type="success" icon="CircleCheck" plain size="small"
 | 
					 | 
				
			||||||
                                        >commit</el-button
 | 
					 | 
				
			||||||
                                    >
 | 
					 | 
				
			||||||
                                </div>
 | 
					                                </div>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                                <div style="float: right" class="fl">
 | 
					                                <div style="float: right" class="fl">
 | 
				
			||||||
@@ -101,7 +109,7 @@
 | 
				
			|||||||
                            </div>
 | 
					                            </div>
 | 
				
			||||||
                        </div>
 | 
					                        </div>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                        <div @click="closeExecBtns" class="mt5 sqlEditor" @contextmenu="showExecBtns">
 | 
					                        <div class="mt5 sqlEditor">
 | 
				
			||||||
                            <textarea ref="codeTextarea"></textarea>
 | 
					                            <textarea ref="codeTextarea"></textarea>
 | 
				
			||||||
                        </div>
 | 
					                        </div>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -145,46 +153,64 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
                <el-tab-pane closable v-for="dt in dataTabs" :key="dt.name" :label="dt.label" :name="dt.name">
 | 
					                <el-tab-pane closable v-for="dt in dataTabs" :key="dt.name" :label="dt.label" :name="dt.name">
 | 
				
			||||||
                    <el-row v-if="dbId">
 | 
					                    <el-row v-if="dbId">
 | 
				
			||||||
                        <el-link @click="onRefresh(dt.name)" icon="refresh" :underline="false" class="ml5"></el-link>
 | 
					                        <el-col :span="8">
 | 
				
			||||||
                        <el-link @click="addRow" class="ml5" type="primary" icon="plus" :underline="false"></el-link>
 | 
					                            <el-link @click="onRefresh(dt.name)" icon="refresh" :underline="false" class="ml5"></el-link>
 | 
				
			||||||
                        <el-link @click="onDeleteData" class="ml5" type="danger" icon="delete" :underline="false"></el-link>
 | 
					                            <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                        <el-tooltip class="box-item" effect="dark" content="commit" placement="top">
 | 
					                            <el-link @click="addRow" type="primary" icon="plus" :underline="false"></el-link>
 | 
				
			||||||
                            <el-link @click="onCommit" class="ml5" type="success" icon="check" :underline="false"></el-link>
 | 
					                            <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
                        </el-tooltip>
 | 
					 | 
				
			||||||
                    </el-row>
 | 
					 | 
				
			||||||
                    <el-row class="mt5">
 | 
					 | 
				
			||||||
                        <el-input
 | 
					 | 
				
			||||||
                            v-model="dt.condition"
 | 
					 | 
				
			||||||
                            placeholder="若需条件过滤,可选择列并点击对应的字段并输入需要过滤的内容点击查询按钮即可"
 | 
					 | 
				
			||||||
                            clearable
 | 
					 | 
				
			||||||
                            size="small"
 | 
					 | 
				
			||||||
                        >
 | 
					 | 
				
			||||||
                            <template #prepend>
 | 
					 | 
				
			||||||
                                <el-popover trigger="click" :width="270" placement="right">
 | 
					 | 
				
			||||||
                                    <template #reference>
 | 
					 | 
				
			||||||
                                        <el-link type="success" :underline="false">选择列</el-link>
 | 
					 | 
				
			||||||
                                    </template>
 | 
					 | 
				
			||||||
                                    <el-table
 | 
					 | 
				
			||||||
                                        :data="getColumns4Map(dt.name)"
 | 
					 | 
				
			||||||
                                        max-height="500"
 | 
					 | 
				
			||||||
                                        size="small"
 | 
					 | 
				
			||||||
                                        @row-click="
 | 
					 | 
				
			||||||
                                            (...event) => {
 | 
					 | 
				
			||||||
                                                onConditionRowClick(event, dt);
 | 
					 | 
				
			||||||
                                            }
 | 
					 | 
				
			||||||
                                        "
 | 
					 | 
				
			||||||
                                    >
 | 
					 | 
				
			||||||
                                        <el-table-column property="columnName" label="列名" show-overflow-tooltip> </el-table-column>
 | 
					 | 
				
			||||||
                                        <el-table-column property="columnComment" label="备注" show-overflow-tooltip> </el-table-column>
 | 
					 | 
				
			||||||
                                    </el-table>
 | 
					 | 
				
			||||||
                                </el-popover>
 | 
					 | 
				
			||||||
                            </template>
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
                            <template #append>
 | 
					                            <el-link @click="onDeleteData" type="danger" icon="delete" :underline="false"></el-link>
 | 
				
			||||||
                                <el-button @click="selectByCondition(dt.name, dt.condition)" icon="search" size="small"></el-button>
 | 
					                            <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
                            </template>
 | 
					
 | 
				
			||||||
                        </el-input>
 | 
					                            <el-tooltip class="box-item" effect="dark" content="commit" placement="top">
 | 
				
			||||||
 | 
					                                <el-link @click="onCommit" type="success" icon="CircleCheck" :underline="false"></el-link>
 | 
				
			||||||
 | 
					                            </el-tooltip>
 | 
				
			||||||
 | 
					                            <el-divider direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                            <el-tooltip class="box-item" effect="dark" content="生成insert sql" placement="top">
 | 
				
			||||||
 | 
					                                <el-link @click="onGenerateInsertSql" type="success" :underline="false">gi</el-link>
 | 
				
			||||||
 | 
					                            </el-tooltip>
 | 
				
			||||||
 | 
					                        </el-col>
 | 
				
			||||||
 | 
					                        <el-col :span="16">
 | 
				
			||||||
 | 
					                            <el-input
 | 
				
			||||||
 | 
					                                v-model="dt.condition"
 | 
				
			||||||
 | 
					                                placeholder="若需条件过滤,可选择列并点击对应的字段并输入需要过滤的内容点击查询按钮即可"
 | 
				
			||||||
 | 
					                                clearable
 | 
				
			||||||
 | 
					                                size="small"
 | 
				
			||||||
 | 
					                                style="width: 100%"
 | 
				
			||||||
 | 
					                            >
 | 
				
			||||||
 | 
					                                <template #prepend>
 | 
				
			||||||
 | 
					                                    <el-popover trigger="click" :width="320" placement="right">
 | 
				
			||||||
 | 
					                                        <template #reference>
 | 
				
			||||||
 | 
					                                            <el-link
 | 
				
			||||||
 | 
					                                                type="success"
 | 
				
			||||||
 | 
					                                                :underline="false"
 | 
				
			||||||
 | 
					                                                >选择列</el-link
 | 
				
			||||||
 | 
					                                            >
 | 
				
			||||||
 | 
					                                        </template>
 | 
				
			||||||
 | 
					                                        <el-table
 | 
				
			||||||
 | 
					                                            :data="getColumns4Map(dt.name)"
 | 
				
			||||||
 | 
					                                            max-height="500"
 | 
				
			||||||
 | 
					                                            size="small"
 | 
				
			||||||
 | 
					                                            @row-click="
 | 
				
			||||||
 | 
					                                                (...event) => {
 | 
				
			||||||
 | 
					                                                    onConditionRowClick(event, dt);
 | 
				
			||||||
 | 
					                                                }
 | 
				
			||||||
 | 
					                                            "
 | 
				
			||||||
 | 
					                                            style="cursor: pointer"
 | 
				
			||||||
 | 
					                                        >
 | 
				
			||||||
 | 
					                                            <el-table-column property="columnName" label="列名" show-overflow-tooltip> </el-table-column>
 | 
				
			||||||
 | 
					                                            <el-table-column property="columnComment" label="备注" show-overflow-tooltip> </el-table-column>
 | 
				
			||||||
 | 
					                                        </el-table>
 | 
				
			||||||
 | 
					                                    </el-popover>
 | 
				
			||||||
 | 
					                                </template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					                                <template #append>
 | 
				
			||||||
 | 
					                                    <el-button @click="selectByCondition(dt.name, dt.condition)" icon="search" size="small"></el-button>
 | 
				
			||||||
 | 
					                                </template>
 | 
				
			||||||
 | 
					                            </el-input>
 | 
				
			||||||
 | 
					                        </el-col>
 | 
				
			||||||
                    </el-row>
 | 
					                    </el-row>
 | 
				
			||||||
                    <el-table
 | 
					                    <el-table
 | 
				
			||||||
                        @cell-dblclick="cellClick"
 | 
					                        @cell-dblclick="cellClick"
 | 
				
			||||||
@@ -233,6 +259,34 @@
 | 
				
			|||||||
                </el-tab-pane>
 | 
					                </el-tab-pane>
 | 
				
			||||||
            </el-tabs>
 | 
					            </el-tabs>
 | 
				
			||||||
        </el-container>
 | 
					        </el-container>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        <el-dialog v-model="conditionDialog.visible" :title="conditionDialog.title" width="420px">
 | 
				
			||||||
 | 
					            <el-row>
 | 
				
			||||||
 | 
					                <el-col :span="5">
 | 
				
			||||||
 | 
					                    <el-select v-model="conditionDialog.condition">
 | 
				
			||||||
 | 
					                        <el-option label="=" value="="> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label="LIKE" value="LIKE"> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label=">" value=">"> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label=">=" value=">="> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label="<" value="<"> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label="<=" value="<="> </el-option>
 | 
				
			||||||
 | 
					                    </el-select>
 | 
				
			||||||
 | 
					                </el-col>
 | 
				
			||||||
 | 
					                <el-col :span="19">
 | 
				
			||||||
 | 
					                    <el-input v-model="conditionDialog.value" :placeholder="conditionDialog.placeholder" />
 | 
				
			||||||
 | 
					                </el-col>
 | 
				
			||||||
 | 
					            </el-row>
 | 
				
			||||||
 | 
					            <template #footer>
 | 
				
			||||||
 | 
					                <span class="dialog-footer">
 | 
				
			||||||
 | 
					                    <el-button @click="onCancelCondition">取消</el-button>
 | 
				
			||||||
 | 
					                    <el-button type="primary" @click="onConfirmCondition">确定</el-button>
 | 
				
			||||||
 | 
					                </span>
 | 
				
			||||||
 | 
					            </template>
 | 
				
			||||||
 | 
					        </el-dialog>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        <el-dialog @close="genSqlDialog.visible = false" v-model="genSqlDialog.visible" title="SQL" width="1000px">
 | 
				
			||||||
 | 
					            <el-input v-model="genSqlDialog.sql" type="textarea" rows="20" />
 | 
				
			||||||
 | 
					        </el-dialog>
 | 
				
			||||||
    </div>
 | 
					    </div>
 | 
				
			||||||
</template>
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -306,12 +360,18 @@ export default defineComponent({
 | 
				
			|||||||
                pageSize: 10,
 | 
					                pageSize: 10,
 | 
				
			||||||
                envId: null,
 | 
					                envId: null,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            btnStyle: {
 | 
					            conditionDialog: {
 | 
				
			||||||
                position: 'absolute',
 | 
					                title: '',
 | 
				
			||||||
                zIndex: 1000,
 | 
					                placeholder: '',
 | 
				
			||||||
                display: 'none',
 | 
					                columnRow: null,
 | 
				
			||||||
                left: '',
 | 
					                dataTab: null,
 | 
				
			||||||
                top: '',
 | 
					                visible: false,
 | 
				
			||||||
 | 
					                condition: '=',
 | 
				
			||||||
 | 
					                value: null,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            genSqlDialog: {
 | 
				
			||||||
 | 
					                visible: false,
 | 
				
			||||||
 | 
					                sql: '',
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            cmOptions: {
 | 
					            cmOptions: {
 | 
				
			||||||
                tabSize: 4,
 | 
					                tabSize: 4,
 | 
				
			||||||
@@ -365,7 +425,7 @@ export default defineComponent({
 | 
				
			|||||||
        const setHeight = () => {
 | 
					        const setHeight = () => {
 | 
				
			||||||
            // 默认300px
 | 
					            // 默认300px
 | 
				
			||||||
            codemirror.setSize('auto', `${window.innerHeight - 538}px`);
 | 
					            codemirror.setSize('auto', `${window.innerHeight - 538}px`);
 | 
				
			||||||
            state.dataTabsTableHeight = window.innerHeight - 258 - 33;
 | 
					            state.dataTabsTableHeight = window.innerHeight - 274;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        /**
 | 
					        /**
 | 
				
			||||||
@@ -434,7 +494,6 @@ export default defineComponent({
 | 
				
			|||||||
            } catch (e: any) {
 | 
					            } catch (e: any) {
 | 
				
			||||||
                state.queryTab.loading = false;
 | 
					                state.queryTab.loading = false;
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
            closeExecBtns();
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // 即只有以该字符串开头的sql才可修改表数据内容
 | 
					            // 即只有以该字符串开头的sql才可修改表数据内容
 | 
				
			||||||
            if (sql.startsWith('SELECT *') || sql.startsWith('select *') || sql.startsWith('SELECT\n  *')) {
 | 
					            if (sql.startsWith('SELECT *') || sql.startsWith('select *') || sql.startsWith('SELECT\n  *')) {
 | 
				
			||||||
@@ -717,23 +776,34 @@ export default defineComponent({
 | 
				
			|||||||
         */
 | 
					         */
 | 
				
			||||||
        const onConditionRowClick = (event: any, dataTab: any) => {
 | 
					        const onConditionRowClick = (event: any, dataTab: any) => {
 | 
				
			||||||
            const row = event[0];
 | 
					            const row = event[0];
 | 
				
			||||||
            ElMessageBox.prompt(`请输入 [${row.columnName}] 的值`, '查询条件', {
 | 
					            state.conditionDialog.title = `请输入 [${row.columnName}] 的值`;
 | 
				
			||||||
                confirmButtonText: '确定',
 | 
					            state.conditionDialog.placeholder = `${row.columnType}  ${row.columnComment}`;
 | 
				
			||||||
                cancelButtonText: '取消',
 | 
					            state.conditionDialog.columnRow = row;
 | 
				
			||||||
                inputPlaceholder: `${row.columnType}  ${row.columnComment}`,
 | 
					            state.conditionDialog.dataTab = dataTab;
 | 
				
			||||||
            })
 | 
					            state.conditionDialog.visible = true;
 | 
				
			||||||
                .then(({ value }) => {
 | 
					        };
 | 
				
			||||||
                    if (!value) {
 | 
					
 | 
				
			||||||
                        value = '';
 | 
					        // 确认条件
 | 
				
			||||||
                    }
 | 
					        const onConfirmCondition = () => {
 | 
				
			||||||
                    let condition = dataTab.condition;
 | 
					            const conditionDialog = state.conditionDialog;
 | 
				
			||||||
                    if (condition) {
 | 
					            const dataTab = state.conditionDialog.dataTab as any;
 | 
				
			||||||
                        condition += ` AND `;
 | 
					            let condition = dataTab.condition;
 | 
				
			||||||
                    }
 | 
					            if (condition) {
 | 
				
			||||||
                    condition += `${row.columnName} = `;
 | 
					                condition += ` AND `;
 | 
				
			||||||
                    dataTab.condition = condition + wrapColumnValue(row, value);
 | 
					            }
 | 
				
			||||||
                })
 | 
					            const row = conditionDialog.columnRow as any;
 | 
				
			||||||
                .catch(() => {});
 | 
					            condition += `${row.columnName} ${conditionDialog.condition} `;
 | 
				
			||||||
 | 
					            dataTab.condition = condition + wrapColumnValue(row, conditionDialog.value);
 | 
				
			||||||
 | 
					            onCancelCondition();
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onCancelCondition = () => {
 | 
				
			||||||
 | 
					            state.conditionDialog.visible = false;
 | 
				
			||||||
 | 
					            state.conditionDialog.title = ``;
 | 
				
			||||||
 | 
					            state.conditionDialog.placeholder = ``;
 | 
				
			||||||
 | 
					            state.conditionDialog.value = null;
 | 
				
			||||||
 | 
					            state.conditionDialog.columnRow = null;
 | 
				
			||||||
 | 
					            state.conditionDialog.dataTab = null;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const onRefresh = async (tableName: string) => {
 | 
					        const onRefresh = async (tableName: string) => {
 | 
				
			||||||
@@ -793,10 +863,10 @@ export default defineComponent({
 | 
				
			|||||||
        const getDefaultSelectSql = (tableName: string, where: string = '', orderBy: string = '', pageNum: number = 1) => {
 | 
					        const getDefaultSelectSql = (tableName: string, where: string = '', orderBy: string = '', pageNum: number = 1) => {
 | 
				
			||||||
            const baseSql = `SELECT * FROM ${tableName} ${where ? 'WHERE ' + where : ''} ${orderBy ? orderBy : ''}`;
 | 
					            const baseSql = `SELECT * FROM ${tableName} ${where ? 'WHERE ' + where : ''} ${orderBy ? orderBy : ''}`;
 | 
				
			||||||
            if (state.dbType == 'mysql') {
 | 
					            if (state.dbType == 'mysql') {
 | 
				
			||||||
                 return `${baseSql} LIMIT ${(pageNum - 1) * state.defalutLimit}, ${state.defalutLimit};`
 | 
					                return `${baseSql} LIMIT ${(pageNum - 1) * state.defalutLimit}, ${state.defalutLimit};`;
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
            if (state.dbType == 'postgres') {
 | 
					            if (state.dbType == 'postgres') {
 | 
				
			||||||
                return `${baseSql} OFFSET ${(pageNum - 1) * state.defalutLimit} LIMIT ${state.defalutLimit};`
 | 
					                return `${baseSql} OFFSET ${(pageNum - 1) * state.defalutLimit} LIMIT ${state.defalutLimit};`;
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
            return baseSql;
 | 
					            return baseSql;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
@@ -928,7 +998,6 @@ export default defineComponent({
 | 
				
			|||||||
            state.queryTab.execRes.tableColumn = [];
 | 
					            state.queryTab.execRes.tableColumn = [];
 | 
				
			||||||
            state.cmOptions.hintOptions.tables = [];
 | 
					            state.cmOptions.hintOptions.tables = [];
 | 
				
			||||||
            tableMap.clear();
 | 
					            tableMap.clear();
 | 
				
			||||||
            closeExecBtns();
 | 
					 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const onDataSelectionChange = (datas: []) => {
 | 
					        const onDataSelectionChange = (datas: []) => {
 | 
				
			||||||
@@ -963,6 +1032,38 @@ export default defineComponent({
 | 
				
			|||||||
            });
 | 
					            });
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onGenerateInsertSql = async () => {
 | 
				
			||||||
 | 
					            const queryTab = isQueryTab();
 | 
				
			||||||
 | 
					            const datas = queryTab ? state.queryTab.selectionDatas : state.dataTabs[state.activeName].selectionDatas;
 | 
				
			||||||
 | 
					            isTrue(datas && datas.length > 0, '请先选择要生成insert语句的数据');
 | 
				
			||||||
 | 
					            const tableName = state.nowTableName;
 | 
				
			||||||
 | 
					            const columns: any = await getColumns(tableName);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            const sqls = [];
 | 
				
			||||||
 | 
					            for (let data of datas) {
 | 
				
			||||||
 | 
					                let colNames = [];
 | 
				
			||||||
 | 
					                let values = [];
 | 
				
			||||||
 | 
					                for (let column of columns) {
 | 
				
			||||||
 | 
					                    const colName = column.columnName;
 | 
				
			||||||
 | 
					                    colNames.push(colName);
 | 
				
			||||||
 | 
					                    values.push(wrapValueByType(data[colName]));
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					                sqls.push(`INSERT INTO ${tableName} (${colNames.join(', ')}) VALUES(${values.join(', ')})`);
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            state.genSqlDialog.sql = sqls.join(';\n') + ';';
 | 
				
			||||||
 | 
					            state.genSqlDialog.visible = true;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const wrapValueByType = (val: any) => {
 | 
				
			||||||
 | 
					            if (val == null) {
 | 
				
			||||||
 | 
					                return 'NULL';
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            if (typeof val == 'number') {
 | 
				
			||||||
 | 
					                return val;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            return `'${val}'`;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        /**
 | 
					        /**
 | 
				
			||||||
         * 是否为查询tab
 | 
					         * 是否为查询tab
 | 
				
			||||||
         */
 | 
					         */
 | 
				
			||||||
@@ -1072,7 +1173,6 @@ export default defineComponent({
 | 
				
			|||||||
            let selectSql = codemirror.getSelection();
 | 
					            let selectSql = codemirror.getSelection();
 | 
				
			||||||
            isTrue(selectSql, '请选中需要格式化的sql');
 | 
					            isTrue(selectSql, '请选中需要格式化的sql');
 | 
				
			||||||
            codemirror.replaceSelection(sqlFormatter(selectSql));
 | 
					            codemirror.replaceSelection(sqlFormatter(selectSql));
 | 
				
			||||||
            closeExecBtns();
 | 
					 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const search = async () => {
 | 
					        const search = async () => {
 | 
				
			||||||
@@ -1080,31 +1180,6 @@ export default defineComponent({
 | 
				
			|||||||
            state.dbs = res.list;
 | 
					            state.dbs = res.list;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        /**
 | 
					 | 
				
			||||||
         * 显示执行sql和格式化按钮
 | 
					 | 
				
			||||||
         */
 | 
					 | 
				
			||||||
        const showExecBtns = (event: any) => {
 | 
					 | 
				
			||||||
            if (event.preventDefault) {
 | 
					 | 
				
			||||||
                event.preventDefault();
 | 
					 | 
				
			||||||
            } else {
 | 
					 | 
				
			||||||
                event.returnValue = false;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            state.btnStyle.display = 'inline';
 | 
					 | 
				
			||||||
            state.btnStyle.left = event.offsetX + 15 + 'px';
 | 
					 | 
				
			||||||
            state.btnStyle.top = event.clientY - 80 + 'px';
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        /**
 | 
					 | 
				
			||||||
         * 关闭执行sql和格式化按钮
 | 
					 | 
				
			||||||
         */
 | 
					 | 
				
			||||||
        const closeExecBtns = () => {
 | 
					 | 
				
			||||||
            if (state.btnStyle.left) {
 | 
					 | 
				
			||||||
                state.btnStyle.display = 'none';
 | 
					 | 
				
			||||||
                state.btnStyle.left = '';
 | 
					 | 
				
			||||||
                state.btnStyle.top = '';
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            codeTextarea,
 | 
					            codeTextarea,
 | 
				
			||||||
@@ -1121,6 +1196,8 @@ export default defineComponent({
 | 
				
			|||||||
            getColumnTip,
 | 
					            getColumnTip,
 | 
				
			||||||
            getColumns4Map,
 | 
					            getColumns4Map,
 | 
				
			||||||
            onConditionRowClick,
 | 
					            onConditionRowClick,
 | 
				
			||||||
 | 
					            onConfirmCondition,
 | 
				
			||||||
 | 
					            onCancelCondition,
 | 
				
			||||||
            changeSqlTemplate,
 | 
					            changeSqlTemplate,
 | 
				
			||||||
            deleteSql,
 | 
					            deleteSql,
 | 
				
			||||||
            saveSql,
 | 
					            saveSql,
 | 
				
			||||||
@@ -1137,8 +1214,7 @@ export default defineComponent({
 | 
				
			|||||||
            onDataSelectionChange,
 | 
					            onDataSelectionChange,
 | 
				
			||||||
            onDeleteData,
 | 
					            onDeleteData,
 | 
				
			||||||
            onTableSortChange,
 | 
					            onTableSortChange,
 | 
				
			||||||
            showExecBtns,
 | 
					            onGenerateInsertSql,
 | 
				
			||||||
            closeExecBtns,
 | 
					 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
});
 | 
					});
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -4,6 +4,8 @@ export const dbApi = {
 | 
				
			|||||||
    // 获取权限列表
 | 
					    // 获取权限列表
 | 
				
			||||||
    dbs: Api.create("/dbs", 'get'),
 | 
					    dbs: Api.create("/dbs", 'get'),
 | 
				
			||||||
    saveDb: Api.create("/dbs", 'post'),
 | 
					    saveDb: Api.create("/dbs", 'post'),
 | 
				
			||||||
 | 
					    getAllDatabase: Api.create("/dbs/databases", 'post'),
 | 
				
			||||||
 | 
					    getDbPwd: Api.create("/dbs/{id}/pwd", 'get'),
 | 
				
			||||||
    deleteDb: Api.create("/dbs/{id}", 'delete'),
 | 
					    deleteDb: Api.create("/dbs/{id}", 'delete'),
 | 
				
			||||||
    dumpDb: Api.create("/dbs/{id}/dump", 'post'),
 | 
					    dumpDb: Api.create("/dbs/{id}/dump", 'post'),
 | 
				
			||||||
    tableInfos: Api.create("/dbs/{id}/t-infos", 'get'),
 | 
					    tableInfos: Api.create("/dbs/{id}/t-infos", 'get'),
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -57,7 +57,7 @@
 | 
				
			|||||||
            </el-row>
 | 
					            </el-row>
 | 
				
			||||||
        </el-dialog>
 | 
					        </el-dialog>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        <el-dialog :title="tree.title" v-model="tree.visible" :close-on-click-modal="false" width="680px">
 | 
					        <el-dialog :title="tree.title" v-model="tree.visible" :close-on-click-modal="false" width="50%">
 | 
				
			||||||
            <el-progress
 | 
					            <el-progress
 | 
				
			||||||
                v-if="uploadProgressShow"
 | 
					                v-if="uploadProgressShow"
 | 
				
			||||||
                style="width: 90%; margin-left: 20px"
 | 
					                style="width: 90%; margin-left: 20px"
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,6 +1,6 @@
 | 
				
			|||||||
<template>
 | 
					<template>
 | 
				
			||||||
    <div>
 | 
					    <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 :model="form" ref="machineForm" :rules="rules" label-width="85px">
 | 
				
			||||||
                <el-form-item prop="projectId" label="项目:" required>
 | 
					                <el-form-item prop="projectId" label="项目:" required>
 | 
				
			||||||
                    <el-select style="width: 100%" v-model="form.projectId" placeholder="请选择项目" @change="changeProject" filterable>
 | 
					                    <el-select style="width: 100%" v-model="form.projectId" placeholder="请选择项目" @change="changeProject" filterable>
 | 
				
			||||||
@@ -35,7 +35,15 @@
 | 
				
			|||||||
                        v-model.trim="form.password"
 | 
					                        v-model.trim="form.password"
 | 
				
			||||||
                        placeholder="请输入密码,修改操作可不填"
 | 
					                        placeholder="请输入密码,修改操作可不填"
 | 
				
			||||||
                        autocomplete="new-password"
 | 
					                        autocomplete="new-password"
 | 
				
			||||||
                    ></el-input>
 | 
					                    >
 | 
				
			||||||
 | 
					                        <template v-if="form.id && form.id != 0" #suffix>
 | 
				
			||||||
 | 
					                            <el-popover @hide="pwd = ''" placement="right" title="原密码" :width="200" trigger="click" :content="pwd">
 | 
				
			||||||
 | 
					                                <template #reference>
 | 
				
			||||||
 | 
					                                    <el-link @click="getPwd" :underline="false" type="primary" class="mr5">原密码</el-link>
 | 
				
			||||||
 | 
					                                </template>
 | 
				
			||||||
 | 
					                            </el-popover>
 | 
				
			||||||
 | 
					                        </template>
 | 
				
			||||||
 | 
					                    </el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item v-if="form.authMethod == 2" prop="password" label="秘钥:">
 | 
					                <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-input type="textarea" :rows="3" v-model="form.password" placeholder="请将私钥文件内容拷贝至此,修改操作可不填"></el-input>
 | 
				
			||||||
@@ -43,6 +51,24 @@
 | 
				
			|||||||
                <el-form-item prop="remark" label="备注:">
 | 
					                <el-form-item prop="remark" label="备注:">
 | 
				
			||||||
                    <el-input type="textarea" v-model="form.remark"></el-input>
 | 
					                    <el-input type="textarea" v-model="form.remark"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </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>
 | 
					            </el-form>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            <template #footer>
 | 
					            <template #footer>
 | 
				
			||||||
@@ -83,6 +109,7 @@ export default defineComponent({
 | 
				
			|||||||
        const state = reactive({
 | 
					        const state = reactive({
 | 
				
			||||||
            dialogVisible: false,
 | 
					            dialogVisible: false,
 | 
				
			||||||
            projects: [],
 | 
					            projects: [],
 | 
				
			||||||
 | 
					            sshTunnelMachineList: [],
 | 
				
			||||||
            form: {
 | 
					            form: {
 | 
				
			||||||
                id: null,
 | 
					                id: null,
 | 
				
			||||||
                projectId: null,
 | 
					                projectId: null,
 | 
				
			||||||
@@ -93,7 +120,10 @@ export default defineComponent({
 | 
				
			|||||||
                username: '',
 | 
					                username: '',
 | 
				
			||||||
                password: '',
 | 
					                password: '',
 | 
				
			||||||
                remark: '',
 | 
					                remark: '',
 | 
				
			||||||
 | 
					                enableSshTunnel: null,
 | 
				
			||||||
 | 
					                sshTunnelMachineId: null,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
 | 
					            pwd: '',
 | 
				
			||||||
            btnLoading: false,
 | 
					            btnLoading: false,
 | 
				
			||||||
            rules: {
 | 
					            rules: {
 | 
				
			||||||
                projectId: [
 | 
					                projectId: [
 | 
				
			||||||
@@ -152,8 +182,24 @@ export default defineComponent({
 | 
				
			|||||||
            } else {
 | 
					            } else {
 | 
				
			||||||
                state.form = { port: 22, authMethod: 1 } 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 getPwd = async () => {
 | 
				
			||||||
 | 
					            state.pwd = await machineApi.getMachinePwd.request({ id: state.form.id });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const changeProject = (projectId: number) => {
 | 
					        const changeProject = (projectId: number) => {
 | 
				
			||||||
            for (let p of state.projects as any) {
 | 
					            for (let p of state.projects as any) {
 | 
				
			||||||
                if (p.id == projectId) {
 | 
					                if (p.id == projectId) {
 | 
				
			||||||
@@ -168,7 +214,15 @@ export default defineComponent({
 | 
				
			|||||||
            }
 | 
					            }
 | 
				
			||||||
            machineForm.value.validate(async (valid: boolean) => {
 | 
					            machineForm.value.validate(async (valid: boolean) => {
 | 
				
			||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    const reqForm = { ...state.form };
 | 
					                    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) {
 | 
					                    if (reqForm.authMethod == 1) {
 | 
				
			||||||
                        reqForm.password = await RsaEncrypt(state.form.password);
 | 
					                        reqForm.password = await RsaEncrypt(state.form.password);
 | 
				
			||||||
                    }
 | 
					                    }
 | 
				
			||||||
@@ -196,6 +250,8 @@ export default defineComponent({
 | 
				
			|||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            machineForm,
 | 
					            machineForm,
 | 
				
			||||||
 | 
					            getSshTunnelMachines,
 | 
				
			||||||
 | 
					            getPwd,
 | 
				
			||||||
            changeProject,
 | 
					            changeProject,
 | 
				
			||||||
            btnOk,
 | 
					            btnOk,
 | 
				
			||||||
            cancel,
 | 
					            cancel,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -42,7 +42,7 @@
 | 
				
			|||||||
                    </template>
 | 
					                    </template>
 | 
				
			||||||
                </el-table-column>
 | 
					                </el-table-column>
 | 
				
			||||||
                <el-table-column prop="name" label="名称" min-width="140" show-overflow-tooltip></el-table-column>
 | 
					                <el-table-column prop="name" label="名称" min-width="140" show-overflow-tooltip></el-table-column>
 | 
				
			||||||
                <el-table-column prop="ip" label="ip:port" min-width="140">
 | 
					                <el-table-column prop="ip" label="ip:port" min-width="150">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-link :disabled="scope.row.status == -1" @click="showMachineStats(scope.row)" type="primary" :underline="false">{{
 | 
					                        <el-link :disabled="scope.row.status == -1" @click="showMachineStats(scope.row)" type="primary" :underline="false">{{
 | 
				
			||||||
                            `${scope.row.ip}:${scope.row.port}`
 | 
					                            `${scope.row.ip}:${scope.row.port}`
 | 
				
			||||||
@@ -68,11 +68,6 @@
 | 
				
			|||||||
                <el-table-column prop="username" label="用户名" min-width="90"></el-table-column>
 | 
					                <el-table-column prop="username" label="用户名" min-width="90"></el-table-column>
 | 
				
			||||||
                <el-table-column prop="projectName" label="项目" min-width="120"></el-table-column>
 | 
					                <el-table-column prop="projectName" label="项目" min-width="120"></el-table-column>
 | 
				
			||||||
                <el-table-column prop="remark" label="备注" min-width="250" show-overflow-tooltip></el-table-column>
 | 
					                <el-table-column prop="remark" label="备注" min-width="250" show-overflow-tooltip></el-table-column>
 | 
				
			||||||
                <el-table-column prop="ip" label="hasCli" width="70">
 | 
					 | 
				
			||||||
                    <template #default="scope">
 | 
					 | 
				
			||||||
                        {{ `${scope.row.hasCli ? '是' : '否'}` }}
 | 
					 | 
				
			||||||
                    </template>
 | 
					 | 
				
			||||||
                </el-table-column>
 | 
					 | 
				
			||||||
                <el-table-column prop="createTime" label="创建时间" min-width="165">
 | 
					                <el-table-column prop="createTime" label="创建时间" min-width="165">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
					                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
				
			||||||
@@ -232,7 +227,6 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        onMounted(async () => {
 | 
					        onMounted(async () => {
 | 
				
			||||||
            search();
 | 
					            search();
 | 
				
			||||||
            state.projects = await projectApi.accountProjects.request(null);
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const choose = (item: any) => {
 | 
					        const choose = (item: any) => {
 | 
				
			||||||
@@ -255,12 +249,18 @@ export default defineComponent({
 | 
				
			|||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const closeCli = async (row: any) => {
 | 
					        const closeCli = async (row: any) => {
 | 
				
			||||||
 | 
					            await ElMessageBox.confirm(`确定关闭该机器客户端连接?`, '提示', {
 | 
				
			||||||
 | 
					                confirmButtonText: '确定',
 | 
				
			||||||
 | 
					                cancelButtonText: '取消',
 | 
				
			||||||
 | 
					                type: 'warning',
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
            await machineApi.closeCli.request({ id: row.id });
 | 
					            await machineApi.closeCli.request({ id: row.id });
 | 
				
			||||||
            ElMessage.success('关闭成功');
 | 
					            ElMessage.success('关闭成功');
 | 
				
			||||||
            search();
 | 
					            search();
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const openFormDialog = (machine: any) => {
 | 
					        const openFormDialog = async (machine: any) => {
 | 
				
			||||||
 | 
					            state.projects = await projectApi.accountProjects.request(null);
 | 
				
			||||||
            let dialogTitle;
 | 
					            let dialogTitle;
 | 
				
			||||||
            if (machine) {
 | 
					            if (machine) {
 | 
				
			||||||
                state.machineEditDialog.data = state.currentData as any;
 | 
					                state.machineEditDialog.data = state.currentData as any;
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -7,9 +7,9 @@
 | 
				
			|||||||
            :before-close="cancel"
 | 
					            :before-close="cancel"
 | 
				
			||||||
            :show-close="true"
 | 
					            :show-close="true"
 | 
				
			||||||
            :destroy-on-close="true"
 | 
					            :destroy-on-close="true"
 | 
				
			||||||
            width="800px"
 | 
					            width="900px"
 | 
				
			||||||
        >
 | 
					        >
 | 
				
			||||||
            <el-form :model="form" ref="mockDataForm" label-width="70px">
 | 
					            <el-form :model="form" ref="scriptForm" label-width="70px" size="small">
 | 
				
			||||||
                <el-form-item prop="method" label="名称">
 | 
					                <el-form-item prop="method" label="名称">
 | 
				
			||||||
                    <el-input v-model.trim="form.name" placeholder="请输入名称"></el-input>
 | 
					                    <el-input v-model.trim="form.name" placeholder="请输入名称"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
@@ -24,8 +24,23 @@
 | 
				
			|||||||
                    </el-select>
 | 
					                    </el-select>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-form-item prop="params" label="参数">
 | 
					                <el-row style="margin-left: 30px; margin-bottom: 5px">
 | 
				
			||||||
                    <el-input v-model="form.params" placeholder="参数数组json,若无可不填"></el-input>
 | 
					                    <el-button @click="onAddParam" size="small" type="success">新增占位符参数</el-button>
 | 
				
			||||||
 | 
					                </el-row>
 | 
				
			||||||
 | 
					                <el-form-item :key="param" v-for="(param, index) in params" prop="params" :label="`参数${index + 1}`">
 | 
				
			||||||
 | 
					                    <el-row>
 | 
				
			||||||
 | 
					                        <el-col :span="5"><el-input v-model="param.model" placeholder="内容中用{{.model}}替换"></el-input></el-col>
 | 
				
			||||||
 | 
					                        <el-divider :span="1" direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					                        <el-col :span="4"><el-input v-model="param.name" placeholder="字段名"></el-input></el-col>
 | 
				
			||||||
 | 
					                        <el-divider :span="1" direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					                        <el-col :span="4"><el-input v-model="param.placeholder" placeholder="字段说明"></el-input></el-col>
 | 
				
			||||||
 | 
					                        <el-divider :span="1" direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					                        <el-col :span="4">
 | 
				
			||||||
 | 
					                            <el-input v-model="param.options" placeholder="可选值 ,分割"></el-input>
 | 
				
			||||||
 | 
					                        </el-col>
 | 
				
			||||||
 | 
					                        <el-divider :span="1" direction="vertical" border-style="dashed" />
 | 
				
			||||||
 | 
					                        <el-col :span="2"><el-button @click="onDeleteParam(index)" size="small" type="danger">删除</el-button></el-col>
 | 
				
			||||||
 | 
					                    </el-row>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-form-item prop="script" label="内容" id="content">
 | 
					                <el-form-item prop="script" label="内容" id="content">
 | 
				
			||||||
@@ -35,13 +50,12 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
            <template #footer>
 | 
					            <template #footer>
 | 
				
			||||||
                <div class="dialog-footer">
 | 
					                <div class="dialog-footer">
 | 
				
			||||||
                    <el-button @click="cancel()" :disabled="submitDisabled" size="small">关 闭</el-button>
 | 
					                    <el-button @click="cancel()" :disabled="submitDisabled">关 闭</el-button>
 | 
				
			||||||
                    <el-button
 | 
					                    <el-button
 | 
				
			||||||
                        v-auth="'machine:script:save'"
 | 
					                        v-auth="'machine:script:save'"
 | 
				
			||||||
                        type="primary"
 | 
					                        type="primary"
 | 
				
			||||||
                        :loading="btnLoading"
 | 
					                        :loading="btnLoading"
 | 
				
			||||||
                        @click="btnOk"
 | 
					                        @click="btnOk"
 | 
				
			||||||
                        size="small"
 | 
					 | 
				
			||||||
                        :disabled="submitDisabled"
 | 
					                        :disabled="submitDisabled"
 | 
				
			||||||
                        >保 存</el-button
 | 
					                        >保 存</el-button
 | 
				
			||||||
                    >
 | 
					                    >
 | 
				
			||||||
@@ -84,41 +98,59 @@ export default defineComponent({
 | 
				
			|||||||
    },
 | 
					    },
 | 
				
			||||||
    setup(props: any, { emit }) {
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
        const { isCommon, machineId } = toRefs(props);
 | 
					        const { isCommon, machineId } = toRefs(props);
 | 
				
			||||||
        const mockDataForm: any = ref(null);
 | 
					        const scriptForm: any = ref(null);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const state = reactive({
 | 
					        const state = reactive({
 | 
				
			||||||
            dialogVisible: false,
 | 
					            dialogVisible: false,
 | 
				
			||||||
            submitDisabled: false,
 | 
					            submitDisabled: false,
 | 
				
			||||||
 | 
					            params: [] as any,
 | 
				
			||||||
            form: {
 | 
					            form: {
 | 
				
			||||||
                id: null,
 | 
					                id: null,
 | 
				
			||||||
                name: '',
 | 
					                name: '',
 | 
				
			||||||
                machineId: 0,
 | 
					                machineId: 0,
 | 
				
			||||||
                description: '',
 | 
					                description: '',
 | 
				
			||||||
                script: '',
 | 
					                script: '',
 | 
				
			||||||
                params: null,
 | 
					                params: '',
 | 
				
			||||||
                type: null,
 | 
					                type: null,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            btnLoading: false,
 | 
					            btnLoading: false,
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        watch(props, (newValue) => {
 | 
					        watch(props, (newValue) => {
 | 
				
			||||||
 | 
					            state.dialogVisible = newValue.visible;
 | 
				
			||||||
 | 
					            if (!newValue.visible) {
 | 
				
			||||||
 | 
					                return;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
            if (newValue.data) {
 | 
					            if (newValue.data) {
 | 
				
			||||||
                state.form = { ...newValue.data };
 | 
					                state.form = { ...newValue.data };
 | 
				
			||||||
 | 
					                if (state.form.params) {
 | 
				
			||||||
 | 
					                    state.params = JSON.parse(state.form.params);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
            } else {
 | 
					            } else {
 | 
				
			||||||
                state.form = {} as any;
 | 
					                state.form = {} as any;
 | 
				
			||||||
                state.form.script = '';
 | 
					                state.form.script = '';
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
            state.dialogVisible = newValue.visible;
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onAddParam = () => {
 | 
				
			||||||
 | 
					            state.params.push({ name: '', model: '', placeholder: '' });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onDeleteParam = (idx: number) => {
 | 
				
			||||||
 | 
					            state.params.splice(idx, 1);
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const btnOk = () => {
 | 
					        const btnOk = () => {
 | 
				
			||||||
            state.form.machineId = isCommon.value ? 9999999 : (machineId.value as any);
 | 
					            state.form.machineId = isCommon.value ? 9999999 : (machineId.value as any);
 | 
				
			||||||
            console.log('machineid:', machineId);
 | 
					            console.log('machineid:', machineId);
 | 
				
			||||||
            mockDataForm.value.validate((valid: any) => {
 | 
					            scriptForm.value.validate((valid: any) => {
 | 
				
			||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    notEmpty(state.form.name, '名称不能为空');
 | 
					                    notEmpty(state.form.name, '名称不能为空');
 | 
				
			||||||
                    notEmpty(state.form.description, '描述不能为空');
 | 
					                    notEmpty(state.form.description, '描述不能为空');
 | 
				
			||||||
                    notEmpty(state.form.script, '内容不能为空');
 | 
					                    notEmpty(state.form.script, '内容不能为空');
 | 
				
			||||||
 | 
					                    if (state.params) {
 | 
				
			||||||
 | 
					                        state.form.params = JSON.stringify(state.params);
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
                    machineApi.saveScript.request(state.form).then(
 | 
					                    machineApi.saveScript.request(state.form).then(
 | 
				
			||||||
                        () => {
 | 
					                        () => {
 | 
				
			||||||
                            ElMessage.success('保存成功');
 | 
					                            ElMessage.success('保存成功');
 | 
				
			||||||
@@ -139,12 +171,15 @@ export default defineComponent({
 | 
				
			|||||||
        const cancel = () => {
 | 
					        const cancel = () => {
 | 
				
			||||||
            emit('update:visible', false);
 | 
					            emit('update:visible', false);
 | 
				
			||||||
            emit('cancel');
 | 
					            emit('cancel');
 | 
				
			||||||
 | 
					            state.params = [];
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            enums,
 | 
					            enums,
 | 
				
			||||||
            mockDataForm,
 | 
					            onAddParam,
 | 
				
			||||||
 | 
					            onDeleteParam,
 | 
				
			||||||
 | 
					            scriptForm,
 | 
				
			||||||
            btnOk,
 | 
					            btnOk,
 | 
				
			||||||
            cancel,
 | 
					            cancel,
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -76,7 +76,24 @@
 | 
				
			|||||||
        <el-dialog title="脚本参数" v-model="scriptParamsDialog.visible" width="400px">
 | 
					        <el-dialog title="脚本参数" v-model="scriptParamsDialog.visible" width="400px">
 | 
				
			||||||
            <el-form ref="paramsForm" :model="scriptParamsDialog.params" label-width="70px" size="small">
 | 
					            <el-form ref="paramsForm" :model="scriptParamsDialog.params" label-width="70px" size="small">
 | 
				
			||||||
                <el-form-item v-for="item in scriptParamsDialog.paramsFormItem" :key="item.name" :prop="item.model" :label="item.name" required>
 | 
					                <el-form-item v-for="item in scriptParamsDialog.paramsFormItem" :key="item.name" :prop="item.model" :label="item.name" required>
 | 
				
			||||||
                    <el-input v-model="scriptParamsDialog.params[item.model]" :placeholder="item.placeholder" autocomplete="off"></el-input>
 | 
					                    <el-input
 | 
				
			||||||
 | 
					                        v-if="!item.options"
 | 
				
			||||||
 | 
					                        v-model="scriptParamsDialog.params[item.model]"
 | 
				
			||||||
 | 
					                        :placeholder="item.placeholder"
 | 
				
			||||||
 | 
					                        autocomplete="off"
 | 
				
			||||||
 | 
					                        clearable
 | 
				
			||||||
 | 
					                    ></el-input>
 | 
				
			||||||
 | 
					                    <el-select
 | 
				
			||||||
 | 
					                        v-else
 | 
				
			||||||
 | 
					                        v-model="scriptParamsDialog.params[item.model]"
 | 
				
			||||||
 | 
					                        :placeholder="item.placeholder"
 | 
				
			||||||
 | 
					                        filterable
 | 
				
			||||||
 | 
					                        autocomplete="off"
 | 
				
			||||||
 | 
					                        clearable
 | 
				
			||||||
 | 
					                        style="width: 100%"
 | 
				
			||||||
 | 
					                    >
 | 
				
			||||||
 | 
					                        <el-option v-for="option in item.options.split(',')" :key="option" :label="option" :value="option" />
 | 
				
			||||||
 | 
					                    </el-select>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
            </el-form>
 | 
					            </el-form>
 | 
				
			||||||
            <template #footer>
 | 
					            <template #footer>
 | 
				
			||||||
@@ -88,7 +105,6 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        <el-dialog title="执行结果" v-model="resultDialog.visible" width="50%">
 | 
					        <el-dialog title="执行结果" v-model="resultDialog.visible" width="50%">
 | 
				
			||||||
            <div style="white-space: pre-line; padding: 10px; color: #000000">
 | 
					            <div style="white-space: pre-line; padding: 10px; color: #000000">
 | 
				
			||||||
                <!-- {{ resultDialog.result }} -->
 | 
					 | 
				
			||||||
                <el-input v-model="resultDialog.result" :rows="20" type="textarea" />
 | 
					                <el-input v-model="resultDialog.result" :rows="20" type="textarea" />
 | 
				
			||||||
            </div>
 | 
					            </div>
 | 
				
			||||||
        </el-dialog>
 | 
					        </el-dialog>
 | 
				
			||||||
@@ -97,12 +113,12 @@
 | 
				
			|||||||
            v-if="terminalDialog.visible"
 | 
					            v-if="terminalDialog.visible"
 | 
				
			||||||
            title="终端"
 | 
					            title="终端"
 | 
				
			||||||
            v-model="terminalDialog.visible"
 | 
					            v-model="terminalDialog.visible"
 | 
				
			||||||
            width="70%"
 | 
					            width="80%"
 | 
				
			||||||
            :close-on-click-modal="false"
 | 
					            :close-on-click-modal="false"
 | 
				
			||||||
            :modal="false"
 | 
					            :modal="false"
 | 
				
			||||||
            @close="closeTermnial"
 | 
					            @close="closeTermnial"
 | 
				
			||||||
        >
 | 
					        >
 | 
				
			||||||
            <ssh-terminal ref="terminal" :cmd="terminalDialog.cmd" :machineId="terminalDialog.machineId" height="600px" />
 | 
					            <ssh-terminal ref="terminal" :cmd="terminalDialog.cmd" :machineId="terminalDialog.machineId" height="560px" />
 | 
				
			||||||
        </el-dialog>
 | 
					        </el-dialog>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        <script-edit
 | 
					        <script-edit
 | 
				
			||||||
@@ -196,8 +212,10 @@ export default defineComponent({
 | 
				
			|||||||
            // 如果存在参数,则弹窗输入参数后执行
 | 
					            // 如果存在参数,则弹窗输入参数后执行
 | 
				
			||||||
            if (script.params) {
 | 
					            if (script.params) {
 | 
				
			||||||
                state.scriptParamsDialog.paramsFormItem = JSON.parse(script.params);
 | 
					                state.scriptParamsDialog.paramsFormItem = JSON.parse(script.params);
 | 
				
			||||||
                state.scriptParamsDialog.visible = true;
 | 
					                if (state.scriptParamsDialog.paramsFormItem && state.scriptParamsDialog.paramsFormItem.length > 0) {
 | 
				
			||||||
                return;
 | 
					                    state.scriptParamsDialog.visible = true;
 | 
				
			||||||
 | 
					                    return;
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            run(script);
 | 
					            run(script);
 | 
				
			||||||
@@ -268,8 +286,6 @@ export default defineComponent({
 | 
				
			|||||||
        const closeTermnial = () => {
 | 
					        const closeTermnial = () => {
 | 
				
			||||||
            state.terminalDialog.visible = false;
 | 
					            state.terminalDialog.visible = false;
 | 
				
			||||||
            state.terminalDialog.machineId = 0;
 | 
					            state.terminalDialog.machineId = 0;
 | 
				
			||||||
            // const t: any = this.$refs['terminal']
 | 
					 | 
				
			||||||
            // t.closeAll()
 | 
					 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        /**
 | 
					        /**
 | 
				
			||||||
@@ -295,8 +311,6 @@ export default defineComponent({
 | 
				
			|||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const submitSuccess = () => {
 | 
					        const submitSuccess = () => {
 | 
				
			||||||
            // this.delChoose()
 | 
					 | 
				
			||||||
            // this.search()
 | 
					 | 
				
			||||||
            getScripts();
 | 
					            getScripts();
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -326,6 +340,7 @@ export default defineComponent({
 | 
				
			|||||||
            context.emit('update:machineId', null);
 | 
					            context.emit('update:machineId', null);
 | 
				
			||||||
            context.emit('cancel');
 | 
					            context.emit('cancel');
 | 
				
			||||||
            state.scriptTable = [];
 | 
					            state.scriptTable = [];
 | 
				
			||||||
 | 
					            state.scriptParamsDialog.paramsFormItem = [];
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@ import { FitAddon } from 'xterm-addon-fit';
 | 
				
			|||||||
import { getSession } from '@/common/utils/storage.ts';
 | 
					import { getSession } from '@/common/utils/storage.ts';
 | 
				
			||||||
import config from '@/common/config';
 | 
					import config from '@/common/config';
 | 
				
			||||||
import { useStore } from '@/store/index.ts';
 | 
					import { useStore } from '@/store/index.ts';
 | 
				
			||||||
import { toRefs, watch, computed, reactive, defineComponent, onMounted, onBeforeUnmount } from 'vue';
 | 
					import { nextTick, toRefs, watch, computed, reactive, defineComponent, onMounted, onBeforeUnmount } from 'vue';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
    name: 'SshTerminal',
 | 
					    name: 'SshTerminal',
 | 
				
			||||||
@@ -27,22 +27,20 @@ export default defineComponent({
 | 
				
			|||||||
            socket: null as any,
 | 
					            socket: null as any,
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const resize = 1;
 | 
				
			||||||
 | 
					        const data = 2;
 | 
				
			||||||
 | 
					        const ping = 3;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        watch(props, (newValue) => {
 | 
					        watch(props, (newValue) => {
 | 
				
			||||||
            state.machineId = newValue.machineId;
 | 
					            state.machineId = newValue.machineId;
 | 
				
			||||||
            state.cmd = newValue.cmd;
 | 
					            state.cmd = newValue.cmd;
 | 
				
			||||||
            state.height = newValue.height;
 | 
					            state.height = newValue.height;
 | 
				
			||||||
            if (state.machineId) {
 | 
					 | 
				
			||||||
                initSocket();
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        onMounted(() => {
 | 
					        onMounted(() => {
 | 
				
			||||||
            state.machineId = props.machineId;
 | 
					            state.machineId = props.machineId;
 | 
				
			||||||
            state.height = props.height;
 | 
					            state.height = props.height;
 | 
				
			||||||
            state.cmd = props.cmd;
 | 
					            state.cmd = props.cmd;
 | 
				
			||||||
            if (state.machineId) {
 | 
					 | 
				
			||||||
                initSocket();
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        onBeforeUnmount(() => {
 | 
					        onBeforeUnmount(() => {
 | 
				
			||||||
@@ -56,13 +54,17 @@ export default defineComponent({
 | 
				
			|||||||
            return store.state.themeConfig.themeConfig;
 | 
					            return store.state.themeConfig.themeConfig;
 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        nextTick(() => {
 | 
				
			||||||
 | 
					            initXterm();
 | 
				
			||||||
 | 
					            initSocket();
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        function initXterm() {
 | 
					        function initXterm() {
 | 
				
			||||||
            const term: any = new Terminal({
 | 
					            const term: any = new Terminal({
 | 
				
			||||||
                fontSize: getThemeConfig.value.terminalFontSize || 15,
 | 
					                fontSize: getThemeConfig.value.terminalFontSize || 15,
 | 
				
			||||||
                // fontWeight: getThemeConfig.value.terminalFontWeight || 'normal',
 | 
					                fontWeight: getThemeConfig.value.terminalFontWeight || 'normal',
 | 
				
			||||||
                fontFamily: 'JetBrainsMono, Consolas, Menlo, Monaco',
 | 
					                fontFamily: 'JetBrainsMono, monaco, Consolas, Lucida Console, monospace',
 | 
				
			||||||
                cursorBlink: true,
 | 
					                cursorBlink: true,
 | 
				
			||||||
                // cursorStyle: 'underline', //光标样式
 | 
					 | 
				
			||||||
                disableStdin: false,
 | 
					                disableStdin: false,
 | 
				
			||||||
                theme: {
 | 
					                theme: {
 | 
				
			||||||
                    foreground: getThemeConfig.value.terminalForeground || '#7e9192', //字体
 | 
					                    foreground: getThemeConfig.value.terminalForeground || '#7e9192', //字体
 | 
				
			||||||
@@ -82,6 +84,14 @@ export default defineComponent({
 | 
				
			|||||||
                try {
 | 
					                try {
 | 
				
			||||||
                    // 窗口大小改变时,触发xterm的resize方法使自适应
 | 
					                    // 窗口大小改变时,触发xterm的resize方法使自适应
 | 
				
			||||||
                    fitAddon.fit();
 | 
					                    fitAddon.fit();
 | 
				
			||||||
 | 
					                    if (state.term) {
 | 
				
			||||||
 | 
					                        state.term.focus();
 | 
				
			||||||
 | 
					                        send({
 | 
				
			||||||
 | 
					                            type: resize,
 | 
				
			||||||
 | 
					                            Cols: parseInt(state.term.cols),
 | 
				
			||||||
 | 
					                            Rows: parseInt(state.term.rows),
 | 
				
			||||||
 | 
					                        });
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
                } catch (e) {
 | 
					                } catch (e) {
 | 
				
			||||||
                    console.log(e);
 | 
					                    console.log(e);
 | 
				
			||||||
                }
 | 
					                }
 | 
				
			||||||
@@ -104,69 +114,52 @@ export default defineComponent({
 | 
				
			|||||||
            term.onData((key: any) => {
 | 
					            term.onData((key: any) => {
 | 
				
			||||||
                sendCmd(key);
 | 
					                sendCmd(key);
 | 
				
			||||||
            });
 | 
					            });
 | 
				
			||||||
            // 为解决窗体resize方法才会向后端发送列数和行数,所以页面加载时也要触发此方法
 | 
					 | 
				
			||||||
            send({
 | 
					 | 
				
			||||||
                type: 'resize',
 | 
					 | 
				
			||||||
                Cols: parseInt(term.cols),
 | 
					 | 
				
			||||||
                Rows: parseInt(term.rows),
 | 
					 | 
				
			||||||
            });
 | 
					 | 
				
			||||||
            // 如果有初始要执行的命令,则发送执行命令
 | 
					 | 
				
			||||||
            if (state.cmd) {
 | 
					 | 
				
			||||||
                sendCmd(state.cmd + ' \r');
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        let pingInterval: any;
 | 
				
			||||||
        function initSocket() {
 | 
					        function initSocket() {
 | 
				
			||||||
            state.socket = new WebSocket(`${config.baseWsUrl}/machines/${state.machineId}/terminal?token=${getSession('token')}`);
 | 
					            state.socket = new WebSocket(
 | 
				
			||||||
 | 
					                `${config.baseWsUrl}/machines/${state.machineId}/terminal?token=${getSession('token')}&cols=${state.term.cols}&rows=${
 | 
				
			||||||
 | 
					                    state.term.rows
 | 
				
			||||||
 | 
					                }`
 | 
				
			||||||
 | 
					            );
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // 监听socket连接
 | 
					            // 监听socket连接
 | 
				
			||||||
            state.socket.onopen = open;
 | 
					            state.socket.onopen = () => {
 | 
				
			||||||
 | 
					                // 如果有初始要执行的命令,则发送执行命令
 | 
				
			||||||
 | 
					                if (state.cmd) {
 | 
				
			||||||
 | 
					                    sendCmd(state.cmd + ' \r');
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					                // 开启心跳
 | 
				
			||||||
 | 
					                pingInterval = setInterval(() => {
 | 
				
			||||||
 | 
					                    send({ type: ping, msg: 'ping' });
 | 
				
			||||||
 | 
					                }, 8000);
 | 
				
			||||||
 | 
					            };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // 监听socket错误信息
 | 
					            // 监听socket错误信息
 | 
				
			||||||
            state.socket.onerror = error;
 | 
					            state.socket.onerror = (e: any) => {
 | 
				
			||||||
            // 监听socket消息
 | 
					                console.log('连接错误', e);
 | 
				
			||||||
            state.socket.onmessage = getMessage;
 | 
					            };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            state.socket.onclose = () => {
 | 
				
			||||||
 | 
					                if (state.term) {
 | 
				
			||||||
 | 
					                    state.term.writeln('\r\n\x1b[31m提示: 连接已关闭...');
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					                if (pingInterval) {
 | 
				
			||||||
 | 
					                    clearInterval(pingInterval);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // 发送socket消息
 | 
					            // 发送socket消息
 | 
				
			||||||
            state.socket.onsend = send;
 | 
					            state.socket.onsend = send;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            // 监听socket消息
 | 
				
			||||||
 | 
					            state.socket.onmessage = getMessage;
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        function open() {
 | 
					        function getMessage(msg: any) {
 | 
				
			||||||
            console.log('socket连接成功');
 | 
					            // msg.data是真正后端返回的数据
 | 
				
			||||||
            initXterm();
 | 
					            state.term.write(msg.data);
 | 
				
			||||||
            //开启心跳
 | 
					 | 
				
			||||||
            //   this.start();
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        function error() {
 | 
					 | 
				
			||||||
            console.log('连接错误');
 | 
					 | 
				
			||||||
            //重连
 | 
					 | 
				
			||||||
            // reconnect();
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        function close() {
 | 
					 | 
				
			||||||
            if (state.socket) {
 | 
					 | 
				
			||||||
                state.socket.close();
 | 
					 | 
				
			||||||
                console.log('socket关闭');
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            //重连
 | 
					 | 
				
			||||||
            //   this.reconnect()
 | 
					 | 
				
			||||||
        }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        function getMessage(msg: string) {
 | 
					 | 
				
			||||||
            //   console.log(msg)
 | 
					 | 
				
			||||||
            state.term.write(msg['data']);
 | 
					 | 
				
			||||||
            //msg是返回的数据
 | 
					 | 
				
			||||||
            //   msg = JSON.parse(msg.data);
 | 
					 | 
				
			||||||
            //   this.socket.send("ping");//有事没事ping一下,看看ws还活着没
 | 
					 | 
				
			||||||
            //   //switch用于处理返回的数据,根据返回数据的格式去判断
 | 
					 | 
				
			||||||
            //   switch (msg["operation"]) {
 | 
					 | 
				
			||||||
            //     case "stdout":
 | 
					 | 
				
			||||||
            //       this.term.write(msg["data"]);//这里write也许不是固定的,失败后找后端看一下该怎么往term里面write
 | 
					 | 
				
			||||||
            //       break;
 | 
					 | 
				
			||||||
            //     default:
 | 
					 | 
				
			||||||
            //       console.error("Unexpected message type:", msg);//但是错误是固定的。。。。
 | 
					 | 
				
			||||||
            //   }
 | 
					 | 
				
			||||||
            //收到服务器信息,心跳重置
 | 
					 | 
				
			||||||
            //   this.reset();
 | 
					 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        function send(msg: any) {
 | 
					        function send(msg: any) {
 | 
				
			||||||
@@ -175,11 +168,18 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        function sendCmd(key: any) {
 | 
					        function sendCmd(key: any) {
 | 
				
			||||||
            send({
 | 
					            send({
 | 
				
			||||||
                type: 'cmd',
 | 
					                type: data,
 | 
				
			||||||
                msg: key,
 | 
					                msg: key,
 | 
				
			||||||
            });
 | 
					            });
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        function close() {
 | 
				
			||||||
 | 
					            if (state.socket) {
 | 
				
			||||||
 | 
					                state.socket.close();
 | 
				
			||||||
 | 
					                console.log('socket关闭');
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        function closeAll() {
 | 
					        function closeAll() {
 | 
				
			||||||
            close();
 | 
					            close();
 | 
				
			||||||
            if (state.term) {
 | 
					            if (state.term) {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -3,6 +3,7 @@ import Api from '@/common/Api';
 | 
				
			|||||||
export const machineApi = {
 | 
					export const machineApi = {
 | 
				
			||||||
    // 获取权限列表
 | 
					    // 获取权限列表
 | 
				
			||||||
    list: Api.create("/machines", 'get'),
 | 
					    list: Api.create("/machines", 'get'),
 | 
				
			||||||
 | 
					    getMachinePwd: Api.create("/machines/{id}/pwd", 'get'),
 | 
				
			||||||
    info: Api.create("/machines/{id}/sysinfo", 'get'),
 | 
					    info: Api.create("/machines/{id}/sysinfo", 'get'),
 | 
				
			||||||
    stats: Api.create("/machines/{id}/stats", 'get'),
 | 
					    stats: Api.create("/machines/{id}/stats", 'get'),
 | 
				
			||||||
    process: Api.create("/machines/{id}/process", 'get'),
 | 
					    process: Api.create("/machines/{id}/process", 'get'),
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -120,7 +120,7 @@
 | 
				
			|||||||
            </template>
 | 
					            </template>
 | 
				
			||||||
        </el-dialog>
 | 
					        </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" />
 | 
					            <json-edit v-model="jsoneditorDialog.doc" />
 | 
				
			||||||
        </el-dialog>
 | 
					        </el-dialog>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -61,7 +61,6 @@ import { mongoApi } from './api';
 | 
				
			|||||||
import { projectApi } from '../project/api.ts';
 | 
					import { projectApi } from '../project/api.ts';
 | 
				
			||||||
import { machineApi } from '../machine/api.ts';
 | 
					import { machineApi } from '../machine/api.ts';
 | 
				
			||||||
import { ElMessage } from 'element-plus';
 | 
					import { ElMessage } from 'element-plus';
 | 
				
			||||||
import { RsaEncrypt } from '@/common/rsa';
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
    name: 'MongoEdit',
 | 
					    name: 'MongoEdit',
 | 
				
			||||||
@@ -181,7 +180,7 @@ export default defineComponent({
 | 
				
			|||||||
            mongoForm.value.validate(async (valid: boolean) => {
 | 
					            mongoForm.value.validate(async (valid: boolean) => {
 | 
				
			||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    const reqForm = { ...state.form };
 | 
					                    const reqForm = { ...state.form };
 | 
				
			||||||
                    reqForm.uri = await RsaEncrypt(reqForm.uri);
 | 
					                    // reqForm.uri = await RsaEncrypt(reqForm.uri);
 | 
				
			||||||
                    mongoApi.saveMongo.request(reqForm).then(() => {
 | 
					                    mongoApi.saveMongo.request(reqForm).then(() => {
 | 
				
			||||||
                        ElMessage.success('保存成功');
 | 
					                        ElMessage.success('保存成功');
 | 
				
			||||||
                        emit('val-change', state.form);
 | 
					                        emit('val-change', state.form);
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -250,7 +250,6 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        onMounted(async () => {
 | 
					        onMounted(async () => {
 | 
				
			||||||
            search();
 | 
					            search();
 | 
				
			||||||
            state.projects = await projectApi.accountProjects.request(null);
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const handlePageChange = (curPage: number) => {
 | 
					        const handlePageChange = (curPage: number) => {
 | 
				
			||||||
@@ -371,7 +370,8 @@ export default defineComponent({
 | 
				
			|||||||
            state.total = res.total;
 | 
					            state.total = res.total;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const editMongo = (isAdd = false) => {
 | 
					        const editMongo = async (isAdd = false) => {
 | 
				
			||||||
 | 
					            state.projects = await projectApi.accountProjects.request(null);
 | 
				
			||||||
            if (isAdd) {
 | 
					            if (isAdd) {
 | 
				
			||||||
                state.mongoEditDialog.data = null;
 | 
					                state.mongoEditDialog.data = null;
 | 
				
			||||||
                state.mongoEditDialog.title = '新增mongo';
 | 
					                state.mongoEditDialog.title = '新增mongo';
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -25,7 +25,7 @@
 | 
				
			|||||||
                </div>
 | 
					                </div>
 | 
				
			||||||
            </div>
 | 
					            </div>
 | 
				
			||||||
            <el-table :data="projects" @current-change="choose" ref="table" style="width: 100%">
 | 
					            <el-table :data="projects" @current-change="choose" ref="table" style="width: 100%">
 | 
				
			||||||
                <el-table-column label="选择" width="50px">
 | 
					                <el-table-column label="选择" width="55px">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
					                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
				
			||||||
                            <i></i>
 | 
					                            <i></i>
 | 
				
			||||||
@@ -152,7 +152,7 @@
 | 
				
			|||||||
                            :remote-method="getAccount"
 | 
					                            :remote-method="getAccount"
 | 
				
			||||||
                            v-model="showMemDialog.memForm.accountId"
 | 
					                            v-model="showMemDialog.memForm.accountId"
 | 
				
			||||||
                            filterable
 | 
					                            filterable
 | 
				
			||||||
                            placeholder="请选择"
 | 
					                            placeholder="请输入账号模糊搜索并选择"
 | 
				
			||||||
                        >
 | 
					                        >
 | 
				
			||||||
                            <el-option v-for="item in showMemDialog.accounts" :key="item.id" :label="item.username" :value="item.id"> </el-option>
 | 
					                            <el-option v-for="item in showMemDialog.accounts" :key="item.id" :label="item.username" :value="item.id"> </el-option>
 | 
				
			||||||
                        </el-select>
 | 
					                        </el-select>
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,313 +0,0 @@
 | 
				
			|||||||
<template>
 | 
					 | 
				
			||||||
    <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" :show-close="false" width="750px" :destroy-on-close="true">
 | 
					 | 
				
			||||||
        <el-form label-width="85px">
 | 
					 | 
				
			||||||
            <el-form-item prop="key" label="key:">
 | 
					 | 
				
			||||||
                <el-input :disabled="operationType == 2" v-model="key.key"></el-input>
 | 
					 | 
				
			||||||
            </el-form-item>
 | 
					 | 
				
			||||||
            <el-form-item prop="timed" label="过期时间:">
 | 
					 | 
				
			||||||
                <el-input v-model.number="key.timed" type="number"></el-input>
 | 
					 | 
				
			||||||
            </el-form-item>
 | 
					 | 
				
			||||||
            <el-form-item prop="dataType" label="数据类型:">
 | 
					 | 
				
			||||||
                <el-select :disabled="operationType == 2" style="width: 100%" v-model="key.type" placeholder="请选择数据类型">
 | 
					 | 
				
			||||||
                    <el-option key="string" label="string" value="string"> </el-option>
 | 
					 | 
				
			||||||
                    <el-option key="hash" label="hash" value="hash"> </el-option>
 | 
					 | 
				
			||||||
                    <el-option key="set" label="set" value="set"> </el-option>
 | 
					 | 
				
			||||||
                </el-select>
 | 
					 | 
				
			||||||
            </el-form-item>
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            <el-form-item v-if="key.type == 'string'" prop="value" label="内容:">
 | 
					 | 
				
			||||||
                <div id="string-value-text" style="width: 100%">
 | 
					 | 
				
			||||||
                    <el-input class="json-text" v-model="string.value" type="textarea" :autosize="{ minRows: 10, maxRows: 20 }"></el-input>
 | 
					 | 
				
			||||||
                    <el-select class="text-type-select" @change="onChangeTextType" v-model="string.type">
 | 
					 | 
				
			||||||
                        <el-option key="text" label="text" value="text"> </el-option>
 | 
					 | 
				
			||||||
                        <el-option key="json" label="json" value="json"> </el-option>
 | 
					 | 
				
			||||||
                    </el-select>
 | 
					 | 
				
			||||||
                </div>
 | 
					 | 
				
			||||||
            </el-form-item>
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            <span v-if="key.type == 'hash'">
 | 
					 | 
				
			||||||
                <el-button @click="onAddHashValue" icon="plus" size="small" plain class="mt10">添加</el-button>
 | 
					 | 
				
			||||||
                <el-table :data="hash.value" stripe style="width: 100%">
 | 
					 | 
				
			||||||
                    <el-table-column prop="key" label="key" width>
 | 
					 | 
				
			||||||
                        <template #default="scope">
 | 
					 | 
				
			||||||
                            <el-input v-model="scope.row.key" clearable size="small"></el-input>
 | 
					 | 
				
			||||||
                        </template>
 | 
					 | 
				
			||||||
                    </el-table-column>
 | 
					 | 
				
			||||||
                    <el-table-column prop="value" label="value" min-width="200">
 | 
					 | 
				
			||||||
                        <template #default="scope">
 | 
					 | 
				
			||||||
                            <el-input
 | 
					 | 
				
			||||||
                                v-model="scope.row.value"
 | 
					 | 
				
			||||||
                                clearable
 | 
					 | 
				
			||||||
                                type="textarea"
 | 
					 | 
				
			||||||
                                :autosize="{ minRows: 2, maxRows: 10 }"
 | 
					 | 
				
			||||||
                                size="small"
 | 
					 | 
				
			||||||
                            ></el-input>
 | 
					 | 
				
			||||||
                        </template>
 | 
					 | 
				
			||||||
                    </el-table-column>
 | 
					 | 
				
			||||||
                    <el-table-column label="操作" width="90">
 | 
					 | 
				
			||||||
                        <template #default="scope">
 | 
					 | 
				
			||||||
                            <el-button type="danger" @click="hash.value.splice(scope.$index, 1)" icon="delete" size="small" plain>删除</el-button>
 | 
					 | 
				
			||||||
                        </template>
 | 
					 | 
				
			||||||
                    </el-table-column>
 | 
					 | 
				
			||||||
                </el-table>
 | 
					 | 
				
			||||||
            </span>
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            <span v-if="key.type == 'set'">
 | 
					 | 
				
			||||||
                <el-button @click="onAddSetValue" icon="plus" size="small" plain class="mt10">添加</el-button>
 | 
					 | 
				
			||||||
                <el-table :data="set.value" stripe style="width: 100%">
 | 
					 | 
				
			||||||
                    <el-table-column prop="value" label="value" min-width="200">
 | 
					 | 
				
			||||||
                        <template #default="scope">
 | 
					 | 
				
			||||||
                            <el-input
 | 
					 | 
				
			||||||
                                v-model="scope.row.value"
 | 
					 | 
				
			||||||
                                clearable
 | 
					 | 
				
			||||||
                                type="textarea"
 | 
					 | 
				
			||||||
                                :autosize="{ minRows: 2, maxRows: 10 }"
 | 
					 | 
				
			||||||
                                size="small"
 | 
					 | 
				
			||||||
                            ></el-input>
 | 
					 | 
				
			||||||
                        </template>
 | 
					 | 
				
			||||||
                    </el-table-column>
 | 
					 | 
				
			||||||
                    <el-table-column label="操作" width="90">
 | 
					 | 
				
			||||||
                        <template #default="scope">
 | 
					 | 
				
			||||||
                            <el-button type="danger" @click="set.value.splice(scope.$index, 1)" icon="delete" size="small" plain>删除</el-button>
 | 
					 | 
				
			||||||
                        </template>
 | 
					 | 
				
			||||||
                    </el-table-column>
 | 
					 | 
				
			||||||
                </el-table>
 | 
					 | 
				
			||||||
            </span>
 | 
					 | 
				
			||||||
        </el-form>
 | 
					 | 
				
			||||||
        <template #footer>
 | 
					 | 
				
			||||||
            <div class="dialog-footer">
 | 
					 | 
				
			||||||
                <el-button @click="cancel()">取 消</el-button>
 | 
					 | 
				
			||||||
                <el-button @click="saveValue" type="primary" v-auth="'redis:data:save'">确 定</el-button>
 | 
					 | 
				
			||||||
            </div>
 | 
					 | 
				
			||||||
        </template>
 | 
					 | 
				
			||||||
    </el-dialog>
 | 
					 | 
				
			||||||
</template>
 | 
					 | 
				
			||||||
<script lang="ts">
 | 
					 | 
				
			||||||
import { defineComponent, reactive, watch, toRefs } from 'vue';
 | 
					 | 
				
			||||||
import { redisApi } from './api';
 | 
					 | 
				
			||||||
import { ElMessage } from 'element-plus';
 | 
					 | 
				
			||||||
import { isTrue, notEmpty } from '@/common/assert';
 | 
					 | 
				
			||||||
import { formatJsonString } from '@/common/utils/format';
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
export default defineComponent({
 | 
					 | 
				
			||||||
    name: 'DateEdit',
 | 
					 | 
				
			||||||
    components: {},
 | 
					 | 
				
			||||||
    props: {
 | 
					 | 
				
			||||||
        visible: {
 | 
					 | 
				
			||||||
            type: Boolean,
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        title: {
 | 
					 | 
				
			||||||
            type: String,
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        redisId: {
 | 
					 | 
				
			||||||
            type: [Number],
 | 
					 | 
				
			||||||
            require: true,
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        keyInfo: {
 | 
					 | 
				
			||||||
            type: [Object],
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        // 操作类型,1:新增,2:修改
 | 
					 | 
				
			||||||
        operationType: {
 | 
					 | 
				
			||||||
            type: [Number],
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        stringValue: {
 | 
					 | 
				
			||||||
            type: [String],
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        setValue: {
 | 
					 | 
				
			||||||
            type: [Array, Object],
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
        hashValue: {
 | 
					 | 
				
			||||||
            type: [Array, Object],
 | 
					 | 
				
			||||||
        },
 | 
					 | 
				
			||||||
    },
 | 
					 | 
				
			||||||
    emits: ['valChange', 'cancel', 'update:visible'],
 | 
					 | 
				
			||||||
    setup(props: any, { emit }) {
 | 
					 | 
				
			||||||
        const state = reactive({
 | 
					 | 
				
			||||||
            dialogVisible: false,
 | 
					 | 
				
			||||||
            operationType: 1,
 | 
					 | 
				
			||||||
            redisId: '',
 | 
					 | 
				
			||||||
            key: {
 | 
					 | 
				
			||||||
                key: '',
 | 
					 | 
				
			||||||
                type: 'string',
 | 
					 | 
				
			||||||
                timed: -1,
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            string: {
 | 
					 | 
				
			||||||
                type: 'text',
 | 
					 | 
				
			||||||
                value: '',
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            hash: {
 | 
					 | 
				
			||||||
                value: [
 | 
					 | 
				
			||||||
                    {
 | 
					 | 
				
			||||||
                        key: '',
 | 
					 | 
				
			||||||
                        value: '',
 | 
					 | 
				
			||||||
                    },
 | 
					 | 
				
			||||||
                ],
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            set: {
 | 
					 | 
				
			||||||
                value: [{ value: '' }],
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
        });
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        const cancel = () => {
 | 
					 | 
				
			||||||
            emit('update:visible', false);
 | 
					 | 
				
			||||||
            emit('cancel');
 | 
					 | 
				
			||||||
            setTimeout(() => {
 | 
					 | 
				
			||||||
                state.key = {
 | 
					 | 
				
			||||||
                    key: '',
 | 
					 | 
				
			||||||
                    type: 'string',
 | 
					 | 
				
			||||||
                    timed: -1,
 | 
					 | 
				
			||||||
                };
 | 
					 | 
				
			||||||
                state.string.value = '';
 | 
					 | 
				
			||||||
                state.string.type = 'text';
 | 
					 | 
				
			||||||
                state.hash.value = [
 | 
					 | 
				
			||||||
                    {
 | 
					 | 
				
			||||||
                        key: '',
 | 
					 | 
				
			||||||
                        value: '',
 | 
					 | 
				
			||||||
                    },
 | 
					 | 
				
			||||||
                ];
 | 
					 | 
				
			||||||
            }, 500);
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.visible,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                state.dialogVisible = val;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.redisId,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                state.redisId = val;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.operationType,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                state.operationType = val;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.keyInfo,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                if (val) {
 | 
					 | 
				
			||||||
                    state.key = { ...val };
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            {
 | 
					 | 
				
			||||||
                deep: true, // 深度监听的参数
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.stringValue,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                if (val) {
 | 
					 | 
				
			||||||
                    state.string.value = val;
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            {
 | 
					 | 
				
			||||||
                deep: true, // 深度监听的参数
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.setValue,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                if (val) {
 | 
					 | 
				
			||||||
                    state.set.value = val;
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            {
 | 
					 | 
				
			||||||
                deep: true, // 深度监听的参数
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        watch(
 | 
					 | 
				
			||||||
            () => props.hashValue,
 | 
					 | 
				
			||||||
            (val) => {
 | 
					 | 
				
			||||||
                if (val) {
 | 
					 | 
				
			||||||
                    state.hash.value = val;
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            {
 | 
					 | 
				
			||||||
                deep: true, // 深度监听的参数
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        );
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        const saveValue = async () => {
 | 
					 | 
				
			||||||
            notEmpty(state.key.key, 'key不能为空');
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            if (state.key.type == 'string') {
 | 
					 | 
				
			||||||
                notEmpty(state.string.value, 'value不能为空');
 | 
					 | 
				
			||||||
                const sv = { value: formatJsonString(state.string.value, true), id: state.redisId };
 | 
					 | 
				
			||||||
                Object.assign(sv, state.key);
 | 
					 | 
				
			||||||
                await redisApi.saveStringValue.request(sv);
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            if (state.key.type == 'hash') {
 | 
					 | 
				
			||||||
                isTrue(state.hash.value.length > 0, 'hash内容不能为空');
 | 
					 | 
				
			||||||
                const sv = { value: state.hash.value, id: state.redisId };
 | 
					 | 
				
			||||||
                Object.assign(sv, state.key);
 | 
					 | 
				
			||||||
                await redisApi.saveHashValue.request(sv);
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            if (state.key.type == 'set') {
 | 
					 | 
				
			||||||
                isTrue(state.set.value.length > 0, 'set内容不能为空');
 | 
					 | 
				
			||||||
                const sv = { value: state.set.value.map((x) => x.value), id: state.redisId };
 | 
					 | 
				
			||||||
                Object.assign(sv, state.key);
 | 
					 | 
				
			||||||
                await redisApi.saveSetValue.request(sv);
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            ElMessage.success('数据保存成功');
 | 
					 | 
				
			||||||
            cancel();
 | 
					 | 
				
			||||||
            emit('valChange');
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        const onAddHashValue = () => {
 | 
					 | 
				
			||||||
            state.hash.value.push({ key: '', value: '' });
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        const onAddSetValue = () => {
 | 
					 | 
				
			||||||
            state.set.value.push({ value: '' });
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        // 更改文本类型
 | 
					 | 
				
			||||||
        const onChangeTextType = (val: string) => {
 | 
					 | 
				
			||||||
            if (val == 'json') {
 | 
					 | 
				
			||||||
                state.string.value = formatJsonString(state.string.value, false);
 | 
					 | 
				
			||||||
                return;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            if (val == 'text') {
 | 
					 | 
				
			||||||
                state.string.value = formatJsonString(state.string.value, true);
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {
 | 
					 | 
				
			||||||
            ...toRefs(state),
 | 
					 | 
				
			||||||
            saveValue,
 | 
					 | 
				
			||||||
            cancel,
 | 
					 | 
				
			||||||
            onAddHashValue,
 | 
					 | 
				
			||||||
            onAddSetValue,
 | 
					 | 
				
			||||||
            onChangeTextType,
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
    },
 | 
					 | 
				
			||||||
});
 | 
					 | 
				
			||||||
</script>
 | 
					 | 
				
			||||||
<style lang="scss">
 | 
					 | 
				
			||||||
#string-value-text {
 | 
					 | 
				
			||||||
    flex-grow: 1;
 | 
					 | 
				
			||||||
    display: flex;
 | 
					 | 
				
			||||||
    position: relative;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
    .text-type-select {
 | 
					 | 
				
			||||||
        position: absolute;
 | 
					 | 
				
			||||||
        z-index: 2;
 | 
					 | 
				
			||||||
        right: 10px;
 | 
					 | 
				
			||||||
        top: 10px;
 | 
					 | 
				
			||||||
        max-width: 70px;
 | 
					 | 
				
			||||||
    }
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
</style>
 | 
					 | 
				
			||||||
@@ -23,7 +23,7 @@
 | 
				
			|||||||
                        <el-form class="search-form" label-position="right" :inline="true" label-width="60px">
 | 
					                        <el-form class="search-form" label-position="right" :inline="true" label-width="60px">
 | 
				
			||||||
                            <el-form-item label="key" label-width="40px">
 | 
					                            <el-form-item label="key" label-width="40px">
 | 
				
			||||||
                                <el-input
 | 
					                                <el-input
 | 
				
			||||||
                                    placeholder="支持*模糊key"
 | 
					                                    placeholder="match 支持*模糊key"
 | 
				
			||||||
                                    style="width: 240px"
 | 
					                                    style="width: 240px"
 | 
				
			||||||
                                    v-model="scanParam.match"
 | 
					                                    v-model="scanParam.match"
 | 
				
			||||||
                                    @clear="clear()"
 | 
					                                    @clear="clear()"
 | 
				
			||||||
@@ -36,7 +36,14 @@
 | 
				
			|||||||
                            <el-form-item>
 | 
					                            <el-form-item>
 | 
				
			||||||
                                <el-button @click="searchKey()" type="success" icon="search" plain></el-button>
 | 
					                                <el-button @click="searchKey()" type="success" icon="search" plain></el-button>
 | 
				
			||||||
                                <el-button @click="scan()" icon="bottom" plain>scan</el-button>
 | 
					                                <el-button @click="scan()" icon="bottom" plain>scan</el-button>
 | 
				
			||||||
                                <el-button type="primary" icon="plus" @click="onAddData(false)" plain></el-button>
 | 
					                                <el-popover placement="right" :width="200" trigger="click">
 | 
				
			||||||
 | 
					                                    <template #reference>
 | 
				
			||||||
 | 
					                                        <el-button type="primary" icon="plus" plain></el-button>
 | 
				
			||||||
 | 
					                                    </template>
 | 
				
			||||||
 | 
					                                    <el-tag @click="onAddData('string')" :color="getTypeColor('string')" style="cursor: pointer">string</el-tag>
 | 
				
			||||||
 | 
					                                    <el-tag @click="onAddData('hash')" :color="getTypeColor('hash')" class="ml5" style="cursor: pointer">hash</el-tag>
 | 
				
			||||||
 | 
					                                    <el-tag @click="onAddData('set')" :color="getTypeColor('set')" class="ml5" style="cursor: pointer">set</el-tag>
 | 
				
			||||||
 | 
					                                </el-popover>
 | 
				
			||||||
                            </el-form-item>
 | 
					                            </el-form-item>
 | 
				
			||||||
                            <div style="float: right">
 | 
					                            <div style="float: right">
 | 
				
			||||||
                                <span>keys: {{ dbsize }}</span>
 | 
					                                <span>keys: {{ dbsize }}</span>
 | 
				
			||||||
@@ -69,17 +76,32 @@
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        <div style="text-align: center; margin-top: 10px"></div>
 | 
					        <div style="text-align: center; margin-top: 10px"></div>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        <!-- <value-dialog v-model:visible="valueDialog.visible" :keyValue="valueDialog.value" /> -->
 | 
					        <hash-value
 | 
				
			||||||
 | 
					            v-model:visible="hashValueDialog.visible"
 | 
				
			||||||
 | 
					            :operationType="dataEdit.operationType"
 | 
				
			||||||
 | 
					            :title="dataEdit.title"
 | 
				
			||||||
 | 
					            :keyInfo="dataEdit.keyInfo"
 | 
				
			||||||
 | 
					            :redisId="scanParam.id"
 | 
				
			||||||
 | 
					            @cancel="onCancelDataEdit"
 | 
				
			||||||
 | 
					            @valChange="searchKey"
 | 
				
			||||||
 | 
					        />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        <data-edit
 | 
					        <string-value
 | 
				
			||||||
            v-model:visible="dataEdit.visible"
 | 
					            v-model:visible="stringValueDialog.visible"
 | 
				
			||||||
 | 
					            :operationType="dataEdit.operationType"
 | 
				
			||||||
 | 
					            :title="dataEdit.title"
 | 
				
			||||||
 | 
					            :keyInfo="dataEdit.keyInfo"
 | 
				
			||||||
 | 
					            :redisId="scanParam.id"
 | 
				
			||||||
 | 
					            @cancel="onCancelDataEdit"
 | 
				
			||||||
 | 
					            @valChange="searchKey"
 | 
				
			||||||
 | 
					        />
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        <set-value
 | 
				
			||||||
 | 
					            v-model:visible="setValueDialog.visible"
 | 
				
			||||||
            :title="dataEdit.title"
 | 
					            :title="dataEdit.title"
 | 
				
			||||||
            :keyInfo="dataEdit.keyInfo"
 | 
					            :keyInfo="dataEdit.keyInfo"
 | 
				
			||||||
            :redisId="scanParam.id"
 | 
					            :redisId="scanParam.id"
 | 
				
			||||||
            :operationType="dataEdit.operationType"
 | 
					            :operationType="dataEdit.operationType"
 | 
				
			||||||
            :stringValue="dataEdit.stringValue"
 | 
					 | 
				
			||||||
            :setValue="dataEdit.setValue"
 | 
					 | 
				
			||||||
            :hashValue="dataEdit.hashValue"
 | 
					 | 
				
			||||||
            @valChange="searchKey"
 | 
					            @valChange="searchKey"
 | 
				
			||||||
            @cancel="onCancelDataEdit"
 | 
					            @cancel="onCancelDataEdit"
 | 
				
			||||||
        />
 | 
					        />
 | 
				
			||||||
@@ -91,13 +113,17 @@ import { redisApi } from './api';
 | 
				
			|||||||
import { toRefs, reactive, defineComponent } from 'vue';
 | 
					import { toRefs, reactive, defineComponent } from 'vue';
 | 
				
			||||||
import { ElMessage, ElMessageBox } from 'element-plus';
 | 
					import { ElMessage, ElMessageBox } from 'element-plus';
 | 
				
			||||||
import ProjectEnvSelect from '../component/ProjectEnvSelect.vue';
 | 
					import ProjectEnvSelect from '../component/ProjectEnvSelect.vue';
 | 
				
			||||||
import DataEdit from './DataEdit.vue';
 | 
					import HashValue from './HashValue.vue';
 | 
				
			||||||
 | 
					import StringValue from './StringValue.vue';
 | 
				
			||||||
 | 
					import SetValue from './SetValue.vue';
 | 
				
			||||||
import { isTrue, notBlank, notNull } from '@/common/assert';
 | 
					import { isTrue, notBlank, notNull } from '@/common/assert';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
    name: 'DataOperation',
 | 
					    name: 'DataOperation',
 | 
				
			||||||
    components: {
 | 
					    components: {
 | 
				
			||||||
        DataEdit,
 | 
					        StringValue,
 | 
				
			||||||
 | 
					        HashValue,
 | 
				
			||||||
 | 
					        SetValue,
 | 
				
			||||||
        ProjectEnvSelect,
 | 
					        ProjectEnvSelect,
 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
    setup() {
 | 
					    setup() {
 | 
				
			||||||
@@ -113,10 +139,6 @@ export default defineComponent({
 | 
				
			|||||||
                count: 10,
 | 
					                count: 10,
 | 
				
			||||||
                cursor: {},
 | 
					                cursor: {},
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            valueDialog: {
 | 
					 | 
				
			||||||
                visible: false,
 | 
					 | 
				
			||||||
                value: {},
 | 
					 | 
				
			||||||
            },
 | 
					 | 
				
			||||||
            dataEdit: {
 | 
					            dataEdit: {
 | 
				
			||||||
                visible: false,
 | 
					                visible: false,
 | 
				
			||||||
                title: '新增数据',
 | 
					                title: '新增数据',
 | 
				
			||||||
@@ -126,9 +148,15 @@ export default defineComponent({
 | 
				
			|||||||
                    timed: -1,
 | 
					                    timed: -1,
 | 
				
			||||||
                    key: '',
 | 
					                    key: '',
 | 
				
			||||||
                },
 | 
					                },
 | 
				
			||||||
                stringValue: '',
 | 
					            },
 | 
				
			||||||
                hashValue: [{ key: '', value: '' }],
 | 
					            hashValueDialog: {
 | 
				
			||||||
                setValue: [{ value: '' }],
 | 
					                visible: false,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            stringValueDialog: {
 | 
				
			||||||
 | 
					                visible: false,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            setValueDialog: {
 | 
				
			||||||
 | 
					                visible: false,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
            keys: [],
 | 
					            keys: [],
 | 
				
			||||||
            dbsize: 0,
 | 
					            dbsize: 0,
 | 
				
			||||||
@@ -158,10 +186,15 @@ export default defineComponent({
 | 
				
			|||||||
        const scan = async () => {
 | 
					        const scan = async () => {
 | 
				
			||||||
            isTrue(state.scanParam.id != null, '请先选择redis');
 | 
					            isTrue(state.scanParam.id != null, '请先选择redis');
 | 
				
			||||||
            notBlank(state.scanParam.count, 'count不能为空');
 | 
					            notBlank(state.scanParam.count, 'count不能为空');
 | 
				
			||||||
            isTrue(state.scanParam.count < 20001, 'count不能超过20000');
 | 
					
 | 
				
			||||||
 | 
					            const match = state.scanParam.match;
 | 
				
			||||||
 | 
					            if (!match || match == '*') {
 | 
				
			||||||
 | 
					                isTrue(state.scanParam.count <= 200, 'match为空或者*时, count不能超过200');
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                isTrue(state.scanParam.count <= 20000, 'count不能超过20000');
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            state.loading = true;
 | 
					            state.loading = true;
 | 
				
			||||||
 | 
					 | 
				
			||||||
            try {
 | 
					            try {
 | 
				
			||||||
                const res = await redisApi.scan.request(state.scanParam);
 | 
					                const res = await redisApi.scan.request(state.scanParam);
 | 
				
			||||||
                state.keys = res.keys;
 | 
					                state.keys = res.keys;
 | 
				
			||||||
@@ -207,60 +240,43 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        const getValue = async (row: any) => {
 | 
					        const getValue = async (row: any) => {
 | 
				
			||||||
            const type = row.type;
 | 
					            const type = row.type;
 | 
				
			||||||
            const key = row.key;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            let res: any;
 | 
					 | 
				
			||||||
            const reqParam = {
 | 
					 | 
				
			||||||
                key: row.key,
 | 
					 | 
				
			||||||
                id: state.scanParam.id,
 | 
					 | 
				
			||||||
            };
 | 
					 | 
				
			||||||
            switch (type) {
 | 
					 | 
				
			||||||
                case 'string':
 | 
					 | 
				
			||||||
                    res = await redisApi.getStringValue.request(reqParam);
 | 
					 | 
				
			||||||
                    break;
 | 
					 | 
				
			||||||
                case 'hash':
 | 
					 | 
				
			||||||
                    res = await redisApi.getHashValue.request(reqParam);
 | 
					 | 
				
			||||||
                    break;
 | 
					 | 
				
			||||||
                case 'set':
 | 
					 | 
				
			||||||
                    res = await redisApi.getSetValue.request(reqParam);
 | 
					 | 
				
			||||||
                    break;
 | 
					 | 
				
			||||||
                default:
 | 
					 | 
				
			||||||
                    res = null;
 | 
					 | 
				
			||||||
                    break;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            notNull(res, '暂不支持该类型数据查看');
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            if (type == 'string') {
 | 
					 | 
				
			||||||
                state.dataEdit.stringValue = res;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            if (type == 'set') {
 | 
					 | 
				
			||||||
                state.dataEdit.setValue = res.map((x: any) => {
 | 
					 | 
				
			||||||
                    return {
 | 
					 | 
				
			||||||
                        value: x,
 | 
					 | 
				
			||||||
                    };
 | 
					 | 
				
			||||||
                });
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
            if (type == 'hash') {
 | 
					 | 
				
			||||||
                const hash = [];
 | 
					 | 
				
			||||||
                //遍历key和value
 | 
					 | 
				
			||||||
                const keys = Object.keys(res);
 | 
					 | 
				
			||||||
                for (let i = 0; i < keys.length; i++) {
 | 
					 | 
				
			||||||
                    const key = keys[i];
 | 
					 | 
				
			||||||
                    const value = res[key];
 | 
					 | 
				
			||||||
                    hash.push({
 | 
					 | 
				
			||||||
                        key,
 | 
					 | 
				
			||||||
                        value,
 | 
					 | 
				
			||||||
                    });
 | 
					 | 
				
			||||||
                }
 | 
					 | 
				
			||||||
                state.dataEdit.hashValue = hash;
 | 
					 | 
				
			||||||
            }
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
            state.dataEdit.keyInfo.type = type;
 | 
					            state.dataEdit.keyInfo.type = type;
 | 
				
			||||||
            state.dataEdit.keyInfo.timed = row.ttl;
 | 
					            state.dataEdit.keyInfo.timed = row.ttl;
 | 
				
			||||||
            state.dataEdit.keyInfo.key = key;
 | 
					            state.dataEdit.keyInfo.key = row.key;
 | 
				
			||||||
            state.dataEdit.operationType = 2;
 | 
					            state.dataEdit.operationType = 2;
 | 
				
			||||||
            state.dataEdit.title = '修改数据';
 | 
					            state.dataEdit.title = '查看数据';
 | 
				
			||||||
            state.dataEdit.visible = true;
 | 
					
 | 
				
			||||||
 | 
					            if (type == 'hash') {
 | 
				
			||||||
 | 
					                state.hashValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else if (type == 'string') {
 | 
				
			||||||
 | 
					                state.stringValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else if (type == 'set') {
 | 
				
			||||||
 | 
					                state.setValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                ElMessage.warning('暂不支持该类型');
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onAddData = (type: string) => {
 | 
				
			||||||
 | 
					            notNull(state.scanParam.id, '请先选择redis');
 | 
				
			||||||
 | 
					            state.dataEdit.operationType = 1;
 | 
				
			||||||
 | 
					            state.dataEdit.title = '新增数据';
 | 
				
			||||||
 | 
					            state.dataEdit.keyInfo.type = type;
 | 
				
			||||||
 | 
					            state.dataEdit.keyInfo.timed = -1;
 | 
				
			||||||
 | 
					            if (type == 'hash') {
 | 
				
			||||||
 | 
					                state.hashValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else if (type == 'string') {
 | 
				
			||||||
 | 
					                state.stringValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else if (type == 'set') {
 | 
				
			||||||
 | 
					                state.setValueDialog.visible = true;
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                ElMessage.warning('暂不支持该类型');
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onCancelDataEdit = () => {
 | 
				
			||||||
 | 
					            state.dataEdit.keyInfo = {} as any;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const del = (key: string) => {
 | 
					        const del = (key: string) => {
 | 
				
			||||||
@@ -331,20 +347,6 @@ export default defineComponent({
 | 
				
			|||||||
            }
 | 
					            }
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const onAddData = () => {
 | 
					 | 
				
			||||||
            notNull(state.scanParam.id, '请先选择redis');
 | 
					 | 
				
			||||||
            state.dataEdit.operationType = 1;
 | 
					 | 
				
			||||||
            state.dataEdit.title = '新增数据';
 | 
					 | 
				
			||||||
            state.dataEdit.visible = true;
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        const onCancelDataEdit = () => {
 | 
					 | 
				
			||||||
            state.dataEdit.keyInfo = {} as any;
 | 
					 | 
				
			||||||
            state.dataEdit.stringValue = '';
 | 
					 | 
				
			||||||
            state.dataEdit.setValue = [];
 | 
					 | 
				
			||||||
            state.dataEdit.hashValue = [];
 | 
					 | 
				
			||||||
        };
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            changeProjectEnv,
 | 
					            changeProjectEnv,
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										254
									
								
								mayfly_go_web/src/views/ops/redis/HashValue.vue
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										254
									
								
								mayfly_go_web/src/views/ops/redis/HashValue.vue
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,254 @@
 | 
				
			|||||||
 | 
					<template>
 | 
				
			||||||
 | 
					    <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" width="800px" :destroy-on-close="true">
 | 
				
			||||||
 | 
					        <el-form label-width="85px">
 | 
				
			||||||
 | 
					            <el-form-item prop="key" label="key:">
 | 
				
			||||||
 | 
					                <el-input :disabled="operationType == 2" v-model="key.key"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="timed" label="过期时间:">
 | 
				
			||||||
 | 
					                <el-input v-model.number="key.timed" type="number"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="dataType" label="数据类型:">
 | 
				
			||||||
 | 
					                <el-input v-model="key.type" disabled></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            <el-row class="mt10">
 | 
				
			||||||
 | 
					                <el-form label-position="right" :inline="true">
 | 
				
			||||||
 | 
					                    <el-form-item label="field" label-width="40px" v-if="operationType == 2">
 | 
				
			||||||
 | 
					                        <el-input placeholder="支持*模糊field" style="width: 140px" v-model="scanParam.match" clearable size="small"></el-input>
 | 
				
			||||||
 | 
					                    </el-form-item>
 | 
				
			||||||
 | 
					                    <el-form-item label="count" v-if="operationType == 2">
 | 
				
			||||||
 | 
					                        <el-input placeholder="count" style="width: 62px" v-model.number="scanParam.count" size="small"></el-input>
 | 
				
			||||||
 | 
					                    </el-form-item>
 | 
				
			||||||
 | 
					                    <el-form-item>
 | 
				
			||||||
 | 
					                        <el-button v-if="operationType == 2" @click="reHscan()" type="success" icon="search" plain size="small"></el-button>
 | 
				
			||||||
 | 
					                        <el-button v-if="operationType == 2" @click="hscan()" icon="bottom" plain size="small">scan</el-button>
 | 
				
			||||||
 | 
					                        <el-button @click="onAddHashValue" icon="plus" size="small" plain>添加</el-button>
 | 
				
			||||||
 | 
					                    </el-form-item>
 | 
				
			||||||
 | 
					                    <div v-if="operationType == 2" class="mt10" style="float: right">
 | 
				
			||||||
 | 
					                        <span>fieldSize: {{ keySize }}</span>
 | 
				
			||||||
 | 
					                    </div>
 | 
				
			||||||
 | 
					                </el-form>
 | 
				
			||||||
 | 
					            </el-row>
 | 
				
			||||||
 | 
					            <el-table :data="hashValues" stripe style="width: 100%">
 | 
				
			||||||
 | 
					                <el-table-column prop="field" label="field" width>
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-input v-model="scope.row.field" clearable size="small"></el-input>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="value" label="value" min-width="200">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-input v-model="scope.row.value" clearable type="textarea" :autosize="{ minRows: 2, maxRows: 10 }" size="small"></el-input>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column label="操作" width="120">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-button v-if="operationType == 2" type="success" @click="hset(scope.row)" icon="check" size="small" plain></el-button>
 | 
				
			||||||
 | 
					                        <el-button type="danger" @click="hdel(scope.row.field, scope.$index)" icon="delete" size="small" plain></el-button>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					            </el-table>
 | 
				
			||||||
 | 
					        </el-form>
 | 
				
			||||||
 | 
					        <template #footer v-if="operationType == 1">
 | 
				
			||||||
 | 
					            <div class="dialog-footer">
 | 
				
			||||||
 | 
					                <el-button @click="cancel()">取 消</el-button>
 | 
				
			||||||
 | 
					                <el-button @click="saveValue" type="primary" v-auth="'redis:data:save'">确 定</el-button>
 | 
				
			||||||
 | 
					            </div>
 | 
				
			||||||
 | 
					        </template>
 | 
				
			||||||
 | 
					    </el-dialog>
 | 
				
			||||||
 | 
					</template>
 | 
				
			||||||
 | 
					<script lang="ts">
 | 
				
			||||||
 | 
					import { defineComponent, reactive, watch, toRefs } from 'vue';
 | 
				
			||||||
 | 
					import { redisApi } from './api';
 | 
				
			||||||
 | 
					import { ElMessage, ElMessageBox } from 'element-plus';
 | 
				
			||||||
 | 
					import { isTrue, notEmpty } from '@/common/assert';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export default defineComponent({
 | 
				
			||||||
 | 
					    name: 'HashValue',
 | 
				
			||||||
 | 
					    components: {},
 | 
				
			||||||
 | 
					    props: {
 | 
				
			||||||
 | 
					        visible: {
 | 
				
			||||||
 | 
					            type: Boolean,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        title: {
 | 
				
			||||||
 | 
					            type: String,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        // 操作类型,1:新增,2:修改
 | 
				
			||||||
 | 
					        operationType: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					            require: true,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        redisId: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					            require: true,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        keyInfo: {
 | 
				
			||||||
 | 
					            type: [Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        hashValue: {
 | 
				
			||||||
 | 
					            type: [Array, Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					    emits: ['valChange', 'cancel', 'update:visible'],
 | 
				
			||||||
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            dialogVisible: false,
 | 
				
			||||||
 | 
					            operationType: 1,
 | 
				
			||||||
 | 
					            redisId: 0,
 | 
				
			||||||
 | 
					            key: {
 | 
				
			||||||
 | 
					                key: '',
 | 
				
			||||||
 | 
					                type: 'hash',
 | 
				
			||||||
 | 
					                timed: -1,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            scanParam: {
 | 
				
			||||||
 | 
					                key: '',
 | 
				
			||||||
 | 
					                id: 0,
 | 
				
			||||||
 | 
					                cursor: 0,
 | 
				
			||||||
 | 
					                match: '',
 | 
				
			||||||
 | 
					                count: 10,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            keySize: 0,
 | 
				
			||||||
 | 
					            hashValues: [
 | 
				
			||||||
 | 
					                {
 | 
				
			||||||
 | 
					                    field: '',
 | 
				
			||||||
 | 
					                    value: '',
 | 
				
			||||||
 | 
					                },
 | 
				
			||||||
 | 
					            ],
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const cancel = () => {
 | 
				
			||||||
 | 
					            emit('update:visible', false);
 | 
				
			||||||
 | 
					            emit('cancel');
 | 
				
			||||||
 | 
					            setTimeout(() => {
 | 
				
			||||||
 | 
					                state.hashValues = [];
 | 
				
			||||||
 | 
					                state.key = {} as any;
 | 
				
			||||||
 | 
					            }, 500);
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(props, async (newValue) => {
 | 
				
			||||||
 | 
					            const visible = newValue.visible;
 | 
				
			||||||
 | 
					            state.redisId = newValue.redisId;
 | 
				
			||||||
 | 
					            state.key = newValue.keyInfo;
 | 
				
			||||||
 | 
					            state.operationType = newValue.operationType;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            if (visible && state.operationType == 2) {
 | 
				
			||||||
 | 
					                state.scanParam.id = props.redisId;
 | 
				
			||||||
 | 
					                state.scanParam.key = state.key.key;
 | 
				
			||||||
 | 
					                await reHscan();
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            state.dialogVisible = visible;
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const reHscan = async () => {
 | 
				
			||||||
 | 
					            state.scanParam.id = state.redisId;
 | 
				
			||||||
 | 
					            state.scanParam.cursor = 0;
 | 
				
			||||||
 | 
					            hscan();
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const hscan = async () => {
 | 
				
			||||||
 | 
					            const match = state.scanParam.match;
 | 
				
			||||||
 | 
					            if (!match || match == '' || match == '*') {
 | 
				
			||||||
 | 
					                if (state.scanParam.count > 100) {
 | 
				
			||||||
 | 
					                    ElMessage.error('match为空或者*时, count不能超过100');
 | 
				
			||||||
 | 
					                    return;
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                if (state.scanParam.count > 1000) {
 | 
				
			||||||
 | 
					                    ElMessage.error('count不能超过1000');
 | 
				
			||||||
 | 
					                    return;
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            const scanRes = await redisApi.hscan.request(state.scanParam);
 | 
				
			||||||
 | 
					            state.scanParam.cursor = scanRes.cursor;
 | 
				
			||||||
 | 
					            state.keySize = scanRes.keySize;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            const keys = scanRes.keys;
 | 
				
			||||||
 | 
					            const hashValue = [];
 | 
				
			||||||
 | 
					            const fieldCount = keys.length / 2;
 | 
				
			||||||
 | 
					            let nextFieldIndex = 0;
 | 
				
			||||||
 | 
					            for (let i = 0; i < fieldCount; i++) {
 | 
				
			||||||
 | 
					                hashValue.push({ field: keys[nextFieldIndex++], value: keys[nextFieldIndex++] });
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            state.hashValues = hashValue;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const hdel = async (field: any, index: any) => {
 | 
				
			||||||
 | 
					            // 如果是新增操作,则直接数组移除即可
 | 
				
			||||||
 | 
					            if (state.operationType == 1) {
 | 
				
			||||||
 | 
					                state.hashValues.splice(index, 1);
 | 
				
			||||||
 | 
					                return;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            await ElMessageBox.confirm(`确定删除[${field}]?`, '提示', {
 | 
				
			||||||
 | 
					                confirmButtonText: '确定',
 | 
				
			||||||
 | 
					                cancelButtonText: '取消',
 | 
				
			||||||
 | 
					                type: 'warning',
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					            await redisApi.hdel.request({
 | 
				
			||||||
 | 
					                id: state.redisId,
 | 
				
			||||||
 | 
					                key: state.key.key,
 | 
				
			||||||
 | 
					                field,
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					            ElMessage.success('删除成功');
 | 
				
			||||||
 | 
					            reHscan();
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const hset = async (row: any) => {
 | 
				
			||||||
 | 
					            await redisApi.saveHashValue.request({
 | 
				
			||||||
 | 
					                id: state.redisId,
 | 
				
			||||||
 | 
					                key: state.key.key,
 | 
				
			||||||
 | 
					                timed: state.key.timed,
 | 
				
			||||||
 | 
					                value: [
 | 
				
			||||||
 | 
					                    {
 | 
				
			||||||
 | 
					                        field: row.field,
 | 
				
			||||||
 | 
					                        value: row.value,
 | 
				
			||||||
 | 
					                    },
 | 
				
			||||||
 | 
					                ],
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					            ElMessage.success('保存成功');
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onAddHashValue = () => {
 | 
				
			||||||
 | 
					            state.hashValues.unshift({ field: '', value: '' });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const saveValue = async () => {
 | 
				
			||||||
 | 
					            notEmpty(state.key.key, 'key不能为空');
 | 
				
			||||||
 | 
					            isTrue(state.hashValues.length > 0, 'hash内容不能为空');
 | 
				
			||||||
 | 
					            const sv = { value: state.hashValues, id: state.redisId };
 | 
				
			||||||
 | 
					            Object.assign(sv, state.key);
 | 
				
			||||||
 | 
					            await redisApi.saveHashValue.request(sv);
 | 
				
			||||||
 | 
					            ElMessage.success('保存成功');
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            cancel();
 | 
				
			||||||
 | 
					            emit('valChange');
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return {
 | 
				
			||||||
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            reHscan,
 | 
				
			||||||
 | 
					            hscan,
 | 
				
			||||||
 | 
					            cancel,
 | 
				
			||||||
 | 
					            hdel,
 | 
				
			||||||
 | 
					            hset,
 | 
				
			||||||
 | 
					            onAddHashValue,
 | 
				
			||||||
 | 
					            saveValue,
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					});
 | 
				
			||||||
 | 
					</script>
 | 
				
			||||||
 | 
					<style lang="scss">
 | 
				
			||||||
 | 
					#string-value-text {
 | 
				
			||||||
 | 
					    flex-grow: 1;
 | 
				
			||||||
 | 
					    display: flex;
 | 
				
			||||||
 | 
					    position: relative;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    .text-type-select {
 | 
				
			||||||
 | 
					        position: absolute;
 | 
				
			||||||
 | 
					        z-index: 2;
 | 
				
			||||||
 | 
					        right: 10px;
 | 
				
			||||||
 | 
					        top: 10px;
 | 
				
			||||||
 | 
					        max-width: 70px;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					</style>
 | 
				
			||||||
@@ -17,12 +17,13 @@
 | 
				
			|||||||
                    <el-select style="width: 100%" v-model="form.mode" placeholder="请选择模式">
 | 
					                    <el-select style="width: 100%" v-model="form.mode" placeholder="请选择模式">
 | 
				
			||||||
                        <el-option label="standalone" value="standalone"> </el-option>
 | 
					                        <el-option label="standalone" value="standalone"> </el-option>
 | 
				
			||||||
                        <el-option label="cluster" value="cluster"> </el-option>
 | 
					                        <el-option label="cluster" value="cluster"> </el-option>
 | 
				
			||||||
 | 
					                        <el-option label="sentinel" value="sentinel"> </el-option>
 | 
				
			||||||
                    </el-select>
 | 
					                    </el-select>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item prop="host" label="host:" required>
 | 
					                <el-form-item prop="host" label="host:" required>
 | 
				
			||||||
                    <el-input
 | 
					                    <el-input
 | 
				
			||||||
                        v-model.trim="form.host"
 | 
					                        v-model.trim="form.host"
 | 
				
			||||||
                        placeholder="请输入host:port,集群模式用','分割"
 | 
					                        placeholder="请输入host:port;sentinel模式为: mastername=sentinelhost:port,若集群或哨兵需设多个节点可使用','分割"
 | 
				
			||||||
                        auto-complete="off"
 | 
					                        auto-complete="off"
 | 
				
			||||||
                        type="textarea"
 | 
					                        type="textarea"
 | 
				
			||||||
                    ></el-input>
 | 
					                    ></el-input>
 | 
				
			||||||
@@ -34,7 +35,14 @@
 | 
				
			|||||||
                        v-model.trim="form.password"
 | 
					                        v-model.trim="form.password"
 | 
				
			||||||
                        placeholder="请输入密码, 修改操作可不填"
 | 
					                        placeholder="请输入密码, 修改操作可不填"
 | 
				
			||||||
                        autocomplete="new-password"
 | 
					                        autocomplete="new-password"
 | 
				
			||||||
                    ></el-input>
 | 
					                        ><template v-if="form.id && form.id != 0" #suffix>
 | 
				
			||||||
 | 
					                            <el-popover @hide="pwd = ''" placement="right" title="原密码" :width="200" trigger="click" :content="pwd">
 | 
				
			||||||
 | 
					                                <template #reference>
 | 
				
			||||||
 | 
					                                    <el-link @click="getPwd" :underline="false" type="primary" class="mr5">原密码</el-link>
 | 
				
			||||||
 | 
					                                </template>
 | 
				
			||||||
 | 
					                            </el-popover>
 | 
				
			||||||
 | 
					                        </template></el-input
 | 
				
			||||||
 | 
					                    >
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item prop="db" label="库号:" required>
 | 
					                <el-form-item prop="db" label="库号:" required>
 | 
				
			||||||
                    <el-input v-model.number="form.db" placeholder="请输入库号"></el-input>
 | 
					                    <el-input v-model.number="form.db" placeholder="请输入库号"></el-input>
 | 
				
			||||||
@@ -106,7 +114,7 @@ export default defineComponent({
 | 
				
			|||||||
                id: null,
 | 
					                id: null,
 | 
				
			||||||
                name: null,
 | 
					                name: null,
 | 
				
			||||||
                mode: 'standalone',
 | 
					                mode: 'standalone',
 | 
				
			||||||
                host: null,
 | 
					                host: '',
 | 
				
			||||||
                password: null,
 | 
					                password: null,
 | 
				
			||||||
                project: null,
 | 
					                project: null,
 | 
				
			||||||
                projectId: null,
 | 
					                projectId: null,
 | 
				
			||||||
@@ -116,6 +124,7 @@ export default defineComponent({
 | 
				
			|||||||
                enableSshTunnel: null,
 | 
					                enableSshTunnel: null,
 | 
				
			||||||
                sshTunnelMachineId: null,
 | 
					                sshTunnelMachineId: null,
 | 
				
			||||||
            },
 | 
					            },
 | 
				
			||||||
 | 
					            pwd: '',
 | 
				
			||||||
            btnLoading: false,
 | 
					            btnLoading: false,
 | 
				
			||||||
            rules: {
 | 
					            rules: {
 | 
				
			||||||
                projectId: [
 | 
					                projectId: [
 | 
				
			||||||
@@ -183,6 +192,10 @@ export default defineComponent({
 | 
				
			|||||||
            state.envs = await projectApi.projectEnvs.request({ projectId });
 | 
					            state.envs = await projectApi.projectEnvs.request({ projectId });
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const getPwd = async () => {
 | 
				
			||||||
 | 
					            state.pwd = await redisApi.getRedisPwd.request({ id: state.form.id });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const changeProject = (projectId: number) => {
 | 
					        const changeProject = (projectId: number) => {
 | 
				
			||||||
            for (let p of state.projects as any) {
 | 
					            for (let p of state.projects as any) {
 | 
				
			||||||
                if (p.id == projectId) {
 | 
					                if (p.id == projectId) {
 | 
				
			||||||
@@ -207,6 +220,10 @@ export default defineComponent({
 | 
				
			|||||||
            redisForm.value.validate(async (valid: boolean) => {
 | 
					            redisForm.value.validate(async (valid: boolean) => {
 | 
				
			||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    const reqForm = { ...state.form };
 | 
					                    const reqForm = { ...state.form };
 | 
				
			||||||
 | 
					                    if (reqForm.mode == 'sentinel' && reqForm.host.split('=').length != 2) {
 | 
				
			||||||
 | 
					                        ElMessage.error('sentinel模式host需为: mastername=sentinelhost:sentinelport模式');
 | 
				
			||||||
 | 
					                        return;
 | 
				
			||||||
 | 
					                    }
 | 
				
			||||||
                    reqForm.password = await RsaEncrypt(reqForm.password);
 | 
					                    reqForm.password = await RsaEncrypt(reqForm.password);
 | 
				
			||||||
                    redisApi.saveRedis.request(reqForm).then(() => {
 | 
					                    redisApi.saveRedis.request(reqForm).then(() => {
 | 
				
			||||||
                        ElMessage.success('保存成功');
 | 
					                        ElMessage.success('保存成功');
 | 
				
			||||||
@@ -234,6 +251,7 @@ export default defineComponent({
 | 
				
			|||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
            redisForm,
 | 
					            redisForm,
 | 
				
			||||||
            getSshTunnelMachines,
 | 
					            getSshTunnelMachines,
 | 
				
			||||||
 | 
					            getPwd,
 | 
				
			||||||
            changeProject,
 | 
					            changeProject,
 | 
				
			||||||
            changeEnv,
 | 
					            changeEnv,
 | 
				
			||||||
            btnOk,
 | 
					            btnOk,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -31,7 +31,13 @@
 | 
				
			|||||||
                <el-table-column prop="creator" label="创建人" min-width="100"></el-table-column>
 | 
					                <el-table-column prop="creator" label="创建人" min-width="100"></el-table-column>
 | 
				
			||||||
                <el-table-column label="更多" min-width="130" fixed="right">
 | 
					                <el-table-column label="更多" min-width="130" fixed="right">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-link v-if="scope.row.mode == 'standalone'" type="primary" @click="info(scope.row)" :underline="false">单机信息</el-link>
 | 
					                        <el-link
 | 
				
			||||||
 | 
					                            v-if="scope.row.mode == 'standalone' || scope.row.mode == 'sentinel'"
 | 
				
			||||||
 | 
					                            type="primary"
 | 
				
			||||||
 | 
					                            @click="info(scope.row)"
 | 
				
			||||||
 | 
					                            :underline="false"
 | 
				
			||||||
 | 
					                            >单机信息</el-link
 | 
				
			||||||
 | 
					                        >
 | 
				
			||||||
                        <el-link @click="onShowClusterInfo(scope.row)" v-if="scope.row.mode == 'cluster'" type="success" :underline="false"
 | 
					                        <el-link @click="onShowClusterInfo(scope.row)" v-if="scope.row.mode == 'cluster'" type="success" :underline="false"
 | 
				
			||||||
                            >集群信息</el-link
 | 
					                            >集群信息</el-link
 | 
				
			||||||
                        >
 | 
					                        >
 | 
				
			||||||
@@ -202,7 +208,6 @@ export default defineComponent({
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
        onMounted(async () => {
 | 
					        onMounted(async () => {
 | 
				
			||||||
            search();
 | 
					            search();
 | 
				
			||||||
            state.projects = await projectApi.accountProjects.request(null);
 | 
					 | 
				
			||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const handlePageChange = (curPage: number) => {
 | 
					        const handlePageChange = (curPage: number) => {
 | 
				
			||||||
@@ -258,7 +263,8 @@ export default defineComponent({
 | 
				
			|||||||
            state.total = res.total;
 | 
					            state.total = res.total;
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const editRedis = (isAdd = false) => {
 | 
					        const editRedis = async (isAdd = false) => {
 | 
				
			||||||
 | 
					            state.projects = await projectApi.accountProjects.request(null);
 | 
				
			||||||
            if (isAdd) {
 | 
					            if (isAdd) {
 | 
				
			||||||
                state.redisEditDialog.data = null;
 | 
					                state.redisEditDialog.data = null;
 | 
				
			||||||
                state.redisEditDialog.title = '新增redis';
 | 
					                state.redisEditDialog.title = '新增redis';
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										157
									
								
								mayfly_go_web/src/views/ops/redis/SetValue.vue
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										157
									
								
								mayfly_go_web/src/views/ops/redis/SetValue.vue
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,157 @@
 | 
				
			|||||||
 | 
					<template>
 | 
				
			||||||
 | 
					    <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" width="800px" :destroy-on-close="true">
 | 
				
			||||||
 | 
					        <el-form label-width="85px">
 | 
				
			||||||
 | 
					            <el-form-item prop="key" label="key:">
 | 
				
			||||||
 | 
					                <el-input :disabled="operationType == 2" v-model="key.key"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="timed" label="过期时间:">
 | 
				
			||||||
 | 
					                <el-input v-model.number="key.timed" type="number"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="dataType" label="数据类型:">
 | 
				
			||||||
 | 
					                <el-input v-model="key.type" disabled></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            <el-button @click="onAddSetValue" icon="plus" size="small" plain class="mt10">添加</el-button>
 | 
				
			||||||
 | 
					            <el-table :data="value" stripe style="width: 100%">
 | 
				
			||||||
 | 
					                <el-table-column prop="value" label="value" min-width="200">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-input v-model="scope.row.value" clearable type="textarea" :autosize="{ minRows: 2, maxRows: 10 }" size="small"></el-input>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column label="操作" width="90">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-button type="danger" @click="set.value.splice(scope.$index, 1)" icon="delete" size="small" plain>删除</el-button>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					            </el-table>
 | 
				
			||||||
 | 
					        </el-form>
 | 
				
			||||||
 | 
					        <template #footer>
 | 
				
			||||||
 | 
					            <div class="dialog-footer">
 | 
				
			||||||
 | 
					                <el-button @click="cancel()">取 消</el-button>
 | 
				
			||||||
 | 
					                <el-button @click="saveValue" type="primary" v-auth="'redis:data:save'">确 定</el-button>
 | 
				
			||||||
 | 
					            </div>
 | 
				
			||||||
 | 
					        </template>
 | 
				
			||||||
 | 
					    </el-dialog>
 | 
				
			||||||
 | 
					</template>
 | 
				
			||||||
 | 
					<script lang="ts">
 | 
				
			||||||
 | 
					import { defineComponent, reactive, watch, toRefs } from 'vue';
 | 
				
			||||||
 | 
					import { redisApi } from './api';
 | 
				
			||||||
 | 
					import { ElMessage } from 'element-plus';
 | 
				
			||||||
 | 
					import { isTrue, notEmpty } from '@/common/assert';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export default defineComponent({
 | 
				
			||||||
 | 
					    name: 'SetValue',
 | 
				
			||||||
 | 
					    components: {},
 | 
				
			||||||
 | 
					    props: {
 | 
				
			||||||
 | 
					        visible: {
 | 
				
			||||||
 | 
					            type: Boolean,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        title: {
 | 
				
			||||||
 | 
					            type: String,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        redisId: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					            require: true,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        keyInfo: {
 | 
				
			||||||
 | 
					            type: [Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        // 操作类型,1:新增,2:修改
 | 
				
			||||||
 | 
					        operationType: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        setValue: {
 | 
				
			||||||
 | 
					            type: [Array, Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					    emits: ['valChange', 'cancel', 'update:visible'],
 | 
				
			||||||
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            dialogVisible: false,
 | 
				
			||||||
 | 
					            operationType: 1,
 | 
				
			||||||
 | 
					            redisId: '',
 | 
				
			||||||
 | 
					            key: {
 | 
				
			||||||
 | 
					                key: '',
 | 
				
			||||||
 | 
					                type: 'string',
 | 
				
			||||||
 | 
					                timed: -1,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            value: [{ value: '' }],
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const cancel = () => {
 | 
				
			||||||
 | 
					            emit('update:visible', false);
 | 
				
			||||||
 | 
					            emit('cancel');
 | 
				
			||||||
 | 
					            setTimeout(() => {
 | 
				
			||||||
 | 
					                state.key = {
 | 
				
			||||||
 | 
					                    key: '',
 | 
				
			||||||
 | 
					                    type: 'string',
 | 
				
			||||||
 | 
					                    timed: -1,
 | 
				
			||||||
 | 
					                };
 | 
				
			||||||
 | 
					                state.value = [];
 | 
				
			||||||
 | 
					            }, 500);
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(props, async (newValue) => {
 | 
				
			||||||
 | 
					            state.dialogVisible = newValue.visible;
 | 
				
			||||||
 | 
					            state.key = newValue.key;
 | 
				
			||||||
 | 
					            state.redisId = newValue.redisId;
 | 
				
			||||||
 | 
					            state.key = newValue.keyInfo;
 | 
				
			||||||
 | 
					            state.operationType = newValue.operationType;
 | 
				
			||||||
 | 
					            // 如果是查看编辑操作,则获取值
 | 
				
			||||||
 | 
					            if (state.dialogVisible && state.operationType == 2) {
 | 
				
			||||||
 | 
					                getSetValue();
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const getSetValue = async () => {
 | 
				
			||||||
 | 
					            const res = await redisApi.getSetValue.request({
 | 
				
			||||||
 | 
					                id: state.redisId,
 | 
				
			||||||
 | 
					                key: state.key.key,
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					            state.value = res.map((x: any) => {
 | 
				
			||||||
 | 
					                return {
 | 
				
			||||||
 | 
					                    value: x,
 | 
				
			||||||
 | 
					                };
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const saveValue = async () => {
 | 
				
			||||||
 | 
					            notEmpty(state.key.key, 'key不能为空');
 | 
				
			||||||
 | 
					            isTrue(state.value.length > 0, 'set内容不能为空');
 | 
				
			||||||
 | 
					            const sv = { value: state.value.map((x) => x.value), id: state.redisId };
 | 
				
			||||||
 | 
					            Object.assign(sv, state.key);
 | 
				
			||||||
 | 
					            await redisApi.saveSetValue.request(sv);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            ElMessage.success('数据保存成功');
 | 
				
			||||||
 | 
					            cancel();
 | 
				
			||||||
 | 
					            emit('valChange');
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const onAddSetValue = () => {
 | 
				
			||||||
 | 
					            state.value.unshift({ value: '' });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return {
 | 
				
			||||||
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            saveValue,
 | 
				
			||||||
 | 
					            cancel,
 | 
				
			||||||
 | 
					            onAddSetValue,
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					});
 | 
				
			||||||
 | 
					</script>
 | 
				
			||||||
 | 
					<style lang="scss">
 | 
				
			||||||
 | 
					#string-value-text {
 | 
				
			||||||
 | 
					    flex-grow: 1;
 | 
				
			||||||
 | 
					    display: flex;
 | 
				
			||||||
 | 
					    position: relative;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    .text-type-select {
 | 
				
			||||||
 | 
					        position: absolute;
 | 
				
			||||||
 | 
					        z-index: 2;
 | 
				
			||||||
 | 
					        right: 10px;
 | 
				
			||||||
 | 
					        top: 10px;
 | 
				
			||||||
 | 
					        max-width: 70px;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					</style>
 | 
				
			||||||
							
								
								
									
										169
									
								
								mayfly_go_web/src/views/ops/redis/StringValue.vue
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										169
									
								
								mayfly_go_web/src/views/ops/redis/StringValue.vue
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,169 @@
 | 
				
			|||||||
 | 
					<template>
 | 
				
			||||||
 | 
					    <el-dialog :title="title" v-model="dialogVisible" :before-close="cancel" width="800px" :destroy-on-close="true">
 | 
				
			||||||
 | 
					        <el-form label-width="85px">
 | 
				
			||||||
 | 
					            <el-form-item prop="key" label="key:">
 | 
				
			||||||
 | 
					                <el-input :disabled="operationType == 2" v-model="key.key"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="timed" label="过期时间:">
 | 
				
			||||||
 | 
					                <el-input v-model.number="key.timed" type="number"></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					            <el-form-item prop="dataType" label="数据类型:">
 | 
				
			||||||
 | 
					                <el-input v-model="key.type" disabled></el-input>
 | 
				
			||||||
 | 
					            </el-form-item>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            <div id="string-value-text" style="width: 100%">
 | 
				
			||||||
 | 
					                <el-input class="json-text" v-model="string.value" type="textarea" :autosize="{ minRows: 10, maxRows: 20 }"></el-input>
 | 
				
			||||||
 | 
					                <el-select class="text-type-select" @change="onChangeTextType" v-model="string.type">
 | 
				
			||||||
 | 
					                    <el-option key="text" label="text" value="text"> </el-option>
 | 
				
			||||||
 | 
					                    <el-option key="json" label="json" value="json"> </el-option>
 | 
				
			||||||
 | 
					                </el-select>
 | 
				
			||||||
 | 
					            </div>
 | 
				
			||||||
 | 
					        </el-form>
 | 
				
			||||||
 | 
					        <template #footer>
 | 
				
			||||||
 | 
					            <div class="dialog-footer">
 | 
				
			||||||
 | 
					                <el-button @click="cancel()">取 消</el-button>
 | 
				
			||||||
 | 
					                <el-button @click="saveValue" type="primary" v-auth="'redis:data:save'">确 定</el-button>
 | 
				
			||||||
 | 
					            </div>
 | 
				
			||||||
 | 
					        </template>
 | 
				
			||||||
 | 
					    </el-dialog>
 | 
				
			||||||
 | 
					</template>
 | 
				
			||||||
 | 
					<script lang="ts">
 | 
				
			||||||
 | 
					import { defineComponent, reactive, watch, toRefs } from 'vue';
 | 
				
			||||||
 | 
					import { redisApi } from './api';
 | 
				
			||||||
 | 
					import { ElMessage } from 'element-plus';
 | 
				
			||||||
 | 
					import { notEmpty } from '@/common/assert';
 | 
				
			||||||
 | 
					import { formatJsonString } from '@/common/utils/format';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export default defineComponent({
 | 
				
			||||||
 | 
					    name: 'StringValue',
 | 
				
			||||||
 | 
					    components: {},
 | 
				
			||||||
 | 
					    props: {
 | 
				
			||||||
 | 
					        visible: {
 | 
				
			||||||
 | 
					            type: Boolean,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        title: {
 | 
				
			||||||
 | 
					            type: String,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        redisId: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					            require: true,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        keyInfo: {
 | 
				
			||||||
 | 
					            type: [Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        // 操作类型,1:新增,2:修改
 | 
				
			||||||
 | 
					        operationType: {
 | 
				
			||||||
 | 
					            type: [Number],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					    emits: ['valChange', 'cancel', 'update:visible'],
 | 
				
			||||||
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            dialogVisible: false,
 | 
				
			||||||
 | 
					            operationType: 1,
 | 
				
			||||||
 | 
					            redisId: '',
 | 
				
			||||||
 | 
					            key: {
 | 
				
			||||||
 | 
					                key: '',
 | 
				
			||||||
 | 
					                type: 'string',
 | 
				
			||||||
 | 
					                timed: -1,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            string: {
 | 
				
			||||||
 | 
					                type: 'text',
 | 
				
			||||||
 | 
					                value: '',
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const cancel = () => {
 | 
				
			||||||
 | 
					            emit('update:visible', false);
 | 
				
			||||||
 | 
					            emit('cancel');
 | 
				
			||||||
 | 
					            setTimeout(() => {
 | 
				
			||||||
 | 
					                state.key = {
 | 
				
			||||||
 | 
					                    key: '',
 | 
				
			||||||
 | 
					                    type: 'string',
 | 
				
			||||||
 | 
					                    timed: -1,
 | 
				
			||||||
 | 
					                };
 | 
				
			||||||
 | 
					                state.string.value = '';
 | 
				
			||||||
 | 
					                state.string.type = 'text';
 | 
				
			||||||
 | 
					            }, 500);
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(
 | 
				
			||||||
 | 
					            () => props.visible,
 | 
				
			||||||
 | 
					            (val) => {
 | 
				
			||||||
 | 
					                state.dialogVisible = val;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        );
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(
 | 
				
			||||||
 | 
					            () => props.redisId,
 | 
				
			||||||
 | 
					            (val) => {
 | 
				
			||||||
 | 
					                state.redisId = val;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        );
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(props, async (newValue) => {
 | 
				
			||||||
 | 
					            state.dialogVisible = newValue.visible;
 | 
				
			||||||
 | 
					            state.key = newValue.key;
 | 
				
			||||||
 | 
					            state.redisId = newValue.redisId;
 | 
				
			||||||
 | 
					            state.key = newValue.keyInfo;
 | 
				
			||||||
 | 
					            state.operationType = newValue.operationType
 | 
				
			||||||
 | 
					            // 如果是查看编辑操作,则获取值
 | 
				
			||||||
 | 
					            if (state.dialogVisible && state.operationType == 2) {
 | 
				
			||||||
 | 
					                getStringValue();
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const getStringValue = async () => {
 | 
				
			||||||
 | 
					            state.string.value = await redisApi.getStringValue.request({
 | 
				
			||||||
 | 
					                id: state.redisId,
 | 
				
			||||||
 | 
					                key: state.key.key,
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const saveValue = async () => {
 | 
				
			||||||
 | 
					            notEmpty(state.key.key, 'key不能为空');
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            notEmpty(state.string.value, 'value不能为空');
 | 
				
			||||||
 | 
					            const sv = { value: formatJsonString(state.string.value, true), id: state.redisId };
 | 
				
			||||||
 | 
					            Object.assign(sv, state.key);
 | 
				
			||||||
 | 
					            await redisApi.saveStringValue.request(sv);
 | 
				
			||||||
 | 
					            ElMessage.success('数据保存成功');
 | 
				
			||||||
 | 
					            cancel();
 | 
				
			||||||
 | 
					            emit('valChange');
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        // 更改文本类型
 | 
				
			||||||
 | 
					        const onChangeTextType = (val: string) => {
 | 
				
			||||||
 | 
					            if (val == 'json') {
 | 
				
			||||||
 | 
					                state.string.value = formatJsonString(state.string.value, false);
 | 
				
			||||||
 | 
					                return;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            if (val == 'text') {
 | 
				
			||||||
 | 
					                state.string.value = formatJsonString(state.string.value, true);
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return {
 | 
				
			||||||
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            saveValue,
 | 
				
			||||||
 | 
					            cancel,
 | 
				
			||||||
 | 
					            onChangeTextType,
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					});
 | 
				
			||||||
 | 
					</script>
 | 
				
			||||||
 | 
					<style lang="scss">
 | 
				
			||||||
 | 
					#string-value-text {
 | 
				
			||||||
 | 
					    flex-grow: 1;
 | 
				
			||||||
 | 
					    display: flex;
 | 
				
			||||||
 | 
					    position: relative;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					    .text-type-select {
 | 
				
			||||||
 | 
					        position: absolute;
 | 
				
			||||||
 | 
					        z-index: 2;
 | 
				
			||||||
 | 
					        right: 10px;
 | 
				
			||||||
 | 
					        top: 10px;
 | 
				
			||||||
 | 
					        max-width: 70px;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					</style>
 | 
				
			||||||
@@ -2,6 +2,7 @@ import Api from '@/common/Api';
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
export const redisApi = {
 | 
					export const redisApi = {
 | 
				
			||||||
    redisList : Api.create("/redis", 'get'),
 | 
					    redisList : Api.create("/redis", 'get'),
 | 
				
			||||||
 | 
					    getRedisPwd: Api.create("/redis/{id}/pwd", 'get'),
 | 
				
			||||||
    redisInfo: Api.create("/redis/{id}/info", 'get'),
 | 
					    redisInfo: Api.create("/redis/{id}/info", 'get'),
 | 
				
			||||||
    clusterInfo: Api.create("/redis/{id}/cluster-info", 'get'),
 | 
					    clusterInfo: Api.create("/redis/{id}/cluster-info", 'get'),
 | 
				
			||||||
    saveRedis: Api.create("/redis", 'post'),
 | 
					    saveRedis: Api.create("/redis", 'post'),
 | 
				
			||||||
@@ -11,6 +12,9 @@ export const redisApi = {
 | 
				
			|||||||
    getStringValue: Api.create("/redis/{id}/string-value", 'get'),
 | 
					    getStringValue: Api.create("/redis/{id}/string-value", 'get'),
 | 
				
			||||||
    saveStringValue: Api.create("/redis/{id}/string-value", 'post'),
 | 
					    saveStringValue: Api.create("/redis/{id}/string-value", 'post'),
 | 
				
			||||||
    getHashValue: Api.create("/redis/{id}/hash-value", 'get'),
 | 
					    getHashValue: Api.create("/redis/{id}/hash-value", 'get'),
 | 
				
			||||||
 | 
					    hscan: Api.create("/redis/{id}/hscan", 'get'),
 | 
				
			||||||
 | 
					    hget: Api.create("/redis/{id}/hget", 'get'),
 | 
				
			||||||
 | 
					    hdel: Api.create("/redis/{id}/hdel", 'delete'),
 | 
				
			||||||
    saveHashValue: Api.create("/redis/{id}/hash-value", 'post'),
 | 
					    saveHashValue: Api.create("/redis/{id}/hash-value", 'post'),
 | 
				
			||||||
    getSetValue: Api.create("/redis/{id}/set-value", 'get'),
 | 
					    getSetValue: Api.create("/redis/{id}/set-value", 'get'),
 | 
				
			||||||
    saveSetValue: Api.create("/redis/{id}/set-value", 'post'),
 | 
					    saveSetValue: Api.create("/redis/{id}/set-value", 'post'),
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,12 +5,9 @@
 | 
				
			|||||||
                <el-form-item prop="username" label="用户名:" required>
 | 
					                <el-form-item prop="username" label="用户名:" required>
 | 
				
			||||||
                    <el-input :disabled="edit" v-model.trim="form.username" placeholder="请输入账号用户名,密码默认与账号名一致" auto-complete="off"></el-input>
 | 
					                    <el-input :disabled="edit" v-model.trim="form.username" placeholder="请输入账号用户名,密码默认与账号名一致" auto-complete="off"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <!-- <el-form-item prop="password" label="密码:" required>
 | 
					                <el-form-item v-if="edit" prop="password" label="密码:" required>
 | 
				
			||||||
                    <el-input type="password" v-model.trim="form.password" placeholder="请输入密码" autocomplete="new-password"></el-input>
 | 
					                    <el-input type="password" v-model.trim="form.password" placeholder="请输入密码" autocomplete="new-password"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
                <el-form-item v-if="!edit" label="确认密码:" required>
 | 
					 | 
				
			||||||
                    <el-input type="password" v-model.trim="form.repassword" placeholder="请输入确认密码" autocomplete="new-password"></el-input>
 | 
					 | 
				
			||||||
                </el-form-item> -->
 | 
					 | 
				
			||||||
            </el-form>
 | 
					            </el-form>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            <template #footer>
 | 
					            <template #footer>
 | 
				
			||||||
@@ -74,6 +71,7 @@ export default defineComponent({
 | 
				
			|||||||
        watch(props, (newValue) => {
 | 
					        watch(props, (newValue) => {
 | 
				
			||||||
            if (newValue.account) {
 | 
					            if (newValue.account) {
 | 
				
			||||||
                state.form = { ...newValue.account };
 | 
					                state.form = { ...newValue.account };
 | 
				
			||||||
 | 
					                state.edit = true;
 | 
				
			||||||
            } else {
 | 
					            } else {
 | 
				
			||||||
                state.form = {} as any;
 | 
					                state.form = {} as any;
 | 
				
			||||||
            }
 | 
					            }
 | 
				
			||||||
@@ -81,11 +79,9 @@ export default defineComponent({
 | 
				
			|||||||
        });
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const btnOk = async () => {
 | 
					        const btnOk = async () => {
 | 
				
			||||||
            let p = state.form.id ? accountApi.update : accountApi.save;
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
            accountForm.value.validate((valid: boolean) => {
 | 
					            accountForm.value.validate((valid: boolean) => {
 | 
				
			||||||
                if (valid) {
 | 
					                if (valid) {
 | 
				
			||||||
                    p.request(state.form).then(() => {
 | 
					                    accountApi.save.request(state.form).then(() => {
 | 
				
			||||||
                        ElMessage.success('操作成功');
 | 
					                        ElMessage.success('操作成功');
 | 
				
			||||||
                        emit('val-change', state.form);
 | 
					                        emit('val-change', state.form);
 | 
				
			||||||
                        state.btnLoading = true;
 | 
					                        state.btnLoading = true;
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,7 +2,7 @@
 | 
				
			|||||||
    <div class="role-list">
 | 
					    <div class="role-list">
 | 
				
			||||||
        <el-card>
 | 
					        <el-card>
 | 
				
			||||||
            <el-button v-auth="'account:add'" type="primary" icon="plus" @click="editAccount(true)">添加</el-button>
 | 
					            <el-button v-auth="'account:add'" type="primary" icon="plus" @click="editAccount(true)">添加</el-button>
 | 
				
			||||||
            <!-- <el-button v-auth="'account:update'" :disabled="chooseId == null" @click="editAccount(false)" type="primary" icon="edit">编辑</el-button> -->
 | 
					            <el-button v-auth="'account:add'" :disabled="chooseId == null" @click="editAccount(false)" type="primary" icon="edit">编辑</el-button>
 | 
				
			||||||
            <el-button v-auth="'account:saveRoles'" :disabled="chooseId == null" @click="roleEdit()" type="success" icon="setting"
 | 
					            <el-button v-auth="'account:saveRoles'" :disabled="chooseId == null" @click="roleEdit()" type="success" icon="setting"
 | 
				
			||||||
                >角色分配</el-button
 | 
					                >角色分配</el-button
 | 
				
			||||||
            >
 | 
					            >
 | 
				
			||||||
@@ -20,7 +20,7 @@
 | 
				
			|||||||
                <el-button @click="search()" type="success" icon="search" size="small"></el-button>
 | 
					                <el-button @click="search()" type="success" icon="search" size="small"></el-button>
 | 
				
			||||||
            </div>
 | 
					            </div>
 | 
				
			||||||
            <el-table :data="datas" ref="table" @current-change="choose" show-overflow-tooltip>
 | 
					            <el-table :data="datas" ref="table" @current-change="choose" show-overflow-tooltip>
 | 
				
			||||||
                <el-table-column label="选择" width="50px">
 | 
					                <el-table-column label="选择" width="55px">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
					                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
				
			||||||
                            <i></i>
 | 
					                            <i></i>
 | 
				
			||||||
@@ -29,20 +29,20 @@
 | 
				
			|||||||
                </el-table-column>
 | 
					                </el-table-column>
 | 
				
			||||||
                <el-table-column prop="username" label="用户名" min-width="115"></el-table-column>
 | 
					                <el-table-column prop="username" label="用户名" min-width="115"></el-table-column>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-table-column align="center" prop="status" label="状态" min-width="63">
 | 
					                <el-table-column align="center" prop="status" label="状态" min-width="65">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-tag v-if="scope.row.status == 1" type="success">正常</el-tag>
 | 
					                        <el-tag v-if="scope.row.status == 1" type="success">正常</el-tag>
 | 
				
			||||||
                        <el-tag v-if="scope.row.status == -1" type="danger">禁用</el-tag>
 | 
					                        <el-tag v-if="scope.row.status == -1" type="danger">禁用</el-tag>
 | 
				
			||||||
                    </template>
 | 
					                    </template>
 | 
				
			||||||
                </el-table-column>
 | 
					                </el-table-column>
 | 
				
			||||||
                <el-table-column min-width="160" prop="lastLoginTime" label="最后登录时间">
 | 
					                <el-table-column min-width="160" prop="lastLoginTime" label="最后登录时间" show-overflow-tooltip>
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        {{ $filters.dateFormat(scope.row.lastLoginTime) }}
 | 
					                        {{ $filters.dateFormat(scope.row.lastLoginTime) }}
 | 
				
			||||||
                    </template>
 | 
					                    </template>
 | 
				
			||||||
                </el-table-column>
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
                <el-table-column min-width="115" prop="creator" label="创建账号"></el-table-column>
 | 
					                <el-table-column min-width="115" prop="creator" label="创建账号"></el-table-column>
 | 
				
			||||||
                <el-table-column min-width="160" prop="createTime" label="创建时间">
 | 
					                <el-table-column min-width="160" prop="createTime" label="创建时间" show-overflow-tooltip>
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
					                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
				
			||||||
                    </template>
 | 
					                    </template>
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -32,6 +32,12 @@ export const accountApi = {
 | 
				
			|||||||
    saveRoles: Api.create("/sys/accounts/roles", 'post')
 | 
					    saveRoles: Api.create("/sys/accounts/roles", 'post')
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export const configApi = {
 | 
				
			||||||
 | 
					    list: Api.create("/sys/configs", 'get'),
 | 
				
			||||||
 | 
					    save: Api.create("/sys/configs", 'post'),
 | 
				
			||||||
 | 
					    getValue: Api.create("/sys/configs/value", 'get'),
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export const logApi = {
 | 
					export const logApi = {
 | 
				
			||||||
    list: Api.create("/syslogs", "get")
 | 
					    list: Api.create("/syslogs", "get")
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										99
									
								
								mayfly_go_web/src/views/system/config/ConfigEdit.vue
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										99
									
								
								mayfly_go_web/src/views/system/config/ConfigEdit.vue
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,99 @@
 | 
				
			|||||||
 | 
					<template>
 | 
				
			||||||
 | 
					    <div>
 | 
				
			||||||
 | 
					        <el-dialog :title="title" v-model="dvisible" :show-close="false" :before-close="cancel" width="500px" :destroy-on-close="true">
 | 
				
			||||||
 | 
					            <el-form ref="configForm" :model="form" label-width="90px">
 | 
				
			||||||
 | 
					                <el-form-item prop="name" label="配置项:" required>
 | 
				
			||||||
 | 
					                    <el-input v-model="form.name"></el-input>
 | 
				
			||||||
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					                <el-form-item prop="key" label="配置key:" required>
 | 
				
			||||||
 | 
					                    <el-input :disabled="form.id != null" v-model="form.key"></el-input>
 | 
				
			||||||
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					                <el-form-item prop="value" label="配置值:" required>
 | 
				
			||||||
 | 
					                    <el-input v-model="form.value"></el-input>
 | 
				
			||||||
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					                <el-form-item label="备注:">
 | 
				
			||||||
 | 
					                    <el-input v-model="form.remark" type="textarea" :rows="2"></el-input>
 | 
				
			||||||
 | 
					                </el-form-item>
 | 
				
			||||||
 | 
					            </el-form>
 | 
				
			||||||
 | 
					            <template #footer>
 | 
				
			||||||
 | 
					                <div class="dialog-footer">
 | 
				
			||||||
 | 
					                    <el-button @click="cancel()">取 消</el-button>
 | 
				
			||||||
 | 
					                    <el-button type="primary" :loading="btnLoading" @click="btnOk">确 定</el-button>
 | 
				
			||||||
 | 
					                </div>
 | 
				
			||||||
 | 
					            </template>
 | 
				
			||||||
 | 
					        </el-dialog>
 | 
				
			||||||
 | 
					    </div>
 | 
				
			||||||
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					<script lang="ts">
 | 
				
			||||||
 | 
					import { ref, toRefs, reactive, watch, defineComponent } from 'vue';
 | 
				
			||||||
 | 
					import { configApi } from '../api';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					export default defineComponent({
 | 
				
			||||||
 | 
					    name: 'ConfigEdit',
 | 
				
			||||||
 | 
					    props: {
 | 
				
			||||||
 | 
					        visible: {
 | 
				
			||||||
 | 
					            type: Boolean,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        data: {
 | 
				
			||||||
 | 
					            type: [Boolean, Object],
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					        title: {
 | 
				
			||||||
 | 
					            type: String,
 | 
				
			||||||
 | 
					        },
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
 | 
					        const configForm: any = ref(null);
 | 
				
			||||||
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            dvisible: false,
 | 
				
			||||||
 | 
					            form: {
 | 
				
			||||||
 | 
					                id: null,
 | 
				
			||||||
 | 
					                name: '',
 | 
				
			||||||
 | 
					                key: '',
 | 
				
			||||||
 | 
					                value: '',
 | 
				
			||||||
 | 
					                remark: '',
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            btnLoading: false,
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        watch(props, (newValue) => {
 | 
				
			||||||
 | 
					            state.dvisible = newValue.visible;
 | 
				
			||||||
 | 
					            if (newValue.data) {
 | 
				
			||||||
 | 
					                state.form = { ...newValue.data };
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                state.form = {} as any;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const cancel = () => {
 | 
				
			||||||
 | 
					            // 更新父组件visible prop对应的值为false
 | 
				
			||||||
 | 
					            emit('update:visible', false);
 | 
				
			||||||
 | 
					            // 若父组件有取消事件,则调用
 | 
				
			||||||
 | 
					            emit('cancel');
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const btnOk = async () => {
 | 
				
			||||||
 | 
					            configForm.value.validate(async (valid: boolean) => {
 | 
				
			||||||
 | 
					                if (valid) {
 | 
				
			||||||
 | 
					                    await configApi.save.request(state.form);
 | 
				
			||||||
 | 
					                    emit('val-change', state.form);
 | 
				
			||||||
 | 
					                    cancel();
 | 
				
			||||||
 | 
					                    state.btnLoading = true;
 | 
				
			||||||
 | 
					                    setTimeout(() => {
 | 
				
			||||||
 | 
					                        state.btnLoading = false;
 | 
				
			||||||
 | 
					                    }, 1000);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return {
 | 
				
			||||||
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            configForm,
 | 
				
			||||||
 | 
					            btnOk,
 | 
				
			||||||
 | 
					            cancel,
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					});
 | 
				
			||||||
 | 
					</script>
 | 
				
			||||||
 | 
					<style lang="scss">
 | 
				
			||||||
 | 
					</style>
 | 
				
			||||||
							
								
								
									
										124
									
								
								mayfly_go_web/src/views/system/config/ConfigList.vue
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										124
									
								
								mayfly_go_web/src/views/system/config/ConfigList.vue
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,124 @@
 | 
				
			|||||||
 | 
					<template>
 | 
				
			||||||
 | 
					    <div class="role-list">
 | 
				
			||||||
 | 
					        <el-card>
 | 
				
			||||||
 | 
					            <el-button type="primary" icon="plus" @click="editConfig(false)">添加</el-button>
 | 
				
			||||||
 | 
					            <el-button :disabled="chooseId == null" @click="editConfig(chooseData)" type="primary" icon="edit">编辑</el-button>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            <el-table :data="configs" @current-change="choose" ref="table" style="width: 100%">
 | 
				
			||||||
 | 
					                <el-table-column label="选择" width="55px">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
				
			||||||
 | 
					                            <i></i>
 | 
				
			||||||
 | 
					                        </el-radio>
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="name" label="配置项"></el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="key" label="配置key"></el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="value" label="配置值" min-width="100px" show-overflow-tooltip></el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="remark" label="备注" min-width="100px" show-overflow-tooltip></el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="updateTime" label="更新时间">
 | 
				
			||||||
 | 
					                    <template #default="scope">
 | 
				
			||||||
 | 
					                        {{ $filters.dateFormat(scope.row.createTime) }}
 | 
				
			||||||
 | 
					                    </template>
 | 
				
			||||||
 | 
					                </el-table-column>
 | 
				
			||||||
 | 
					                <el-table-column prop="modifier" label="修改者" 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>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        <config-edit :title="configEdit.title" v-model:visible="configEdit.visible" :data="configEdit.config" @val-change="configEditChange" />
 | 
				
			||||||
 | 
					    </div>
 | 
				
			||||||
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					<script lang="ts">
 | 
				
			||||||
 | 
					import { toRefs, reactive, onMounted, defineComponent } from 'vue';
 | 
				
			||||||
 | 
					import ConfigEdit from './ConfigEdit.vue';
 | 
				
			||||||
 | 
					import { configApi } from '../api';
 | 
				
			||||||
 | 
					import { ElMessage, ElMessageBox } from 'element-plus';
 | 
				
			||||||
 | 
					export default defineComponent({
 | 
				
			||||||
 | 
					    name: 'ConfigList',
 | 
				
			||||||
 | 
					    components: {
 | 
				
			||||||
 | 
					        ConfigEdit,
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					    setup() {
 | 
				
			||||||
 | 
					        const state = reactive({
 | 
				
			||||||
 | 
					            dialogFormVisible: false,
 | 
				
			||||||
 | 
					            currentEditPermissions: false,
 | 
				
			||||||
 | 
					            query: {
 | 
				
			||||||
 | 
					                pageNum: 1,
 | 
				
			||||||
 | 
					                pageSize: 10,
 | 
				
			||||||
 | 
					                name: null,
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					            total: 0,
 | 
				
			||||||
 | 
					            configs: [],
 | 
				
			||||||
 | 
					            chooseId: null,
 | 
				
			||||||
 | 
					            chooseData: null,
 | 
				
			||||||
 | 
					            configEdit: {
 | 
				
			||||||
 | 
					                title: '配置修改',
 | 
				
			||||||
 | 
					                visible: false,
 | 
				
			||||||
 | 
					                config: {},
 | 
				
			||||||
 | 
					            },
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        onMounted(() => {
 | 
				
			||||||
 | 
					            search();
 | 
				
			||||||
 | 
					        });
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const search = async () => {
 | 
				
			||||||
 | 
					            let res = await configApi.list.request(state.query);
 | 
				
			||||||
 | 
					            state.configs = res.list;
 | 
				
			||||||
 | 
					            state.total = res.total;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const handlePageChange = (curPage: number) => {
 | 
				
			||||||
 | 
					            state.query.pageNum = curPage;
 | 
				
			||||||
 | 
					            search();
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const choose = (item: any) => {
 | 
				
			||||||
 | 
					            if (!item) {
 | 
				
			||||||
 | 
					                return;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					            state.chooseId = item.id;
 | 
				
			||||||
 | 
					            state.chooseData = item;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const configEditChange = () => {
 | 
				
			||||||
 | 
					            ElMessage.success('修改成功!');
 | 
				
			||||||
 | 
					            state.chooseId = null;
 | 
				
			||||||
 | 
					            state.chooseData = null;
 | 
				
			||||||
 | 
					            search();
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        const editConfig = (data: any) => {
 | 
				
			||||||
 | 
					            if (data) {
 | 
				
			||||||
 | 
					                state.configEdit.config = data;
 | 
				
			||||||
 | 
					            } else {
 | 
				
			||||||
 | 
					                state.configEdit.config = false;
 | 
				
			||||||
 | 
					            }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            state.configEdit.visible = true;
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					        return {
 | 
				
			||||||
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            search,
 | 
				
			||||||
 | 
					            handlePageChange,
 | 
				
			||||||
 | 
					            choose,
 | 
				
			||||||
 | 
					            configEditChange,
 | 
				
			||||||
 | 
					            editConfig,
 | 
				
			||||||
 | 
					        };
 | 
				
			||||||
 | 
					    },
 | 
				
			||||||
 | 
					});
 | 
				
			||||||
 | 
					</script>
 | 
				
			||||||
 | 
					<style lang="scss">
 | 
				
			||||||
 | 
					</style>
 | 
				
			||||||
@@ -1,7 +1,7 @@
 | 
				
			|||||||
<template>
 | 
					<template>
 | 
				
			||||||
    <div class="role-dialog">
 | 
					    <div class="role-dialog">
 | 
				
			||||||
        <el-dialog :title="title" v-model="dvisible" :show-close="false" :before-close="cancel" width="500px" :destroy-on-close="true">
 | 
					        <el-dialog :title="title" v-model="dvisible" :show-close="false" :before-close="cancel" width="500px" :destroy-on-close="true">
 | 
				
			||||||
            <el-form :model="form" label-width="90px">
 | 
					            <el-form ref="roleForm" :model="form" label-width="90px">
 | 
				
			||||||
                <el-form-item prop="name" label="角色名称:" required>
 | 
					                <el-form-item prop="name" label="角色名称:" required>
 | 
				
			||||||
                    <el-input v-model="form.name" auto-complete="off"></el-input>
 | 
					                    <el-input v-model="form.name" auto-complete="off"></el-input>
 | 
				
			||||||
                </el-form-item>
 | 
					                </el-form-item>
 | 
				
			||||||
@@ -28,7 +28,7 @@
 | 
				
			|||||||
</template>
 | 
					</template>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
<script lang="ts">
 | 
					<script lang="ts">
 | 
				
			||||||
import { toRefs, reactive, watch, defineComponent } from 'vue';
 | 
					import { ref, toRefs, reactive, watch, defineComponent } from 'vue';
 | 
				
			||||||
import { roleApi } from '../api';
 | 
					import { roleApi } from '../api';
 | 
				
			||||||
 | 
					
 | 
				
			||||||
export default defineComponent({
 | 
					export default defineComponent({
 | 
				
			||||||
@@ -45,6 +45,7 @@ export default defineComponent({
 | 
				
			|||||||
        },
 | 
					        },
 | 
				
			||||||
    },
 | 
					    },
 | 
				
			||||||
    setup(props: any, { emit }) {
 | 
					    setup(props: any, { emit }) {
 | 
				
			||||||
 | 
					        const roleForm: any = ref(null);
 | 
				
			||||||
        const state = reactive({
 | 
					        const state = reactive({
 | 
				
			||||||
            dvisible: false,
 | 
					            dvisible: false,
 | 
				
			||||||
            form: {
 | 
					            form: {
 | 
				
			||||||
@@ -73,18 +74,22 @@ export default defineComponent({
 | 
				
			|||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        const btnOk = async () => {
 | 
					        const btnOk = async () => {
 | 
				
			||||||
            // let p = state.form.id ? roleApi.update : roleApi.save;
 | 
					            roleForm.value.validate(async (valid: boolean) => {
 | 
				
			||||||
            await roleApi.save.request(state.form);
 | 
					                if (valid) {
 | 
				
			||||||
            emit('val-change', state.form);
 | 
					                    await roleApi.save.request(state.form);
 | 
				
			||||||
            cancel();
 | 
					                    emit('val-change', state.form);
 | 
				
			||||||
            state.btnLoading = true;
 | 
					                    cancel();
 | 
				
			||||||
            setTimeout(() => {
 | 
					                    state.btnLoading = true;
 | 
				
			||||||
                state.btnLoading = false;
 | 
					                    setTimeout(() => {
 | 
				
			||||||
            }, 1000);
 | 
					                        state.btnLoading = false;
 | 
				
			||||||
 | 
					                    }, 1000);
 | 
				
			||||||
 | 
					                }
 | 
				
			||||||
 | 
					            });
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        return {
 | 
					        return {
 | 
				
			||||||
            ...toRefs(state),
 | 
					            ...toRefs(state),
 | 
				
			||||||
 | 
					            roleForm,
 | 
				
			||||||
            btnOk,
 | 
					            btnOk,
 | 
				
			||||||
            cancel,
 | 
					            cancel,
 | 
				
			||||||
        };
 | 
					        };
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -20,7 +20,7 @@
 | 
				
			|||||||
                <el-button @click="search" type="success" icon="search"></el-button>
 | 
					                <el-button @click="search" type="success" icon="search"></el-button>
 | 
				
			||||||
            </div>
 | 
					            </div>
 | 
				
			||||||
            <el-table :data="roles" @current-change="choose" ref="table" style="width: 100%">
 | 
					            <el-table :data="roles" @current-change="choose" ref="table" style="width: 100%">
 | 
				
			||||||
                <el-table-column label="选择" width="50px">
 | 
					                <el-table-column label="选择" width="55px">
 | 
				
			||||||
                    <template #default="scope">
 | 
					                    <template #default="scope">
 | 
				
			||||||
                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
					                        <el-radio v-model="chooseId" :label="scope.row.id">
 | 
				
			||||||
                            <i></i>
 | 
					                            <i></i>
 | 
				
			||||||
 
 | 
				
			|||||||
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							@@ -1,7 +1,3 @@
 | 
				
			|||||||
app:
 | 
					 | 
				
			||||||
  name: mayfly-go
 | 
					 | 
				
			||||||
  version: 1.2.3
 | 
					 | 
				
			||||||
  
 | 
					 | 
				
			||||||
server:
 | 
					server:
 | 
				
			||||||
  # debug release test
 | 
					  # debug release test
 | 
				
			||||||
  model: release
 | 
					  model: release
 | 
				
			||||||
@@ -11,24 +7,14 @@ server:
 | 
				
			|||||||
    enable: false
 | 
					    enable: false
 | 
				
			||||||
    key-file: ./default.key
 | 
					    key-file: ./default.key
 | 
				
			||||||
    cert-file: ./default.pem
 | 
					    cert-file: ./default.pem
 | 
				
			||||||
  # 静态资源
 | 
					 | 
				
			||||||
  static:
 | 
					 | 
				
			||||||
    - relative-path: /assets
 | 
					 | 
				
			||||||
      root: ./static/assets
 | 
					 | 
				
			||||||
  # 静态文件
 | 
					 | 
				
			||||||
  static-file:
 | 
					 | 
				
			||||||
    - relative-path: /
 | 
					 | 
				
			||||||
      filepath: ./static/index.html
 | 
					 | 
				
			||||||
    - relative-path: /favicon.ico
 | 
					 | 
				
			||||||
      filepath: ./static/favicon.ico
 | 
					 | 
				
			||||||
    - relative-path: /config.js
 | 
					 | 
				
			||||||
      filepath: ./static/config.js
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
jwt:
 | 
					jwt:
 | 
				
			||||||
  key: mykey
 | 
					  # jwt key,不设置默认使用随机字符串
 | 
				
			||||||
 | 
					  key: 
 | 
				
			||||||
  # 过期时间单位分钟
 | 
					  # 过期时间单位分钟
 | 
				
			||||||
  expire-time: 1440
 | 
					  expire-time: 1440
 | 
				
			||||||
 | 
					# 资源密码aes加密key
 | 
				
			||||||
 | 
					aes:
 | 
				
			||||||
 | 
					  key: 1111111111111111
 | 
				
			||||||
mysql:
 | 
					mysql:
 | 
				
			||||||
  host: localhost:3306
 | 
					  host: localhost:3306
 | 
				
			||||||
  username: root
 | 
					  username: root
 | 
				
			||||||
@@ -36,7 +22,6 @@ mysql:
 | 
				
			|||||||
  db-name: mayfly-go
 | 
					  db-name: mayfly-go
 | 
				
			||||||
  config: charset=utf8&loc=Local&parseTime=true
 | 
					  config: charset=utf8&loc=Local&parseTime=true
 | 
				
			||||||
  max-idle-conns: 5
 | 
					  max-idle-conns: 5
 | 
				
			||||||
 | 
					 | 
				
			||||||
log:
 | 
					log:
 | 
				
			||||||
   # 日志等级, trace, debug, info, warn, error, fatal
 | 
					   # 日志等级, trace, debug, info, warn, error, fatal
 | 
				
			||||||
  level: info
 | 
					  level: info
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -3,23 +3,23 @@ module mayfly-go
 | 
				
			|||||||
go 1.18
 | 
					go 1.18
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
	github.com/dgrijalva/jwt-go v3.2.0+incompatible // jwt
 | 
					 | 
				
			||||||
	github.com/gin-gonic/gin v1.8.1
 | 
						github.com/gin-gonic/gin v1.8.1
 | 
				
			||||||
	github.com/go-redis/redis/v8 v8.11.5
 | 
						github.com/go-redis/redis/v8 v8.11.5
 | 
				
			||||||
	github.com/go-sql-driver/mysql v1.6.0
 | 
						github.com/go-sql-driver/mysql v1.6.0
 | 
				
			||||||
 | 
						github.com/golang-jwt/jwt/v4 v4.4.2
 | 
				
			||||||
	github.com/gorilla/websocket v1.5.0
 | 
						github.com/gorilla/websocket v1.5.0
 | 
				
			||||||
	github.com/lib/pq v1.10.6
 | 
						github.com/lib/pq v1.10.6
 | 
				
			||||||
	github.com/mojocn/base64Captcha v1.3.5 // 验证码
 | 
						github.com/mojocn/base64Captcha v1.3.5 // 验证码
 | 
				
			||||||
	github.com/pkg/sftp v1.13.5
 | 
						github.com/pkg/sftp v1.13.5
 | 
				
			||||||
	github.com/robfig/cron/v3 v3.0.1 // 定时任务
 | 
						github.com/robfig/cron/v3 v3.0.1 // 定时任务
 | 
				
			||||||
	github.com/sirupsen/logrus v1.8.1
 | 
						github.com/sirupsen/logrus v1.9.0
 | 
				
			||||||
	github.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2
 | 
						github.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2
 | 
				
			||||||
	go.mongodb.org/mongo-driver v1.9.1 // mongo
 | 
						go.mongodb.org/mongo-driver v1.9.1 // mongo
 | 
				
			||||||
	golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d // ssh
 | 
						golang.org/x/crypto v0.0.0-20220722155217-630584e8d5aa // ssh
 | 
				
			||||||
	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b
 | 
						gopkg.in/yaml.v3 v3.0.1
 | 
				
			||||||
	// gorm
 | 
						// gorm
 | 
				
			||||||
	gorm.io/driver/mysql v1.3.4
 | 
						gorm.io/driver/mysql v1.3.5
 | 
				
			||||||
	gorm.io/gorm v1.23.5
 | 
						gorm.io/gorm v1.23.8
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
@@ -34,7 +34,7 @@ require (
 | 
				
			|||||||
	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
 | 
						github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
 | 
				
			||||||
	github.com/golang/snappy v0.0.1 // indirect
 | 
						github.com/golang/snappy v0.0.1 // indirect
 | 
				
			||||||
	github.com/jinzhu/inflection v1.0.0 // indirect
 | 
						github.com/jinzhu/inflection v1.0.0 // indirect
 | 
				
			||||||
	github.com/jinzhu/now v1.1.4 // indirect
 | 
						github.com/jinzhu/now v1.1.5 // indirect
 | 
				
			||||||
	github.com/json-iterator/go v1.1.12 // indirect
 | 
						github.com/json-iterator/go v1.1.12 // indirect
 | 
				
			||||||
	github.com/klauspost/compress v1.13.6 // indirect
 | 
						github.com/klauspost/compress v1.13.6 // indirect
 | 
				
			||||||
	github.com/kr/fs v0.1.0 // indirect
 | 
						github.com/kr/fs v0.1.0 // indirect
 | 
				
			||||||
@@ -52,7 +52,7 @@ require (
 | 
				
			|||||||
	golang.org/x/image v0.0.0-20220302094943-723b81ca9867 // indirect
 | 
						golang.org/x/image v0.0.0-20220302094943-723b81ca9867 // indirect
 | 
				
			||||||
	golang.org/x/net v0.0.0-20220225172249-27dd8689420f // indirect
 | 
						golang.org/x/net v0.0.0-20220225172249-27dd8689420f // indirect
 | 
				
			||||||
	golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9 // indirect
 | 
						golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9 // indirect
 | 
				
			||||||
	golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5 // indirect
 | 
						golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect
 | 
				
			||||||
	golang.org/x/text v0.3.7 // indirect
 | 
						golang.org/x/text v0.3.7 // indirect
 | 
				
			||||||
	golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 // indirect
 | 
						golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 // indirect
 | 
				
			||||||
	google.golang.org/protobuf v1.28.0 // indirect
 | 
						google.golang.org/protobuf v1.28.0 // indirect
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,16 +2,25 @@ package initialize
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"io/fs"
 | 
				
			||||||
	common_router "mayfly-go/internal/common/router"
 | 
						common_router "mayfly-go/internal/common/router"
 | 
				
			||||||
	devops_router "mayfly-go/internal/devops/router"
 | 
						devops_router "mayfly-go/internal/devops/router"
 | 
				
			||||||
	sys_router "mayfly-go/internal/sys/router"
 | 
						sys_router "mayfly-go/internal/sys/router"
 | 
				
			||||||
	"mayfly-go/pkg/config"
 | 
						"mayfly-go/pkg/config"
 | 
				
			||||||
	"mayfly-go/pkg/middleware"
 | 
						"mayfly-go/pkg/middleware"
 | 
				
			||||||
 | 
						"mayfly-go/static"
 | 
				
			||||||
	"net/http"
 | 
						"net/http"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/gin-gonic/gin"
 | 
						"github.com/gin-gonic/gin"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func WrapStaticHandler(h http.Handler) gin.HandlerFunc {
 | 
				
			||||||
 | 
						return func(c *gin.Context) {
 | 
				
			||||||
 | 
							c.Writer.Header().Set("Cache-Control", `public, max-age=31536000`)
 | 
				
			||||||
 | 
							h.ServeHTTP(c.Writer, c.Request)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func InitRouter() *gin.Engine {
 | 
					func InitRouter() *gin.Engine {
 | 
				
			||||||
	// server配置
 | 
						// server配置
 | 
				
			||||||
	serverConfig := config.Conf.Server
 | 
						serverConfig := config.Conf.Server
 | 
				
			||||||
@@ -25,12 +34,21 @@ func InitRouter() *gin.Engine {
 | 
				
			|||||||
		g.JSON(http.StatusNotFound, gin.H{"code": 404, "msg": fmt.Sprintf("not found '%s:%s'", g.Request.Method, g.Request.URL.Path)})
 | 
							g.JSON(http.StatusNotFound, gin.H{"code": 404, "msg": fmt.Sprintf("not found '%s:%s'", g.Request.Method, g.Request.URL.Path)})
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 使用embed打包静态资源至二进制文件中
 | 
				
			||||||
 | 
						fsys, _ := fs.Sub(static.Static, "static")
 | 
				
			||||||
 | 
						fileServer := http.FileServer(http.FS(fsys))
 | 
				
			||||||
 | 
						handler := WrapStaticHandler(fileServer)
 | 
				
			||||||
 | 
						router.GET("/", handler)
 | 
				
			||||||
 | 
						router.GET("/favicon.ico", handler)
 | 
				
			||||||
 | 
						router.GET("/config.js", handler)
 | 
				
			||||||
 | 
						// 所有/assets/**开头的都是静态资源文件
 | 
				
			||||||
 | 
						router.GET("/assets/*file", handler)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 设置静态资源
 | 
						// 设置静态资源
 | 
				
			||||||
	if staticConfs := serverConfig.Static; staticConfs != nil {
 | 
						if staticConfs := serverConfig.Static; staticConfs != nil {
 | 
				
			||||||
		for _, scs := range *staticConfs {
 | 
							for _, scs := range *staticConfs {
 | 
				
			||||||
			router.Static(scs.RelativePath, scs.Root)
 | 
								router.Static(scs.RelativePath, scs.Root)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	// 设置静态文件
 | 
						// 设置静态文件
 | 
				
			||||||
	if staticFileConfs := serverConfig.StaticFile; staticFileConfs != nil {
 | 
						if staticFileConfs := serverConfig.StaticFile; staticFileConfs != nil {
 | 
				
			||||||
@@ -38,6 +56,7 @@ func InitRouter() *gin.Engine {
 | 
				
			|||||||
			router.StaticFile(sfs.RelativePath, sfs.Filepath)
 | 
								router.StaticFile(sfs.RelativePath, sfs.Filepath)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 是否允许跨域
 | 
						// 是否允许跨域
 | 
				
			||||||
	if serverConfig.Cors {
 | 
						if serverConfig.Cors {
 | 
				
			||||||
		router.Use(middleware.Cors())
 | 
							router.Use(middleware.Cors())
 | 
				
			||||||
@@ -55,6 +74,7 @@ func InitRouter() *gin.Engine {
 | 
				
			|||||||
		sys_router.InitRoleRouter(api)
 | 
							sys_router.InitRoleRouter(api)
 | 
				
			||||||
		sys_router.InitSystemRouter(api)
 | 
							sys_router.InitSystemRouter(api)
 | 
				
			||||||
		sys_router.InitSyslogRouter(api)
 | 
							sys_router.InitSyslogRouter(api)
 | 
				
			||||||
 | 
							sys_router.InitSysConfigRouter(api)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		devops_router.InitProjectRouter(api)
 | 
							devops_router.InitProjectRouter(api)
 | 
				
			||||||
		devops_router.InitDbRouter(api)
 | 
							devops_router.InitDbRouter(api)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										35
									
								
								server/internal/common/utils/pwd.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								server/internal/common/utils/pwd.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,35 @@
 | 
				
			|||||||
 | 
					package utils
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/config"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 使用config.yml的aes.key进行密码加密
 | 
				
			||||||
 | 
					func PwdAesEncrypt(password string) string {
 | 
				
			||||||
 | 
						if password == "" {
 | 
				
			||||||
 | 
							return ""
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						aes := config.Conf.Aes
 | 
				
			||||||
 | 
						if aes == nil {
 | 
				
			||||||
 | 
							return password
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						encryptPwd, err := aes.EncryptBase64([]byte(password))
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "密码加密失败: %s")
 | 
				
			||||||
 | 
						return encryptPwd
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 使用config.yml的aes.key进行密码解密
 | 
				
			||||||
 | 
					func PwdAesDecrypt(encryptPwd string) string {
 | 
				
			||||||
 | 
						if encryptPwd == "" {
 | 
				
			||||||
 | 
							return ""
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						aes := config.Conf.Aes
 | 
				
			||||||
 | 
						if aes == nil {
 | 
				
			||||||
 | 
							return encryptPwd
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						decryptPwd, err := aes.DecryptBase64(encryptPwd)
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "密码解密失败: %s")
 | 
				
			||||||
 | 
						// 解密后的密码
 | 
				
			||||||
 | 
						return string(decryptPwd)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										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 = 4 * time.Minute
 | 
				
			||||||
 | 
					// DbConnExpireTime      = 2 * time.Minute
 | 
				
			||||||
 | 
					// RedisConnExpireTime   = 2 * time.Minute
 | 
				
			||||||
 | 
					// MongoConnExpireTime   = 2 * time.Minute
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
@@ -55,21 +55,40 @@ func (d *Db) Save(rc *ctx.ReqCtx) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	// 密码脱敏记录日志
 | 
						// 密码脱敏记录日志
 | 
				
			||||||
	form.Password = "****"
 | 
						form.Password = "****"
 | 
				
			||||||
 | 
					 | 
				
			||||||
	// if form.Type == "mysql" && form.EnableSSH == 1 {
 | 
					 | 
				
			||||||
	// 	// originSSHPwd, err := utils.DefaultRsaDecrypt(form.SSHPass, true)
 | 
					 | 
				
			||||||
	// 	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
					 | 
				
			||||||
	// 	// db.SSHPass = originSSHPwd
 | 
					 | 
				
			||||||
	// 	// 密码脱敏记录日志
 | 
					 | 
				
			||||||
	// 	form.SSHPass = "****"
 | 
					 | 
				
			||||||
	// }
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	rc.ReqParam = form
 | 
						rc.ReqParam = form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	db.SetBaseInfo(rc.LoginAccount)
 | 
						db.SetBaseInfo(rc.LoginAccount)
 | 
				
			||||||
	d.DbApp.Save(db)
 | 
						d.DbApp.Save(db)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取数据库实例密码,由于数据库是加密存储,故提供该接口展示原文密码
 | 
				
			||||||
 | 
					func (d *Db) GetDbPwd(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						dbId := GetDbId(rc.GinCtx)
 | 
				
			||||||
 | 
						dbEntity := d.DbApp.GetById(dbId, "Password")
 | 
				
			||||||
 | 
						dbEntity.PwdDecrypt()
 | 
				
			||||||
 | 
						rc.ResData = dbEntity.Password
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取数据库实例的所有数据库名
 | 
				
			||||||
 | 
					func (d *Db) GetDatabaseNames(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
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 如果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) {
 | 
					func (d *Db) DeleteDb(rc *ctx.ReqCtx) {
 | 
				
			||||||
	dbId := GetDbId(rc.GinCtx)
 | 
						dbId := GetDbId(rc.GinCtx)
 | 
				
			||||||
	d.DbApp.Delete(dbId)
 | 
						d.DbApp.Delete(dbId)
 | 
				
			||||||
@@ -78,19 +97,19 @@ func (d *Db) DeleteDb(rc *ctx.ReqCtx) {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *Db) TableInfos(rc *ctx.ReqCtx) {
 | 
					func (d *Db) TableInfos(rc *ctx.ReqCtx) {
 | 
				
			||||||
	rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetTableInfos()
 | 
						rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetMeta().GetTableInfos()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *Db) TableIndex(rc *ctx.ReqCtx) {
 | 
					func (d *Db) TableIndex(rc *ctx.ReqCtx) {
 | 
				
			||||||
	tn := rc.GinCtx.Query("tableName")
 | 
						tn := rc.GinCtx.Query("tableName")
 | 
				
			||||||
	biz.NotEmpty(tn, "tableName不能为空")
 | 
						biz.NotEmpty(tn, "tableName不能为空")
 | 
				
			||||||
	rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetTableIndex(tn)
 | 
						rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetMeta().GetTableIndex(tn)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *Db) GetCreateTableDdl(rc *ctx.ReqCtx) {
 | 
					func (d *Db) GetCreateTableDdl(rc *ctx.ReqCtx) {
 | 
				
			||||||
	tn := rc.GinCtx.Query("tableName")
 | 
						tn := rc.GinCtx.Query("tableName")
 | 
				
			||||||
	biz.NotEmpty(tn, "tableName不能为空")
 | 
						biz.NotEmpty(tn, "tableName不能为空")
 | 
				
			||||||
	rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetCreateTableDdl(tn)
 | 
						rc.ResData = d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetMeta().GetCreateTableDdl(tn)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *Db) ExecSql(rc *ctx.ReqCtx) {
 | 
					func (d *Db) ExecSql(rc *ctx.ReqCtx) {
 | 
				
			||||||
@@ -218,11 +237,12 @@ func (d *Db) DumpSql(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	writer.WriteString(fmt.Sprintf("\n-- 导出数据库: %s ", db))
 | 
						writer.WriteString(fmt.Sprintf("\n-- 导出数据库: %s ", db))
 | 
				
			||||||
	writer.WriteString("\n-- ----------------------------\n")
 | 
						writer.WriteString("\n-- ----------------------------\n")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						dbmeta := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx)).GetMeta()
 | 
				
			||||||
	for _, table := range tables {
 | 
						for _, table := range tables {
 | 
				
			||||||
		if needStruct {
 | 
							if needStruct {
 | 
				
			||||||
			writer.WriteString(fmt.Sprintf("\n-- ----------------------------\n-- 表结构: %s \n-- ----------------------------\n", table))
 | 
								writer.WriteString(fmt.Sprintf("\n-- ----------------------------\n-- 表结构: %s \n-- ----------------------------\n", table))
 | 
				
			||||||
			writer.WriteString(fmt.Sprintf("DROP TABLE IF EXISTS `%s`;\n", table))
 | 
								writer.WriteString(fmt.Sprintf("DROP TABLE IF EXISTS `%s`;\n", table))
 | 
				
			||||||
			writer.WriteString(dbInstance.GetCreateTableDdl(table)[0]["Create Table"].(string) + ";\n")
 | 
								writer.WriteString(dbmeta.GetCreateTableDdl(table)[0]["Create Table"].(string) + ";\n")
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if !needData {
 | 
							if !needData {
 | 
				
			||||||
@@ -284,7 +304,7 @@ func (d *Db) DumpSql(rc *ctx.ReqCtx) {
 | 
				
			|||||||
func (d *Db) TableMA(rc *ctx.ReqCtx) {
 | 
					func (d *Db) TableMA(rc *ctx.ReqCtx) {
 | 
				
			||||||
	dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
						dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
				
			||||||
	biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
						biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
				
			||||||
	rc.ResData = dbi.GetTableMetedatas()
 | 
						rc.ResData = dbi.GetMeta().GetTables()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// @router /api/db/:dbId/c-metadata [get]
 | 
					// @router /api/db/:dbId/c-metadata [get]
 | 
				
			||||||
@@ -295,16 +315,17 @@ func (d *Db) ColumnMA(rc *ctx.ReqCtx) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
						dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
				
			||||||
	biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
						biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
				
			||||||
	rc.ResData = dbi.GetColumnMetadatas(tn)
 | 
						rc.ResData = dbi.GetMeta().GetColumns(tn)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// @router /api/db/:dbId/hint-tables [get]
 | 
					// @router /api/db/:dbId/hint-tables [get]
 | 
				
			||||||
func (d *Db) HintTables(rc *ctx.ReqCtx) {
 | 
					func (d *Db) HintTables(rc *ctx.ReqCtx) {
 | 
				
			||||||
	dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
						dbi := d.DbApp.GetDbInstance(GetIdAndDb(rc.GinCtx))
 | 
				
			||||||
	biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
						biz.ErrIsNilAppendErr(d.ProjectApp.CanAccess(rc.LoginAccount.Id, dbi.ProjectId), "%s")
 | 
				
			||||||
	// 获取所有表
 | 
					 | 
				
			||||||
	tables := dbi.GetTableMetedatas()
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						dm := dbi.GetMeta()
 | 
				
			||||||
 | 
						// 获取所有表
 | 
				
			||||||
 | 
						tables := dm.GetTables()
 | 
				
			||||||
	tableNames := make([]string, 0)
 | 
						tableNames := make([]string, 0)
 | 
				
			||||||
	for _, v := range tables {
 | 
						for _, v := range tables {
 | 
				
			||||||
		tableNames = append(tableNames, v["tableName"].(string))
 | 
							tableNames = append(tableNames, v["tableName"].(string))
 | 
				
			||||||
@@ -319,7 +340,7 @@ func (d *Db) HintTables(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 获取所有表下的所有列信息
 | 
						// 获取所有表下的所有列信息
 | 
				
			||||||
	columnMds := dbi.GetColumnMetadatas(tableNames...)
 | 
						columnMds := dm.GetColumns(tableNames...)
 | 
				
			||||||
	for _, v := range columnMds {
 | 
						for _, v := range columnMds {
 | 
				
			||||||
		tName := v["tableName"].(string)
 | 
							tName := v["tableName"].(string)
 | 
				
			||||||
		if res[tName] == nil {
 | 
							if res[tName] == nil {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,7 +9,7 @@ type DbForm struct {
 | 
				
			|||||||
	Username  string `binding:"required" json:"username"`
 | 
						Username  string `binding:"required" json:"username"`
 | 
				
			||||||
	Password  string `json:"password"`
 | 
						Password  string `json:"password"`
 | 
				
			||||||
	Params    string `json:"params"`
 | 
						Params    string `json:"params"`
 | 
				
			||||||
	Database  string `binding:"required" json:"database"`
 | 
						Database  string `json:"database"`
 | 
				
			||||||
	ProjectId uint64 `binding:"required" json:"projectId"`
 | 
						ProjectId uint64 `binding:"required" json:"projectId"`
 | 
				
			||||||
	Project   string `json:"project"`
 | 
						Project   string `json:"project"`
 | 
				
			||||||
	Env       string `json:"env"`
 | 
						Env       string `json:"env"`
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,16 +1,18 @@
 | 
				
			|||||||
package form
 | 
					package form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type MachineForm struct {
 | 
					type MachineForm struct {
 | 
				
			||||||
	Id          uint64 `json:"id"`
 | 
						Id                 uint64 `json:"id"`
 | 
				
			||||||
	ProjectId   uint64 `json:"projectId"`
 | 
						ProjectId          uint64 `json:"projectId"`
 | 
				
			||||||
	ProjectName string `json:"projectName"`
 | 
						ProjectName        string `json:"projectName"`
 | 
				
			||||||
	Name        string `json:"name" binding:"required"`
 | 
						Name               string `json:"name" binding:"required"`
 | 
				
			||||||
	Ip          string `json:"ip" binding:"required"`       // IP地址
 | 
						Ip                 string `json:"ip" binding:"required"`       // IP地址
 | 
				
			||||||
	Username    string `json:"username" binding:"required"` // 用户名
 | 
						Username           string `json:"username" binding:"required"` // 用户名
 | 
				
			||||||
	AuthMethod  int8   `json:"authMethod" binding:"required"`
 | 
						AuthMethod         int8   `json:"authMethod" binding:"required"`
 | 
				
			||||||
	Password    string `json:"password"`
 | 
						Password           string `json:"password"`
 | 
				
			||||||
	Port        int    `json:"port" binding:"required"` // 端口号
 | 
						Port               int    `json:"port" binding:"required"` // 端口号
 | 
				
			||||||
	Remark      string `json:"remark"`
 | 
						Remark             string `json:"remark"`
 | 
				
			||||||
 | 
						EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
				
			||||||
 | 
						SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type MachineRunForm struct {
 | 
					type MachineRunForm struct {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -72,6 +72,14 @@ func (m *Machine) SaveMachine(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	m.MachineApp.Save(me)
 | 
						m.MachineApp.Save(me)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取机器实例密码,由于数据库是加密存储,故提供该接口展示原文密码
 | 
				
			||||||
 | 
					func (m *Machine) GetMachinePwd(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						mid := GetMachineId(rc.GinCtx)
 | 
				
			||||||
 | 
						me := m.MachineApp.GetById(mid, "Password")
 | 
				
			||||||
 | 
						me.PwdDecrypt()
 | 
				
			||||||
 | 
						rc.ResData = me.Password
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (m *Machine) ChangeStatus(rc *ctx.ReqCtx) {
 | 
					func (m *Machine) ChangeStatus(rc *ctx.ReqCtx) {
 | 
				
			||||||
	g := rc.GinCtx
 | 
						g := rc.GinCtx
 | 
				
			||||||
	id := uint64(ginx.PathParamInt(g, "machineId"))
 | 
						id := uint64(ginx.PathParamInt(g, "machineId"))
 | 
				
			||||||
@@ -130,7 +138,7 @@ func (m *Machine) KillProcess(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	cli := m.MachineApp.GetCli(GetMachineId(rc.GinCtx))
 | 
						cli := m.MachineApp.GetCli(GetMachineId(rc.GinCtx))
 | 
				
			||||||
	biz.ErrIsNilAppendErr(m.ProjectApp.CanAccess(rc.LoginAccount.Id, cli.GetMachine().ProjectId), "%s")
 | 
						biz.ErrIsNilAppendErr(m.ProjectApp.CanAccess(rc.LoginAccount.Id, cli.GetMachine().ProjectId), "%s")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	_, err := cli.Run("kill -9 " + pid)
 | 
						_, err := cli.Run("sudo kill -9 " + pid)
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "终止进程失败: %s")
 | 
						biz.ErrIsNilAppendErr(err, "终止进程失败: %s")
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -152,21 +160,16 @@ func (m *Machine) WsSSH(g *gin.Context) {
 | 
				
			|||||||
		panic(biz.NewBizErr("\033[1;31m您没有权限操作该机器终端,请重新登录后再试~\033[0m"))
 | 
							panic(biz.NewBizErr("\033[1;31m您没有权限操作该机器终端,请重新登录后再试~\033[0m"))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	cols := ginx.QueryInt(g, "cols", 80)
 | 
					 | 
				
			||||||
	rows := ginx.QueryInt(g, "rows", 40)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	cli := m.MachineApp.GetCli(GetMachineId(g))
 | 
						cli := m.MachineApp.GetCli(GetMachineId(g))
 | 
				
			||||||
	biz.ErrIsNilAppendErr(m.ProjectApp.CanAccess(rc.LoginAccount.Id, cli.GetMachine().ProjectId), "%s")
 | 
						biz.ErrIsNilAppendErr(m.ProjectApp.CanAccess(rc.LoginAccount.Id, cli.GetMachine().ProjectId), "%s")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	sws, err := machine.NewLogicSshWsSession(cols, rows, cli, wsConn)
 | 
						cols := ginx.QueryInt(g, "cols", 80)
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "\033[1;31m连接失败:%s\033[0m")
 | 
						rows := ginx.QueryInt(g, "rows", 40)
 | 
				
			||||||
	defer sws.Close()
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	quitChan := make(chan bool, 3)
 | 
						mts, err := machine.NewTerminalSession(utils.RandString(16), wsConn, cli, rows, cols)
 | 
				
			||||||
	sws.Start(quitChan)
 | 
						biz.ErrIsNilAppendErr(err, "\033[1;31m连接失败: %s\033[0m")
 | 
				
			||||||
	go sws.Wait(quitChan)
 | 
						mts.Start()
 | 
				
			||||||
 | 
						defer mts.Stop()
 | 
				
			||||||
	<-quitChan
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func GetMachineId(g *gin.Context) uint64 {
 | 
					func GetMachineId(g *gin.Context) uint64 {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -9,6 +9,7 @@ import (
 | 
				
			|||||||
	"mayfly-go/pkg/ctx"
 | 
						"mayfly-go/pkg/ctx"
 | 
				
			||||||
	"mayfly-go/pkg/ginx"
 | 
						"mayfly-go/pkg/ginx"
 | 
				
			||||||
	"mayfly-go/pkg/utils"
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
 | 
						"regexp"
 | 
				
			||||||
	"strconv"
 | 
						"strconv"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/gin-gonic/gin"
 | 
						"github.com/gin-gonic/gin"
 | 
				
			||||||
@@ -34,14 +35,15 @@ func (m *Mongo) Save(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	form := &form.Mongo{}
 | 
						form := &form.Mongo{}
 | 
				
			||||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
						ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	rc.ReqParam = form
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	mongo := new(entity.Mongo)
 | 
						mongo := new(entity.Mongo)
 | 
				
			||||||
	utils.Copy(mongo, form)
 | 
						utils.Copy(mongo, form)
 | 
				
			||||||
	// 解密uri,并使用解密后的赋值
 | 
					
 | 
				
			||||||
	originUri, err := utils.DefaultRsaDecrypt(form.Uri, true)
 | 
						// 密码脱敏记录日志
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "解密uri错误: %s")
 | 
						form.Uri = func(str string) string {
 | 
				
			||||||
	mongo.Uri = originUri
 | 
							reg := regexp.MustCompile(`(^mongodb://.+?:)(.+)(@.+$)`)
 | 
				
			||||||
 | 
							return reg.ReplaceAllString(str, `${1}****${3}`)
 | 
				
			||||||
 | 
						}(form.Uri)
 | 
				
			||||||
 | 
						rc.ReqParam = form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	mongo.SetBaseInfo(rc.LoginAccount)
 | 
						mongo.SetBaseInfo(rc.LoginAccount)
 | 
				
			||||||
	m.MongoApp.Save(mongo)
 | 
						m.MongoApp.Save(mongo)
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -52,6 +52,14 @@ func (r *Redis) Save(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	r.RedisApp.Save(redis)
 | 
						r.RedisApp.Save(redis)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取redis实例密码,由于数据库是加密存储,故提供该接口展示原文密码
 | 
				
			||||||
 | 
					func (r *Redis) GetRedisPwd(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						rid := uint64(ginx.PathParamInt(rc.GinCtx, "id"))
 | 
				
			||||||
 | 
						re := r.RedisApp.GetById(rid, "Password")
 | 
				
			||||||
 | 
						re.PwdDecrypt()
 | 
				
			||||||
 | 
						rc.ResData = re.Password
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (r *Redis) DeleteRedis(rc *ctx.ReqCtx) {
 | 
					func (r *Redis) DeleteRedis(rc *ctx.ReqCtx) {
 | 
				
			||||||
	r.RedisApp.Delete(uint64(ginx.PathParamInt(rc.GinCtx, "id")))
 | 
						r.RedisApp.Delete(uint64(ginx.PathParamInt(rc.GinCtx, "id")))
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -185,7 +193,7 @@ func (r *Redis) Scan(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	kis := make([]*vo.KeyInfo, 0)
 | 
						kis := make([]*vo.KeyInfo, 0)
 | 
				
			||||||
	var cursorRes map[string]uint64 = make(map[string]uint64)
 | 
						var cursorRes map[string]uint64 = make(map[string]uint64)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if ri.Mode == "" || ri.Mode == entity.RedisModeStandalone {
 | 
						if ri.Mode == "" || ri.Mode == entity.RedisModeStandalone || ri.Mode == entity.RedisModeSentinel {
 | 
				
			||||||
		redisAddr := ri.Cli.Options().Addr
 | 
							redisAddr := ri.Cli.Options().Addr
 | 
				
			||||||
		keys, cursor := ri.Scan(form.Cursor[redisAddr], form.Match, form.Count)
 | 
							keys, cursor := ri.Scan(form.Cursor[redisAddr], form.Match, form.Count)
 | 
				
			||||||
		cursorRes[redisAddr] = cursor
 | 
							cursorRes[redisAddr] = cursor
 | 
				
			||||||
@@ -277,13 +285,6 @@ func (r *Redis) GetStringValue(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	rc.ResData = str
 | 
						rc.ResData = str
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (r *Redis) GetHashValue(rc *ctx.ReqCtx) {
 | 
					 | 
				
			||||||
	ri, key := r.checkKey(rc)
 | 
					 | 
				
			||||||
	res, err := ri.GetCmdable().HGetAll(context.TODO(), key).Result()
 | 
					 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "获取hash值失败: %s")
 | 
					 | 
				
			||||||
	rc.ResData = res
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (r *Redis) SetStringValue(rc *ctx.ReqCtx) {
 | 
					func (r *Redis) SetStringValue(rc *ctx.ReqCtx) {
 | 
				
			||||||
	g := rc.GinCtx
 | 
						g := rc.GinCtx
 | 
				
			||||||
	keyValue := new(form.StringValue)
 | 
						keyValue := new(form.StringValue)
 | 
				
			||||||
@@ -297,6 +298,45 @@ func (r *Redis) SetStringValue(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	rc.ResData = str
 | 
						rc.ResData = str
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *Redis) Hscan(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						ri, key := r.checkKey(rc)
 | 
				
			||||||
 | 
						g := rc.GinCtx
 | 
				
			||||||
 | 
						count := ginx.QueryInt(g, "count", 10)
 | 
				
			||||||
 | 
						match := g.Query("match")
 | 
				
			||||||
 | 
						cursor := ginx.QueryInt(g, "cursor", 0)
 | 
				
			||||||
 | 
						contextTodo := context.TODO()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						cmdable := ri.GetCmdable()
 | 
				
			||||||
 | 
						keys, nextCursor, err := cmdable.HScan(contextTodo, key, uint64(cursor), match, int64(count)).Result()
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "hcan err: %s")
 | 
				
			||||||
 | 
						keySize, err := cmdable.HLen(contextTodo, key).Result()
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "hlen err: %s")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						rc.ResData = map[string]interface{}{
 | 
				
			||||||
 | 
							"keys":    keys,
 | 
				
			||||||
 | 
							"cursor":  nextCursor,
 | 
				
			||||||
 | 
							"keySize": keySize,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *Redis) Hdel(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						ri, key := r.checkKey(rc)
 | 
				
			||||||
 | 
						field := rc.GinCtx.Query("field")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						delRes, err := ri.GetCmdable().HDel(context.TODO(), key, field).Result()
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "hdel err: %s")
 | 
				
			||||||
 | 
						rc.ResData = delRes
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *Redis) Hget(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						ri, key := r.checkKey(rc)
 | 
				
			||||||
 | 
						field := rc.GinCtx.Query("field")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						res, err := ri.GetCmdable().HGet(context.TODO(), key, field).Result()
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "hget err: %s")
 | 
				
			||||||
 | 
						rc.ResData = res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (r *Redis) SetHashValue(rc *ctx.ReqCtx) {
 | 
					func (r *Redis) SetHashValue(rc *ctx.ReqCtx) {
 | 
				
			||||||
	g := rc.GinCtx
 | 
						g := rc.GinCtx
 | 
				
			||||||
	hashValue := new(form.HashValue)
 | 
						hashValue := new(form.HashValue)
 | 
				
			||||||
@@ -307,13 +347,12 @@ func (r *Redis) SetHashValue(rc *ctx.ReqCtx) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	cmd := ri.GetCmdable()
 | 
						cmd := ri.GetCmdable()
 | 
				
			||||||
	key := hashValue.Key
 | 
						key := hashValue.Key
 | 
				
			||||||
	// 简单处理->先删除,后新增
 | 
						contextTodo := context.TODO()
 | 
				
			||||||
	cmd.Del(context.TODO(), key)
 | 
					 | 
				
			||||||
	for _, v := range hashValue.Value {
 | 
						for _, v := range hashValue.Value {
 | 
				
			||||||
		res := cmd.HSet(context.TODO(), key, v["key"].(string), v["value"])
 | 
							res := cmd.HSet(contextTodo, key, v["field"].(string), v["value"])
 | 
				
			||||||
		biz.ErrIsNilAppendErr(res.Err(), "保存hash值失败: %s")
 | 
							biz.ErrIsNilAppendErr(res.Err(), "保存hash值失败: %s")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if hashValue.Timed != -1 {
 | 
						if hashValue.Timed != 0 && hashValue.Timed != -1 {
 | 
				
			||||||
		cmd.Expire(context.TODO(), key, time.Second*time.Duration(hashValue.Timed))
 | 
							cmd.Expire(context.TODO(), key, time.Second*time.Duration(hashValue.Timed))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -10,8 +10,8 @@ type Redis struct {
 | 
				
			|||||||
	ProjectId          *int64     `json:"projectId"`
 | 
						ProjectId          *int64     `json:"projectId"`
 | 
				
			||||||
	Project            *string    `json:"project"`
 | 
						Project            *string    `json:"project"`
 | 
				
			||||||
	Mode               *string    `json:"mode"`
 | 
						Mode               *string    `json:"mode"`
 | 
				
			||||||
	EnableSshTunnel    *int8      `orm:"column(enable_ssh_tunnel)" json:"enableSshTunnel"`        // 是否启用ssh隧道
 | 
						EnableSshTunnel    *int8      `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
				
			||||||
	SshTunnelMachineId *uint64    `orm:"column(ssh_tunnel_machine_id)" json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
						SshTunnelMachineId *uint64    `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
				
			||||||
	Remark             *string    `json:"remark"`
 | 
						Remark             *string    `json:"remark"`
 | 
				
			||||||
	Env                *string    `json:"env"`
 | 
						Env                *string    `json:"env"`
 | 
				
			||||||
	EnvId              *int64     `json:"envId"`
 | 
						EnvId              *int64     `json:"envId"`
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -15,23 +15,25 @@ type AccountVO struct {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
type MachineVO struct {
 | 
					type MachineVO struct {
 | 
				
			||||||
	//models.BaseModel
 | 
						//models.BaseModel
 | 
				
			||||||
	Id          *uint64    `json:"id"`
 | 
						Id                 *uint64    `json:"id"`
 | 
				
			||||||
	ProjectId   uint64     `json:"projectId"`
 | 
						ProjectId          uint64     `json:"projectId"`
 | 
				
			||||||
	ProjectName string     `json:"projectName"`
 | 
						ProjectName        string     `json:"projectName"`
 | 
				
			||||||
	Name        *string    `json:"name"`
 | 
						Name               *string    `json:"name"`
 | 
				
			||||||
	Username    *string    `json:"username"`
 | 
						Username           *string    `json:"username"`
 | 
				
			||||||
	Ip          *string    `json:"ip"`
 | 
						Ip                 *string    `json:"ip"`
 | 
				
			||||||
	Port        *int       `json:"port"`
 | 
						Port               *int       `json:"port"`
 | 
				
			||||||
	AuthMethod  *int8      `json:"authMethod"`
 | 
						AuthMethod         *int8      `json:"authMethod"`
 | 
				
			||||||
	Status      *int8      `json:"status"`
 | 
						Status             *int8      `json:"status"`
 | 
				
			||||||
	CreateTime  *time.Time `json:"createTime"`
 | 
						EnableSshTunnel    *int8      `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
				
			||||||
	Creator     *string    `json:"creator"`
 | 
						SshTunnelMachineId *uint64    `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
				
			||||||
	CreatorId   *int64     `json:"creatorId"`
 | 
						CreateTime         *time.Time `json:"createTime"`
 | 
				
			||||||
	UpdateTime  *time.Time `json:"updateTime"`
 | 
						Creator            *string    `json:"creator"`
 | 
				
			||||||
	Modifier    *string    `json:"modifier"`
 | 
						CreatorId          *int64     `json:"creatorId"`
 | 
				
			||||||
	ModifierId  *int64     `json:"modifierId"`
 | 
						UpdateTime         *time.Time `json:"updateTime"`
 | 
				
			||||||
	HasCli      bool       `json:"hasCli" gorm:"-"`
 | 
						Modifier           *string    `json:"modifier"`
 | 
				
			||||||
	Remark      *string    `json:"remark"`
 | 
						ModifierId         *int64     `json:"modifierId"`
 | 
				
			||||||
 | 
						HasCli             bool       `json:"hasCli" gorm:"-"`
 | 
				
			||||||
 | 
						Remark             *string    `json:"remark"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type MachineScriptVO struct {
 | 
					type MachineScriptVO struct {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,6 +5,7 @@ import (
 | 
				
			|||||||
	"database/sql"
 | 
						"database/sql"
 | 
				
			||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"mayfly-go/internal/constant"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/repository"
 | 
						"mayfly-go/internal/devops/domain/repository"
 | 
				
			||||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
						"mayfly-go/internal/devops/infrastructure/machine"
 | 
				
			||||||
@@ -23,7 +24,6 @@ import (
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	"github.com/go-sql-driver/mysql"
 | 
						"github.com/go-sql-driver/mysql"
 | 
				
			||||||
	"github.com/lib/pq"
 | 
						"github.com/lib/pq"
 | 
				
			||||||
	"golang.org/x/crypto/ssh"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Db interface {
 | 
					type Db interface {
 | 
				
			||||||
@@ -47,6 +47,9 @@ type Db interface {
 | 
				
			|||||||
	// @param id 数据库实例id
 | 
						// @param id 数据库实例id
 | 
				
			||||||
	// @param db 数据库
 | 
						// @param db 数据库
 | 
				
			||||||
	GetDbInstance(id uint64, db string) *DbInstance
 | 
						GetDbInstance(id uint64, db string) *DbInstance
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取数据库实例的所有数据库列表
 | 
				
			||||||
 | 
						GetDatabases(entity *entity.Db) []string
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type dbAppImpl struct {
 | 
					type dbAppImpl struct {
 | 
				
			||||||
@@ -94,6 +97,7 @@ func (d *dbAppImpl) Save(dbEntity *entity.Db) {
 | 
				
			|||||||
	if dbEntity.Id == 0 {
 | 
						if dbEntity.Id == 0 {
 | 
				
			||||||
		biz.NotEmpty(dbEntity.Password, "密码不能为空")
 | 
							biz.NotEmpty(dbEntity.Password, "密码不能为空")
 | 
				
			||||||
		biz.IsTrue(err != nil, "该数据库实例已存在")
 | 
							biz.IsTrue(err != nil, "该数据库实例已存在")
 | 
				
			||||||
 | 
							dbEntity.PwdEncrypt()
 | 
				
			||||||
		d.dbRepo.Insert(dbEntity)
 | 
							d.dbRepo.Insert(dbEntity)
 | 
				
			||||||
		return
 | 
							return
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -126,6 +130,7 @@ func (d *dbAppImpl) Save(dbEntity *entity.Db) {
 | 
				
			|||||||
		d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: dbId, Db: v.(string)})
 | 
							d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: dbId, Db: v.(string)})
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						dbEntity.PwdEncrypt()
 | 
				
			||||||
	d.dbRepo.Update(dbEntity)
 | 
						d.dbRepo.Update(dbEntity)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -141,11 +146,35 @@ func (d *dbAppImpl) Delete(id uint64) {
 | 
				
			|||||||
	d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: id})
 | 
						d.dbSqlRepo.DeleteBy(&entity.DbSql{DbId: id})
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d *dbAppImpl) GetDatabases(ed *entity.Db) []string {
 | 
				
			||||||
 | 
						ed.Network = ed.GetNetwork()
 | 
				
			||||||
 | 
						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
 | 
					var mutex sync.Mutex
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
					func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
				
			||||||
	mutex.Lock()
 | 
					 | 
				
			||||||
	defer mutex.Unlock()
 | 
					 | 
				
			||||||
	// Id不为0,则为需要缓存
 | 
						// Id不为0,则为需要缓存
 | 
				
			||||||
	needCache := id != 0
 | 
						needCache := id != 0
 | 
				
			||||||
	if needCache {
 | 
						if needCache {
 | 
				
			||||||
@@ -154,43 +183,19 @@ func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
				
			|||||||
			return load.(*DbInstance)
 | 
								return load.(*DbInstance)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						mutex.Lock()
 | 
				
			||||||
 | 
						defer mutex.Unlock()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	d := da.GetById(id)
 | 
						d := da.GetById(id)
 | 
				
			||||||
 | 
						// 密码解密
 | 
				
			||||||
 | 
						d.PwdDecrypt()
 | 
				
			||||||
	biz.NotNil(d, "数据库信息不存在")
 | 
						biz.NotNil(d, "数据库信息不存在")
 | 
				
			||||||
	biz.IsTrue(strings.Contains(d.Database, db), "未配置该库的操作权限")
 | 
						biz.IsTrue(strings.Contains(d.Database, db), "未配置该库的操作权限")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	cacheKey := GetDbCacheKey(id, db)
 | 
						cacheKey := GetDbCacheKey(id, db)
 | 
				
			||||||
	dbi := &DbInstance{Id: cacheKey, Type: d.Type, ProjectId: d.ProjectId}
 | 
						dbi := &DbInstance{Id: cacheKey, Type: d.Type, ProjectId: d.ProjectId, sshTunnelMachineId: d.SshTunnelMachineId}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	//SSH Conect
 | 
						DB, err := GetDbConn(d, db)
 | 
				
			||||||
	if d.EnableSshTunnel == 1 && d.SshTunnelMachineId != 0 {
 | 
					 | 
				
			||||||
		me := MachineApp.GetById(d.SshTunnelMachineId)
 | 
					 | 
				
			||||||
		biz.NotNil(me, "隧道机器信息不存在")
 | 
					 | 
				
			||||||
		sshClient, err := machine.GetSshClient(me)
 | 
					 | 
				
			||||||
		biz.ErrIsNilAppendErr(err, "ssh隧道连接失败: %s")
 | 
					 | 
				
			||||||
		dbi.sshTunnel = sshClient
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		if d.Type == entity.DbTypeMysql {
 | 
					 | 
				
			||||||
			mysql.RegisterDialContext(d.Network, func(ctx context.Context, addr string) (net.Conn, error) {
 | 
					 | 
				
			||||||
				return sshClient.Dial("tcp", addr)
 | 
					 | 
				
			||||||
			})
 | 
					 | 
				
			||||||
		} else if d.Type == entity.DbTypePostgres {
 | 
					 | 
				
			||||||
			_, err := pq.DialOpen(&PqSqlDialer{sshTunnel: sshClient}, getDsn(d))
 | 
					 | 
				
			||||||
			if err != nil {
 | 
					 | 
				
			||||||
				dbi.Close()
 | 
					 | 
				
			||||||
				panic(biz.NewBizErr(fmt.Sprintf("postgres隧道连接失败: %s", err.Error())))
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// 将数据库替换为要访问的数据库,原本数据库为空格拼接的所有库
 | 
					 | 
				
			||||||
	d.Database = db
 | 
					 | 
				
			||||||
	DB, err := sql.Open(d.Type, getDsn(d))
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		dbi.Close()
 | 
					 | 
				
			||||||
		panic(biz.NewBizErr(fmt.Sprintf("Open %s failed, err:%v\n", d.Type, err)))
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	err = DB.Ping()
 | 
					 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		dbi.Close()
 | 
							dbi.Close()
 | 
				
			||||||
		global.Log.Errorf("连接db失败: %s:%d/%s", d.Host, d.Port, db)
 | 
							global.Log.Errorf("连接db失败: %s:%d/%s", d.Host, d.Port, db)
 | 
				
			||||||
@@ -212,32 +217,32 @@ func (da *dbAppImpl) GetDbInstance(id uint64, db string) *DbInstance {
 | 
				
			|||||||
	return dbi
 | 
						return dbi
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type PqSqlDialer struct {
 | 
					 | 
				
			||||||
	sshTunnel *ssh.Client
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (pd *PqSqlDialer) Dial(network, address string) (net.Conn, error) {
 | 
					 | 
				
			||||||
	if sshConn, err := pd.sshTunnel.Dial(network, 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)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
//------------------------------------------------------------------------------
 | 
					//------------------------------------------------------------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 客户端连接缓存,30分钟内没有访问则会被关闭, key为数据库实例id:数据库
 | 
					// 单次最大查询数据集
 | 
				
			||||||
var dbCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
					const Max_Rows = 2000
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 客户端连接缓存,指定时间内没有访问则会被关闭, key为数据库实例id:数据库
 | 
				
			||||||
 | 
					var dbCache = cache.NewTimedCache(constant.DbConnExpireTime, 5*time.Second).
 | 
				
			||||||
	WithUpdateAccessTime(true).
 | 
						WithUpdateAccessTime(true).
 | 
				
			||||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
						OnEvicted(func(key interface{}, value interface{}) {
 | 
				
			||||||
		global.Log.Info(fmt.Sprintf("删除db连接缓存 id = %s", key))
 | 
							global.Log.Info(fmt.Sprintf("删除db连接缓存 id = %s", key))
 | 
				
			||||||
		value.(*DbInstance).Close()
 | 
							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 {
 | 
					func GetDbCacheKey(dbId uint64, db string) string {
 | 
				
			||||||
	return fmt.Sprintf("%d:%s", dbId, db)
 | 
						return fmt.Sprintf("%d:%s", dbId, db)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -250,55 +255,44 @@ func GetDbInstanceByCache(id string) *DbInstance {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func TestConnection(d *entity.Db) {
 | 
					func TestConnection(d *entity.Db) {
 | 
				
			||||||
	//SSH Conect
 | 
						// 验证第一个库是否可以连接即可
 | 
				
			||||||
 | 
						DB, err := GetDbConn(d, strings.Split(d.Database, " ")[0])
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "数据库连接失败: %s")
 | 
				
			||||||
 | 
						defer DB.Close()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取数据库连接
 | 
				
			||||||
 | 
					func GetDbConn(d *entity.Db, db string) (*sql.DB, error) {
 | 
				
			||||||
 | 
						// SSH Conect
 | 
				
			||||||
	if d.EnableSshTunnel == 1 && d.SshTunnelMachineId != 0 {
 | 
						if d.EnableSshTunnel == 1 && d.SshTunnelMachineId != 0 {
 | 
				
			||||||
		me := MachineApp.GetById(d.SshTunnelMachineId)
 | 
							sshTunnelMachine := MachineApp.GetSshTunnelMachine(d.SshTunnelMachineId)
 | 
				
			||||||
		sshClient, err := machine.GetSshClient(me)
 | 
					 | 
				
			||||||
		biz.ErrIsNilAppendErr(err, "ssh隧道连接失败: %s")
 | 
					 | 
				
			||||||
		defer sshClient.Close()
 | 
					 | 
				
			||||||
		if d.Type == entity.DbTypeMysql {
 | 
							if d.Type == entity.DbTypeMysql {
 | 
				
			||||||
			mysql.RegisterDialContext(d.Network, func(ctx context.Context, addr string) (net.Conn, error) {
 | 
								mysql.RegisterDialContext(d.Network, func(ctx context.Context, addr string) (net.Conn, error) {
 | 
				
			||||||
				return sshClient.Dial("tcp", addr)
 | 
									return sshTunnelMachine.GetDialConn("tcp", addr)
 | 
				
			||||||
			})
 | 
								})
 | 
				
			||||||
		} else if d.Type == entity.DbTypePostgres {
 | 
							} else if d.Type == entity.DbTypePostgres {
 | 
				
			||||||
			_, err := pq.DialOpen(&PqSqlDialer{sshTunnel: sshClient}, getDsn(d))
 | 
								_, err := pq.DialOpen(&PqSqlDialer{sshTunnelMachine: sshTunnelMachine}, getDsn(d, db))
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				panic(biz.NewBizErr(fmt.Sprintf("postgres隧道连接失败: %s", err.Error())))
 | 
									panic(biz.NewBizErr(fmt.Sprintf("postgres隧道连接失败: %s", err.Error())))
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 验证第一个库是否可以连接即可
 | 
						DB, err := sql.Open(d.Type, getDsn(d, db))
 | 
				
			||||||
	d.Database = strings.Split(d.Database, " ")[0]
 | 
						if err != nil {
 | 
				
			||||||
	DB, err := sql.Open(d.Type, getDsn(d))
 | 
							return nil, err
 | 
				
			||||||
	biz.ErrIsNil(err, "Open %s failed, err:%v\n", d.Type, err)
 | 
						}
 | 
				
			||||||
	defer DB.Close()
 | 
						err = DB.Ping()
 | 
				
			||||||
	perr := DB.Ping()
 | 
						if err != nil {
 | 
				
			||||||
	biz.ErrIsNilAppendErr(perr, "数据库连接失败: %s")
 | 
							DB.Close()
 | 
				
			||||||
}
 | 
							return nil, err
 | 
				
			||||||
 | 
					 | 
				
			||||||
// db实例
 | 
					 | 
				
			||||||
type DbInstance struct {
 | 
					 | 
				
			||||||
	Id        string
 | 
					 | 
				
			||||||
	Type      string
 | 
					 | 
				
			||||||
	ProjectId uint64
 | 
					 | 
				
			||||||
	db        *sql.DB
 | 
					 | 
				
			||||||
	sshTunnel *ssh.Client
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// 执行查询语句
 | 
					 | 
				
			||||||
// 依次返回 列名数组,结果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非查询语句")
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	rows, err := d.db.Query(execSql)
 | 
						return DB, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func SelectDataByDb(db *sql.DB, selectSql string) ([]string, []map[string]interface{}, error) {
 | 
				
			||||||
 | 
						rows, err := db.Query(selectSql)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return nil, nil, err
 | 
							return nil, nil, err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -324,7 +318,11 @@ func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interfac
 | 
				
			|||||||
	colNames := make([]string, 0)
 | 
						colNames := make([]string, 0)
 | 
				
			||||||
	// 是否第一次遍历,列名数组只需第一次遍历时加入
 | 
						// 是否第一次遍历,列名数组只需第一次遍历时加入
 | 
				
			||||||
	isFirst := true
 | 
						isFirst := true
 | 
				
			||||||
 | 
						rowNum := 0
 | 
				
			||||||
	for rows.Next() {
 | 
						for rows.Next() {
 | 
				
			||||||
 | 
							rowNum++
 | 
				
			||||||
 | 
							biz.IsTrue(rowNum <= Max_Rows, "结果集 > 2000, 请完善条件或分页信息")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// 不Scan也会导致等待,该链接实际处于未工作的状态,然后也会导致连接数迅速达到最大
 | 
							// 不Scan也会导致等待,该链接实际处于未工作的状态,然后也会导致连接数迅速达到最大
 | 
				
			||||||
		err := rows.Scan(scans...)
 | 
							err := rows.Scan(scans...)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
@@ -338,6 +336,7 @@ func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interfac
 | 
				
			|||||||
			colName := colType.Name()
 | 
								colName := colType.Name()
 | 
				
			||||||
			// 字段类型名
 | 
								// 字段类型名
 | 
				
			||||||
			colScanType := colType.ScanType().Name()
 | 
								colScanType := colType.ScanType().Name()
 | 
				
			||||||
 | 
								// 如果是第一行,则将列名加入到列信息中,由于map是无序的,所有需要返回列名的有序数组
 | 
				
			||||||
			if isFirst {
 | 
								if isFirst {
 | 
				
			||||||
				colNames = append(colNames, colName)
 | 
									colNames = append(colNames, colName)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
@@ -383,6 +382,45 @@ func (d *DbInstance) SelectData(execSql string) ([]string, []map[string]interfac
 | 
				
			|||||||
	return colNames, result, nil
 | 
						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
 | 
					// 执行 update, insert, delete,建表等sql
 | 
				
			||||||
// 返回影响条数和错误
 | 
					// 返回影响条数和错误
 | 
				
			||||||
func (d *DbInstance) Exec(sql string) (int64, error) {
 | 
					func (d *DbInstance) Exec(sql string) (int64, error) {
 | 
				
			||||||
@@ -393,25 +431,33 @@ func (d *DbInstance) Exec(sql string) (int64, error) {
 | 
				
			|||||||
	return res.RowsAffected()
 | 
						return res.RowsAffected()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取数据库元信息实现接口
 | 
				
			||||||
 | 
					func (di *DbInstance) GetMeta() DbMetadata {
 | 
				
			||||||
 | 
						dbType := di.Type
 | 
				
			||||||
 | 
						if dbType == entity.DbTypeMysql {
 | 
				
			||||||
 | 
							return &MysqlMetadata{di: di}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if dbType == entity.DbTypePostgres {
 | 
				
			||||||
 | 
							return &PgsqlMetadata{di: di}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 关闭连接
 | 
					// 关闭连接
 | 
				
			||||||
func (d *DbInstance) Close() {
 | 
					func (d *DbInstance) Close() {
 | 
				
			||||||
	if d.db != nil {
 | 
						if d.db != nil {
 | 
				
			||||||
		if err := d.db.Close(); err != nil {
 | 
							if err := d.db.Close(); err != nil {
 | 
				
			||||||
			global.Log.Errorf("关闭数据库实例[%s]连接失败: %s", d.Id, err.Error())
 | 
								global.Log.Errorf("关闭数据库实例[%s]连接失败: %s", d.Id, err.Error())
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
							d.db = nil
 | 
				
			||||||
	if d.sshTunnel != nil {
 | 
					 | 
				
			||||||
		if err := d.sshTunnel.Close(); err != nil {
 | 
					 | 
				
			||||||
			global.Log.Errorf("关闭数据库实例[%s]的ssh隧道失败: %s", d.Id, err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取dataSourceName
 | 
					// 获取dataSourceName
 | 
				
			||||||
func getDsn(d *entity.Db) string {
 | 
					func getDsn(d *entity.Db, db string) string {
 | 
				
			||||||
	var dsn string
 | 
						var dsn string
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
						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, d.Database)
 | 
							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 != "" {
 | 
							if d.Params != "" {
 | 
				
			||||||
			dsn = fmt.Sprintf("%s&%s", dsn, d.Params)
 | 
								dsn = fmt.Sprintf("%s&%s", dsn, d.Params)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
@@ -419,7 +465,7 @@ func getDsn(d *entity.Db) string {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if d.Type == entity.DbTypePostgres {
 | 
						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, d.Database)
 | 
							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 != "" {
 | 
							if d.Params != "" {
 | 
				
			||||||
			dsn = fmt.Sprintf("%s %s", dsn, strings.Join(strings.Split(d.Params, "&"), " "))
 | 
								dsn = fmt.Sprintf("%s %s", dsn, strings.Join(strings.Split(d.Params, "&"), " "))
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
@@ -433,8 +479,32 @@ func CloseDb(dbId uint64, db string) {
 | 
				
			|||||||
	dbCache.Delete(GetDbCacheKey(dbId, db))
 | 
						dbCache.Delete(GetDbCacheKey(dbId, db))
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
//-----------------------------------元数据-------------------------------------------
 | 
					// -----------------------------------元数据-------------------------------------------
 | 
				
			||||||
 | 
					// 数据库元信息接口(表、列等元信息)
 | 
				
			||||||
 | 
					type DbMetadata interface {
 | 
				
			||||||
 | 
						// 获取表基础元信息, 如表名等
 | 
				
			||||||
 | 
						GetTables() []map[string]interface{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取列元信息, 如列名等
 | 
				
			||||||
 | 
						GetColumns(tableNames ...string) []map[string]interface{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取表主键字段名,默认第一个字段
 | 
				
			||||||
 | 
						GetPrimaryKey(tablename string) string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取表信息,比GetTables获取更详细的表信息
 | 
				
			||||||
 | 
						GetTableInfos() []map[string]interface{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取表索引信息
 | 
				
			||||||
 | 
						GetTableIndex(tableName string) []map[string]interface{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取建表ddl
 | 
				
			||||||
 | 
						GetCreateTableDdl(tableName string) []map[string]interface{}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 默认每次查询列元信息数量
 | 
				
			||||||
 | 
					const DEFAULT_COLUMN_SIZE = 2000
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ---------------------------------- mysql元数据 -----------------------------------
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
	// mysql 表信息元数据
 | 
						// mysql 表信息元数据
 | 
				
			||||||
	MYSQL_TABLE_MA = `SELECT table_name tableName, engine, table_comment tableComment, 
 | 
						MYSQL_TABLE_MA = `SELECT table_name tableName, engine, table_comment tableComment, 
 | 
				
			||||||
@@ -453,9 +523,6 @@ const (
 | 
				
			|||||||
	FROM information_schema.STATISTICS 
 | 
						FROM information_schema.STATISTICS 
 | 
				
			||||||
    WHERE table_schema = (SELECT database()) AND table_name = '%s' LIMIT 500`
 | 
					    WHERE table_schema = (SELECT database()) AND table_name = '%s' LIMIT 500`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 默认每次查询列元信息数量
 | 
					 | 
				
			||||||
	DEFAULT_COLUMN_SIZE = 2000
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// mysql 列信息元数据
 | 
						// mysql 列信息元数据
 | 
				
			||||||
	MYSQL_COLUMN_MA = `SELECT table_name tableName, column_name columnName, column_type columnType,
 | 
						MYSQL_COLUMN_MA = `SELECT table_name tableName, column_name columnName, column_type columnType,
 | 
				
			||||||
	column_comment columnComment, column_key columnKey, extra, is_nullable nullable from information_schema.columns
 | 
						column_comment columnComment, column_key columnKey, extra, is_nullable nullable from information_schema.columns
 | 
				
			||||||
@@ -466,6 +533,74 @@ const (
 | 
				
			|||||||
	WHERE table_name in (%s) AND table_schema = (SELECT database())`
 | 
						WHERE table_name in (%s) AND table_schema = (SELECT database())`
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type MysqlMetadata struct {
 | 
				
			||||||
 | 
						di *DbInstance
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取表基础元信息, 如表名等
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetTables() []map[string]interface{} {
 | 
				
			||||||
 | 
						_, res, _ := mm.di.SelectData(MYSQL_TABLE_MA)
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取列元信息, 如列名等
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetColumns(tableNames ...string) []map[string]interface{} {
 | 
				
			||||||
 | 
						var sql, tableName string
 | 
				
			||||||
 | 
						for i := 0; i < len(tableNames); i++ {
 | 
				
			||||||
 | 
							if i != 0 {
 | 
				
			||||||
 | 
								tableName = tableName + ", "
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							tableName = tableName + "'" + tableNames[i] + "'"
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						pageNum := 1
 | 
				
			||||||
 | 
						// 如果大于一个表,则统计列数并分页获取
 | 
				
			||||||
 | 
						if len(tableNames) > 1 {
 | 
				
			||||||
 | 
							countSql := fmt.Sprintf(MYSQL_COLOUMN_MA_COUNT, tableName)
 | 
				
			||||||
 | 
							_, countRes, _ := mm.di.SelectData(countSql)
 | 
				
			||||||
 | 
							// 查询出所有列信息总数,手动分页获取所有数据
 | 
				
			||||||
 | 
							maCount := int(countRes[0]["maNum"].(int64))
 | 
				
			||||||
 | 
							// 计算需要查询的页数
 | 
				
			||||||
 | 
							pageNum = maCount / DEFAULT_COLUMN_SIZE
 | 
				
			||||||
 | 
							if maCount%DEFAULT_COLUMN_SIZE > 0 {
 | 
				
			||||||
 | 
								pageNum++
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						res := make([]map[string]interface{}, 0)
 | 
				
			||||||
 | 
						for index := 0; index < pageNum; index++ {
 | 
				
			||||||
 | 
							sql = fmt.Sprintf(MYSQL_COLUMN_MA, tableName, index*DEFAULT_COLUMN_SIZE, DEFAULT_COLUMN_SIZE)
 | 
				
			||||||
 | 
							_, result, err := mm.di.SelectData(sql)
 | 
				
			||||||
 | 
							biz.ErrIsNilAppendErr(err, "获取数据库列信息失败: %s")
 | 
				
			||||||
 | 
							res = append(res, result...)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取表主键字段名,默认第一个字段
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetPrimaryKey(tablename string) string {
 | 
				
			||||||
 | 
						return mm.GetColumns(tablename)[0]["columnName"].(string)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取表信息,比GetTableMetedatas获取更详细的表信息
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetTableInfos() []map[string]interface{} {
 | 
				
			||||||
 | 
						_, res, _ := mm.di.SelectData(MYSQL_TABLE_INFO)
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取表索引信息
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetTableIndex(tableName string) []map[string]interface{} {
 | 
				
			||||||
 | 
						_, res, _ := mm.di.SelectData(MYSQL_INDEX_INFO)
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 获取建表ddl
 | 
				
			||||||
 | 
					func (mm *MysqlMetadata) GetCreateTableDdl(tableName string) []map[string]interface{} {
 | 
				
			||||||
 | 
						_, res, _ := mm.di.SelectData(fmt.Sprintf("show create table %s ", tableName))
 | 
				
			||||||
 | 
						return res
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// ---------------------------------- pgsql元数据 -----------------------------------
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
	// postgres 表信息元数据
 | 
						// postgres 表信息元数据
 | 
				
			||||||
	PGSQL_TABLE_MA = `SELECT obj_description(c.oid) AS "tableComment", c.relname AS "tableName" FROM pg_class c 
 | 
						PGSQL_TABLE_MA = `SELECT obj_description(c.oid) AS "tableComment", c.relname AS "tableName" FROM pg_class c 
 | 
				
			||||||
@@ -512,18 +647,18 @@ const (
 | 
				
			|||||||
	`
 | 
						`
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *DbInstance) GetTableMetedatas() []map[string]interface{} {
 | 
					type PgsqlMetadata struct {
 | 
				
			||||||
	var sql string
 | 
						di *DbInstance
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
					}
 | 
				
			||||||
		sql = MYSQL_TABLE_MA
 | 
					
 | 
				
			||||||
	} else if d.Type == "postgres" {
 | 
					// 获取表基础元信息, 如表名等
 | 
				
			||||||
		sql = PGSQL_TABLE_MA
 | 
					func (pm *PgsqlMetadata) GetTables() []map[string]interface{} {
 | 
				
			||||||
	}
 | 
						_, res, _ := pm.di.SelectData(PGSQL_TABLE_MA)
 | 
				
			||||||
	_, res, _ := d.SelectData(sql)
 | 
					 | 
				
			||||||
	return res
 | 
						return res
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *DbInstance) GetColumnMetadatas(tableNames ...string) []map[string]interface{} {
 | 
					// 获取列元信息, 如列名等
 | 
				
			||||||
 | 
					func (pm *PgsqlMetadata) GetColumns(tableNames ...string) []map[string]interface{} {
 | 
				
			||||||
	var sql, tableName string
 | 
						var sql, tableName string
 | 
				
			||||||
	for i := 0; i < len(tableNames); i++ {
 | 
						for i := 0; i < len(tableNames); i++ {
 | 
				
			||||||
		if i != 0 {
 | 
							if i != 0 {
 | 
				
			||||||
@@ -532,68 +667,48 @@ func (d *DbInstance) GetColumnMetadatas(tableNames ...string) []map[string]inter
 | 
				
			|||||||
		tableName = tableName + "'" + tableNames[i] + "'"
 | 
							tableName = tableName + "'" + tableNames[i] + "'"
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	var countSqlTmp string
 | 
						pageNum := 1
 | 
				
			||||||
	var sqlTmp string
 | 
						// 如果大于一个表,则统计列数并分页获取
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
						if len(tableNames) > 1 {
 | 
				
			||||||
		countSqlTmp = MYSQL_COLOUMN_MA_COUNT
 | 
							countSql := fmt.Sprintf(PGSQL_COLUMN_MA_COUNT, tableName)
 | 
				
			||||||
		sqlTmp = MYSQL_COLUMN_MA
 | 
							_, countRes, _ := pm.di.SelectData(countSql)
 | 
				
			||||||
	} else if d.Type == entity.DbTypePostgres {
 | 
							// 查询出所有列信息总数,手动分页获取所有数据
 | 
				
			||||||
		countSqlTmp = PGSQL_COLUMN_MA_COUNT
 | 
							maCount := int(countRes[0]["maNum"].(int64))
 | 
				
			||||||
		sqlTmp = PGSQL_COLUMN_MA
 | 
							// 计算需要查询的页数
 | 
				
			||||||
	}
 | 
							pageNum = maCount / DEFAULT_COLUMN_SIZE
 | 
				
			||||||
 | 
							if maCount%DEFAULT_COLUMN_SIZE > 0 {
 | 
				
			||||||
	countSql := fmt.Sprintf(countSqlTmp, tableName)
 | 
								pageNum++
 | 
				
			||||||
	_, countRes, _ := d.SelectData(countSql)
 | 
							}
 | 
				
			||||||
	// 查询出所有列信息总数,手动分页获取所有数据
 | 
					 | 
				
			||||||
	maCount := int(countRes[0]["maNum"].(int64))
 | 
					 | 
				
			||||||
	// 计算需要查询的页数
 | 
					 | 
				
			||||||
	pageNum := maCount / DEFAULT_COLUMN_SIZE
 | 
					 | 
				
			||||||
	if maCount%DEFAULT_COLUMN_SIZE > 0 {
 | 
					 | 
				
			||||||
		pageNum++
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	res := make([]map[string]interface{}, 0)
 | 
						res := make([]map[string]interface{}, 0)
 | 
				
			||||||
	for index := 0; index < pageNum; index++ {
 | 
						for index := 0; index < pageNum; index++ {
 | 
				
			||||||
		sql = fmt.Sprintf(sqlTmp, tableName, index*DEFAULT_COLUMN_SIZE, DEFAULT_COLUMN_SIZE)
 | 
							sql = fmt.Sprintf(PGSQL_COLUMN_MA, tableName, index*DEFAULT_COLUMN_SIZE, DEFAULT_COLUMN_SIZE)
 | 
				
			||||||
		_, result, err := d.SelectData(sql)
 | 
							_, result, err := pm.di.SelectData(sql)
 | 
				
			||||||
		biz.ErrIsNilAppendErr(err, "获取数据库列信息失败: %s")
 | 
							biz.ErrIsNilAppendErr(err, "获取数据库列信息失败: %s")
 | 
				
			||||||
		res = append(res, result...)
 | 
							res = append(res, result...)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return res
 | 
						return res
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取表的主键,目前默认第一列为主键
 | 
					// 获取表主键字段名,默认第一个字段
 | 
				
			||||||
func (d *DbInstance) GetPrimaryKey(tablename string) string {
 | 
					func (pm *PgsqlMetadata) GetPrimaryKey(tablename string) string {
 | 
				
			||||||
	return d.GetColumnMetadatas(tablename)[0]["columnName"].(string)
 | 
						return pm.GetColumns(tablename)[0]["columnName"].(string)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *DbInstance) GetTableInfos() []map[string]interface{} {
 | 
					// 获取表信息,比GetTables获取更详细的表信息
 | 
				
			||||||
	var sql string
 | 
					func (pm *PgsqlMetadata) GetTableInfos() []map[string]interface{} {
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
						_, res, _ := pm.di.SelectData(PGSQL_TABLE_INFO)
 | 
				
			||||||
		sql = MYSQL_TABLE_INFO
 | 
					 | 
				
			||||||
	} else if d.Type == entity.DbTypePostgres {
 | 
					 | 
				
			||||||
		sql = PGSQL_TABLE_INFO
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_, res, _ := d.SelectData(sql)
 | 
					 | 
				
			||||||
	return res
 | 
						return res
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *DbInstance) GetTableIndex(tableName string) []map[string]interface{} {
 | 
					// 获取表索引信息
 | 
				
			||||||
	var sql string
 | 
					func (pm *PgsqlMetadata) GetTableIndex(tableName string) []map[string]interface{} {
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
						_, res, _ := pm.di.SelectData(PGSQL_INDEX_INFO)
 | 
				
			||||||
		sql = fmt.Sprintf(MYSQL_INDEX_INFO, tableName)
 | 
					 | 
				
			||||||
	} else if d.Type == entity.DbTypePostgres {
 | 
					 | 
				
			||||||
		sql = fmt.Sprintf(PGSQL_INDEX_INFO, tableName)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_, res, _ := d.SelectData(sql)
 | 
					 | 
				
			||||||
	return res
 | 
						return res
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (d *DbInstance) GetCreateTableDdl(tableName string) []map[string]interface{} {
 | 
					// 获取建表ddl
 | 
				
			||||||
	var sql string
 | 
					func (mm *PgsqlMetadata) GetCreateTableDdl(tableName string) []map[string]interface{} {
 | 
				
			||||||
	if d.Type == entity.DbTypeMysql {
 | 
						return nil
 | 
				
			||||||
		sql = fmt.Sprintf("show create table %s ", tableName)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	_, res, _ := d.SelectData(sql)
 | 
					 | 
				
			||||||
	return res
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -3,6 +3,7 @@ package application
 | 
				
			|||||||
import (
 | 
					import (
 | 
				
			||||||
	"encoding/json"
 | 
						"encoding/json"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/repository"
 | 
						"mayfly-go/internal/devops/domain/repository"
 | 
				
			||||||
	"mayfly-go/internal/devops/infrastructure/persistence"
 | 
						"mayfly-go/internal/devops/infrastructure/persistence"
 | 
				
			||||||
@@ -89,7 +90,7 @@ func doUpdate(update *sqlparser.Update, dbInstance *DbInstance, dbSqlExec *entit
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 获取表主键列名,排除使用别名
 | 
						// 获取表主键列名,排除使用别名
 | 
				
			||||||
	primaryKey := dbInstance.GetPrimaryKey(tableName)
 | 
						primaryKey := dbInstance.GetMeta().GetPrimaryKey(tableName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	updateColumnsAndPrimaryKey := strings.Join(updateColumns, ",") + "," + primaryKey
 | 
						updateColumnsAndPrimaryKey := strings.Join(updateColumns, ",") + "," + primaryKey
 | 
				
			||||||
	// 查询要更新字段数据的旧值,以及主键值
 | 
						// 查询要更新字段数据的旧值,以及主键值
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -32,6 +32,9 @@ type Machine interface {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	// 获取机器连接
 | 
						// 获取机器连接
 | 
				
			||||||
	GetCli(id uint64) *machine.Cli
 | 
						GetCli(id uint64) *machine.Cli
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取ssh隧道机器连接
 | 
				
			||||||
 | 
						GetSshTunnelMachine(id uint64) *machine.SshTunnelMachine
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type machineAppImpl struct {
 | 
					type machineAppImpl struct {
 | 
				
			||||||
@@ -53,7 +56,7 @@ func (m *machineAppImpl) Count(condition *entity.Machine) int64 {
 | 
				
			|||||||
func (m *machineAppImpl) Save(me *entity.Machine) {
 | 
					func (m *machineAppImpl) Save(me *entity.Machine) {
 | 
				
			||||||
	// ’修改机器信息且密码不为空‘ or ‘新增’需要测试是否可连接
 | 
						// ’修改机器信息且密码不为空‘ or ‘新增’需要测试是否可连接
 | 
				
			||||||
	if (me.Id != 0 && me.Password != "") || me.Id == 0 {
 | 
						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}
 | 
						oldMachine := &entity.Machine{Ip: me.Ip, Port: me.Port, Username: me.Username}
 | 
				
			||||||
@@ -66,11 +69,13 @@ func (m *machineAppImpl) Save(me *entity.Machine) {
 | 
				
			|||||||
		}
 | 
							}
 | 
				
			||||||
		// 关闭连接
 | 
							// 关闭连接
 | 
				
			||||||
		machine.DeleteCli(me.Id)
 | 
							machine.DeleteCli(me.Id)
 | 
				
			||||||
 | 
							me.PwdEncrypt()
 | 
				
			||||||
		m.machineRepo.UpdateById(me)
 | 
							m.machineRepo.UpdateById(me)
 | 
				
			||||||
	} else {
 | 
						} else {
 | 
				
			||||||
		biz.IsTrue(err != nil, "该机器信息已存在")
 | 
							biz.IsTrue(err != nil, "该机器信息已存在")
 | 
				
			||||||
		// 新增机器,默认启用状态
 | 
							// 新增机器,默认启用状态
 | 
				
			||||||
		me.Status = entity.MachineStatusEnable
 | 
							me.Status = entity.MachineStatusEnable
 | 
				
			||||||
 | 
							me.PwdEncrypt()
 | 
				
			||||||
		m.machineRepo.Create(me)
 | 
							m.machineRepo.Create(me)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -120,9 +125,21 @@ func (m *machineAppImpl) GetById(id uint64, cols ...string) *entity.Machine {
 | 
				
			|||||||
func (m *machineAppImpl) GetCli(id uint64) *machine.Cli {
 | 
					func (m *machineAppImpl) GetCli(id uint64) *machine.Cli {
 | 
				
			||||||
	cli, err := machine.GetCli(id, func(machineId uint64) *entity.Machine {
 | 
						cli, err := machine.GetCli(id, func(machineId uint64) *entity.Machine {
 | 
				
			||||||
		machine := m.GetById(machineId)
 | 
							machine := m.GetById(machineId)
 | 
				
			||||||
 | 
							machine.PwdDecrypt()
 | 
				
			||||||
		biz.IsTrue(machine.Status == entity.MachineStatusEnable, "该机器已被停用")
 | 
							biz.IsTrue(machine.Status == entity.MachineStatusEnable, "该机器已被停用")
 | 
				
			||||||
		return machine
 | 
							return machine
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "获取客户端错误: %s")
 | 
						biz.ErrIsNilAppendErr(err, "获取客户端错误: %s")
 | 
				
			||||||
	return cli
 | 
						return cli
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *machineAppImpl) GetSshTunnelMachine(id uint64) *machine.SshTunnelMachine {
 | 
				
			||||||
 | 
						sshTunnel, err := machine.GetSshTunnelMachine(id, func(machineId uint64) *entity.Machine {
 | 
				
			||||||
 | 
							machine := m.GetById(machineId)
 | 
				
			||||||
 | 
							machine.PwdDecrypt()
 | 
				
			||||||
 | 
							biz.IsTrue(machine.Status == entity.MachineStatusEnable, "该机器已被停用")
 | 
				
			||||||
 | 
							return machine
 | 
				
			||||||
 | 
						})
 | 
				
			||||||
 | 
						biz.ErrIsNilAppendErr(err, "获取ssh隧道连接失败: %s")
 | 
				
			||||||
 | 
						return sshTunnel
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,6 +2,7 @@ package application
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"context"
 | 
						"context"
 | 
				
			||||||
 | 
						"mayfly-go/internal/constant"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/repository"
 | 
						"mayfly-go/internal/devops/domain/repository"
 | 
				
			||||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
						"mayfly-go/internal/devops/infrastructure/machine"
 | 
				
			||||||
@@ -12,11 +13,11 @@ import (
 | 
				
			|||||||
	"mayfly-go/pkg/model"
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
	"mayfly-go/pkg/utils"
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
	"net"
 | 
						"net"
 | 
				
			||||||
 | 
						"regexp"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"go.mongodb.org/mongo-driver/mongo"
 | 
						"go.mongodb.org/mongo-driver/mongo"
 | 
				
			||||||
	"go.mongodb.org/mongo-driver/mongo/options"
 | 
						"go.mongodb.org/mongo-driver/mongo/options"
 | 
				
			||||||
	"golang.org/x/crypto/ssh"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Mongo interface {
 | 
					type Mongo interface {
 | 
				
			||||||
@@ -95,14 +96,27 @@ func (d *mongoAppImpl) GetMongoCli(id uint64) *mongo.Client {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
// -----------------------------------------------------------
 | 
					// -----------------------------------------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
//mongo客户端连接缓存,30分钟内没有访问则会被关闭
 | 
					// mongo客户端连接缓存,指定时间内没有访问则会被关闭
 | 
				
			||||||
var mongoCliCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
					var mongoCliCache = cache.NewTimedCache(constant.MongoConnExpireTime, 5*time.Second).
 | 
				
			||||||
	WithUpdateAccessTime(true).
 | 
						WithUpdateAccessTime(true).
 | 
				
			||||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
						OnEvicted(func(key interface{}, value interface{}) {
 | 
				
			||||||
		global.Log.Info("删除mongo连接缓存: id = ", key)
 | 
							global.Log.Info("删除mongo连接缓存: id = ", key)
 | 
				
			||||||
		value.(*MongoInstance).Close()
 | 
							value.(*MongoInstance).Close()
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					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的连接实例
 | 
					// 获取mongo的连接实例
 | 
				
			||||||
func GetMongoInstance(mongoId uint64, getMongoEntity func(uint64) *entity.Mongo) (*MongoInstance, error) {
 | 
					func GetMongoInstance(mongoId uint64, getMongoEntity func(uint64) *entity.Mongo) (*MongoInstance, error) {
 | 
				
			||||||
	mi, err := mongoCliCache.ComputeIfAbsent(mongoId, func(_ interface{}) (interface{}, error) {
 | 
						mi, err := mongoCliCache.ComputeIfAbsent(mongoId, func(_ interface{}) (interface{}, error) {
 | 
				
			||||||
@@ -124,10 +138,10 @@ func DeleteMongoCache(mongoId uint64) {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type MongoInstance struct {
 | 
					type MongoInstance struct {
 | 
				
			||||||
	Id        uint64
 | 
						Id                 uint64
 | 
				
			||||||
	ProjectId uint64
 | 
						ProjectId          uint64
 | 
				
			||||||
	Cli       *mongo.Client
 | 
						Cli                *mongo.Client
 | 
				
			||||||
	sshTunnel *ssh.Client
 | 
						sshTunnelMachineId uint64
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (mi *MongoInstance) Close() {
 | 
					func (mi *MongoInstance) Close() {
 | 
				
			||||||
@@ -135,11 +149,7 @@ func (mi *MongoInstance) Close() {
 | 
				
			|||||||
		if err := mi.Cli.Disconnect(context.Background()); err != nil {
 | 
							if err := mi.Cli.Disconnect(context.Background()); err != nil {
 | 
				
			||||||
			global.Log.Errorf("关闭mongo实例[%d]连接失败: %s", mi.Id, err)
 | 
								global.Log.Errorf("关闭mongo实例[%d]连接失败: %s", mi.Id, err)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
							mi.Cli = nil
 | 
				
			||||||
	if mi.sshTunnel != nil {
 | 
					 | 
				
			||||||
		if err := mi.sshTunnel.Close(); err != nil {
 | 
					 | 
				
			||||||
			global.Log.Errorf("关闭mongo实例[%d]的ssh隧道失败: %s", mi.Id, err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -154,33 +164,34 @@ func connect(me *entity.Mongo) (*MongoInstance, error) {
 | 
				
			|||||||
		SetMaxPoolSize(1)
 | 
							SetMaxPoolSize(1)
 | 
				
			||||||
	// 启用ssh隧道则连接隧道机器
 | 
						// 启用ssh隧道则连接隧道机器
 | 
				
			||||||
	if me.EnableSshTunnel == 1 {
 | 
						if me.EnableSshTunnel == 1 {
 | 
				
			||||||
		machineEntity := MachineApp.GetById(4)
 | 
							mongoInstance.sshTunnelMachineId = me.SshTunnelMachineId
 | 
				
			||||||
		sshClient, err := machine.GetSshClient(machineEntity)
 | 
							mongoOptions.SetDialer(&MongoSshDialer{machineId: me.SshTunnelMachineId})
 | 
				
			||||||
		biz.ErrIsNilAppendErr(err, "ssh隧道连接失败: %s")
 | 
					 | 
				
			||||||
		mongoInstance.sshTunnel = sshClient
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		mongoOptions.SetDialer(&MongoSshDialer{sshTunnel: sshClient})
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	client, err := mongo.Connect(ctx, mongoOptions)
 | 
						client, err := mongo.Connect(ctx, mongoOptions)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
 | 
							mongoInstance.Close()
 | 
				
			||||||
		return nil, err
 | 
							return nil, err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if err = client.Ping(context.TODO(), nil); err != nil {
 | 
						if err = client.Ping(context.TODO(), nil); err != nil {
 | 
				
			||||||
 | 
							mongoInstance.Close()
 | 
				
			||||||
		return nil, err
 | 
							return nil, err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	global.Log.Infof("连接mongo: %s", me.Uri)
 | 
						global.Log.Infof("连接mongo: %s", func(str string) string {
 | 
				
			||||||
 | 
							reg := regexp.MustCompile(`(^mongodb://.+?:)(.+)(@.+$)`)
 | 
				
			||||||
 | 
							return reg.ReplaceAllString(str, `${1}****${3}`)
 | 
				
			||||||
 | 
						}(me.Uri))
 | 
				
			||||||
	mongoInstance.Cli = client
 | 
						mongoInstance.Cli = client
 | 
				
			||||||
	return mongoInstance, err
 | 
						return mongoInstance, err
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type MongoSshDialer struct {
 | 
					type MongoSshDialer struct {
 | 
				
			||||||
	sshTunnel *ssh.Client
 | 
						machineId uint64
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (sd *MongoSshDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
 | 
					func (sd *MongoSshDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
 | 
				
			||||||
	if sshConn, err := sd.sshTunnel.Dial(network, address); err == nil {
 | 
						if sshConn, err := MachineApp.GetSshTunnelMachine(sd.machineId).GetDialConn(network, address); err == nil {
 | 
				
			||||||
		// 将ssh conn包装,否则内部部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
							// 将ssh conn包装,否则内部部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
				
			||||||
		return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
							return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
				
			||||||
	} else {
 | 
						} else {
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -3,6 +3,7 @@ package application
 | 
				
			|||||||
import (
 | 
					import (
 | 
				
			||||||
	"context"
 | 
						"context"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"mayfly-go/internal/constant"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/repository"
 | 
						"mayfly-go/internal/devops/domain/repository"
 | 
				
			||||||
	"mayfly-go/internal/devops/infrastructure/machine"
 | 
						"mayfly-go/internal/devops/infrastructure/machine"
 | 
				
			||||||
@@ -17,7 +18,6 @@ import (
 | 
				
			|||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/go-redis/redis/v8"
 | 
						"github.com/go-redis/redis/v8"
 | 
				
			||||||
	"golang.org/x/crypto/ssh"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Redis interface {
 | 
					type Redis interface {
 | 
				
			||||||
@@ -80,6 +80,7 @@ func (r *redisAppImpl) Save(re *entity.Redis) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	if re.Id == 0 {
 | 
						if re.Id == 0 {
 | 
				
			||||||
		biz.IsTrue(err != nil, "该库已存在")
 | 
							biz.IsTrue(err != nil, "该库已存在")
 | 
				
			||||||
 | 
							re.PwdEncrypt()
 | 
				
			||||||
		r.redisRepo.Insert(re)
 | 
							r.redisRepo.Insert(re)
 | 
				
			||||||
	} else {
 | 
						} else {
 | 
				
			||||||
		// 如果存在该库,则校验修改的库是否为该库
 | 
							// 如果存在该库,则校验修改的库是否为该库
 | 
				
			||||||
@@ -88,6 +89,7 @@ func (r *redisAppImpl) Save(re *entity.Redis) {
 | 
				
			|||||||
		}
 | 
							}
 | 
				
			||||||
		// 先关闭数据库连接
 | 
							// 先关闭数据库连接
 | 
				
			||||||
		CloseRedis(re.Id)
 | 
							CloseRedis(re.Id)
 | 
				
			||||||
 | 
							re.PwdEncrypt()
 | 
				
			||||||
		r.redisRepo.Update(re)
 | 
							r.redisRepo.Update(re)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -110,6 +112,7 @@ func (r *redisAppImpl) GetRedisInstance(id uint64) *RedisInstance {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
	// 缓存不存在,则回调获取redis信息
 | 
						// 缓存不存在,则回调获取redis信息
 | 
				
			||||||
	re := r.GetById(id)
 | 
						re := r.GetById(id)
 | 
				
			||||||
 | 
						re.PwdDecrypt()
 | 
				
			||||||
	biz.NotNil(re, "redis信息不存在")
 | 
						biz.NotNil(re, "redis信息不存在")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	redisMode := re.Mode
 | 
						redisMode := re.Mode
 | 
				
			||||||
@@ -130,6 +133,14 @@ func (r *redisAppImpl) GetRedisInstance(id uint64) *RedisInstance {
 | 
				
			|||||||
			ri.Close()
 | 
								ri.Close()
 | 
				
			||||||
			panic(biz.NewBizErr(fmt.Sprintf("redis集群连接失败: %s", e.Error())))
 | 
								panic(biz.NewBizErr(fmt.Sprintf("redis集群连接失败: %s", e.Error())))
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
						} else if redisMode == entity.RedisModeSentinel {
 | 
				
			||||||
 | 
							ri = getRedisSentinelCient(re)
 | 
				
			||||||
 | 
							// 测试连接
 | 
				
			||||||
 | 
							_, e := ri.Cli.Ping(context.Background()).Result()
 | 
				
			||||||
 | 
							if e != nil {
 | 
				
			||||||
 | 
								ri.Close()
 | 
				
			||||||
 | 
								panic(biz.NewBizErr(fmt.Sprintf("redis sentinel连接失败: %s", e.Error())))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	global.Log.Infof("连接redis: %s", re.Host)
 | 
						global.Log.Infof("连接redis: %s", re.Host)
 | 
				
			||||||
@@ -151,9 +162,8 @@ func getRedisCient(re *entity.Redis) *RedisInstance {
 | 
				
			|||||||
		WriteTimeout: -1,
 | 
							WriteTimeout: -1,
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if re.EnableSshTunnel == 1 {
 | 
						if re.EnableSshTunnel == 1 {
 | 
				
			||||||
		sshClient, dialerFunc := getRedisDialer(re.SshTunnelMachineId)
 | 
							ri.sshTunnelMachineId = re.SshTunnelMachineId
 | 
				
			||||||
		ri.sshTunnel = sshClient
 | 
							redisOptions.Dialer = getRedisDialer(re.SshTunnelMachineId)
 | 
				
			||||||
		redisOptions.Dialer = dialerFunc
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	ri.Cli = redis.NewClient(redisOptions)
 | 
						ri.Cli = redis.NewClient(redisOptions)
 | 
				
			||||||
	return ri
 | 
						return ri
 | 
				
			||||||
@@ -168,21 +178,38 @@ func getRedisClusterClient(re *entity.Redis) *RedisInstance {
 | 
				
			|||||||
		DialTimeout: 8 * time.Second,
 | 
							DialTimeout: 8 * time.Second,
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if re.EnableSshTunnel == 1 {
 | 
						if re.EnableSshTunnel == 1 {
 | 
				
			||||||
		sshClient, dialerFunc := getRedisDialer(re.SshTunnelMachineId)
 | 
							ri.sshTunnelMachineId = re.SshTunnelMachineId
 | 
				
			||||||
		ri.sshTunnel = sshClient
 | 
							redisClusterOptions.Dialer = getRedisDialer(re.SshTunnelMachineId)
 | 
				
			||||||
		redisClusterOptions.Dialer = dialerFunc
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	ri.ClusterCli = redis.NewClusterClient(redisClusterOptions)
 | 
						ri.ClusterCli = redis.NewClusterClient(redisClusterOptions)
 | 
				
			||||||
	return ri
 | 
						return ri
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func getRedisDialer(machineId uint64) (*ssh.Client, func(ctx context.Context, network, addr string) (net.Conn, error)) {
 | 
					func getRedisSentinelCient(re *entity.Redis) *RedisInstance {
 | 
				
			||||||
	me := MachineApp.GetById(machineId)
 | 
						ri := &RedisInstance{Id: re.Id, ProjectId: re.ProjectId, Mode: re.Mode}
 | 
				
			||||||
	sshClient, err := machine.GetSshClient(me)
 | 
						// sentinel模式host为 masterName=host:port,host:port
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "ssh隧道连接失败: %s")
 | 
						masterNameAndHosts := strings.Split(re.Host, "=")
 | 
				
			||||||
 | 
						sentinelOptions := &redis.FailoverOptions{
 | 
				
			||||||
 | 
							MasterName:    masterNameAndHosts[0],
 | 
				
			||||||
 | 
							SentinelAddrs: strings.Split(masterNameAndHosts[1], ","),
 | 
				
			||||||
 | 
							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
 | 
				
			||||||
 | 
							sentinelOptions.Dialer = getRedisDialer(re.SshTunnelMachineId)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						ri.Cli = redis.NewFailoverClient(sentinelOptions)
 | 
				
			||||||
 | 
						return ri
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return sshClient, func(_ context.Context, network, addr string) (net.Conn, error) {
 | 
					func getRedisDialer(machineId uint64) func(ctx context.Context, network, addr string) (net.Conn, error) {
 | 
				
			||||||
		if sshConn, err := sshClient.Dial(network, addr); err == nil {
 | 
						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
 | 
								// 将ssh conn包装,否则redis内部设置超时会报错,ssh conn不支持设置超时会返回错误: ssh: tcpChan: deadline not supported
 | 
				
			||||||
			return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
								return &utils.WrapSshConn{Conn: sshConn}, nil
 | 
				
			||||||
		} else {
 | 
							} else {
 | 
				
			||||||
@@ -193,8 +220,8 @@ func getRedisDialer(machineId uint64) (*ssh.Client, func(ctx context.Context, ne
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
//------------------------------------------------------------------------------
 | 
					//------------------------------------------------------------------------------
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// redis客户端连接缓存,30分钟内没有访问则会被关闭
 | 
					// redis客户端连接缓存,指定时间内没有访问则会被关闭
 | 
				
			||||||
var redisCache = cache.NewTimedCache(30*time.Minute, 5*time.Second).
 | 
					var redisCache = cache.NewTimedCache(constant.RedisConnExpireTime, 5*time.Second).
 | 
				
			||||||
	WithUpdateAccessTime(true).
 | 
						WithUpdateAccessTime(true).
 | 
				
			||||||
	OnEvicted(func(key interface{}, value interface{}) {
 | 
						OnEvicted(func(key interface{}, value interface{}) {
 | 
				
			||||||
		global.Log.Info(fmt.Sprintf("删除redis连接缓存 id = %d", key))
 | 
							global.Log.Info(fmt.Sprintf("删除redis连接缓存 id = %d", key))
 | 
				
			||||||
@@ -206,6 +233,19 @@ func CloseRedis(id uint64) {
 | 
				
			|||||||
	redisCache.Delete(id)
 | 
						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) {
 | 
					func TestRedisConnection(re *entity.Redis) {
 | 
				
			||||||
	var cmd redis.Cmdable
 | 
						var cmd redis.Cmdable
 | 
				
			||||||
	if re.Mode == "" || re.Mode == entity.RedisModeStandalone {
 | 
						if re.Mode == "" || re.Mode == entity.RedisModeStandalone {
 | 
				
			||||||
@@ -216,6 +256,10 @@ func TestRedisConnection(re *entity.Redis) {
 | 
				
			|||||||
		ccli := getRedisClusterClient(re)
 | 
							ccli := getRedisClusterClient(re)
 | 
				
			||||||
		defer ccli.Close()
 | 
							defer ccli.Close()
 | 
				
			||||||
		cmd = ccli.ClusterCli
 | 
							cmd = ccli.ClusterCli
 | 
				
			||||||
 | 
						} else if re.Mode == entity.RedisModeSentinel {
 | 
				
			||||||
 | 
							rcli := getRedisSentinelCient(re)
 | 
				
			||||||
 | 
							defer rcli.Close()
 | 
				
			||||||
 | 
							cmd = rcli.Cli
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 测试连接
 | 
						// 测试连接
 | 
				
			||||||
@@ -225,21 +269,21 @@ func TestRedisConnection(re *entity.Redis) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
// redis实例
 | 
					// redis实例
 | 
				
			||||||
type RedisInstance struct {
 | 
					type RedisInstance struct {
 | 
				
			||||||
	Id         uint64
 | 
						Id                 uint64
 | 
				
			||||||
	ProjectId  uint64
 | 
						ProjectId          uint64
 | 
				
			||||||
	Mode       string
 | 
						Mode               string
 | 
				
			||||||
	Cli        *redis.Client
 | 
						Cli                *redis.Client
 | 
				
			||||||
	ClusterCli *redis.ClusterClient
 | 
						ClusterCli         *redis.ClusterClient
 | 
				
			||||||
	sshTunnel  *ssh.Client
 | 
						sshTunnelMachineId uint64
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取命令执行接口的具体实现
 | 
					// 获取命令执行接口的具体实现
 | 
				
			||||||
func (r *RedisInstance) GetCmdable() redis.Cmdable {
 | 
					func (r *RedisInstance) GetCmdable() redis.Cmdable {
 | 
				
			||||||
	redisMode := r.Mode
 | 
						redisMode := r.Mode
 | 
				
			||||||
	if redisMode == "" || redisMode == entity.RedisModeStandalone {
 | 
						if redisMode == "" || redisMode == entity.RedisModeStandalone || r.Mode == entity.RedisModeSentinel {
 | 
				
			||||||
		return r.Cli
 | 
							return r.Cli
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if r.Mode == entity.RedisModeCluster {
 | 
						if redisMode == entity.RedisModeCluster {
 | 
				
			||||||
		return r.ClusterCli
 | 
							return r.ClusterCli
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	return nil
 | 
						return nil
 | 
				
			||||||
@@ -252,19 +296,16 @@ func (r *RedisInstance) Scan(cursor uint64, match string, count int64) ([]string
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (r *RedisInstance) Close() {
 | 
					func (r *RedisInstance) Close() {
 | 
				
			||||||
	if r.Mode == entity.RedisModeStandalone {
 | 
						if r.Mode == entity.RedisModeStandalone || r.Mode == entity.RedisModeSentinel {
 | 
				
			||||||
		if err := r.Cli.Close(); err != nil {
 | 
							if err := r.Cli.Close(); err != nil {
 | 
				
			||||||
			global.Log.Errorf("关闭redis单机实例[%d]连接失败: %s", r.Id, err.Error())
 | 
								global.Log.Errorf("关闭redis单机实例[%d]连接失败: %s", r.Id, err.Error())
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							r.Cli = nil
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if r.Mode == entity.RedisModeCluster {
 | 
						if r.Mode == entity.RedisModeCluster {
 | 
				
			||||||
		if err := r.ClusterCli.Close(); err != nil {
 | 
							if err := r.ClusterCli.Close(); err != nil {
 | 
				
			||||||
			global.Log.Errorf("关闭redis集群实例[%d]连接失败: %s", r.Id, err.Error())
 | 
								global.Log.Errorf("关闭redis集群实例[%d]连接失败: %s", r.Id, err.Error())
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
							r.ClusterCli = nil
 | 
				
			||||||
	if r.sshTunnel != nil {
 | 
					 | 
				
			||||||
		if err := r.sshTunnel.Close(); err != nil {
 | 
					 | 
				
			||||||
			global.Log.Errorf("关闭redis实例[%d]的ssh隧道失败: %s", r.Id, err.Error())
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -2,6 +2,7 @@ package entity
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"mayfly-go/internal/common/utils"
 | 
				
			||||||
	"mayfly-go/pkg/model"
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -27,9 +28,9 @@ type Db struct {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取数据库连接网络, 若没有使用ssh隧道,则直接返回。否则返回拼接的网络需要注册至指定dial
 | 
					// 获取数据库连接网络, 若没有使用ssh隧道,则直接返回。否则返回拼接的网络需要注册至指定dial
 | 
				
			||||||
func (d Db) GetNetwork() string {
 | 
					func (d *Db) GetNetwork() string {
 | 
				
			||||||
	network := d.Network
 | 
						network := d.Network
 | 
				
			||||||
	if d.EnableSshTunnel == -1 {
 | 
						if d.EnableSshTunnel == 0 || d.EnableSshTunnel == -1 {
 | 
				
			||||||
		if network == "" {
 | 
							if network == "" {
 | 
				
			||||||
			return "tcp"
 | 
								return "tcp"
 | 
				
			||||||
		} else {
 | 
							} else {
 | 
				
			||||||
@@ -39,6 +40,16 @@ func (d Db) GetNetwork() string {
 | 
				
			|||||||
	return fmt.Sprintf("%s+ssh:%d", d.Type, d.SshTunnelMachineId)
 | 
						return fmt.Sprintf("%s+ssh:%d", d.Type, d.SshTunnelMachineId)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d *Db) PwdEncrypt() {
 | 
				
			||||||
 | 
						// 密码替换为加密后的密码
 | 
				
			||||||
 | 
						d.Password = utils.PwdAesEncrypt(d.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (d *Db) PwdDecrypt() {
 | 
				
			||||||
 | 
						// 密码替换为解密后的密码
 | 
				
			||||||
 | 
						d.Password = utils.PwdAesDecrypt(d.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
	DbTypeMysql    = "mysql"
 | 
						DbTypeMysql    = "mysql"
 | 
				
			||||||
	DbTypePostgres = "postgres"
 | 
						DbTypePostgres = "postgres"
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,21 +1,24 @@
 | 
				
			|||||||
package entity
 | 
					package entity
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/common/utils"
 | 
				
			||||||
	"mayfly-go/pkg/model"
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Machine struct {
 | 
					type Machine struct {
 | 
				
			||||||
	model.Model
 | 
						model.Model
 | 
				
			||||||
	ProjectId   uint64 `json:"projectId"`
 | 
						ProjectId          uint64 `json:"projectId"`
 | 
				
			||||||
	ProjectName string `json:"projectName"`
 | 
						ProjectName        string `json:"projectName"`
 | 
				
			||||||
	Name        string `json:"name"`
 | 
						Name               string `json:"name"`
 | 
				
			||||||
	Ip          string `json:"ip"`         // IP地址
 | 
						Ip                 string `json:"ip"`         // IP地址
 | 
				
			||||||
	Username    string `json:"username"`   // 用户名
 | 
						Username           string `json:"username"`   // 用户名
 | 
				
			||||||
	AuthMethod  int8   `json:"authMethod"` // 授权认证方式
 | 
						AuthMethod         int8   `json:"authMethod"` // 授权认证方式
 | 
				
			||||||
	Password    string `json:"-"`
 | 
						Password           string `json:"-"`
 | 
				
			||||||
	Port        int    `json:"port"`   // 端口号
 | 
						Port               int    `json:"port"`               // 端口号
 | 
				
			||||||
	Status      int8   `json:"status"` // 状态 1:启用;2:停用
 | 
						Status             int8   `json:"status"`             // 状态 1:启用;2:停用
 | 
				
			||||||
	Remark      string `json:"remark"` // 备注
 | 
						Remark             string `json:"remark"`             // 备注
 | 
				
			||||||
 | 
						EnableSshTunnel    int8   `json:"enableSshTunnel"`    // 是否启用ssh隧道
 | 
				
			||||||
 | 
						SshTunnelMachineId uint64 `json:"sshTunnelMachineId"` // ssh隧道机器id
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
@@ -24,3 +27,13 @@ const (
 | 
				
			|||||||
	MachineAuthMethodPassword  int8 = 1  // 密码登录
 | 
						MachineAuthMethodPassword  int8 = 1  // 密码登录
 | 
				
			||||||
	MachineAuthMethodPublicKey int8 = 2  // 公钥免密登录
 | 
						MachineAuthMethodPublicKey int8 = 2  // 公钥免密登录
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *Machine) PwdEncrypt() {
 | 
				
			||||||
 | 
						// 密码替换为加密后的密码
 | 
				
			||||||
 | 
						m.Password = utils.PwdAesEncrypt(m.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *Machine) PwdDecrypt() {
 | 
				
			||||||
 | 
						// 密码替换为解密后的密码
 | 
				
			||||||
 | 
						m.Password = utils.PwdAesDecrypt(m.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,6 +1,7 @@
 | 
				
			|||||||
package entity
 | 
					package entity
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/common/utils"
 | 
				
			||||||
	"mayfly-go/pkg/model"
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -23,4 +24,15 @@ type Redis struct {
 | 
				
			|||||||
const (
 | 
					const (
 | 
				
			||||||
	RedisModeStandalone = "standalone"
 | 
						RedisModeStandalone = "standalone"
 | 
				
			||||||
	RedisModeCluster    = "cluster"
 | 
						RedisModeCluster    = "cluster"
 | 
				
			||||||
 | 
						RedisModeSentinel   = "sentinel"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *Redis) PwdEncrypt() {
 | 
				
			||||||
 | 
						// 密码替换为加密后的密码
 | 
				
			||||||
 | 
						r.Password = utils.PwdAesEncrypt(r.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *Redis) PwdDecrypt() {
 | 
				
			||||||
 | 
						// 密码替换为解密后的密码
 | 
				
			||||||
 | 
						r.Password = utils.PwdAesDecrypt(r.Password)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -3,6 +3,7 @@ package machine
 | 
				
			|||||||
import (
 | 
					import (
 | 
				
			||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"mayfly-go/internal/constant"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/pkg/biz"
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
	"mayfly-go/pkg/cache"
 | 
						"mayfly-go/pkg/cache"
 | 
				
			||||||
@@ -18,10 +19,12 @@ import (
 | 
				
			|||||||
// 客户端信息
 | 
					// 客户端信息
 | 
				
			||||||
type Cli struct {
 | 
					type Cli struct {
 | 
				
			||||||
	machine *entity.Machine
 | 
						machine *entity.Machine
 | 
				
			||||||
	// ssh客户端
 | 
					 | 
				
			||||||
	client *ssh.Client
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	sftpClient *sftp.Client
 | 
						client     *ssh.Client  // ssh客户端
 | 
				
			||||||
 | 
						sftpClient *sftp.Client // sftp客户端
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						enableSshTunnel    int8
 | 
				
			||||||
 | 
						sshTunnelMachineId uint64
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
//连接
 | 
					//连接
 | 
				
			||||||
@@ -39,7 +42,7 @@ func (c *Cli) connect() error {
 | 
				
			|||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 关闭client和并从缓存中移除
 | 
					// 关闭client并从缓存中移除,如果使用隧道则也关闭
 | 
				
			||||||
func (c *Cli) Close() {
 | 
					func (c *Cli) Close() {
 | 
				
			||||||
	m := c.machine
 | 
						m := c.machine
 | 
				
			||||||
	global.Log.Info(fmt.Sprintf("关闭机器客户端连接-> id: %d, name: %s, ip: %s", m.Id, m.Name, m.Ip))
 | 
						global.Log.Info(fmt.Sprintf("关闭机器客户端连接-> id: %d, name: %s, ip: %s", m.Id, m.Name, m.Ip))
 | 
				
			||||||
@@ -51,6 +54,9 @@ func (c *Cli) Close() {
 | 
				
			|||||||
		c.sftpClient.Close()
 | 
							c.sftpClient.Close()
 | 
				
			||||||
		c.sftpClient = nil
 | 
							c.sftpClient = nil
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						if c.enableSshTunnel == 1 {
 | 
				
			||||||
 | 
							CloseSshTunnelMachine(c.sshTunnelMachineId, c.machine.Id)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取sftp client
 | 
					// 获取sftp client
 | 
				
			||||||
@@ -105,13 +111,26 @@ func (c *Cli) GetMachine() *entity.Machine {
 | 
				
			|||||||
	return c.machine
 | 
						return c.machine
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 机器客户端连接缓存,45分钟内没有访问则会被关闭
 | 
					// 机器客户端连接缓存,指定时间内没有访问则会被关闭
 | 
				
			||||||
var cliCache = cache.NewTimedCache(45*time.Minute, 5*time.Second).
 | 
					var cliCache = cache.NewTimedCache(constant.MachineConnExpireTime, 5*time.Second).
 | 
				
			||||||
	WithUpdateAccessTime(true).
 | 
						WithUpdateAccessTime(true).
 | 
				
			||||||
	OnEvicted(func(_, value interface{}) {
 | 
						OnEvicted(func(_, value interface{}) {
 | 
				
			||||||
		value.(*Cli).Close()
 | 
							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的客户端连接
 | 
					// 是否存在指定id的客户端连接
 | 
				
			||||||
func HasCli(machineId uint64) bool {
 | 
					func HasCli(machineId uint64) bool {
 | 
				
			||||||
	if _, ok := cliCache.Get(machineId); ok {
 | 
						if _, ok := cliCache.Get(machineId); ok {
 | 
				
			||||||
@@ -128,10 +147,18 @@ func DeleteCli(id uint64) {
 | 
				
			|||||||
// 从缓存中获取客户端信息,不存在则回调获取机器信息函数,并新建
 | 
					// 从缓存中获取客户端信息,不存在则回调获取机器信息函数,并新建
 | 
				
			||||||
func GetCli(machineId uint64, getMachine func(uint64) *entity.Machine) (*Cli, error) {
 | 
					func GetCli(machineId uint64, getMachine func(uint64) *entity.Machine) (*Cli, error) {
 | 
				
			||||||
	cli, err := cliCache.ComputeIfAbsent(machineId, func(_ interface{}) (interface{}, error) {
 | 
						cli, err := cliCache.ComputeIfAbsent(machineId, func(_ interface{}) (interface{}, error) {
 | 
				
			||||||
		c, err := newClient(getMachine(machineId))
 | 
							me := getMachine(machineId)
 | 
				
			||||||
 | 
							err := IfUseSshTunnelChangeIpPort(me, getMachine)
 | 
				
			||||||
		if err != nil {
 | 
							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
 | 
								return nil, err
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							c.enableSshTunnel = me.EnableSshTunnel
 | 
				
			||||||
 | 
							c.sshTunnelMachineId = me.SshTunnelMachineId
 | 
				
			||||||
		return c, nil
 | 
							return c, nil
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -141,9 +168,20 @@ func GetCli(machineId uint64, getMachine func(uint64) *entity.Machine) (*Cli, er
 | 
				
			|||||||
	return nil, err
 | 
						return nil, err
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 测试连接
 | 
					// 测试连接,使用传值的方式,而非引用。因为如果使用了ssh隧道,则ip和端口会变为本地映射地址与端口
 | 
				
			||||||
func TestConn(m *entity.Machine) error {
 | 
					func TestConn(me entity.Machine, getSshTunnelMachine func(uint64) *entity.Machine) error {
 | 
				
			||||||
	sshClient, err := GetSshClient(m)
 | 
						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 {
 | 
						if err != nil {
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
@@ -151,6 +189,27 @@ func TestConn(m *entity.Machine) error {
 | 
				
			|||||||
	return nil
 | 
						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) {
 | 
					func GetSshClient(m *entity.Machine) (*ssh.Client, error) {
 | 
				
			||||||
	config := ssh.ClientConfig{
 | 
						config := ssh.ClientConfig{
 | 
				
			||||||
		User: m.Username,
 | 
							User: m.Username,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,19 +0,0 @@
 | 
				
			|||||||
package machine
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const StatsShell = `
 | 
					 | 
				
			||||||
cat /proc/uptime
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
/bin/hostname -f
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
cat /proc/loadavg
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
cat /proc/meminfo
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
df -B1
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
/sbin/ip -o addr
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
/bin/cat /proc/net/dev
 | 
					 | 
				
			||||||
echo '-----'
 | 
					 | 
				
			||||||
top -b -n 1 | grep Cpu
 | 
					 | 
				
			||||||
`
 | 
					 | 
				
			||||||
							
								
								
									
										242
									
								
								server/internal/devops/infrastructure/machine/sshtunnel.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										242
									
								
								server/internal/devops/infrastructure/machine/sshtunnel.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,242 @@
 | 
				
			|||||||
 | 
					package machine
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/global"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/scheduler"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
 | 
						"net"
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"sync"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"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隧道机器是否还有被使用")
 | 
				
			||||||
 | 
						// 每十分钟检查一次隧道机器是否还有被使用
 | 
				
			||||||
 | 
						scheduler.AddFun("@every 10m", func() {
 | 
				
			||||||
 | 
							if !mutex.TryLock() {
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							defer mutex.Unlock()
 | 
				
			||||||
 | 
							// 遍历隧道机器,都未被使用将会被关闭
 | 
				
			||||||
 | 
							for mid, sshTunnelMachine := range sshTunnelMachines {
 | 
				
			||||||
 | 
								global.Log.Debugf("开始定时检查ssh隧道机器[%d]是否还有被使用...", mid)
 | 
				
			||||||
 | 
								hasUse := false
 | 
				
			||||||
 | 
								for _, checkUseFunc := range checkSshTunnelMachineHasUseFuncs {
 | 
				
			||||||
 | 
									// 如果一个在使用则返回不关闭,不继续后续检查
 | 
				
			||||||
 | 
									if checkUseFunc(mid) {
 | 
				
			||||||
 | 
										hasUse = true
 | 
				
			||||||
 | 
										break
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								if !hasUse {
 | 
				
			||||||
 | 
									// 都未被使用,则关闭
 | 
				
			||||||
 | 
									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)
 | 
				
			||||||
 | 
							err := stm.SshClient.Close()
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								global.Log.Errorf("关闭ssh隧道机器[%d]发生错误: %s", stm.machineId, err.Error())
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						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)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -53,6 +53,24 @@ type Stats struct {
 | 
				
			|||||||
	CPU          CPUInfo // or []CPUInfo to get all the cpu-core's stats?
 | 
						CPU          CPUInfo // or []CPUInfo to get all the cpu-core's stats?
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const StatsShell = `
 | 
				
			||||||
 | 
					cat /proc/uptime
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					/bin/hostname -f
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					cat /proc/loadavg
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					cat /proc/meminfo
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					df -B1
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					/sbin/ip -o addr
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					/bin/cat /proc/net/dev
 | 
				
			||||||
 | 
					echo '-----'
 | 
				
			||||||
 | 
					top -b -n 1 | grep Cpu
 | 
				
			||||||
 | 
					`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (c *Cli) GetAllStats() *Stats {
 | 
					func (c *Cli) GetAllStats() *Stats {
 | 
				
			||||||
	res, _ := c.Run(StatsShell)
 | 
						res, _ := c.Run(StatsShell)
 | 
				
			||||||
	infos := strings.Split(*res, "-----")
 | 
						infos := strings.Split(*res, "-----")
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										74
									
								
								server/internal/devops/infrastructure/machine/terminal.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										74
									
								
								server/internal/devops/infrastructure/machine/terminal.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,74 @@
 | 
				
			|||||||
 | 
					package machine
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"bufio"
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"golang.org/x/crypto/ssh"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Terminal struct {
 | 
				
			||||||
 | 
						SshSession   *ssh.Session
 | 
				
			||||||
 | 
						StdinPipe    io.WriteCloser
 | 
				
			||||||
 | 
						StdoutReader *bufio.Reader
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 新建机器ssh终端
 | 
				
			||||||
 | 
					func NewTerminal(cli *Cli) (*Terminal, error) {
 | 
				
			||||||
 | 
						sshSession, err := cli.GetSession()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						stdoutPipe, err := sshSession.StdoutPipe()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						stdoutReader := bufio.NewReader(stdoutPipe)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						stdinPipe, err := sshSession.StdinPipe()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						terminal := Terminal{
 | 
				
			||||||
 | 
							SshSession:   sshSession,
 | 
				
			||||||
 | 
							StdinPipe:    stdinPipe,
 | 
				
			||||||
 | 
							StdoutReader: stdoutReader,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return &terminal, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) Write(p []byte) (int, error) {
 | 
				
			||||||
 | 
						return t.StdinPipe.Write(p)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) ReadRune() (r rune, size int, err error) {
 | 
				
			||||||
 | 
						return t.StdoutReader.ReadRune()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) Close() error {
 | 
				
			||||||
 | 
						if t.SshSession != nil {
 | 
				
			||||||
 | 
							return t.SshSession.Close()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) WindowChange(h int, w int) error {
 | 
				
			||||||
 | 
						return t.SshSession.WindowChange(h, w)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) RequestPty(term string, h, w int) error {
 | 
				
			||||||
 | 
						modes := ssh.TerminalModes{
 | 
				
			||||||
 | 
							ssh.ECHO:          1,
 | 
				
			||||||
 | 
							ssh.TTY_OP_ISPEED: 14400,
 | 
				
			||||||
 | 
							ssh.TTY_OP_OSPEED: 14400,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return t.SshSession.RequestPty(term, h, w, modes)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (t *Terminal) Shell() error {
 | 
				
			||||||
 | 
						return t.SshSession.Shell()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -0,0 +1,175 @@
 | 
				
			|||||||
 | 
					package machine
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"context"
 | 
				
			||||||
 | 
						"encoding/json"
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/global"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
						"unicode/utf8"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/gorilla/websocket"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						Resize = 1
 | 
				
			||||||
 | 
						Data   = 2
 | 
				
			||||||
 | 
						Ping   = 3
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type TerminalSession struct {
 | 
				
			||||||
 | 
						ID       string
 | 
				
			||||||
 | 
						wsConn   *websocket.Conn
 | 
				
			||||||
 | 
						terminal *Terminal
 | 
				
			||||||
 | 
						ctx      context.Context
 | 
				
			||||||
 | 
						cancel   context.CancelFunc
 | 
				
			||||||
 | 
						dataChan chan rune
 | 
				
			||||||
 | 
						tick     *time.Ticker
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func NewTerminalSession(sessionId string, ws *websocket.Conn, cli *Cli, rows, cols int) (*TerminalSession, error) {
 | 
				
			||||||
 | 
						terminal, err := NewTerminal(cli)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						err = terminal.RequestPty("xterm-256color", rows, cols)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						err = terminal.Shell()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return nil, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						ctx, cancel := context.WithCancel(context.Background())
 | 
				
			||||||
 | 
						tick := time.NewTicker(time.Millisecond * time.Duration(60))
 | 
				
			||||||
 | 
						ts := &TerminalSession{
 | 
				
			||||||
 | 
							ID:       sessionId,
 | 
				
			||||||
 | 
							wsConn:   ws,
 | 
				
			||||||
 | 
							terminal: terminal,
 | 
				
			||||||
 | 
							ctx:      ctx,
 | 
				
			||||||
 | 
							cancel:   cancel,
 | 
				
			||||||
 | 
							dataChan: make(chan rune),
 | 
				
			||||||
 | 
							tick:     tick,
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return ts, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r TerminalSession) Start() {
 | 
				
			||||||
 | 
						go r.readFormTerminal()
 | 
				
			||||||
 | 
						go r.writeToWebsocket()
 | 
				
			||||||
 | 
						r.receiveWsMsg()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r TerminalSession) Stop() {
 | 
				
			||||||
 | 
						global.Log.Debug("close machine ssh terminal session")
 | 
				
			||||||
 | 
						r.tick.Stop()
 | 
				
			||||||
 | 
						r.cancel()
 | 
				
			||||||
 | 
						if r.wsConn != nil {
 | 
				
			||||||
 | 
							r.wsConn.Close()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if r.terminal != nil {
 | 
				
			||||||
 | 
							if err := r.terminal.Close(); err != nil {
 | 
				
			||||||
 | 
								global.Log.Errorf("关闭机器ssh终端失败: %s", err.Error())
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (ts TerminalSession) readFormTerminal() {
 | 
				
			||||||
 | 
						for {
 | 
				
			||||||
 | 
							select {
 | 
				
			||||||
 | 
							case <-ts.ctx.Done():
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								rn, size, err := ts.terminal.ReadRune()
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									if err != io.EOF {
 | 
				
			||||||
 | 
										global.Log.Error("机器ssh终端读取消息失败: ", err)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								if size > 0 {
 | 
				
			||||||
 | 
									ts.dataChan <- rn
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (ts TerminalSession) writeToWebsocket() {
 | 
				
			||||||
 | 
						var buf []byte
 | 
				
			||||||
 | 
						for {
 | 
				
			||||||
 | 
							select {
 | 
				
			||||||
 | 
							case <-ts.ctx.Done():
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							case <-ts.tick.C:
 | 
				
			||||||
 | 
								if len(buf) > 0 {
 | 
				
			||||||
 | 
									s := string(buf)
 | 
				
			||||||
 | 
									if err := WriteMessage(ts.wsConn, s); err != nil {
 | 
				
			||||||
 | 
										global.Log.Error("机器ssh终端发送消息至websocket失败: ", err)
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									buf = []byte{}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							case data := <-ts.dataChan:
 | 
				
			||||||
 | 
								if data != utf8.RuneError {
 | 
				
			||||||
 | 
									p := make([]byte, utf8.RuneLen(data))
 | 
				
			||||||
 | 
									utf8.EncodeRune(p, data)
 | 
				
			||||||
 | 
									buf = append(buf, p...)
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									buf = append(buf, []byte("@")...)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type WsMsg struct {
 | 
				
			||||||
 | 
						Type int    `json:"type"`
 | 
				
			||||||
 | 
						Msg  string `json:"msg"`
 | 
				
			||||||
 | 
						Cols int    `json:"cols"`
 | 
				
			||||||
 | 
						Rows int    `json:"rows"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (ts *TerminalSession) receiveWsMsg() {
 | 
				
			||||||
 | 
						wsConn := ts.wsConn
 | 
				
			||||||
 | 
						for {
 | 
				
			||||||
 | 
							select {
 | 
				
			||||||
 | 
							case <-ts.ctx.Done():
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							default:
 | 
				
			||||||
 | 
								// read websocket msg
 | 
				
			||||||
 | 
								_, wsData, err := wsConn.ReadMessage()
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									global.Log.Debug("机器ssh终端读取websocket消息失败: ", err)
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								// 解析消息
 | 
				
			||||||
 | 
								msgObj := WsMsg{}
 | 
				
			||||||
 | 
								if err := json.Unmarshal(wsData, &msgObj); err != nil {
 | 
				
			||||||
 | 
									global.Log.Error("机器ssh终端消息解析失败: ", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								switch msgObj.Type {
 | 
				
			||||||
 | 
								case Resize:
 | 
				
			||||||
 | 
									if msgObj.Cols > 0 && msgObj.Rows > 0 {
 | 
				
			||||||
 | 
										if err := ts.terminal.WindowChange(msgObj.Rows, msgObj.Cols); err != nil {
 | 
				
			||||||
 | 
											global.Log.Error("ssh pty change windows size failed")
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								case Data:
 | 
				
			||||||
 | 
									_, err := ts.terminal.Write([]byte(msgObj.Msg))
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										global.Log.Debug("机器ssh终端写入消息失败: %s", err)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								case Ping:
 | 
				
			||||||
 | 
									_, err := ts.terminal.SshSession.SendRequest("ping", true, nil)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										WriteMessage(wsConn, "\r\n\033[1;31m提示: 终端连接已断开...\033[0m")
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func WriteMessage(ws *websocket.Conn, msg string) error {
 | 
				
			||||||
 | 
						return ws.WriteMessage(websocket.TextMessage, []byte(msg))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -1,195 +0,0 @@
 | 
				
			|||||||
package machine
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
import (
 | 
					 | 
				
			||||||
	"bytes"
 | 
					 | 
				
			||||||
	"encoding/json"
 | 
					 | 
				
			||||||
	"io"
 | 
					 | 
				
			||||||
	"mayfly-go/pkg/global"
 | 
					 | 
				
			||||||
	"sync"
 | 
					 | 
				
			||||||
	"time"
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	"github.com/gorilla/websocket"
 | 
					 | 
				
			||||||
	"golang.org/x/crypto/ssh"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type safeBuffer struct {
 | 
					 | 
				
			||||||
	buffer bytes.Buffer
 | 
					 | 
				
			||||||
	mu     sync.Mutex
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (w *safeBuffer) Write(p []byte) (int, error) {
 | 
					 | 
				
			||||||
	w.mu.Lock()
 | 
					 | 
				
			||||||
	defer w.mu.Unlock()
 | 
					 | 
				
			||||||
	return w.buffer.Write(p)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
func (w *safeBuffer) Bytes() []byte {
 | 
					 | 
				
			||||||
	w.mu.Lock()
 | 
					 | 
				
			||||||
	defer w.mu.Unlock()
 | 
					 | 
				
			||||||
	return w.buffer.Bytes()
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
func (w *safeBuffer) Reset() {
 | 
					 | 
				
			||||||
	w.mu.Lock()
 | 
					 | 
				
			||||||
	defer w.mu.Unlock()
 | 
					 | 
				
			||||||
	w.buffer.Reset()
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
const (
 | 
					 | 
				
			||||||
	wsMsgCmd    = "cmd"
 | 
					 | 
				
			||||||
	wsMsgResize = "resize"
 | 
					 | 
				
			||||||
)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type WsMsg struct {
 | 
					 | 
				
			||||||
	Type string `json:"type"`
 | 
					 | 
				
			||||||
	Msg  string `json:"msg"`
 | 
					 | 
				
			||||||
	Cols int    `json:"cols"`
 | 
					 | 
				
			||||||
	Rows int    `json:"rows"`
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
type LogicSshWsSession struct {
 | 
					 | 
				
			||||||
	stdinPipe       io.WriteCloser
 | 
					 | 
				
			||||||
	comboOutput     *safeBuffer //ssh 终端混合输出
 | 
					 | 
				
			||||||
	inputFilterBuff *safeBuffer //用来过滤输入的命令和ssh_filter配置对比的
 | 
					 | 
				
			||||||
	session         *ssh.Session
 | 
					 | 
				
			||||||
	wsConn          *websocket.Conn
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func NewLogicSshWsSession(cols, rows int, cli *Cli, wsConn *websocket.Conn) (*LogicSshWsSession, error) {
 | 
					 | 
				
			||||||
	sshSession, err := cli.GetSession()
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	stdinP, err := sshSession.StdinPipe()
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	comboWriter := new(safeBuffer)
 | 
					 | 
				
			||||||
	inputBuf := new(safeBuffer)
 | 
					 | 
				
			||||||
	//ssh.stdout and stderr will write output into comboWriter
 | 
					 | 
				
			||||||
	sshSession.Stdout = comboWriter
 | 
					 | 
				
			||||||
	sshSession.Stderr = comboWriter
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	modes := ssh.TerminalModes{
 | 
					 | 
				
			||||||
		ssh.ECHO:          1,     // disable echo
 | 
					 | 
				
			||||||
		ssh.TTY_OP_ISPEED: 14400, // input speed = 14.4kbaud
 | 
					 | 
				
			||||||
		ssh.TTY_OP_OSPEED: 14400, // output speed = 14.4kbaud
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	// Request pseudo terminal
 | 
					 | 
				
			||||||
	if err := sshSession.RequestPty("xterm", rows, cols, modes); err != nil {
 | 
					 | 
				
			||||||
		return nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	// Start remote shell
 | 
					 | 
				
			||||||
	if err := sshSession.Shell(); err != nil {
 | 
					 | 
				
			||||||
		return nil, err
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	return &LogicSshWsSession{
 | 
					 | 
				
			||||||
		stdinPipe:       stdinP,
 | 
					 | 
				
			||||||
		comboOutput:     comboWriter,
 | 
					 | 
				
			||||||
		inputFilterBuff: inputBuf,
 | 
					 | 
				
			||||||
		session:         sshSession,
 | 
					 | 
				
			||||||
		wsConn:          wsConn,
 | 
					 | 
				
			||||||
	}, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
//Close 关闭
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) Close() {
 | 
					 | 
				
			||||||
	if sws.session != nil {
 | 
					 | 
				
			||||||
		sws.session.Close()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if sws.comboOutput != nil {
 | 
					 | 
				
			||||||
		sws.comboOutput = nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) Start(quitChan chan bool) {
 | 
					 | 
				
			||||||
	go sws.receiveWsMsg(quitChan)
 | 
					 | 
				
			||||||
	go sws.sendComboOutput(quitChan)
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
//receiveWsMsg  receive websocket msg do some handling then write into ssh.session.stdin
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) receiveWsMsg(exitCh chan bool) {
 | 
					 | 
				
			||||||
	wsConn := sws.wsConn
 | 
					 | 
				
			||||||
	//tells other go routine quit
 | 
					 | 
				
			||||||
	defer setQuit(exitCh)
 | 
					 | 
				
			||||||
	for {
 | 
					 | 
				
			||||||
		select {
 | 
					 | 
				
			||||||
		case <-exitCh:
 | 
					 | 
				
			||||||
			return
 | 
					 | 
				
			||||||
		default:
 | 
					 | 
				
			||||||
			//read websocket msg
 | 
					 | 
				
			||||||
			_, wsData, err := wsConn.ReadMessage()
 | 
					 | 
				
			||||||
			if err != nil {
 | 
					 | 
				
			||||||
				if websocket.IsCloseError(err, websocket.CloseGoingAway, websocket.CloseNoStatusReceived) {
 | 
					 | 
				
			||||||
					return
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				global.Log.Error("reading webSocket message failed: ", err)
 | 
					 | 
				
			||||||
				return
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			//unmashal bytes into struct
 | 
					 | 
				
			||||||
			msgObj := WsMsg{}
 | 
					 | 
				
			||||||
			if err := json.Unmarshal(wsData, &msgObj); err != nil {
 | 
					 | 
				
			||||||
				global.Log.Error("unmarshal websocket message failed:", err)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			switch msgObj.Type {
 | 
					 | 
				
			||||||
			case wsMsgResize:
 | 
					 | 
				
			||||||
				//handle xterm.js size change
 | 
					 | 
				
			||||||
				if msgObj.Cols > 0 && msgObj.Rows > 0 {
 | 
					 | 
				
			||||||
					if err := sws.session.WindowChange(msgObj.Rows, msgObj.Cols); err != nil {
 | 
					 | 
				
			||||||
						global.Log.Error("ssh pty change windows size failed")
 | 
					 | 
				
			||||||
					}
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
			case wsMsgCmd:
 | 
					 | 
				
			||||||
				sws.sendWebsocketInputCommandToSshSessionStdinPipe([]byte(msgObj.Msg))
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
//sendWebsocketInputCommandToSshSessionStdinPipe
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) sendWebsocketInputCommandToSshSessionStdinPipe(cmdBytes []byte) {
 | 
					 | 
				
			||||||
	if _, err := sws.stdinPipe.Write(cmdBytes); err != nil {
 | 
					 | 
				
			||||||
		global.Log.Error("ws cmd bytes write to ssh.stdin pipe failed")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) sendComboOutput(exitCh chan bool) {
 | 
					 | 
				
			||||||
	wsConn := sws.wsConn
 | 
					 | 
				
			||||||
	//todo 优化成一个方法
 | 
					 | 
				
			||||||
	//tells other go routine quit
 | 
					 | 
				
			||||||
	defer setQuit(exitCh)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	//every 120ms write combine output bytes into websocket response
 | 
					 | 
				
			||||||
	tick := time.NewTicker(time.Millisecond * time.Duration(60))
 | 
					 | 
				
			||||||
	//for range time.Tick(120 * time.Millisecond){}
 | 
					 | 
				
			||||||
	defer tick.Stop()
 | 
					 | 
				
			||||||
	for {
 | 
					 | 
				
			||||||
		select {
 | 
					 | 
				
			||||||
		case <-tick.C:
 | 
					 | 
				
			||||||
			if sws.comboOutput == nil {
 | 
					 | 
				
			||||||
				return
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			bs := sws.comboOutput.Bytes()
 | 
					 | 
				
			||||||
			if len(bs) > 0 {
 | 
					 | 
				
			||||||
				err := wsConn.WriteMessage(websocket.TextMessage, bs)
 | 
					 | 
				
			||||||
				if err != nil {
 | 
					 | 
				
			||||||
					global.Log.Error("ssh sending combo output to webSocket failed")
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				sws.comboOutput.buffer.Reset()
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
		case <-exitCh:
 | 
					 | 
				
			||||||
			return
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func (sws *LogicSshWsSession) Wait(quitChan chan bool) {
 | 
					 | 
				
			||||||
	if err := sws.session.Wait(); err != nil {
 | 
					 | 
				
			||||||
		setQuit(quitChan)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func setQuit(ch chan bool) {
 | 
					 | 
				
			||||||
	ch <- true
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
@@ -4,6 +4,7 @@ import (
 | 
				
			|||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/entity"
 | 
						"mayfly-go/internal/devops/domain/entity"
 | 
				
			||||||
	"mayfly-go/internal/devops/domain/repository"
 | 
						"mayfly-go/internal/devops/domain/repository"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
	"mayfly-go/pkg/model"
 | 
						"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) {
 | 
					func (m *machineRepo) Create(entity *entity.Machine) {
 | 
				
			||||||
	model.Insert(entity)
 | 
						biz.ErrIsNilAppendErr(model.Insert(entity), "创建机器信息失败: %s")
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (m *machineRepo) UpdateById(entity *entity.Machine) {
 | 
					func (m *machineRepo) UpdateById(entity *entity.Machine) {
 | 
				
			||||||
	model.UpdateById(entity)
 | 
						biz.ErrIsNilAppendErr(model.UpdateById(entity), "更新机器信息失败: %s")
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,27 +0,0 @@
 | 
				
			|||||||
package scheduler
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func init() {
 | 
					 | 
				
			||||||
	SaveMachineMonitor()
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
func SaveMachineMonitor() {
 | 
					 | 
				
			||||||
	AddFun("@every 60s", func() {
 | 
					 | 
				
			||||||
		// for _, m := range models.GetNeedMonitorMachine() {
 | 
					 | 
				
			||||||
		// 	m := m
 | 
					 | 
				
			||||||
		// 	go func() {
 | 
					 | 
				
			||||||
		// 		cli, err := machine.GetCli(uint64(utils.GetInt4Map(m, "id")))
 | 
					 | 
				
			||||||
		// 		if err != nil {
 | 
					 | 
				
			||||||
		// 			mlog.Log.Error("获取客户端失败:", err.Error())
 | 
					 | 
				
			||||||
		// 			return
 | 
					 | 
				
			||||||
		// 		}
 | 
					 | 
				
			||||||
		// 		mm := cli.GetMonitorInfo()
 | 
					 | 
				
			||||||
		// 		if mm != nil {
 | 
					 | 
				
			||||||
		// 			err := model.Insert(mm)
 | 
					 | 
				
			||||||
		// 			if err != nil {
 | 
					 | 
				
			||||||
		// 				mlog.Log.Error("保存机器监控信息失败: ", err.Error())
 | 
					 | 
				
			||||||
		// 			}
 | 
					 | 
				
			||||||
		// 		}
 | 
					 | 
				
			||||||
		// 	}()
 | 
					 | 
				
			||||||
		// }
 | 
					 | 
				
			||||||
	})
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
@@ -20,8 +20,7 @@ func InitDbRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
		}
 | 
							}
 | 
				
			||||||
		// 获取所有数据库列表
 | 
							// 获取所有数据库列表
 | 
				
			||||||
		db.GET("", func(c *gin.Context) {
 | 
							db.GET("", func(c *gin.Context) {
 | 
				
			||||||
			rc := ctx.NewReqCtxWithGin(c)
 | 
								ctx.NewReqCtxWithGin(c).Handle(d.Dbs)
 | 
				
			||||||
			rc.Handle(d.Dbs)
 | 
					 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		saveDb := ctx.NewLogInfo("保存数据库信息").WithSave(true)
 | 
							saveDb := ctx.NewLogInfo("保存数据库信息").WithSave(true)
 | 
				
			||||||
@@ -31,6 +30,16 @@ func InitDbRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
				Handle(d.Save)
 | 
									Handle(d.Save)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// 获取数据库实例的所有数据库名
 | 
				
			||||||
 | 
							db.POST("databases", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).
 | 
				
			||||||
 | 
									Handle(d.GetDatabaseNames)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							db.GET(":dbId/pwd", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(d.GetDbPwd)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		deleteDb := ctx.NewLogInfo("删除数据库信息").WithSave(true)
 | 
							deleteDb := ctx.NewLogInfo("删除数据库信息").WithSave(true)
 | 
				
			||||||
		db.DELETE(":dbId", func(c *gin.Context) {
 | 
							db.DELETE(":dbId", func(c *gin.Context) {
 | 
				
			||||||
			ctx.NewReqCtxWithGin(c).
 | 
								ctx.NewReqCtxWithGin(c).
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -20,6 +20,10 @@ func InitMachineRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
			ctx.NewReqCtxWithGin(c).Handle(m.Machines)
 | 
								ctx.NewReqCtxWithGin(c).Handle(m.Machines)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							machines.GET(":machineId/pwd", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(m.GetMachinePwd)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		machines.GET(":machineId/stats", func(c *gin.Context) {
 | 
							machines.GET(":machineId/stats", func(c *gin.Context) {
 | 
				
			||||||
			ctx.NewReqCtxWithGin(c).Handle(m.MachineStats)
 | 
								ctx.NewReqCtxWithGin(c).Handle(m.MachineStats)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -26,6 +26,10 @@ func InitRedisRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
			ctx.NewReqCtxWithGin(c).WithLog(save).Handle(rs.Save)
 | 
								ctx.NewReqCtxWithGin(c).WithLog(save).Handle(rs.Save)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							redis.GET(":id/pwd", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(rs.GetRedisPwd)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		delRedis := ctx.NewLogInfo("删除redis信息").WithSave(true)
 | 
							delRedis := ctx.NewLogInfo("删除redis信息").WithSave(true)
 | 
				
			||||||
		redis.DELETE(":id", func(c *gin.Context) {
 | 
							redis.DELETE(":id", func(c *gin.Context) {
 | 
				
			||||||
			ctx.NewReqCtxWithGin(c).WithLog(delRedis).Handle(rs.DeleteRedis)
 | 
								ctx.NewReqCtxWithGin(c).WithLog(delRedis).Handle(rs.DeleteRedis)
 | 
				
			||||||
@@ -60,9 +64,17 @@ func InitRedisRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
			ctx.NewReqCtxWithGin(c).Handle(rs.SetStringValue)
 | 
								ctx.NewReqCtxWithGin(c).Handle(rs.SetStringValue)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// 获取hash类型值
 | 
							// hscan
 | 
				
			||||||
		redis.GET(":id/hash-value", func(c *gin.Context) {
 | 
							redis.GET(":id/hscan", func(c *gin.Context) {
 | 
				
			||||||
			ctx.NewReqCtxWithGin(c).Handle(rs.GetHashValue)
 | 
								ctx.NewReqCtxWithGin(c).Handle(rs.Hscan)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							redis.GET(":id/hget", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(rs.Hget)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							redis.DELETE(":id/hdel", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(rs.Hdel)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// 设置hash类型值
 | 
							// 设置hash类型值
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -23,6 +23,7 @@ type Account struct {
 | 
				
			|||||||
	ResourceApp application.Resource
 | 
						ResourceApp application.Resource
 | 
				
			||||||
	RoleApp     application.Role
 | 
						RoleApp     application.Role
 | 
				
			||||||
	MsgApp      application.Msg
 | 
						MsgApp      application.Msg
 | 
				
			||||||
 | 
						ConfigApp   application.Config
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
/**   登录者个人相关操作   **/
 | 
					/**   登录者个人相关操作   **/
 | 
				
			||||||
@@ -32,14 +33,19 @@ func (a *Account) Login(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	loginForm := &form.LoginForm{}
 | 
						loginForm := &form.LoginForm{}
 | 
				
			||||||
	ginx.BindJsonAndValid(rc.GinCtx, loginForm)
 | 
						ginx.BindJsonAndValid(rc.GinCtx, loginForm)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 校验验证码
 | 
						// 判断是否有开启登录验证码校验
 | 
				
			||||||
	biz.IsTrue(captcha.Verify(loginForm.Cid, loginForm.Captcha), "验证码错误")
 | 
						if a.ConfigApp.GetConfig(entity.ConfigKeyUseLoginCaptcha).BoolValue(true) {
 | 
				
			||||||
 | 
							// 校验验证码
 | 
				
			||||||
 | 
							biz.IsTrue(captcha.Verify(loginForm.Cid, loginForm.Captcha), "验证码错误")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	originPwd, err := utils.DefaultRsaDecrypt(loginForm.Password, true)
 | 
						originPwd, err := utils.DefaultRsaDecrypt(loginForm.Password, true)
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
						biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	account := &entity.Account{Username: loginForm.Username, Password: utils.Md5(originPwd)}
 | 
						account := &entity.Account{Username: loginForm.Username}
 | 
				
			||||||
	biz.ErrIsNil(a.AccountApp.GetAccount(account, "Id", "Username", "Status", "LastLoginTime", "LastLoginIp"), "用户名或密码错误")
 | 
						err = a.AccountApp.GetAccount(account, "Id", "Username", "Password", "Status", "LastLoginTime", "LastLoginIp")
 | 
				
			||||||
 | 
						biz.ErrIsNil(err, "用户名或密码错误")
 | 
				
			||||||
 | 
						biz.IsTrue(utils.CheckPwdHash(originPwd, account.Password), "用户名或密码错误")
 | 
				
			||||||
	biz.IsTrue(account.IsEnable(), "该账号不可用")
 | 
						biz.IsTrue(account.IsEnable(), "该账号不可用")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 校验密码强度是否符合
 | 
						// 校验密码强度是否符合
 | 
				
			||||||
@@ -86,8 +92,11 @@ func (a *Account) ChangePassword(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	originOldPwd, err := utils.DefaultRsaDecrypt(form.OldPassword, true)
 | 
						originOldPwd, err := utils.DefaultRsaDecrypt(form.OldPassword, true)
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "解密旧密码错误: %s")
 | 
						biz.ErrIsNilAppendErr(err, "解密旧密码错误: %s")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	account := &entity.Account{Username: form.Username, Password: utils.Md5(originOldPwd)}
 | 
						account := &entity.Account{Username: form.Username}
 | 
				
			||||||
	biz.ErrIsNil(a.AccountApp.GetAccount(account, "Id", "Username", "Status"), "旧密码不正确")
 | 
						err = a.AccountApp.GetAccount(account, "Id", "Username", "Password", "Status")
 | 
				
			||||||
 | 
						biz.ErrIsNil(err, "旧密码错误")
 | 
				
			||||||
 | 
						biz.IsTrue(utils.CheckPwdHash(originOldPwd, account.Password), "旧密码错误")
 | 
				
			||||||
 | 
						biz.IsTrue(account.IsEnable(), "该账号不可用")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	originNewPwd, err := utils.DefaultRsaDecrypt(form.NewPassword, true)
 | 
						originNewPwd, err := utils.DefaultRsaDecrypt(form.NewPassword, true)
 | 
				
			||||||
	biz.ErrIsNilAppendErr(err, "解密新密码错误: %s")
 | 
						biz.ErrIsNilAppendErr(err, "解密新密码错误: %s")
 | 
				
			||||||
@@ -95,7 +104,7 @@ func (a *Account) ChangePassword(rc *ctx.ReqCtx) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	updateAccount := new(entity.Account)
 | 
						updateAccount := new(entity.Account)
 | 
				
			||||||
	updateAccount.Id = account.Id
 | 
						updateAccount.Id = account.Id
 | 
				
			||||||
	updateAccount.Password = utils.Md5(originNewPwd)
 | 
						updateAccount.Password = utils.PwdHash(originNewPwd)
 | 
				
			||||||
	a.AccountApp.Update(updateAccount)
 | 
						a.AccountApp.Update(updateAccount)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 赋值loginAccount 主要用于记录操作日志,因为操作日志保存请求上下文没有该信息不保存日志
 | 
						// 赋值loginAccount 主要用于记录操作日志,因为操作日志保存请求上下文没有该信息不保存日志
 | 
				
			||||||
@@ -140,17 +149,6 @@ func (a *Account) saveLogin(account *entity.Account, ip string) {
 | 
				
			|||||||
	loginMsg.Creator = account.Username
 | 
						loginMsg.Creator = account.Username
 | 
				
			||||||
	loginMsg.CreatorId = account.Id
 | 
						loginMsg.CreatorId = account.Id
 | 
				
			||||||
	a.MsgApp.Create(loginMsg)
 | 
						a.MsgApp.Create(loginMsg)
 | 
				
			||||||
 | 
					 | 
				
			||||||
	// bodyMap, err := httpclient.NewRequest(fmt.Sprintf("http://ip-api.com/json/%s?lang=zh-CN", ip)).Get().BodyToMap()
 | 
					 | 
				
			||||||
	// if err != nil {
 | 
					 | 
				
			||||||
	// 	global.Log.Errorf("获取客户端ip地址信息失败:%s", err.Error())
 | 
					 | 
				
			||||||
	// 	return
 | 
					 | 
				
			||||||
	// }
 | 
					 | 
				
			||||||
	// if bodyMap["status"].(string) == "fail" {
 | 
					 | 
				
			||||||
	// 	return
 | 
					 | 
				
			||||||
	// }
 | 
					 | 
				
			||||||
	// msg := fmt.Sprintf("%s于%s-%s登录", account.Username, bodyMap["regionName"], bodyMap["city"])
 | 
					 | 
				
			||||||
	// global.Log.Info(msg)
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 获取个人账号信息
 | 
					// 获取个人账号信息
 | 
				
			||||||
@@ -176,7 +174,7 @@ func (a *Account) UpdateAccount(rc *ctx.ReqCtx) {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	if updateAccount.Password != "" {
 | 
						if updateAccount.Password != "" {
 | 
				
			||||||
		biz.IsTrue(CheckPasswordLever(updateAccount.Password), "密码强度必须8位以上且包含字⺟⼤⼩写+数字+特殊符号")
 | 
							biz.IsTrue(CheckPasswordLever(updateAccount.Password), "密码强度必须8位以上且包含字⺟⼤⼩写+数字+特殊符号")
 | 
				
			||||||
		updateAccount.Password = utils.Md5(updateAccount.Password)
 | 
							updateAccount.Password = utils.PwdHash(updateAccount.Password)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	a.AccountApp.Update(updateAccount)
 | 
						a.AccountApp.Update(updateAccount)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@@ -198,8 +196,8 @@ func (a *Account) Accounts(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	rc.ResData = a.AccountApp.GetPageList(condition, ginx.GetPageParam(rc.GinCtx), new([]vo.AccountManageVO))
 | 
						rc.ResData = a.AccountApp.GetPageList(condition, ginx.GetPageParam(rc.GinCtx), new([]vo.AccountManageVO))
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// @router /accounts [get]
 | 
					// @router /accounts
 | 
				
			||||||
func (a *Account) CreateAccount(rc *ctx.ReqCtx) {
 | 
					func (a *Account) SaveAccount(rc *ctx.ReqCtx) {
 | 
				
			||||||
	form := &form.AccountCreateForm{}
 | 
						form := &form.AccountCreateForm{}
 | 
				
			||||||
	ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
						ginx.BindJsonAndValid(rc.GinCtx, form)
 | 
				
			||||||
	rc.ReqParam = form
 | 
						rc.ReqParam = form
 | 
				
			||||||
@@ -207,7 +205,16 @@ func (a *Account) CreateAccount(rc *ctx.ReqCtx) {
 | 
				
			|||||||
	account := &entity.Account{}
 | 
						account := &entity.Account{}
 | 
				
			||||||
	utils.Copy(account, form)
 | 
						utils.Copy(account, form)
 | 
				
			||||||
	account.SetBaseInfo(rc.LoginAccount)
 | 
						account.SetBaseInfo(rc.LoginAccount)
 | 
				
			||||||
	a.AccountApp.Create(account)
 | 
					
 | 
				
			||||||
 | 
						if account.Id == 0 {
 | 
				
			||||||
 | 
							a.AccountApp.Create(account)
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							if account.Password != "" {
 | 
				
			||||||
 | 
								biz.IsTrue(CheckPasswordLever(account.Password), "密码强度必须8位以上且包含字⺟⼤⼩写+数字+特殊符号")
 | 
				
			||||||
 | 
								account.Password = utils.PwdHash(account.Password)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							a.AccountApp.Update(account)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (a *Account) ChangeStatus(rc *ctx.ReqCtx) {
 | 
					func (a *Account) ChangeStatus(rc *ctx.ReqCtx) {
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										39
									
								
								server/internal/sys/api/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								server/internal/sys/api/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,39 @@
 | 
				
			|||||||
 | 
					package api
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/api/form"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/application"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/entity"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/ctx"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/ginx"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Config struct {
 | 
				
			||||||
 | 
						ConfigApp application.Config
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Config) Configs(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						g := rc.GinCtx
 | 
				
			||||||
 | 
						condition := &entity.Config{Key: g.Query("key")}
 | 
				
			||||||
 | 
						rc.ResData = c.ConfigApp.GetPageList(condition, ginx.GetPageParam(g), new([]entity.Config))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Config) GetConfigValueByKey(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						key := rc.GinCtx.Query("key")
 | 
				
			||||||
 | 
						biz.NotEmpty(key, "key不能为空")
 | 
				
			||||||
 | 
						rc.ResData = c.ConfigApp.GetConfig(key).Value
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (c *Config) SaveConfig(rc *ctx.ReqCtx) {
 | 
				
			||||||
 | 
						g := rc.GinCtx
 | 
				
			||||||
 | 
						form := &form.ConfigForm{}
 | 
				
			||||||
 | 
						ginx.BindJsonAndValid(g, form)
 | 
				
			||||||
 | 
						rc.ReqParam = form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						config := new(entity.Config)
 | 
				
			||||||
 | 
						utils.Copy(config, form)
 | 
				
			||||||
 | 
						config.SetBaseInfo(rc.LoginAccount)
 | 
				
			||||||
 | 
						c.ConfigApp.Save(config)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -1,7 +1,9 @@
 | 
				
			|||||||
package form
 | 
					package form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type AccountCreateForm struct {
 | 
					type AccountCreateForm struct {
 | 
				
			||||||
 | 
						Id       uint64
 | 
				
			||||||
	Username *string `json:"username" binding:"required,min=4,max=16"`
 | 
						Username *string `json:"username" binding:"required,min=4,max=16"`
 | 
				
			||||||
 | 
						Password *string `json:"password"`
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type AccountUpdateForm struct {
 | 
					type AccountUpdateForm struct {
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										9
									
								
								server/internal/sys/api/form/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								server/internal/sys/api/form/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
				
			|||||||
 | 
					package form
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type ConfigForm struct {
 | 
				
			||||||
 | 
						Id     int
 | 
				
			||||||
 | 
						Name   string `binding:"required"`
 | 
				
			||||||
 | 
						Key    string `binding:"required"`
 | 
				
			||||||
 | 
						Value  string
 | 
				
			||||||
 | 
						Remark string `json:"remark"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -4,6 +4,6 @@ package form
 | 
				
			|||||||
type LoginForm struct {
 | 
					type LoginForm struct {
 | 
				
			||||||
	Username string `json:"username" binding:"required"`
 | 
						Username string `json:"username" binding:"required"`
 | 
				
			||||||
	Password string `binding:"required"`
 | 
						Password string `binding:"required"`
 | 
				
			||||||
	Captcha  string `binding:"required"`
 | 
						Captcha  string
 | 
				
			||||||
	Cid      string `binding:"required"`
 | 
						Cid      string
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -43,7 +43,7 @@ func (a *accountAppImpl) GetPageList(condition *entity.Account, pageParam *model
 | 
				
			|||||||
func (a *accountAppImpl) Create(account *entity.Account) {
 | 
					func (a *accountAppImpl) Create(account *entity.Account) {
 | 
				
			||||||
	biz.IsTrue(a.GetAccount(&entity.Account{Username: account.Username}) != nil, "该账号用户名已存在")
 | 
						biz.IsTrue(a.GetAccount(&entity.Account{Username: account.Username}) != nil, "该账号用户名已存在")
 | 
				
			||||||
	// 默认密码为账号用户名
 | 
						// 默认密码为账号用户名
 | 
				
			||||||
	account.Password = utils.Md5(account.Username)
 | 
						account.Password = utils.PwdHash(account.Username)
 | 
				
			||||||
	account.Status = entity.AccountEnableStatus
 | 
						account.Status = entity.AccountEnableStatus
 | 
				
			||||||
	a.accountRepo.Insert(account)
 | 
						a.accountRepo.Insert(account)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										46
									
								
								server/internal/sys/application/config_app.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										46
									
								
								server/internal/sys/application/config_app.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,46 @@
 | 
				
			|||||||
 | 
					package application
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/entity"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/repository"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/infrastructure/persistence"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/global"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Config interface {
 | 
				
			||||||
 | 
						GetPageList(condition *entity.Config, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						Save(config *entity.Config)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 获取指定key的配置信息, 不会返回nil, 若不存在则值都默认值即空字符串
 | 
				
			||||||
 | 
						GetConfig(key string) *entity.Config
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type configAppImpl struct {
 | 
				
			||||||
 | 
						configRepo repository.Config
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var ConfigApp Config = &configAppImpl{
 | 
				
			||||||
 | 
						configRepo: persistence.ConfigDao,
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (a *configAppImpl) GetPageList(condition *entity.Config, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult {
 | 
				
			||||||
 | 
						return a.configRepo.GetPageList(condition, pageParam, toEntity)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (a *configAppImpl) Save(config *entity.Config) {
 | 
				
			||||||
 | 
						if config.Id == 0 {
 | 
				
			||||||
 | 
							a.configRepo.Insert(config)
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							a.configRepo.Update(config)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (a *configAppImpl) GetConfig(key string) *entity.Config {
 | 
				
			||||||
 | 
						config := &entity.Config{Key: key}
 | 
				
			||||||
 | 
						if err := a.configRepo.GetConfig(config, "Id", "Key", "Value"); err != nil {
 | 
				
			||||||
 | 
							global.Log.Warnf("不存在key = [%s] 的系统配置", key)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return config
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										29
									
								
								server/internal/sys/domain/entity/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										29
									
								
								server/internal/sys/domain/entity/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,29 @@
 | 
				
			|||||||
 | 
					package entity
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import "mayfly-go/pkg/model"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						ConfigKeyUseLoginCaptcha string = "UseLoginCaptcha" // 是否使用登录验证码
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Config struct {
 | 
				
			||||||
 | 
						model.Model
 | 
				
			||||||
 | 
						Name   string `json:"name"` // 配置名
 | 
				
			||||||
 | 
						Key    string `json:"key"`  // 配置key
 | 
				
			||||||
 | 
						Value  string `json:"value"`
 | 
				
			||||||
 | 
						Remark string `json:"remark"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (a *Config) TableName() string {
 | 
				
			||||||
 | 
						return "t_sys_config"
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 若配置信息不存在, 则返回传递的默认值.
 | 
				
			||||||
 | 
					// 否则只有value == "1"为true,其他为false
 | 
				
			||||||
 | 
					func (c *Config) BoolValue(defaultValue bool) bool {
 | 
				
			||||||
 | 
						// 如果值不存在,则返回默认值
 | 
				
			||||||
 | 
						if c.Id == 0 {
 | 
				
			||||||
 | 
							return defaultValue
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return c.Value == "1"
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										18
									
								
								server/internal/sys/domain/repository/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										18
									
								
								server/internal/sys/domain/repository/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,18 @@
 | 
				
			|||||||
 | 
					package repository
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/entity"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Config interface {
 | 
				
			||||||
 | 
						GetPageList(condition *entity.Config, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						Insert(config *entity.Config)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						Update(config *entity.Config)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						GetConfig(config *entity.Config, cols ...string) error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						GetByCondition(condition *entity.Config, cols ...string) error
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -0,0 +1,32 @@
 | 
				
			|||||||
 | 
					package persistence
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/entity"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/domain/repository"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type configRepo struct{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var ConfigDao repository.Config = &configRepo{}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *configRepo) GetPageList(condition *entity.Config, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult {
 | 
				
			||||||
 | 
						return model.GetPage(pageParam, condition, toEntity)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *configRepo) Insert(config *entity.Config) {
 | 
				
			||||||
 | 
						biz.ErrIsNil(model.Insert(config), "新增系统配置失败")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *configRepo) Update(config *entity.Config) {
 | 
				
			||||||
 | 
						biz.ErrIsNil(model.UpdateById(config), "更新系统配置失败")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (m *configRepo) GetConfig(condition *entity.Config, cols ...string) error {
 | 
				
			||||||
 | 
						return model.GetBy(condition, cols...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (r *configRepo) GetByCondition(condition *entity.Config, cols ...string) error {
 | 
				
			||||||
 | 
						return model.GetBy(condition, cols...)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -15,6 +15,7 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
		ResourceApp: application.ResourceApp,
 | 
							ResourceApp: application.ResourceApp,
 | 
				
			||||||
		RoleApp:     application.RoleApp,
 | 
							RoleApp:     application.RoleApp,
 | 
				
			||||||
		MsgApp:      application.MsgApp,
 | 
							MsgApp:      application.MsgApp,
 | 
				
			||||||
 | 
							ConfigApp:   application.ConfigApp,
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	{
 | 
						{
 | 
				
			||||||
		// 用户登录
 | 
							// 用户登录
 | 
				
			||||||
@@ -61,7 +62,7 @@ func InitAccountRouter(router *gin.RouterGroup) {
 | 
				
			|||||||
			ctx.NewReqCtxWithGin(c).
 | 
								ctx.NewReqCtxWithGin(c).
 | 
				
			||||||
				WithRequiredPermission(addAccountPermission).
 | 
									WithRequiredPermission(addAccountPermission).
 | 
				
			||||||
				WithLog(createAccount).
 | 
									WithLog(createAccount).
 | 
				
			||||||
				Handle(a.CreateAccount)
 | 
									Handle(a.SaveAccount)
 | 
				
			||||||
		})
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		changeStatus := ctx.NewLogInfo("修改账号状态").WithSave(true)
 | 
							changeStatus := ctx.NewLogInfo("修改账号状态").WithSave(true)
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										30
									
								
								server/internal/sys/router/config.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										30
									
								
								server/internal/sys/router/config.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,30 @@
 | 
				
			|||||||
 | 
					package router
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/api"
 | 
				
			||||||
 | 
						"mayfly-go/internal/sys/application"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/ctx"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						"github.com/gin-gonic/gin"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func InitSysConfigRouter(router *gin.RouterGroup) {
 | 
				
			||||||
 | 
						r := &api.Config{ConfigApp: application.ConfigApp}
 | 
				
			||||||
 | 
						db := router.Group("sys/configs")
 | 
				
			||||||
 | 
						{
 | 
				
			||||||
 | 
							db.GET("", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).Handle(r.Configs)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							db.GET("/value", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).WithNeedToken(false).Handle(r.GetConfigValueByKey)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							saveConfig := ctx.NewLogInfo("保存系统配置信息").WithSave(true)
 | 
				
			||||||
 | 
							db.POST("", func(c *gin.Context) {
 | 
				
			||||||
 | 
								ctx.NewReqCtxWithGin(c).
 | 
				
			||||||
 | 
									WithLog(saveConfig).
 | 
				
			||||||
 | 
									Handle(r.SaveConfig)
 | 
				
			||||||
 | 
							})
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -5,7 +5,5 @@ import (
 | 
				
			|||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func main() {
 | 
					func main() {
 | 
				
			||||||
	starter.PrintBanner()
 | 
					 | 
				
			||||||
	starter.InitDb()
 | 
					 | 
				
			||||||
	starter.RunWebServer()
 | 
						starter.RunWebServer()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -30,8 +30,8 @@ CREATE TABLE `t_db` (
 | 
				
			|||||||
  `database` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '数据库,空格分割多个数据库',
 | 
					  `database` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '数据库,空格分割多个数据库',
 | 
				
			||||||
  `params` varchar(125) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '其他连接参数',
 | 
					  `params` varchar(125) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '其他连接参数',
 | 
				
			||||||
  `network` varchar(20) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `network` varchar(20) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
					  `enable_ssh_tunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
				
			||||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
					  `ssh_tunnel_machine_id` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
				
			||||||
  `project_id` bigint(20) DEFAULT NULL,
 | 
					  `project_id` bigint(20) DEFAULT NULL,
 | 
				
			||||||
  `project` varchar(64) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `project` varchar(64) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
  `env_id` bigint(20) DEFAULT NULL COMMENT '环境id',
 | 
					  `env_id` bigint(20) DEFAULT NULL COMMENT '环境id',
 | 
				
			||||||
@@ -111,6 +111,8 @@ CREATE TABLE `t_machine` (
 | 
				
			|||||||
  `username` varchar(12) COLLATE utf8mb4_bin NOT NULL,
 | 
					  `username` varchar(12) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
  `auth_method` tinyint(2) NULL DEFAULT NULL COMMENT '1.密码登录2.publickey登录',
 | 
					  `auth_method` tinyint(2) NULL DEFAULT NULL COMMENT '1.密码登录2.publickey登录',
 | 
				
			||||||
  `password` varchar(3200) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `password` varchar(3200) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
 | 
					  `enable_ssh_tunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
				
			||||||
 | 
					  `ssh_tunnel_machine_id` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
				
			||||||
  `status` tinyint(2) NOT NULL COMMENT '状态: 1:启用; -1:禁用',
 | 
					  `status` tinyint(2) NOT NULL COMMENT '状态: 1:启用; -1:禁用',
 | 
				
			||||||
  `remark` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `remark` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
  `need_monitor` tinyint(2) DEFAULT NULL,
 | 
					  `need_monitor` tinyint(2) DEFAULT NULL,
 | 
				
			||||||
@@ -258,11 +260,11 @@ DROP TABLE IF EXISTS `t_redis`;
 | 
				
			|||||||
CREATE TABLE `t_redis` (
 | 
					CREATE TABLE `t_redis` (
 | 
				
			||||||
  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
					  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
				
			||||||
  `host` varchar(255) COLLATE utf8mb4_bin NOT NULL,
 | 
					  `host` varchar(255) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
  `password` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `password` varchar(100) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
  `db` int(32) DEFAULT NULL,
 | 
					  `db` int(32) DEFAULT NULL,
 | 
				
			||||||
  `mode` varchar(32) DEFAULT NULL,
 | 
					  `mode` varchar(32) DEFAULT NULL,
 | 
				
			||||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
					  `enable_ssh_tunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
				
			||||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
					  `ssh_tunnel_machine_id` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
				
			||||||
  `remark` varchar(125) DEFAULT NULL,
 | 
					  `remark` varchar(125) DEFAULT NULL,
 | 
				
			||||||
  `project_id` bigint(20) DEFAULT NULL,
 | 
					  `project_id` bigint(20) DEFAULT NULL,
 | 
				
			||||||
  `project` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `project` varchar(32) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
@@ -284,7 +286,7 @@ CREATE TABLE `t_redis` (
 | 
				
			|||||||
DROP TABLE IF EXISTS `t_sys_account`;
 | 
					DROP TABLE IF EXISTS `t_sys_account`;
 | 
				
			||||||
CREATE TABLE `t_sys_account` (
 | 
					CREATE TABLE `t_sys_account` (
 | 
				
			||||||
  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
					  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
				
			||||||
  `username` varchar(12) COLLATE utf8mb4_bin NOT NULL,
 | 
					  `username` varchar(30) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
  `password` varchar(64) COLLATE utf8mb4_bin NOT NULL,
 | 
					  `password` varchar(64) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
  `status` tinyint(4) DEFAULT NULL,
 | 
					  `status` tinyint(4) DEFAULT NULL,
 | 
				
			||||||
  `last_login_time` datetime DEFAULT NULL COMMENT '最后登录时间',
 | 
					  `last_login_time` datetime DEFAULT NULL COMMENT '最后登录时间',
 | 
				
			||||||
@@ -302,7 +304,7 @@ CREATE TABLE `t_sys_account` (
 | 
				
			|||||||
-- Records of t_sys_account
 | 
					-- Records of t_sys_account
 | 
				
			||||||
-- ----------------------------
 | 
					-- ----------------------------
 | 
				
			||||||
BEGIN;
 | 
					BEGIN;
 | 
				
			||||||
INSERT INTO `t_sys_account` VALUES (1, 'admin', 'e10adc3949ba59abbe56e057f20f883e', 1, '2021-11-17 16:30:02', '12.0.216.228', '2020-01-01 19:00:00', 1, 'admin', '2020-01-01 19:00:00', 1, 'admin');
 | 
					INSERT INTO `t_sys_account` VALUES (1, 'admin', '$2a$10$w3Wky2U.tinvR7c/s0aKPuwZsIu6pM1/DMJalwBDMbE6niHIxVrrm', 1, '2021-11-17 16:30:02', '12.0.216.228', '2020-01-01 19:00:00', 1, 'admin', '2020-01-01 19:00:00', 1, 'admin');
 | 
				
			||||||
COMMIT;
 | 
					COMMIT;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
-- ----------------------------
 | 
					-- ----------------------------
 | 
				
			||||||
@@ -435,6 +437,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 (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 (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 (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');
 | 
					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');
 | 
				
			||||||
 | 
					INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (86, 4, 1, 1, '系统配置', 'configs', 5, '{\"component\":\"ConfigList\",\"icon\":\"Setting\",\"isKeepAlive\":true,\"routeName\":\"ConfigList\"}', 1, 'admin', 1, 'admin', '2022-08-25 22:18:55', '2022-08-25 22:19:18');
 | 
				
			||||||
 | 
					INSERT INTO `t_sys_resource`(`id`, `pid`, `type`, `status`, `name`, `code`, `weight`, `meta`, `creator_id`, `creator`, `modifier_id`, `modifier`, `create_time`, `update_time`) VALUES (87, 86, 2, 1, '基本权限', 'config:base', 1, 'null', 1, 'admin', 1, 'admin', '2022-08-25 22:19:35', '2022-08-25 22:19:35');
 | 
				
			||||||
COMMIT;
 | 
					COMMIT;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
-- ----------------------------
 | 
					-- ----------------------------
 | 
				
			||||||
@@ -640,6 +644,8 @@ INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`,
 | 
				
			|||||||
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 (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 (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');
 | 
					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');
 | 
				
			||||||
 | 
					INSERT INTO `t_sys_role_resource`(`id`, `role_id`, `resource_id`, `creator_id`, `creator`, `create_time`) VALUES (509, 1, 86, 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 (510, 1, 87, 1, 'admin', '2022-07-14 11:10:11');
 | 
				
			||||||
COMMIT;
 | 
					COMMIT;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
-- ----------------------------
 | 
					-- ----------------------------
 | 
				
			||||||
@@ -668,8 +674,8 @@ CREATE TABLE `t_mongo` (
 | 
				
			|||||||
  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
					  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
				
			||||||
  `name` varchar(36) COLLATE utf8mb4_bin NOT NULL COMMENT '名称',
 | 
					  `name` varchar(36) COLLATE utf8mb4_bin NOT NULL COMMENT '名称',
 | 
				
			||||||
  `uri` varchar(255) COLLATE utf8mb4_bin NOT NULL COMMENT '连接uri',
 | 
					  `uri` varchar(255) COLLATE utf8mb4_bin NOT NULL COMMENT '连接uri',
 | 
				
			||||||
  `enableSshTunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
					  `enable_ssh_tunnel` tinyint(2) DEFAULT NULL COMMENT '是否启用ssh隧道',
 | 
				
			||||||
  `sshTunnelMachineId` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
					  `ssh_tunnel_machine_id` bigint(20) DEFAULT NULL COMMENT 'ssh隧道的机器id',
 | 
				
			||||||
  `project_id` bigint(20) NOT NULL,
 | 
					  `project_id` bigint(20) NOT NULL,
 | 
				
			||||||
  `project` varchar(36) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
					  `project` varchar(36) COLLATE utf8mb4_bin DEFAULT NULL,
 | 
				
			||||||
  `env_id` bigint(20) DEFAULT NULL,
 | 
					  `env_id` bigint(20) DEFAULT NULL,
 | 
				
			||||||
@@ -683,5 +689,32 @@ CREATE TABLE `t_mongo` (
 | 
				
			|||||||
  PRIMARY KEY (`id`)
 | 
					  PRIMARY KEY (`id`)
 | 
				
			||||||
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;
 | 
					) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					-- ----------------------------
 | 
				
			||||||
 | 
					-- Table structure for t_sys_config
 | 
				
			||||||
 | 
					-- ----------------------------
 | 
				
			||||||
 | 
					DROP TABLE IF EXISTS `t_sys_config`;
 | 
				
			||||||
 | 
					CREATE TABLE `t_sys_config` (
 | 
				
			||||||
 | 
					  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,
 | 
				
			||||||
 | 
					  `name` varchar(60) COLLATE utf8mb4_bin NOT NULL COMMENT '配置名',
 | 
				
			||||||
 | 
					  `key` varchar(120) COLLATE utf8mb4_bin NOT NULL COMMENT '配置key',
 | 
				
			||||||
 | 
					  `value` varchar(500) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '配置value',
 | 
				
			||||||
 | 
					  `remark` varchar(255) COLLATE utf8mb4_bin DEFAULT NULL COMMENT '备注',
 | 
				
			||||||
 | 
					  `create_time` datetime NOT NULL,
 | 
				
			||||||
 | 
					  `creator_id` bigint(20) NOT NULL,
 | 
				
			||||||
 | 
					  `creator` varchar(36) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
 | 
					  `update_time` datetime NOT NULL,
 | 
				
			||||||
 | 
					  `modifier_id` bigint(20) NOT NULL,
 | 
				
			||||||
 | 
					  `modifier` varchar(36) COLLATE utf8mb4_bin NOT NULL,
 | 
				
			||||||
 | 
					  PRIMARY KEY (`id`)
 | 
				
			||||||
 | 
					) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					-- ----------------------------
 | 
				
			||||||
 | 
					-- Records of t_sys_config
 | 
				
			||||||
 | 
					-- ----------------------------
 | 
				
			||||||
 | 
					BEGIN;
 | 
				
			||||||
 | 
					INSERT INTO `t_sys_config` VALUES (1, '是否启用登录验证码', 'UseLoginCaptcha', '1', '1: 启用、0: 不启用', '2022-08-25 22:27:17', 1, 'admin', '2022-08-26 10:26:56', 1, 'admin');
 | 
				
			||||||
 | 
					INSERT INTO `t_sys_config` VALUES (2, '是否启用水印', 'UseWartermark', '1', '1: 启用、0: 不启用', '2022-08-25 23:36:35', 1, 'admin', '2022-08-26 10:02:52', 1, 'admin');
 | 
				
			||||||
 | 
					COMMIT;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
SET FOREIGN_KEY_CHECKS = 1;
 | 
					SET FOREIGN_KEY_CHECKS = 1;
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										27
									
								
								server/pkg/config/aes.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								server/pkg/config/aes.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,27 @@
 | 
				
			|||||||
 | 
					package config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/utils/assert"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					type Aes struct {
 | 
				
			||||||
 | 
						Key string `yaml:"key"`
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// 编码并base64
 | 
				
			||||||
 | 
					func (a *Aes) EncryptBase64(data []byte) (string, error) {
 | 
				
			||||||
 | 
						return utils.AesEncryptBase64(data, []byte(a.Key))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// base64解码后再aes解码
 | 
				
			||||||
 | 
					func (a *Aes) DecryptBase64(data string) ([]byte, error) {
 | 
				
			||||||
 | 
						return utils.AesDecryptBase64(data, []byte(a.Key))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (j *Aes) Valid() {
 | 
				
			||||||
 | 
						aesKeyLen := len(j.Key)
 | 
				
			||||||
 | 
						assert.IsTrue(aesKeyLen == 16 || aesKeyLen == 24 || aesKeyLen == 32,
 | 
				
			||||||
 | 
							fmt.Sprintf("config.yml之 [aes.key] 长度需为16、24、32位长度, 当前为%d位", aesKeyLen))
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@@ -2,11 +2,11 @@ package config
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
import "fmt"
 | 
					import "fmt"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type App struct {
 | 
					const (
 | 
				
			||||||
	Name    string `yaml:"name"`
 | 
						AppName = "mayfly-go"
 | 
				
			||||||
	Version string `yaml:"version"`
 | 
						Version = "v1.2.8"
 | 
				
			||||||
}
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (a *App) GetAppInfo() string {
 | 
					func GetAppInfo() string {
 | 
				
			||||||
	return fmt.Sprintf("[%s:%s]", a.Name, a.Version)
 | 
						return fmt.Sprintf("[%s:%s]", AppName, Version)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -11,12 +11,12 @@ import (
 | 
				
			|||||||
// 配置文件映射对象
 | 
					// 配置文件映射对象
 | 
				
			||||||
var Conf *Config
 | 
					var Conf *Config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func init() {
 | 
					func Init() {
 | 
				
			||||||
	configFilePath := flag.String("e", "./config.yml", "配置文件路径,默认为可执行文件目录")
 | 
						configFilePath := flag.String("e", "./config.yml", "配置文件路径,默认为可执行文件目录")
 | 
				
			||||||
	flag.Parse()
 | 
						flag.Parse()
 | 
				
			||||||
	// 获取启动参数中,配置文件的绝对路径
 | 
						// 获取启动参数中,配置文件的绝对路径
 | 
				
			||||||
	path, _ := filepath.Abs(*configFilePath)
 | 
						path, _ := filepath.Abs(*configFilePath)
 | 
				
			||||||
	startConfigParam = &CmdConfigParam{ConfigFilePath: path}
 | 
						startConfigParam := &CmdConfigParam{ConfigFilePath: path}
 | 
				
			||||||
	// 读取配置文件信息
 | 
						// 读取配置文件信息
 | 
				
			||||||
	yc := &Config{}
 | 
						yc := &Config{}
 | 
				
			||||||
	if err := utils.LoadYml(startConfigParam.ConfigFilePath, yc); err != nil {
 | 
						if err := utils.LoadYml(startConfigParam.ConfigFilePath, yc); err != nil {
 | 
				
			||||||
@@ -32,14 +32,11 @@ type CmdConfigParam struct {
 | 
				
			|||||||
	ConfigFilePath string // -e  配置文件路径
 | 
						ConfigFilePath string // -e  配置文件路径
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 启动可执行文件时的参数
 | 
					 | 
				
			||||||
var startConfigParam *CmdConfigParam
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// yaml配置文件映射对象
 | 
					// yaml配置文件映射对象
 | 
				
			||||||
type Config struct {
 | 
					type Config struct {
 | 
				
			||||||
	App    *App    `yaml:"app"`
 | 
					 | 
				
			||||||
	Server *Server `yaml:"server"`
 | 
						Server *Server `yaml:"server"`
 | 
				
			||||||
	Jwt    *Jwt    `yaml:"jwt"`
 | 
						Jwt    *Jwt    `yaml:"jwt"`
 | 
				
			||||||
 | 
						Aes    *Aes    `yaml:"aes"`
 | 
				
			||||||
	Redis  *Redis  `yaml:"redis"`
 | 
						Redis  *Redis  `yaml:"redis"`
 | 
				
			||||||
	Mysql  *Mysql  `yaml:"mysql"`
 | 
						Mysql  *Mysql  `yaml:"mysql"`
 | 
				
			||||||
	Log    *Log    `yaml:"log"`
 | 
						Log    *Log    `yaml:"log"`
 | 
				
			||||||
@@ -49,14 +46,7 @@ type Config struct {
 | 
				
			|||||||
func (c *Config) Valid() {
 | 
					func (c *Config) Valid() {
 | 
				
			||||||
	assert.IsTrue(c.Jwt != nil, "配置文件的[jwt]信息不能为空")
 | 
						assert.IsTrue(c.Jwt != nil, "配置文件的[jwt]信息不能为空")
 | 
				
			||||||
	c.Jwt.Valid()
 | 
						c.Jwt.Valid()
 | 
				
			||||||
}
 | 
						if c.Aes != nil {
 | 
				
			||||||
 | 
							c.Aes.Valid()
 | 
				
			||||||
// 获取执行可执行文件时,指定的启动参数
 | 
						}
 | 
				
			||||||
func getStartConfig() *CmdConfigParam {
 | 
					 | 
				
			||||||
	configFilePath := flag.String("e", "./config.yml", "配置文件路径,默认为可执行文件目录")
 | 
					 | 
				
			||||||
	flag.Parse()
 | 
					 | 
				
			||||||
	// 获取配置文件绝对路径
 | 
					 | 
				
			||||||
	path, _ := filepath.Abs(*configFilePath)
 | 
					 | 
				
			||||||
	sc := &CmdConfigParam{ConfigFilePath: path}
 | 
					 | 
				
			||||||
	return sc
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -8,6 +8,5 @@ type Jwt struct {
 | 
				
			|||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func (j *Jwt) Valid() {
 | 
					func (j *Jwt) Valid() {
 | 
				
			||||||
	assert.IsTrue(j.Key != "", "config.yml之 [jwt.key] 不能为空")
 | 
					 | 
				
			||||||
	assert.IsTrue(j.ExpireTime != 0, "config.yml之 [jwt.expire-time] 不能为空")
 | 
						assert.IsTrue(j.ExpireTime != 0, "config.yml之 [jwt.expire-time] 不能为空")
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -5,15 +5,22 @@ import (
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	"mayfly-go/pkg/biz"
 | 
						"mayfly-go/pkg/biz"
 | 
				
			||||||
	"mayfly-go/pkg/config"
 | 
						"mayfly-go/pkg/config"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/global"
 | 
				
			||||||
	"mayfly-go/pkg/model"
 | 
						"mayfly-go/pkg/model"
 | 
				
			||||||
 | 
						"mayfly-go/pkg/utils"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/dgrijalva/jwt-go"
 | 
						"github.com/golang-jwt/jwt/v4"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
var (
 | 
					func InitTokenConfig() {
 | 
				
			||||||
	JwtKey  = config.Conf.Jwt.Key
 | 
						JwtKey = config.Conf.Jwt.Key
 | 
				
			||||||
	ExpTime = config.Conf.Jwt.ExpireTime
 | 
						ExpTime = config.Conf.Jwt.ExpireTime
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var (
 | 
				
			||||||
 | 
						JwtKey  string
 | 
				
			||||||
 | 
						ExpTime uint64
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// 创建用户token
 | 
					// 创建用户token
 | 
				
			||||||
@@ -26,6 +33,11 @@ func CreateToken(userId uint64, username string) string {
 | 
				
			|||||||
		"exp":      time.Now().Add(time.Minute * time.Duration(ExpTime)).Unix(),
 | 
							"exp":      time.Now().Add(time.Minute * time.Duration(ExpTime)).Unix(),
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 如果配置文件中的jwt key为空,则随机生成字符串
 | 
				
			||||||
 | 
						if JwtKey == "" {
 | 
				
			||||||
 | 
							JwtKey = utils.RandString(32)
 | 
				
			||||||
 | 
							global.Log.Infof("config.yml未配置jwt.key, 随机生成key为: %s", JwtKey)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	// 使用自定义字符串加密 and get the complete encoded token as a string
 | 
						// 使用自定义字符串加密 and get the complete encoded token as a string
 | 
				
			||||||
	tokenString, err := token.SignedString([]byte(JwtKey))
 | 
						tokenString, err := token.SignedString([]byte(JwtKey))
 | 
				
			||||||
	biz.ErrIsNil(err, "token创建失败")
 | 
						biz.ErrIsNil(err, "token创建失败")
 | 
				
			||||||
 
 | 
				
			|||||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user