feat: redis支持flushdb、菜单资源支持拖拽排序、禁用等

server/internal/sys/application/resource.go

@@ -4,8 +4,11 @@ import (
 	"mayfly-go/internal/sys/domain/entity"
 	"mayfly-go/internal/sys/domain/repository"
 	"mayfly-go/pkg/biz"
+	"mayfly-go/pkg/global"
 	"mayfly-go/pkg/model"
+	"mayfly-go/pkg/utils"
 	"strings"
+	"time"
 )
 
 type Resource interface {
@@ -17,6 +20,10 @@ type Resource interface {
 
 	Save(entity *entity.Resource)
 
+	ChangeStatus(resourceId uint64, status int8)
+
+	Sort(re *entity.Resource)
+
 	Delete(id uint64)
 
 	GetAccountResources(accountId uint64, toEntity any)
@@ -45,6 +52,7 @@ func (r *resourceAppImpl) GetByIdIn(ids []uint64, toEntity any, orderBy ...strin
 }
 
 func (r *resourceAppImpl) Save(resource *entity.Resource) {
+	// 更新操作
 	if resource.Id != 0 {
 		if resource.Code != "" {
 			oldRes := r.GetById(resource.Id, "Code")
@@ -54,17 +62,88 @@ func (r *resourceAppImpl) Save(resource *entity.Resource) {
 			}
 		}
 		model.UpdateById(resource)
-	} else {
-		if pid := resource.Pid; pid != 0 {
-			biz.IsTrue(r.GetById(uint64(pid)) != nil, "该父资源不存在")
-		}
-		// 默认启用状态
-		if resource.Status == 0 {
-			resource.Status = entity.ResourceStatusEnable
-		}
-		r.checkCode(resource.Code)
-		model.Insert(resource)
+		return
 	}
+
+	// 生成随机八位唯一标识符
+	ui := utils.RandString(8)
+	if pid := resource.Pid; pid != 0 {
+		pResource := r.GetById(uint64(pid))
+		biz.IsTrue(pResource != nil, "该父资源不存在")
+		resource.UiPath = pResource.UiPath + ui + entity.ResourceUiPathSp
+	} else {
+		resource.UiPath = ui + entity.ResourceUiPathSp
+	}
+	// 默认启用状态
+	if resource.Status == 0 {
+		resource.Status = entity.ResourceStatusEnable
+	}
+	r.checkCode(resource.Code)
+	resource.Weight = int(time.Now().Unix())
+	model.Insert(resource)
+}
+
+func (r *resourceAppImpl) ChangeStatus(resourceId uint64, status int8) {
+	resource := r.resourceRepo.GetById(resourceId)
+	biz.NotNil(resource, "资源不存在")
+	resource.Status = status
+	r.resourceRepo.UpdateByUiPathLike(resource)
+}
+
+func (r *resourceAppImpl) Sort(sortResource *entity.Resource) {
+	resource := r.resourceRepo.GetById(sortResource.Id)
+	biz.NotNil(resource, "资源不存在")
+	// 未改变父节点，则更新排序值即可
+	if sortResource.Pid == resource.Pid {
+		saveE := &entity.Resource{Weight: sortResource.Weight}
+		saveE.Id = sortResource.Id
+		r.Save(saveE)
+		return
+	}
+
+	// 若资源原本唯一标识路径为：xxxx/yyyy/zzzz/，则获取其父节点路径标识 xxxx/yyyy/ 与自身节点标识 zzzz/
+	splitStr := strings.Split(resource.UiPath, entity.ResourceUiPathSp)
+	// 获取 zzzz/
+	resourceUi := splitStr[len(splitStr)-2] + entity.ResourceUiPathSp
+	// 获取父资源路径 xxxx/yyyy/
+	var parentResourceUiPath string
+	if len(splitStr) > 2 {
+		parentResourceUiPath = strings.Split(resource.UiPath, resourceUi)[0]
+	} else {
+		parentResourceUiPath = resourceUi
+	}
+
+	newParentResourceUiPath := ""
+	if sortResource.Pid != 0 {
+		newParentResource := r.resourceRepo.GetById(uint64(sortResource.Pid))
+		biz.NotNil(newParentResource, "父资源不存在")
+		newParentResourceUiPath = newParentResource.UiPath
+	}
+
+	children := r.resourceRepo.GetChildren(resource.UiPath)
+	for _, v := range children {
+		if v.Id == sortResource.Id {
+			continue
+		}
+		updateUiPath := &entity.Resource{}
+		updateUiPath.Id = v.Id
+		if parentResourceUiPath == resourceUi {
+			updateUiPath.UiPath = newParentResourceUiPath + v.UiPath
+		} else {
+			updateUiPath.UiPath = strings.ReplaceAll(v.UiPath, parentResourceUiPath, newParentResourceUiPath)
+		}
+		r.Save(updateUiPath)
+	}
+
+	// 更新零值使用map，因为pid=0表示根节点
+	updateMap := map[string]interface{}{
+		"pid":     sortResource.Pid,
+		"weight":  sortResource.Weight,
+		"ui_path": newParentResourceUiPath + resourceUi,
+	}
+	condition := new(entity.Resource)
+	condition.Id = sortResource.Id
+	global.Db.Model(condition).Updates(updateMap)
 }
 
 func (r *resourceAppImpl) checkCode(code string) {
@@ -73,19 +152,18 @@ func (r *resourceAppImpl) checkCode(code string) {
 }
 
 func (r *resourceAppImpl) Delete(id uint64) {
-	// 查找pid == id的资源
-	condition := &entity.Resource{Pid: int(id)}
-	var resources resourceList
-	r.resourceRepo.GetResourceList(condition, &resources)
+	resource := r.resourceRepo.GetById(id)
+	biz.NotNil(resource, "资源不存在")
 
-	biz.IsTrue(len(resources) == 0, "请先删除该资源的所有子资源")
-	model.DeleteById(condition, id)
-	// 删除角色关联的资源信息
-	model.DeleteByCondition(&entity.RoleResource{ResourceId: id})
+	// 删除当前节点及其所有子节点
+	children := r.resourceRepo.GetChildren(resource.UiPath)
+	for _, v := range children {
+		r.resourceRepo.Delete(v.Id)
+		// 删除角色关联的资源信息
+		model.DeleteByCondition(&entity.RoleResource{ResourceId: v.Id})
+	}
 }
 
 func (r *resourceAppImpl) GetAccountResources(accountId uint64, toEntity any) {
 	r.resourceRepo.GetAccountResources(accountId, toEntity)
 }
-
-type resourceList []entity.Resource