pcm-coordinator/api/internal/pkg/scheduler/scheduler.go

package scheduler

import (
	"encoding/json"
	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/types"
	"gorm.io/gorm"
)

type scheduler struct {
	task            *types.TaskInfo
	participantIds  []int64
	scheduleService scheduleService
}

func NewScheduler(scheduleService scheduleService, val string) (*scheduler, error) {
	var task *types.TaskInfo
	err := json.Unmarshal([]byte(val), &task)
	if err != nil {
		return nil, errors.New("create scheduler failed : " + err.Error())
	}
	return &scheduler{task: task, scheduleService: scheduleService}, nil
}

func (s scheduler) MatchLabels(dbEngin *gorm.DB) {
	//if len(task.MatchLabels) != 0 {
	//	participantId, err := scheduler.MatchLabels(l.svcCtx.DbEngin, task)
	//	if err != nil {
	//		return err
	//	}
	//}
	var ids []int64
	count := 0
	for key := range s.task.MatchLabels {
		var participantId []int64
		dbEngin.Raw("select participant_id from sc_participant_label_info where `key` = ? and value = ?", key, s.task.MatchLabels[key]).Scan(&participantId)
		if count == 0 {
			ids = participantId
		}
		//if len(participantId) == 0 || len(ids) == 0 {
		//	return nil, nil
		//}
		ids = intersect(ids, participantId)
		count++
	}
	s.participantIds = micsSlice(ids, 1)
}

func (s scheduler) AssignAndSchedule() {

}

func (s scheduler) SaveToDb(dbEngin *gorm.DB) error {
	if len(s.participantIds) == 0 {
		return errors.New("participantIds 为空")
	}
	structForDb, err := s.scheduleService.getNewStructForDb(s.task, s.participantIds)
	if err != nil {
		return err
	}
	tx := dbEngin.Create(structForDb)
	if tx.Error != nil {
		logx.Error(tx.Error)
		return tx.Error
	}
	return nil
}