增加SSH认证建议接口

This commit is contained in:
刘祥超
2021-08-14 21:33:17 +08:00
parent 5a22146309
commit 71d2671c04
2 changed files with 61 additions and 0 deletions

View File

@@ -265,3 +265,35 @@ func (this *NodeGrantService) TestNodeGrant(ctx context.Context, req *pb.TestNod
resp.IsOk = true
return resp, nil
}
// FindSuggestNodeGrants 查找集群推荐的认证
func (this *NodeGrantService) FindSuggestNodeGrants(ctx context.Context, req *pb.FindSuggestNodeGrantsRequest) (*pb.FindSuggestNodeGrantsResponse, error) {
_, err := this.ValidateAdmin(ctx, 0)
if err != nil {
return nil, err
}
var pbGrants = []*pb.NodeGrant{}
var tx = this.NullTx()
grantIds, err := models.SharedNodeLoginDAO.FindFrequentGrantIds(tx, req.NodeClusterId, req.NsClusterId)
if err != nil {
return nil, err
}
for _, grantId := range grantIds {
grant, err := models.SharedNodeGrantDAO.FindEnabledNodeGrant(tx, grantId)
if err != nil {
return nil, err
}
if grant != nil {
pbGrants = append(pbGrants, &pb.NodeGrant{
Id: int64(grant.Id),
Name: grant.Name,
Method: grant.Method,
Username: grant.Username,
Su: grant.Su == 1,
Description: grant.Description,
})
}
}
return &pb.FindSuggestNodeGrantsResponse{NodeGrants: pbGrants}, nil
}