pcm-coordinator/internal/scheduler/service/cloud_service.go

package service

import (
	"fmt"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/config"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/database"
	cloudservice "gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/cloud"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/scheduler/service/collector"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
	"strconv"
	"sync"
)

const (
	PcmK8s = "k8s"
)

type CloudService struct {
	CloudExecutorAdapterMap map[string]collector.CloudCollector
	Storage                 *database.CloudStorage
	LocalCache              map[string]interface{}
	Conf                    *config.Config
	TaskSyncLock            sync.Mutex
}

// NewCloudService 创建并初始化CloudService实例
func NewCloudService(conf *config.Config, storages *database.CloudStorage, localCache map[string]interface{}) (*CloudService, error) {
	cloudService := &CloudService{
		CloudExecutorAdapterMap: make(map[string]collector.CloudCollector),
		Storage:                 storages,
		LocalCache:              localCache,
		Conf:                    conf,
	}

	if err := cloudService.initAdapters(); err != nil {
		return nil, err
	}

	return cloudService, nil
}

// initAdapters 初始化所有适配器
func (s *CloudService) initAdapters() error {
	adapters, err := s.loadAdapters()
	if err != nil {
		return err
	}
	for _, adapter := range adapters {
		if err := s.processAdapter(*adapter); err != nil {
			return err
		}
	}

	return nil
}

// loadAdapters 从存储中加载适配器
func (s *CloudService) loadAdapters() ([]*types.AdapterInfo, error) {
	return s.Storage.GetAdaptersByType("0")
}

// processAdapter 处理单个适配器
func (s *CloudService) processAdapter(adapter types.AdapterInfo) error {
	if adapter.Id == "" {
		return nil
	}

	executor, err := s.createExecutor(adapter)
	if err != nil {
		return err
	}

	if executor != nil {
		s.CloudExecutorAdapterMap[adapter.Id] = executor
	}

	return nil
}

// createExecutor 根据适配器类型创建对应的执行器
func (s *CloudService) createExecutor(adapter types.AdapterInfo) (collector.CloudCollector, error) {
	switch adapter.Nickname {
	case PcmK8s:
		return s.CreateK8sExecutor(adapter)
	// 可以在这里添加其他类型的适配器
	default:
		return nil, nil // 或者返回错误，取决于业务需求
	}
}

// CreateSlurmExecutor 创建Slurm执行器
func (s *CloudService) CreateK8sExecutor(adapter types.AdapterInfo) (collector.CloudCollector, error) {
	id, err := strconv.ParseInt(adapter.Id, 10, 64)
	if err != nil {
		return nil, fmt.Errorf("failed to parse adapter ID %s: %v", adapter.Id, err)
	}
	return cloudservice.NewCloud(adapter.Server, id, adapter.Nickname), nil
}