package domain import ( "admin/apps/game/api" "admin/apps/game/domain/entity" "admin/apps/game/domain/projects" "admin/apps/game/domain/repo" "admin/apps/game/model" "admin/apps/game/model/dto" "admin/internal/consts" "admin/internal/errcode" "admin/lib/xlog" "fmt" "gorm.io/gorm" "strings" ) type CommonResourceService struct { projectRepo repo.IProjectRepo } func initCommonResourcesRepo(db *gorm.DB) { r(consts.ResourcesName_Project, "项目管理", repo.NewCommonResourceRepo(db, &model.Project{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) { serverRepo := r(consts.ResourcesName_Server, "服务器管理", repo.NewCommonResourceRepo(db, &model.Server{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) serverRepo.GlobalBtns = []*api.ResourceBtnInfo{ {Key: consts.BtnKeyGlobal_Server_DownAll, Name: "一键停服", BtnColorType: "info"}, {Key: consts.BtnKeyGlobal_Server_UpAll, Name: "一键起服", BtnColorType: "warning"}, } serverRepo.RowBtns = []*api.ResourceBtnInfo{ {Key: consts.BtnKeyRow_Server_Down, Name: "停服", BtnColorType: "info"}, {Key: consts.BtnKeyRow_Server_Up, Name: "起服", BtnColorType: "warning"}, } } r(consts.ResourcesName_Account, "账号列表", repo.NewCommonResourceRepo(db, &model.Account{}), ShowMethod_Get) // 账号管理不需要在后台读写数据,都是通过项目api拉 r(consts.ResourcesName_SupportAccount, "扶持账号", repo.NewCommonResourceRepo(db, &model.SupportAccount{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Delete) // 扶持账号,只是标记哪些账号是扶持好 r(consts.ResourcesName_Role, "角色列表", repo.NewCommonResourceRepo(db, &model.Role{}), ShowMethod_Get) // 角色管理不需要在后台读写数据,都是通过项目api拉 r(consts.ResourcesName_WhiteList, "白名单", repo.NewCommonResourceRepo(db, &model.WhiteList{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Delete) r(consts.ResourcesName_Ban, "封禁管理", repo.NewCommonResourceRepo(db, &model.Ban{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) r(consts.ResourcesName_MailRole, "个人邮件", repo.NewCommonResourceRepo(db, &model.RoleMail{}), ShowMethod_Get|ShowMethod_Post) // 个人邮件发放就没法撤回? r(consts.ResourcesName_MailGlobal, "全服邮件", repo.NewCommonResourceRepo(db, &model.GlobalMail{}), ShowMethod_Get|ShowMethod_Post) // 直接删除,别修改了,玩家收到的更乱 r(consts.ResourcesName_CDKey, "礼包码", repo.NewCommonResourceRepo(db, &model.CDKey{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) r(consts.ResourcesName_ItemBag, "道具礼包", repo.NewCommonResourceRepo(db, &model.ItemBag{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) r(consts.ResourcesName_Notice, "公告(暂无)", repo.NewCommonResourceRepo(db, &model.Notice{}), ShowMethod_Get|ShowMethod_Post|ShowMethod_Put|ShowMethod_Delete) r(consts.ResourcesName_DevicePush, "设备推送(暂无)", repo.NewCommonResourceRepo(db, &model.DevicePush{}), ShowMethod_Get) } func NewCommonResourceService(db *gorm.DB) *CommonResourceService { initCommonResourcesRepo(db) svc := &CommonResourceService{ projectRepo: repo.NewProjectRepo(db), } return svc } func (svc *CommonResourceService) List(projectId int, resource string, listParams *dto.CommonListReq) (int, []*dto.CommonDtoFieldDesc, []dto.CommonDtoValues, error) { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return 0, nil, nil, err } if resource != consts.ResourcesName_Project && !find { return 0, nil, nil, errcode.New(errcode.ServerError, "not found project %v db data", projectId) } rRepo := findCommResourceRepo(resource) totalCount, fieldsDescInfo, etList, err := rRepo.List(projectEt, listParams) if err != nil { return 0, nil, nil, err } retList := make([]dto.CommonDtoValues, 0, len(etList)) for _, v := range etList { retList = append(retList, v.ToCommonDto()) } // 执行各个项目特定的钩子方法 if hook, ok := projects.GetProjectResourceHook(projectEt, resource).(projects.IPostResourceOpListHook); ok { totalCount1, fieldsDescInfo1, retList1, err1 := hook.List(projectEt, resource, listParams, fieldsDescInfo, totalCount, retList) if err1 != nil { xlog.Warnf("list resource %v, hook return error:%v", resource, err1) return totalCount, fieldsDescInfo, retList, nil } return totalCount1, fieldsDescInfo1, retList1, nil } return totalCount, fieldsDescInfo, retList, nil } func (svc *CommonResourceService) GetById(projectId int, resource string, id int) ([]*dto.CommonDtoFieldDesc, dto.CommonDtoValues, *entity.CommonResource, bool, error) { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return nil, nil, nil, false, err } if resource != consts.ResourcesName_Project && !find { return nil, nil, nil, false, errcode.New(errcode.ServerError, "not found project %v db data", projectId) } fieldsDescInfo, et, find, err := findCommResourceRepo(resource).GetById(projectEt, id) if err != nil { return nil, nil, nil, false, err } return fieldsDescInfo, et.ToCommonDto(), et, find, nil } func (svc *CommonResourceService) Create(projectId int, resource string, dtoObj dto.CommonDtoValues) (dto.CommonDtoValues, error) { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return nil, err } if resource != consts.ResourcesName_Project && !find { return nil, errcode.New(errcode.ServerError, "not found project %v db data", projectId) } createOne := func(obj dto.CommonDtoValues) (dto.CommonDtoValues, error) { et, err := findCommResourceRepo(resource).Create(projectEt, resource, obj) if err != nil { return nil, err } // 返回新的实体,插入数据库之后会填入数据库id,所以要返给前端刷新id数据 // 这里转换一次新的数据传输对象,因为上一步走了创建,会给dto分配id newObj := et.ToCommonDto() // 执行各个项目特定的钩子方法 if hook, ok := projects.GetProjectResourceHook(projectEt, resource).(projects.IPostResourceOpCreateHook); ok { err = hook.Create(projectEt, resource, newObj) if err != nil { return nil, err } return newObj, nil } return newObj, nil } var newDtoObj dto.CommonDtoValues if resource == consts.ResourcesName_SupportAccount { account, findAccount := dtoObj["Account"] if findAccount { for _, ac := range strings.Split(account.(string), ",") { dtoObj["Account"] = ac tmpObj, err := createOne(dtoObj) if err != nil { xlog.Errorf("create support account %v db error:%v", ac, err) err = nil // 忽略多个插入的报错 } else { newDtoObj = tmpObj } } } else { return nil, errcode.New(errcode.ParamsInvalid, "account empty:%+v", dtoObj) } } else { newDtoObj, err = createOne(dtoObj) } return newDtoObj, err } func (svc *CommonResourceService) Edit(projectId int, resource string, dtoObj dto.CommonDtoValues) error { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return err } if resource != consts.ResourcesName_Project && !find { return errcode.New(errcode.ServerError, "not found project %v db data", projectId) } err = findCommResourceRepo(resource).Edit(projectEt, dtoObj) if err != nil { return err } // 执行各个项目特定的钩子方法 if hook, ok := projects.GetProjectResourceHook(projectEt, resource).(projects.IPostResourceOpEditHook); ok { err = hook.Edit(projectEt, resource, dtoObj) if err != nil { return err } return nil } return nil } func (svc *CommonResourceService) Delete(projectId int, resource string, id int) (*entity.CommonResource, error) { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return nil, err } if resource != consts.ResourcesName_Project && !find { return nil, errcode.New(errcode.ServerError, "not found project %v db data", projectId) } oldEt, find, err := findCommResourceRepo(resource).Delete(projectEt, id) if err != nil { return nil, err } // 执行各个项目特定的钩子方法 if find { if hook, ok := projects.GetProjectResourceHook(projectEt, resource).(projects.IPostResourceOpDeleteHook); ok { err = hook.Delete(projectEt, resource, oldEt.ToCommonDto()) if err != nil { return oldEt, err } return oldEt, nil } } return oldEt, nil } func (svc *CommonResourceService) RowsSelection(projectId int, resourceName string, params *dto.CommonRowsSelectionReq) (*dto.CommonRowsSelectionRsp, error) { _, projectEt, find, err := svc.projectRepo.GetById(projectId) if err != nil { return nil, err } if resourceName != consts.ResourcesName_Project && !find { return nil, errcode.New(errcode.ServerError, "not found project %v db data", projectId) } if hook, ok := projects.GetProjectResourceHook(projectEt, resourceName).(projects.IPostResourceOpRowsHook); ok { return hook.RowsSelection(projectEt, resourceName, params.BtnKey, params.Rows) } return nil, nil } func (svc *CommonResourceService) GetSupportResourcesList(permissions []string) []*api.ResourceInitInfo { list := make([]*api.ResourceInitInfo, 0, len(commResourcesRepo)) for _, v := range commResourcesRepo { list = append(list, &api.ResourceInitInfo{ Resource: v.Resource, Desc: v.Desc, ShowMethods: v.ShowMethods, GlobalBtns: v.GlobalBtns, RowBtns: v.RowBtns, }) } return list } const ( ShowMethod_Get = 1 ShowMethod_Post = 1 << 1 ShowMethod_Put = 1 << 2 ShowMethod_Delete = 1 << 3 ) type resourceRepoInfo struct { Resource string Desc string Repo repo.ICommonResourceRepo ShowMethods []string showMethods int GlobalBtns []*api.ResourceBtnInfo RowBtns []*api.ResourceBtnInfo } var commResourcesRepo = make([]*resourceRepoInfo, 0) func r(resource, desc string, repo repo.ICommonResourceRepo, showMethods int) *resourceRepoInfo { curRepo := &resourceRepoInfo{ Resource: resource, Desc: desc, Repo: repo, ShowMethods: make([]string, 0), } if showMethods&ShowMethod_Get == ShowMethod_Get { curRepo.ShowMethods = append(curRepo.ShowMethods, "get") } if showMethods&ShowMethod_Post == ShowMethod_Post { curRepo.ShowMethods = append(curRepo.ShowMethods, "post") } if showMethods&ShowMethod_Put == ShowMethod_Put { curRepo.ShowMethods = append(curRepo.ShowMethods, "put") } if showMethods&ShowMethod_Delete == ShowMethod_Delete { curRepo.ShowMethods = append(curRepo.ShowMethods, "delete") } commResourcesRepo = append(commResourcesRepo, curRepo) return curRepo } func findCommResourceRepo(resource string) repo.ICommonResourceRepo { for _, v := range commResourcesRepo { if v.Resource == resource { return v.Repo } } panic(fmt.Errorf("not found Resource:%v", resource)) return nil }