lisi
/
busniess-user-center
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: lisi:9868e9a3a8374d8142c6be19e19f24d2c17242bb
Branches Tags
lisi:master
..
compare: lisi:819af3e7190eeef2023a6f1e0b58e7d99f48aa46
Branches Tags
lisi:master

No commits in common. "9868e9a3a8374d8142c6be19e19f24d2c17242bb" and "819af3e7190eeef2023a6f1e0b58e7d99f48aa46" have entirely different histories.
9868e9a3a8 ... 819af3e719

7 changed files with 11 additions and 490 deletions
Show Stats Expand all files Collapse all files

16
internal/models/organization/organization.go
View File

@ -1,16 +0,0 @@
package organization
import "time"
type Organization struct {
ID uint `json:"id"` // id
Name string `json:"name"` // 组织名
ParentID uint `json:"parent_id"` // 上级组织id
Sort uint `json:"sort"` // 层级序号
Status int `json:"status"` // 状态:0-无效,1-有效
Path string `json:"path"` // 全路径
CreatedOn time.Time `json:"created_on"` // 记录创建时间
Children []Organization `json:"children"`
}
type OrgTree []Organization

35
internal/models/organization/request.go
View File

@ -1,35 +0,0 @@
package organization
type CreateOrgReq struct {
Name string `json:"name" binding:"required"`
ParentId uint `json:"parent_id" binding:"required"`
Sort int `json:"sort" binding:"required"`
}
type DelOrgReq struct {
Id uint `json:"id" binding:"required"`
}
type GetOrgReq struct {
Id uint `from:"id" json:"id" binding:"required"`
}
type MoveOrgReq struct {
Id uint `json:"id" binding:"required"`
DestId uint `json:"dest_id" binding:"required"`
}
type SaveOrgReq struct {
Id uint `json:"id" binding:"required"`
Name string `json:"name" binding:"required"`
ParentId uint `json:"parent_id" binding:"required"`
Sort int `json:"sort" binding:"required"`
}
type DisableOrgReq struct {
Id uint `json:"id" binding:"required"`
}
type EnableOrgReq struct {
Id uint `json:"id" binding:"required"`
}

58
internal/repo/organization.go
View File

@ -2,24 +2,12 @@
package repo package repo
import ( import (
"context"
"time" "time"
"github.com/samber/do" "github.com/samber/do"
"gorm.io/gorm" "gorm.io/gorm"
) )
func init(){
do.Provide[OrganizationRepo](nil,NewOrganization)
}
type OrganizationStatus int
const(
OrganizationDisableStatus OrganizationStatus = 0
OrganizationEnableStatus OrganizationStatus = 1
)
// 组织表 // 组织表
type Organization struct { type Organization struct {
ID uint `gorm:"column:id;primary_key;AUTO_INCREMENT"` // id ID uint `gorm:"column:id;primary_key;AUTO_INCREMENT"` // id
@ -39,13 +27,7 @@ func (m *Organization) TableName() string {
} }
type OrganizationRepo interface{ type OrganizationRepo interface{
Create(ctx context.Context,org Organization)(id uint,err error)
Save(ctx context.Context,org Organization)(error)
GetById(ctx context.Context,id uint)(org Organization,err error)
DelById(ctx context.Context,id uint)(err error)
GetOrgs(ctx context.Context)(org []Organization,err error)
ExistOrgId(ctx context.Context,id uint)(bool,error)
SetStatus(ctx context.Context,id uint,status OrganizationStatus)error
} }
type orginizationRepo struct{ type orginizationRepo struct{
@ -56,42 +38,4 @@ func NewOrganization(i *do.Injector)(OrganizationRepo,error){
return &orginizationRepo{ return &orginizationRepo{
db:do.MustInvoke[*gorm.DB](i), db:do.MustInvoke[*gorm.DB](i),
},nil },nil
}
func (o *orginizationRepo)Create(ctx context.Context,org Organization)(id uint,err error){
err = o.db.Create(org).Error
id = org.ID
return
}
func (o *orginizationRepo)Save(ctx context.Context,org Organization)(error){
return o.db.Save(&org).Error
}
func (o *orginizationRepo)GetById(ctx context.Context,id uint)(org Organization,err error){
err = o.db.Where("id = ?",id).Take(&org).Error
return
}
func (o *orginizationRepo)DelById(ctx context.Context,id uint)(err error){
return o.db.Where("id = ?",id).Delete(&Organization{}).Error
}
func (o *orginizationRepo)GetOrgs(ctx context.Context)(orgs []Organization,err error){
err = o.db.Model(&Organization{}).Find(&orgs).Error
return
}
func (o *orginizationRepo)ExistOrgId(ctx context.Context,id uint)(bool,error){
var count int64 = 0
err := o.db.Where("id = ?",id).Count(&count).Error
if count >0{
return false,err
}
return true,err
}
func (o *orginizationRepo)SetStatus(ctx context.Context,id uint,status OrganizationStatus)error{
return o.db.Where("id = ?",id).Update("status",status).Error
} }

17
internal/service/organization/interface.go
View File

@ -1,17 +0,0 @@
package organization
import (
orgModel "busniess-user-center/internal/models/organization"
"context"
)
type OrganizationService interface {
CreateOrganization(ctx context.Context, info orgModel.CreateOrgReq) error
SaveOrganization(ctx context.Context, info orgModel.SaveOrgReq) error
DelOrganization(ctx context.Context, info orgModel.DelOrgReq) error
Organization(ctx context.Context, info orgModel.GetOrgReq) (orgModel.Organization, error)
MoveOrganization(ctx context.Context, info orgModel.MoveOrgReq) error
OrganizationTree(ctx context.Context) (orgModel.OrgTree, error)
DisableOrganization(ctx context.Context) error
EnableOrganization(ctx context.Context) error
}

130
internal/service/organization/organization.go
View File

@ -2,16 +2,11 @@ package organization
import ( import (
"busniess-user-center/config" "busniess-user-center/config"
orgModel "busniess-user-center/internal/models/organization"
"busniess-user-center/internal/repo" "busniess-user-center/internal/repo"
"busniess-user-center/pkg/redis" "busniess-user-center/pkg/redis"
"context"
"fmt"
"github.com/jinzhu/copier"
"github.com/samber/do" "github.com/samber/do"
"go.uber.org/zap" "go.uber.org/zap"
"gorm.io/gorm"
) )
func init() { func init() {
@ -19,124 +14,19 @@ func init() {
} }
type organizationService struct { type organizationService struct {
logger *zap.SugaredLogger logger *zap.SugaredLogger
orgRepo repo.OrganizationRepo appRepo repo.OrganizationRepo
redis *redis.Redis menuRepo repo.MenuRepo
conf *config.AppConfig redis *redis.Redis
conf *config.AppConfig
} }
func NewOrganizationService(i *do.Injector) (OrganizationService, error) { func NewOrganizationService(i *do.Injector) (OrganizationService, error) {
return &organizationService{ return &organizationService{
logger: do.MustInvoke[*zap.SugaredLogger](i), logger: do.MustInvoke[*zap.SugaredLogger](i),
orgRepo: do.MustInvoke[repo.OrganizationRepo](i), appRepo: do.MustInvoke[repo.OrganizationRepo](i),
redis: do.MustInvoke[*redis.Redis](i), menuRepo: do.MustInvoke[repo.MenuRepo](i),
conf: do.MustInvoke[*config.AppConfig](i), redis: do.MustInvoke[*redis.Redis](i),
conf: do.MustInvoke[*config.AppConfig](i),
}, nil }, nil
} }
func (o *organizationService) CreateOrganization(ctx context.Context, info *orgModel.CreateOrgReq) error {
if info.ParentId > 0 {
if exist, err := o.orgRepo.ExistOrgId(ctx, info.ParentId); err != nil {
return err
} else if !exist {
return fmt.Errorf("父组织%d不存在", info.ParentId)
}
}
dbOrg := repo.Organization{
Name: info.Name,
ParentID: info.ParentId,
Sort: uint(info.Sort),
Status: int(repo.OrganizationEnableStatus),
}
_, err := o.orgRepo.Create(ctx, dbOrg)
return err
}
func (o *organizationService) SaveOrganization(ctx context.Context, info *orgModel.SaveOrgReq) error {
dbOrg, err := o.orgRepo.GetById(ctx, info.Id)
if err != nil && err != gorm.ErrRecordNotFound {
return err
}
if err == gorm.ErrRecordNotFound {
return fmt.Errorf("组织不存在")
}
dbOrg.Name = info.Name
err = o.orgRepo.Save(ctx, dbOrg)
return err
}
func (o *organizationService) DelOrganization(ctx context.Context, info *orgModel.DelOrgReq) error {
// todo 判断权限
// todo 判断组织下面有没有成员
// todo 删除组织授权和相应的用户
return o.orgRepo.DelById(ctx, info.Id)
}
func (o *organizationService) Organization(ctx context.Context, info *orgModel.GetOrgReq) (org orgModel.Organization, err error) {
dbOrg, err := o.orgRepo.GetById(ctx, info.Id)
if err != nil && err != gorm.ErrRecordNotFound {
return
}
if err != gorm.ErrRecordNotFound {
err = fmt.Errorf("组织不存在")
return
}
copier.Copy(org, dbOrg)
return
}
func (o *organizationService) MoveOrganization(ctx context.Context, info *orgModel.MoveOrgReq) error {
// todo 判断权限
dbOrg, err := o.orgRepo.GetById(ctx, info.Id)
if err != nil && err != gorm.ErrRecordNotFound {
return err
}
if err == gorm.ErrRecordNotFound {
return fmt.Errorf("组织%d不存在", info.Id)
}
if dbOrg.ID == info.DestId || dbOrg.ParentID == info.DestId {
return nil
}
_, err = o.orgRepo.GetById(ctx, info.DestId)
if err != nil && err != gorm.ErrRecordNotFound {
return err
}
if err != gorm.ErrRecordNotFound {
return fmt.Errorf("目标组织%d不存在", info.DestId)
}
dbOrg.ParentID = info.DestId
return o.orgRepo.Save(ctx, dbOrg)
}
func (o *organizationService) OrganizationTree(ctx context.Context) (orgModel.OrgTree, error) {
dbOrgs, err := o.orgRepo.GetOrgs(ctx)
if err != nil {
return nil, err
}
tree := getOrgTree(dbOrgs)
return tree, nil
}
func (o *organizationService) DisableOrganization(ctx context.Context, info orgModel.DisableOrgReq) error {
// todo 判断有没有权限
// 判断是否成员
return o.orgRepo.SetStatus(ctx, info.Id, repo.OrganizationDisableStatus)
}
func (o *organizationService) EnableOrganization(ctx context.Context, info orgModel.EnableOrgReq) error {
// todo 判断有没有权限
return o.orgRepo.SetStatus(ctx, info.Id, repo.OrganizationEnableStatus)
}

171
internal/service/organization/util.go
View File

@ -1,171 +0,0 @@
package organization
import (
orgModel "busniess-user-center/internal/models/organization"
"busniess-user-center/internal/repo"
"sort"
)
func getOrgParentsWihtSelf(orgId uint, allOrgs []repo.Organization) (orgs []repo.Organization) {
orgMap := make(map[uint]repo.Organization, 0)
orgChildMap := make(map[uint][]repo.Organization, 0)
for _, org := range allOrgs {
if org.Status == 1 {
orgMap[org.ID] = org
orgChildMap[org.ParentID] = append(orgChildMap[org.ParentID], org)
}
}
orgs = getParents(orgId, orgMap)
if org, ok := orgMap[orgId]; ok {
orgs = append(orgs, org)
}
return
}
func getOrgParents(orgId uint, allOrgs []repo.Organization) (orgs []repo.Organization) {
orgMap := make(map[uint]repo.Organization, 0)
orgChildMap := make(map[uint][]repo.Organization, 0)
for _, org := range allOrgs {
if org.Status == 1 {
orgMap[org.ID] = org
orgChildMap[org.ParentID] = append(orgChildMap[org.ParentID], org)
}
}
orgs = getParents(orgId, orgMap)
return
}
func getParents(orgId uint, orgParentMap map[uint]repo.Organization) (orgs []repo.Organization) {
org, ok := orgParentMap[orgId]
if !ok {
return
}
if org.ParentID == 0 {
return
}
parentOrg, ok := orgParentMap[org.ParentID]
if ok {
orgs = append(orgs, parentOrg)
orgs = append(orgs, getParents(parentOrg.ID, orgParentMap)...)
}
return
}
func getOrgMap(allOrgs []repo.Organization) map[uint]repo.Organization {
orgMap := make(map[uint]repo.Organization, 0)
for _, org := range allOrgs {
if org.Status == 1 {
orgMap[org.ID] = org
}
}
return orgMap
}
func getOrgChildrenWithSelf(orgId uint, allOrgs []repo.Organization) (orgs []repo.Organization) {
orgMap := make(map[uint]repo.Organization, 0)
orgChildMap := make(map[uint][]repo.Organization, 0)
for _, org := range allOrgs {
if org.Status == 1 {
orgMap[org.ID] = org
orgChildMap[org.ParentID] = append(orgChildMap[org.ParentID], org)
}
}
orgs = getChildren(orgId, orgChildMap)
if org, ok := orgMap[orgId]; ok {
orgs = append(orgs, org)
}
return
}
func getOrgChildren(orgId uint, allOrgs []repo.Organization) (orgs []repo.Organization) {
orgMap := make(map[uint]repo.Organization, 0)
orgChildMap := make(map[uint][]repo.Organization, 0)
for _, org := range allOrgs {
if org.Status == 1 {
orgMap[org.ID] = org
orgChildMap[org.ParentID] = append(orgChildMap[org.ParentID], org)
}
}
orgs = getChildren(orgId, orgChildMap)
return
}
func getChildren(orgId uint, orgChildrenMap map[uint][]repo.Organization) (orgs []repo.Organization) {
if children, ok := orgChildrenMap[orgId]; ok {
for _, child := range children {
orgs = append(orgs, getChildren(child.ID, orgChildrenMap)...)
orgs = append(orgs, child)
}
}
return
}
func getOrgTree(orgs []repo.Organization) orgModel.OrgTree {
// top
topOrgs := []orgModel.Organization{}
parentMap := make(map[uint][]orgModel.Organization)
for _, org := range orgs {
mOrg := convertOrgDTM(org)
if org.ParentID == 0 {
topOrgs = append(topOrgs, mOrg)
}
if _, ok := parentMap[org.ID]; !ok {
parentMap[org.ID] = []orgModel.Organization{mOrg}
} else {
parentMap[org.ID] = append(parentMap[org.ID], mOrg)
}
}
for _, subOrgs := range parentMap {
for _, org := range subOrgs {
children := parentMap[org.ID]
sort.Slice(children, func(i, j int) bool {
return children[i].Sort < children[j].Sort
})
org.Children = children
}
}
for _, org := range topOrgs {
children := parentMap[org.ID]
sort.Slice(children, func(i, j int) bool {
return children[i].Sort < children[j].Sort
})
org.Children = children
}
sort.Slice(topOrgs, func(i, j int) bool {
return topOrgs[i].Sort < topOrgs[j].Sort
})
return topOrgs
}
func convertOrgDTM(org repo.Organization) orgModel.Organization {
return orgModel.Organization{
ID: org.ID, // id
Name: org.Name, // 组织名
ParentID: org.ParentID, // 上级组织id
Sort: org.Sort, // 层级序号
Status: org.Status, // 状态:0-无效,1-有效
Path: org.Path, // 全路径
CreatedOn: org.CreatedOn, // 记录创建时间
}
}

74
server/organization/organization.go
View File

@ -1,74 +0,0 @@
package organization
import (
"context"
orgModel "busniess-user-center/internal/models/organization"
organizationService "busniess-user-center/internal/service/organization"
ginUtil "busniess-user-center/pkg/utils/gin"
"github.com/gin-gonic/gin"
"github.com/samber/do"
"go.uber.org/zap"
)
func init() {
do.Provide(nil, NewOrganizationServer)
}
type OrganizationServer struct {
organizationService organizationService.OrganizationService
logger *zap.SugaredLogger
}
func NewOrganizationServer(i *do.Injector) (*OrganizationServer, error) {
return &OrganizationServer{
organizationService: do.MustInvoke[organizationService.OrganizationService](i),
logger: do.MustInvoke[*zap.SugaredLogger](i),
}, nil
}
func RegisterRoute(api *gin.RouterGroup) {
server := do.MustInvoke[*OrganizationServer](nil)
api.POST("/create", ginUtil.WrapNoRsp(server.Create))
api.POST("/save", ginUtil.WrapNoRsp(server.SaveOrganization))
api.POST("/delete", ginUtil.WrapNoRsp(server.DelOrganization))
api.GET("/get", ginUtil.Wrap(server.Organization))
api.POST("/move", ginUtil.WrapNoRsp(server.MoveOrganization))
api.GET("/orgs", ginUtil.WrapNoReq(server.OrganizationTree))
api.POST("/disenable", ginUtil.WrapNo(server.DisableOrganization))
api.POST("/enable", ginUtil.WrapNo(server.EnableOrganization))
}
func (u *OrganizationServer) Create(ctx context.Context, req *orgModel.CreateOrgReq) (err error) {
// 转换dto
return u.organizationService.CreateOrganization(ctx, req)
}
func (u *OrganizationServer) SaveOrganization(ctx context.Context, info *orgModel.SaveOrgReq) error {
return u.organizationService.SaveOrganization(ctx, info)
}
func (u *OrganizationServer) DelOrganization(ctx context.Context, info *orgModel.DelOrgReq) error {
return u.organizationService.DelOrganization(ctx, info)
}
func (u *OrganizationServer) Organization(ctx context.Context, info *orgModel.GetOrgReq) (orgModel.Organization, error) {
return u.organizationService.Organization(ctx, info)
}
func (u *OrganizationServer) MoveOrganization(ctx context.Context, info *orgModel.MoveOrgReq) error {
return u.organizationService.MoveOrganization(ctx, info)
}
func (u *OrganizationServer) OrganizationTree(ctx context.Context) (orgModel.OrgTree, error) {
return u.organizationService.OrganizationTree(ctx)
}
func (u *OrganizationServer) DisableOrganization(ctx context.Context) error {
return u.organizationService.DisableOrganization(ctx)
}
func (u *OrganizationServer) EnableOrganization(ctx context.Context) error {
return u.organizationService.EnableOrganization(ctx)
}