505 lines
14 KiB
Go
505 lines
14 KiB
Go
package assets
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
err1 "errors"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/controller"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/model"
|
||
node2 "git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/model/app_manage/node"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/model/app_manage/workspace"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/pkg/errors"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/pkg/request"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/service"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/service/app_manage/assets"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/service/app_manage/common"
|
||
"git.inspur.com/sbg-jszt/cfn/cfn-schedule/internal/validator"
|
||
"github.com/gin-gonic/gin"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
type MachinePageBody struct {
|
||
PageInfo *request.PageInfo `json:"pageInfo" binding:"required"`
|
||
Name string `json:"name"`
|
||
Version string `json:"version"`
|
||
GroupName string `json:"groupName"`
|
||
}
|
||
|
||
type T struct {
|
||
GroupName string `json:"groupName"`
|
||
Version string `json:"version"`
|
||
Name string `json:"name"`
|
||
PageInfo struct {
|
||
CurrentPage int `json:"currentPage"`
|
||
Order string `json:"order"`
|
||
OrderField string `json:"order_field"`
|
||
PageSize int `json:"pageSize"`
|
||
} `json:"pageInfo"`
|
||
}
|
||
|
||
func PageMachine(c *gin.Context) {
|
||
var pageBody MachinePageBody
|
||
err := c.ShouldBindJSON(&pageBody)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.InvalidParameter, err, "请检查参数！")
|
||
return
|
||
}
|
||
//pageBody.PageInfo.Order = "create_time_millis desc"
|
||
n := node2.NewMachine()
|
||
fields := map[string]interface{}{}
|
||
if pageBody.Name != "" {
|
||
fields["name"] = pageBody.Name
|
||
}
|
||
if pageBody.Version != "" {
|
||
fields["cfn_agent_version"] = pageBody.Version
|
||
}
|
||
if pageBody.GroupName != "" {
|
||
fields["group_name"] = pageBody.GroupName
|
||
}
|
||
page := &model.PageConfig{}
|
||
page.Covert(pageBody.PageInfo)
|
||
result, err := n.Page(n, page, fields)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "内部错误！")
|
||
return
|
||
}
|
||
page.Data = result
|
||
controller.Success(c, page)
|
||
}
|
||
|
||
func EditMachine(c *gin.Context) {
|
||
n := &node2.MachineNode{}
|
||
err := validator.CheckPostParams(c, n)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.InvalidParameter, err, "请检查参数！")
|
||
return
|
||
}
|
||
if n.Id == "" {
|
||
n.Handler = n
|
||
fields := map[string]interface{}{}
|
||
fields["cfn_agent_url"] = n.CfnAgentUrl
|
||
list, err := n.GetAll(n, fields, nil)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
if len(list.([]node2.MachineNode)) > 0 {
|
||
controller.FailCode(c, errors.InvalidParameter, err, "对应的节点已经存在啦")
|
||
return
|
||
}
|
||
}
|
||
err = assets.TestMachineNode(n)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "节点连接失败，请检查节点是否在线")
|
||
return
|
||
}
|
||
status := 1
|
||
n.Status = status
|
||
// testHttpProxy()
|
||
if n.Id == "" {
|
||
m := 0
|
||
n.TransportMode = m
|
||
n.SetUserName(service.GetUserName(c))
|
||
_, err = n.Create(n)
|
||
} else {
|
||
n.SetModifyUserName(service.GetUserName(c))
|
||
column := []string{"name", "group_name", "cfn_agent_url", "cfn_agent_username", "cfn_agent_password", "template_node", "cfn_agent_timeout", "cfn_agent_http_proxy", "cfn_agent_http_proxy_type", "transport_encryption"}
|
||
_, err = n.Update(n, column)
|
||
}
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
controller.Success(c, "")
|
||
}
|
||
|
||
func DeleteMachine(c *gin.Context) {
|
||
id := c.PostForm("id")
|
||
if id == "" {
|
||
controller.FailCode(c, errors.InvalidParameter, nil, "请检查参数！")
|
||
return
|
||
}
|
||
m := &node2.MachineNode{}
|
||
m.Id = id
|
||
err := m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.InvalidParameter, nil, "请检查参数！")
|
||
return
|
||
}
|
||
n := node2.New()
|
||
fields := make(map[string]interface{})
|
||
fields["machine_id"] = id
|
||
list, err := n.GetAll(n, fields, nil)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
count := len(list.([]node2.Node))
|
||
if count > 0 {
|
||
controller.FailCode(c, errors.ServerError, err, "当前机器还关联"+strconv.Itoa(count)+"个节点，不能删除")
|
||
return
|
||
}
|
||
|
||
_, err = m.Delete(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
controller.Success(c, "")
|
||
}
|
||
|
||
func Distribute(c *gin.Context) {
|
||
ids := c.PostForm("ids")
|
||
if ids == "" {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
mids := strings.Split(ids, ",")
|
||
if len(mids) == 0 {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
workspaceId := c.PostForm("workspaceId")
|
||
w := &workspace.Workspace{}
|
||
w.Id = workspaceId
|
||
err := w.GetById(w)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "工作空间不存在！")
|
||
return
|
||
}
|
||
|
||
fields := map[string]interface{}{}
|
||
fields["workspace_id"] = workspaceId
|
||
for _, mid := range mids {
|
||
m := &node2.MachineNode{}
|
||
m.Id = mid
|
||
err = m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "机器不存在！")
|
||
return
|
||
}
|
||
n := node2.New()
|
||
fields["machine_id"] = mid
|
||
list, _ := n.GetAll(n, fields, nil)
|
||
if len(list.([]node2.Node)) > 0 {
|
||
continue
|
||
}
|
||
|
||
n = &node2.Node{}
|
||
n.MachineId = m.Id
|
||
n.WorkspaceId = workspaceId
|
||
n.Name = m.Name
|
||
n.OpenStatus = 1
|
||
n.Group = m.Group
|
||
n.Url = m.CfnAgentUrl
|
||
n.LoginName = m.CfnAgentUsername
|
||
n.LoginPwd = m.CfnAgentPassword
|
||
n.SetUserName(service.GetUserName(c))
|
||
_, err = n.Create(n)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
}
|
||
controller.Success(c, "")
|
||
}
|
||
|
||
func ListMachineGroup(c *gin.Context) {
|
||
m := &node2.MachineNode{}
|
||
result, _ := m.ListGroup()
|
||
controller.Success(c, result)
|
||
}
|
||
|
||
func ListTemplateNode(c *gin.Context) {
|
||
m := node2.NewMachine()
|
||
fields := map[string]interface{}{}
|
||
fields["template_node"] = 1
|
||
result, err := m.GetAll(m, fields, nil)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "数据库错误！")
|
||
return
|
||
}
|
||
controller.Success(c, result)
|
||
}
|
||
|
||
func Whitelist(c *gin.Context) {
|
||
machineId := c.PostForm("machineId")
|
||
m := &node2.MachineNode{}
|
||
m.Id = machineId
|
||
err := m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "没有对应的机器")
|
||
return
|
||
}
|
||
|
||
result := assets.GetAgentWhitelist(m)
|
||
controller.Success(c, result)
|
||
}
|
||
|
||
func SaveWhitelist(c *gin.Context) {
|
||
ids := c.PostForm("ids")
|
||
if ids == "" {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
mids := strings.Split(ids, ",")
|
||
if len(mids) == 0 {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
formData := make(map[string]string)
|
||
for key, value := range c.Request.PostForm {
|
||
formData[key] = value[0]
|
||
}
|
||
body, _ := json.Marshal(formData)
|
||
for _, mid := range mids {
|
||
m := &node2.MachineNode{}
|
||
m.Id = mid
|
||
err := m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "没有对应的机器")
|
||
return
|
||
}
|
||
resp, err := common.Request(m, "", common.WhitelistDirectory_Submit, nil, nil, bytes.NewReader(body))
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "分发节点失败")
|
||
return
|
||
}
|
||
if !resp.IsBusinessSuccess() {
|
||
controller.FailCode(c, errors.ServerError, err1.New(resp.Result.Msg), "分发节点失败")
|
||
return
|
||
}
|
||
}
|
||
controller.Success(c, "")
|
||
}
|
||
|
||
const config_data = `jpom:
|
||
# jpom 数据存储路径, 如果调试模式运行默认路径为【${user.home}/jpom/】,安装运行默认为jar包文件的父级
|
||
path:
|
||
# 集群配置
|
||
cluster:
|
||
# 集群Id，默认为 default 不区分大小写，只能是字母或者数字，长度小于 20
|
||
id: default
|
||
# 心跳监控时间 （需要大于零） 单位秒 最小配置 5 秒
|
||
heart-second: 30
|
||
node:
|
||
# 节点心跳监控时间 （需要大于零） 单位秒 最小配置 5秒
|
||
heart-second: 30
|
||
# 节点统计日志保留天数，如果小于等于 0 不自动删除
|
||
stat-log-keep-days: 3
|
||
# 上传文件的超时时间 单位秒,最短5秒钟
|
||
upload-file-timeout: 300
|
||
# 节点文件分片上传大小，单位 M，建议小于 5MB（需要考虑插件端上传文件大小限制）
|
||
upload-file-slice-size: 1
|
||
# 节点文件分片上传并发数,最小1 最大 服务端 CPU 核心数
|
||
upload-file-concurrent: 2
|
||
# web socket 消息最大长度
|
||
web-socket-message-size-limit: 5MB
|
||
system:
|
||
# cron 定时器是否开启匹配秒
|
||
timer-match-second: false
|
||
# 旧包文件保留个数
|
||
old-jars-count: 2
|
||
# 系统日志编码格式
|
||
log-charset: UTF-8
|
||
# 控制台编码格式
|
||
console-charset:
|
||
# 在线升级允许降级-操作
|
||
allowed-downgrade: false
|
||
user:
|
||
# 用户连续登录失败次数，超过此数将被限制登录
|
||
always-login-error: 5
|
||
# IP连续登录失败次数，超过此数将被限制登录
|
||
always-ip-login-error: 10
|
||
# 当ip连续登录失败，锁定对应IP时长，5h、1d
|
||
ip-error-lock-time: 5h
|
||
# 是否强制提醒用户开启 mfa
|
||
force-mfa: false
|
||
#登录token失效时间(单位：小时),默认为24
|
||
token-expired: 24
|
||
#登录token失效后自动续签时间（单位：分钟），默认为60，不自动续签
|
||
token-renewal: 60
|
||
# jwt 签名(加密)的key 长度建议控制到 16位
|
||
token-jwt-key:
|
||
web:
|
||
# 前端接口 超时时间 单位秒(最小 5 秒)
|
||
api-timeout: 20
|
||
# 禁用页面引导导航
|
||
disabled-guide: false
|
||
# 禁用登录图形验证码 （一般用于服务器没有字体或者开启了两部验证后才关闭图形验证码）
|
||
disabled-captcha: false
|
||
# 前端消息弹出位置，可选 topLeft topRight bottomLeft bottomRight
|
||
notification-placement:
|
||
# 前端消息传输需要编码或者加密 目前支持：NONE、BASE64
|
||
transport-encryption: NONE
|
||
# 查看日志时初始读取最后多少行（默认10，0不读取）
|
||
init-read-line: 10
|
||
db:
|
||
# 数据库默认 支持 ：H2、MYSQL
|
||
mode: H2
|
||
# 日志存储条数，将自动清理旧数据,配置小于等于零则不清理
|
||
log-storage-count: 10000
|
||
# H2 模式无需配置 mysql 配置 jdbc 地址
|
||
url:
|
||
# 数据库账号 默认 jpom
|
||
user-name: jpom
|
||
# 数据库密码 默认 jpom 如果自行配置请保证密码强度
|
||
user-pwd: jpom
|
||
# h2 数据库缓存大小 kilobyte 1KB 1,024 megabyte 1MB 1,048,576
|
||
cache-size: 50MB
|
||
# 自动备份间隔天数 小于等于 0 不自动备份
|
||
auto-backup-interval-day: 1
|
||
# 自动备份保留天数 小于等于 0，不自动删除自动备份数据
|
||
auto-backup-reserve-day: 5
|
||
# 数据库连接池相关配置
|
||
max-active: 500
|
||
initial-size: 10
|
||
max-wait: 10
|
||
min-idle: 1
|
||
# 控制台是否打印 sql 信息
|
||
show-sql: false
|
||
# 构建相关配置
|
||
build:
|
||
# 最多保存多少份历史记录
|
||
max-history-count: 1000
|
||
# 单个最多保存多少份历史记录
|
||
item-max-history-count: 50
|
||
# 构建命令是否检查 存在删除命令
|
||
check-delete-command: true
|
||
# 构建线程池大小,小于 1 则为不限制，默认大小为 5
|
||
pool-size: 5
|
||
# 构建任务等待数量，超过此数量将取消构建任务，值最小为 1
|
||
pool-wait-queue: 10
|
||
# 日志显示 压缩折叠显示进度比例 范围 1-100
|
||
log-reduce-progress-ratio: 5
|
||
file-storage:
|
||
# 文件中心存储路径
|
||
save-pah:
|
||
scan-static-dir-cron: 0 0/1 * * *
|
||
# 是否开启静态文件目录监听
|
||
watch-monitor-static-dir: true
|
||
# 监听静态文件目录层级
|
||
watch-monitor-max-depth: 5
|
||
assets:
|
||
# 监控线程池大小,小于等于0 为CPU核心数
|
||
monitor-pool-size: 0
|
||
# 监控任务等待数量，超过此数量将取消监控任务，值最小为 1
|
||
monitor-pool-wait-queue: 500
|
||
# ssh 资产
|
||
ssh:
|
||
# 监控频率
|
||
monitor-cron: 0 0/1 * * * ?
|
||
# 指定分组不启用监控功能（如果想禁用所有配置 * 即可）
|
||
disable-monitor-group-name:
|
||
- 禁用监控
|
||
# docker 资产
|
||
docker:
|
||
# 监控频率
|
||
monitor-cron: 0 0/1 * * * ?
|
||
server:
|
||
#运行端口号
|
||
port: 2122
|
||
servlet:
|
||
session:
|
||
cookie:
|
||
name: JPOMID-SERVER
|
||
timeout: 1H
|
||
encoding:
|
||
charset: UTF-8
|
||
force: true
|
||
enabled: true
|
||
forceRequest: true
|
||
forceResponse: true
|
||
compression:
|
||
# gzip 压缩
|
||
enabled: true
|
||
#mime-types: application/json,application/xml,text/html,text/xml,text/plain,application/javascript,image/png
|
||
min-response-size: 2048
|
||
tomcat:
|
||
uri-encoding: UTF-8
|
||
spring:
|
||
web:
|
||
resources:
|
||
static-locations: classpath:/dist/
|
||
cache:
|
||
period: 1D
|
||
mvc:
|
||
throw-exception-if-no-handler-found: true
|
||
log-request-details: true
|
||
servlet:
|
||
multipart:
|
||
# 上传文件大小限制
|
||
max-request-size: 2GB
|
||
max-file-size: 1GB
|
||
h2:
|
||
console:
|
||
# 是否开启 web 访问数据库（url: http://${ip}:${port}/h2-console）
|
||
enabled: false
|
||
# 是否允许远程访问（开启此配置有安全风险），默认为 false（当部署到服务器上之后，是否可以通过其他浏览器访问数据库）
|
||
settings:
|
||
web-allow-others: false
|
||
`
|
||
|
||
func ConfigData(c *gin.Context) {
|
||
machineId := c.PostForm("machineId")
|
||
m := &node2.MachineNode{}
|
||
m.Id = machineId
|
||
err := m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "没有对应的机器")
|
||
return
|
||
}
|
||
resp, err := common.Request3(m, "", common.SystemGetConfig, nil)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "分发节点失败")
|
||
return
|
||
}
|
||
if !resp.IsBusinessSuccess() {
|
||
controller.FailCode(c, errors.ServerError, err1.New(resp.Result.Msg), "分发节点失败")
|
||
return
|
||
}
|
||
controller.Success(c, resp.Result.Data)
|
||
}
|
||
|
||
// todo 暂不允许修改运行配置
|
||
func SaveConfig(c *gin.Context) {
|
||
ids := c.PostForm("ids")
|
||
if ids == "" {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
mids := strings.Split(ids, ",")
|
||
if len(mids) == 0 {
|
||
controller.Success(c, "")
|
||
return
|
||
}
|
||
|
||
content := c.PostForm("content")
|
||
restart := c.PostForm("restart")
|
||
|
||
formData := make(map[string]string)
|
||
formData["content"] = content
|
||
formData["restart"] = restart
|
||
|
||
for _, mid := range mids {
|
||
m := &node2.MachineNode{}
|
||
m.Id = mid
|
||
err := m.GetById(m)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "没有对应的机器")
|
||
return
|
||
}
|
||
resp, err := common.Request4(m, common.SystemSaveConfig, formData)
|
||
if err != nil {
|
||
controller.FailCode(c, errors.ServerError, err, "分发节点失败")
|
||
return
|
||
}
|
||
if !resp.IsSuccess() {
|
||
controller.FailCode(c, errors.ServerError, err1.New(resp.Result.Msg), "分发节点失败")
|
||
return
|
||
}
|
||
}
|
||
controller.Success(c, "修改成功")
|
||
}
|