zishuo/oneterm
1
0
Fork 0
mirror of https://github.com/veops/oneterm.git synced 2025-10-31 10:56:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(backend): asset and account

Browse Source
This commit is contained in:
pycook
2025-05-05 11:15:42 +08:00
parent bf3bd1dc40
commit b12040b3c4
15 changed files with 825 additions and 341 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
backend/internal/api/controller/authorization.go
View File

@@ -159,7 +159,7 @@ func getNodeAssetAccoutIdsByAction(ctx context.Context, action string) (nodeIds,
}
nodes = lo.Filter(nodes, func(n *model.Node, _ int) bool { return lo.Contains(resIds, n.ResourceId) })
nodeIds = lo.Map(nodes, func(n *model.Node, _ int) int { return n.Id })
nodeIds, err = handleSelfChild(ctx, nodeIds...)
nodeIds, err = repository.HandleSelfChild(ctx, nodeIds...)
return
})
@@ -403,7 +403,7 @@ func hasAuthorization(ctx *gin.Context, sess *gsession.Session) (ok bool) {
ctx.Set(kAuthorizationIds, authIds)
nodeIds, assetIds, accountIds := getIdsByAuthorizationIds(ctx)
tmp, err := handleSelfChild(ctx, nodeIds...)
tmp, err := repository.HandleSelfChild(ctx, nodeIds...)
if err != nil {
logger.L().Error("", zap.Error(err))
return
@@ -421,3 +421,15 @@ func hasAuthorization(ctx *gin.Context, sess *gsession.Session) (ok bool) {
return lo.Contains(ids, sess.AssetId)
}
func getIdsByAuthorizationIds(ctx *gin.Context) (nodeIds, assetIds, accountIds []int) {
authorizationIds, ok := ctx.Value(kAuthorizationIds).([]*model.AuthorizationIds)
if !ok || len(authorizationIds) == 0 {
return
}
return assetService.GetIdsByAuthorizationIds(ctx, authorizationIds)
}
func getAssetIdsByNodeAccount(ctx context.Context, nodeIds, accountIds []int) ([]int, error) {
return assetService.GetAssetIdsByNodeAccount(ctx, nodeIds, accountIds)
}