pcm-coordinator/api/internal/mqs/ScheduleAi.go

/*

 Copyright (c) [2023] [pcm]
 [pcm-coordinator] is licensed under Mulan PSL v2.
 You can use this software according to the terms and conditions of the Mulan PSL v2.
 You may obtain a copy of Mulan PSL v2 at:
          http://license.coscl.org.cn/MulanPSL2
 THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
 EITHER EXPaRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
 MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 See the Mulan PSL v2 for more details.

*/

package mqs

import (
	"context"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/schedulers"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/scheduler/service"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc"
)

/*
*
Listening to the payment flow status change notification message queue
*/
type AiQueue struct {
	ctx       context.Context
	svcCtx    *svc.ServiceContext
	scheduler *scheduler.Scheduler
}

func NewAiMq(ctx context.Context, svcCtx *svc.ServiceContext) *AiQueue {
	aiExecutorMap, aiCollectorMap := service.InitAiClusterMap(ctx, svcCtx)
	return &AiQueue{
		ctx:       ctx,
		svcCtx:    svcCtx,
		scheduler: scheduler.NewScheduler2(aiCollectorMap, nil, aiExecutorMap),
	}
}

func (l *AiQueue) Consume(val string) error {
	// 接受消息, 根据标签筛选过滤
	aiSchdl, _ := schedulers.NewAiScheduler(val, l.scheduler)

	// 调度算法
	err := l.scheduler.AssignAndSchedule(aiSchdl)
	if err != nil {
		return err
	}

	return nil
}