floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added shuguangai cluster resources function 

Former-commit-id: 3f03d5c3a29e5e12f399e3da7087ab92e5f802e5
This commit is contained in:
tzwang 2024-01-31 17:45:14 +08:00
parent b04a3f1255
commit 3cc16100fa
3 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/internal/scheduler/service/aiService.go
View File

@ -19,7 +19,7 @@ var (
"Hanwuji": OCTOPUS, "Hanwuji": OCTOPUS,
"Suiyan": OCTOPUS, "Suiyan": OCTOPUS,
"Sailingsi": OCTOPUS, "Sailingsi": OCTOPUS,
"modelarts-CloudBrain2": MODELARTS, "Modelarts-CloudBrain2": MODELARTS,
"ShuguangAi": SHUGUANGAI, "ShuguangAi": SHUGUANGAI,
} }
) )

4
api/internal/scheduler/strategy/test/strategy_test.go
View File

@ -15,7 +15,7 @@ func TestReplication(t *testing.T) {
{Name: "test2", Participant_id: 2}, {Name: "test2", Participant_id: 2},
{Name: "test3", Participant_id: 3}, {Name: "test3", Participant_id: 3},
} }
res := []*collector.ResourceSpecs{ rsc := []*collector.ResourceSpecs{
{ {
ParticipantId: 1, ParticipantId: 1,
Name: "test1", Name: "test1",
@ -47,7 +47,7 @@ func TestReplication(t *testing.T) {
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
params := &param.Params{Resources: res} params := &param.Params{Resources: rsc}
repl := strategy.NewReplicationStrategy(&param.ReplicationParams{Params: params, Replicas: tt.replica}) repl := strategy.NewReplicationStrategy(&param.ReplicationParams{Params: params, Replicas: tt.replica})
schedule, err := repl.Schedule() schedule, err := repl.Schedule()
if err != nil { if err != nil {

23
api/internal/storeLink/shuguangai.go
View File

@ -22,6 +22,7 @@ import (
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/service/collector" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/service/collector"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc" "gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc"
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils" "gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils"
"strconv"
"strings" "strings"
) )
@ -173,6 +174,28 @@ func (o *ShuguangAi) QuerySpecs() (interface{}, error) {
} }
func (o *ShuguangAi) GetResourceSpecs() (*collector.ResourceSpecs, error) { func (o *ShuguangAi) GetResourceSpecs() (*collector.ResourceSpecs, error) {
userReq := &hpcAC.GetUserInfoReq{}
userinfo, err := o.svcCtx.ACRpc.GetUserInfo(o.ctx, userReq)
if err != nil {
return nil, err
}
limitReq := &hpcAC.QueueReq{}
_, err = o.svcCtx.ACRpc.QueryUserQuotasLimit(o.ctx, limitReq)
if err != nil {
return nil, err
}
diskReq := &hpcAC.ParaStorQuotaReq{}
_, err = o.svcCtx.ACRpc.ParaStorQuota(o.ctx, diskReq)
if err != nil {
return nil, err
}
balance, _ := strconv.ParseFloat(userinfo.Data.AccountBalance, 64)
_ = &collector.ResourceSpecs{
ParticipantId: o.participantId,
Name: o.platform,
Balance: balance,
}
return nil, nil return nil, nil
} }