pcm-coordinator/pkg/scheduler/aiScheduler.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 scheduler

import (
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/pkg/response"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/models"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/scheduler/algorithm/providerPricing"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/scheduler/collector"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/scheduler/strategies"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils"
)

type AiScheduler struct {
	yamlString string
	collector  collector.ResourceCollector
}

func NewAiScheduler(val string) *AiScheduler {
	return &AiScheduler{yamlString: val}
}

func (as *AiScheduler) getNewStructForDb(task *response.TaskInfo, resource string, participantId int64) (interface{}, error) {
	ai := models.Ai{
		ParticipantId: participantId,
		TaskId:        task.TaskId,
		Status:        "Saved",
		YamlString:    as.yamlString,
	}
	utils.Convert(task.Metadata, &ai)
	return ai, nil
}

func (as *AiScheduler) pickOptimalStrategy() (strategies.Strategy, error) {
	//a, b := as.genTaskAndProviders()

	return nil, nil
}

func (as *AiScheduler) genTaskAndProviders() (*providerPricing.Task, []*providerPricing.Provider) {
	return nil, nil
}