290 lines
8.9 KiB
Go
290 lines
8.9 KiB
Go
package service
|
|
|
|
import (
|
|
"ego/internal/model"
|
|
"ego/internal/serializer"
|
|
"ego/internal/types"
|
|
"ego/pkg/logger"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// SysDeployProjectService 部署项目服务
|
|
type SysDeployProjectService struct {
|
|
Db *gorm.DB
|
|
}
|
|
|
|
// NewSysDeployProjectService 构建部署项目服务
|
|
func NewSysDeployProjectService(db *gorm.DB) *SysDeployProjectService {
|
|
return &SysDeployProjectService{
|
|
Db: db,
|
|
}
|
|
}
|
|
|
|
// Create 创建部署项目记录
|
|
func (s *SysDeployProjectService) Create(c *gin.Context) serializer.Response {
|
|
var deployProject model.SysDeployProject
|
|
if err := c.ShouldBind(&deployProject); err != nil {
|
|
logger.Error(c, "参数绑定失败!")
|
|
return serializer.ParamErr("参数绑定失败!", err)
|
|
}
|
|
|
|
// 验证必填字段
|
|
if deployProject.Domain == "" {
|
|
logger.Error(c, "域名不能为空!")
|
|
return serializer.ParamErr("域名不能为空!", fmt.Errorf("域名不能为空"))
|
|
}
|
|
|
|
// 检查域名是否已存在
|
|
var existingDeployProject model.SysDeployProject
|
|
if err := s.Db.Where("domain = ? AND del_flag = ?", deployProject.Domain, "0").First(&existingDeployProject).Error; err == nil {
|
|
logger.Error(c, "域名已存在!")
|
|
return serializer.ParamErr("域名已存在，请使用其他域名!", fmt.Errorf("域名已存在"))
|
|
}
|
|
|
|
// 生成部署ID
|
|
if id, err := SysSequenceServiceBuilder(deployProject.TableName()).GenerateId(); err == nil {
|
|
deployProject.DeployId = id
|
|
} else {
|
|
return serializer.DBErr("序列生成失败!", err)
|
|
}
|
|
|
|
// 设置默认值
|
|
now := time.Now()
|
|
deployProject.CreateTime = &now
|
|
deployProject.Status = model.DeployProjectStatusNormal
|
|
deployProject.DeployStatus = model.DeployStatusNotDeployed
|
|
deployProject.DelFlag = "0"
|
|
|
|
// 获取当前用户
|
|
if createBy := c.GetString("id"); createBy != "" {
|
|
deployProject.CreateBy = createBy
|
|
}
|
|
|
|
if err := s.Db.Create(&deployProject).Error; err != nil {
|
|
logger.Error(c, "创建部署项目记录失败!")
|
|
return serializer.DBErr("创建部署项目记录失败!", err)
|
|
}
|
|
|
|
return serializer.Succ("创建部署项目记录成功!", deployProject)
|
|
}
|
|
|
|
// GetByID 根据ID获取部署项目记录
|
|
func (s *SysDeployProjectService) GetByID(c *gin.Context) serializer.Response {
|
|
// 获取当前用户ID
|
|
currentUserId := c.GetString("id")
|
|
if currentUserId == "" {
|
|
return serializer.ParamErr("用户信息获取失败!", nil)
|
|
}
|
|
|
|
var deployProject model.SysDeployProject
|
|
if err := s.Db.Where("deploy_id = ? AND del_flag = ? AND create_by = ?", c.Param("id"), "0", currentUserId).First(&deployProject).Error; err != nil {
|
|
logger.Error(c, "获取部署项目记录失败!")
|
|
return serializer.DBErr("获取部署项目记录失败!", err)
|
|
}
|
|
return serializer.Succ("查询成功!", deployProject)
|
|
}
|
|
|
|
// UpdateByID 根据ID更新部署项目记录
|
|
func (s *SysDeployProjectService) UpdateByID(c *gin.Context) serializer.Response {
|
|
var deployProject model.SysDeployProject
|
|
if err := c.ShouldBind(&deployProject); err != nil {
|
|
logger.Error(c, "参数绑定失败!")
|
|
return serializer.ParamErr("参数绑定失败!", err)
|
|
}
|
|
|
|
id := deployProject.DeployId
|
|
if id == "" {
|
|
logger.Error(c, "id 不可为空!")
|
|
return serializer.ParamErr("id不可为空!", fmt.Errorf("id不可为空"))
|
|
}
|
|
|
|
// 获取当前用户ID
|
|
currentUserId := c.GetString("id")
|
|
if currentUserId == "" {
|
|
return serializer.ParamErr("用户信息获取失败!", nil)
|
|
}
|
|
|
|
// 检查权限：只能更新自己创建的数据
|
|
var existingDeployProject model.SysDeployProject
|
|
if err := s.Db.Where("deploy_id = ? AND del_flag = ? AND create_by = ?", id, "0", currentUserId).First(&existingDeployProject).Error; err != nil {
|
|
logger.Error(c, "部署项目记录不存在或无权限访问!")
|
|
return serializer.ParamErr("部署项目记录不存在或无权限访问!", err)
|
|
}
|
|
|
|
// 设置更新时间
|
|
now := time.Now()
|
|
deployProject.UpdateTime = &now
|
|
|
|
// 获取当前用户
|
|
deployProject.UpdateBy = currentUserId
|
|
|
|
if err := s.Db.Model(&deployProject).Where("deploy_id = ? AND del_flag = ? AND create_by = ?", id, "0", currentUserId).Updates(&deployProject).Error; err != nil {
|
|
logger.Error(c, "更新部署项目记录失败!")
|
|
return serializer.DBErr("更新部署项目记录失败!", err)
|
|
}
|
|
return serializer.Succ("更新部署项目记录成功!", deployProject)
|
|
}
|
|
|
|
// DeleteByID 根据ID删除部署项目记录
|
|
func (s *SysDeployProjectService) DeleteByID(c *gin.Context) serializer.Response {
|
|
id := c.Param("id")
|
|
if id == "" {
|
|
logger.Error(c, "id 不可为空!")
|
|
return serializer.ParamErr("id不可为空!", fmt.Errorf("id不可为空"))
|
|
}
|
|
|
|
// 获取当前用户ID
|
|
currentUserId := c.GetString("id")
|
|
if currentUserId == "" {
|
|
return serializer.ParamErr("用户信息获取失败!", nil)
|
|
}
|
|
|
|
// 软删除
|
|
data := map[string]any{
|
|
"del_flag": "1",
|
|
"update_time": time.Now(),
|
|
"update_by": currentUserId,
|
|
}
|
|
|
|
// 删除已经部署的文件夹
|
|
deployProject := model.SysDeployProject{}
|
|
if err := s.Db.Where("deploy_id = ? AND create_by = ?", id, currentUserId).First(&deployProject).Error; err != nil {
|
|
logger.Error(c, "获取部署项目记录失败或无权限访问!")
|
|
return serializer.DBErr("获取部署项目记录失败或无权限访问!", err)
|
|
}
|
|
|
|
// 删除 /home/:projectName
|
|
err := os.RemoveAll(filepath.Join("/home", deployProject.Domain))
|
|
if err != nil {
|
|
logger.Error(c, "删除部署文件夹失败!")
|
|
return serializer.DBErr("删除部署文件夹失败!", err)
|
|
}
|
|
|
|
// 同时软删除关联的文件记录
|
|
fileData := map[string]any{
|
|
"del_flag": "1",
|
|
"update_time": time.Now(),
|
|
"update_by": currentUserId,
|
|
}
|
|
if err := s.Db.Model(&model.SysDeployFile{}).Where("parent_id = ?", id).Updates(fileData).Error; err != nil {
|
|
logger.Error(c, "删除关联文件记录失败!")
|
|
return serializer.DBErr("删除关联文件记录失败!", err)
|
|
}
|
|
|
|
// 删除数据库记录
|
|
if err := s.Db.Model(&model.SysDeployProject{}).Where("deploy_id = ? AND create_by = ?", id, currentUserId).Updates(data).Error; err != nil {
|
|
logger.Error(c, "删除部署项目记录失败!")
|
|
return serializer.DBErr("删除部署项目记录失败!", err)
|
|
}
|
|
|
|
return serializer.Succ("删除部署项目记录成功!", nil)
|
|
}
|
|
|
|
// GetByCondition 条件查询部署项目记录
|
|
func (s *SysDeployProjectService) GetByCondition(c *gin.Context) serializer.Response {
|
|
var p types.Params
|
|
if err := c.ShouldBind(&p); err != nil {
|
|
return serializer.ParamErr("参数绑定失败!", err)
|
|
}
|
|
|
|
queryStr, args, err := p.ConvertToGormConditions()
|
|
if err != nil {
|
|
logger.Error(c, "参数绑定失败!")
|
|
return serializer.ParamErr("参数绑定失败!", err)
|
|
}
|
|
|
|
var total int64
|
|
var deployProjects []model.SysDeployProject
|
|
offset := (p.Page - 1) * p.Limit
|
|
|
|
// 获取当前用户ID
|
|
currentUserId := c.GetString("id")
|
|
if currentUserId == "" {
|
|
return serializer.ParamErr("用户信息获取失败!", nil)
|
|
}
|
|
|
|
// 构建基础查询
|
|
db := s.Db.Model(&model.SysDeployProject{})
|
|
|
|
// 如果有查询条件，添加条件
|
|
if queryStr != "" {
|
|
db = db.Where(queryStr, args...)
|
|
}
|
|
|
|
// 添加用户权限过滤：只能查询自己创建的数据
|
|
db = db.Where("create_by = ?", currentUserId)
|
|
|
|
// 排序
|
|
if p.Sort != "" {
|
|
db = db.Order(p.Sort)
|
|
} else {
|
|
db = db.Order("create_time DESC")
|
|
}
|
|
|
|
// 执行分页查询
|
|
if err := db.Where("del_flag = ?", "0").Offset(offset).Limit(p.Limit).Find(&deployProjects).Error; err != nil {
|
|
logger.Error(c, "获取部署项目记录失败!")
|
|
return serializer.DBErr("获取部署项目记录失败!", err)
|
|
}
|
|
|
|
// 执行总数查询
|
|
if err := db.Where("del_flag = ?", "0").Count(&total).Error; err != nil {
|
|
logger.Error(c, "获取部署项目记录总数失败!")
|
|
return serializer.DBErr("获取部署项目记录总数失败!", err)
|
|
}
|
|
|
|
return serializer.Succ("查询成功!", gin.H{
|
|
"total": total,
|
|
"items": deployProjects,
|
|
"page": p.Page,
|
|
"limit": p.Limit,
|
|
})
|
|
}
|
|
|
|
// UpdateDeployStatus 更新部署状态
|
|
func (s *SysDeployProjectService) UpdateDeployStatus(deployId, status, errorMsg string) error {
|
|
data := map[string]any{
|
|
"deploy_status": status,
|
|
"update_time": time.Now(),
|
|
}
|
|
|
|
if status == model.DeployStatusSuccess {
|
|
now := time.Now()
|
|
data["deploy_time"] = &now
|
|
}
|
|
|
|
if errorMsg != "" {
|
|
data["error_msg"] = errorMsg
|
|
}
|
|
|
|
return s.Db.Model(&model.SysDeployProject{}).Where("deploy_id = ?", deployId).Updates(data).Error
|
|
}
|
|
|
|
// UpdateAccessInfo 更新访问信息
|
|
func (s *SysDeployProjectService) UpdateAccessInfo(deployId string) error {
|
|
now := time.Now()
|
|
return s.Db.Model(&model.SysDeployProject{}).
|
|
Where("deploy_id = ?", deployId).
|
|
Updates(map[string]any{
|
|
"last_access_time": &now,
|
|
"access_count": gorm.Expr("access_count + 1"),
|
|
}).Error
|
|
}
|
|
|
|
// GetByDomain 根据域名获取部署项目记录
|
|
func (s *SysDeployProjectService) GetByDomain(domain string) (*model.SysDeployProject, error) {
|
|
var deployProject model.SysDeployProject
|
|
err := s.Db.Where("domain = ? AND status = ? AND del_flag = ?",
|
|
domain, model.DeployProjectStatusNormal, "0").First(&deployProject).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &deployProject, nil
|
|
}
|