JCC-CSScheduler/common/globals/pools.go

64 lines
1.6 KiB
Go

package schglb
import (
pcmsdk "gitlink.org.cn/cloudream/common/sdks/pcm"
schsdk "gitlink.org.cn/cloudream/common/sdks/scheduler"
"gitlink.org.cn/cloudream/common/sdks/storage/cdsapi"
uopsdk "gitlink.org.cn/cloudream/common/sdks/unifyops"
scmq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq"
advmq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/advisor"
cltmq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/collector"
exemq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/executor"
mgrmq "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/manager"
)
var ExecutorMQPool exemq.Pool
var AdvisorMQPool advmq.Pool
var CollectorMQPool cltmq.Pool
var ManagerMQPool mgrmq.Pool
func InitMQPool(cfg *scmq.Config) {
ExecutorMQPool = exemq.NewPool(cfg)
AdvisorMQPool = advmq.NewPool(cfg)
CollectorMQPool = cltmq.NewPool(cfg)
ManagerMQPool = mgrmq.NewPool(cfg)
}
var CloudreamStoragePool cdsapi.Pool
var CloudreamStorageConfig *cdsapi.Config
func InitCloudreamStoragePool(cfg *cdsapi.Config) {
CloudreamStorageConfig = cfg
CloudreamStoragePool = cdsapi.NewPool(cfg)
}
var UnifyOpsPool uopsdk.Pool
func IniUnifyOpsPool(cfg *uopsdk.Config) {
UnifyOpsPool = uopsdk.NewPool(cfg)
}
var PCMPool pcmsdk.Pool
func InitPCMPool(cfg *pcmsdk.Config) {
PCMPool = pcmsdk.NewPool(cfg)
}
var CDSRclone schsdk.Rclone
func InitRcloneConfig(rcloneID string, rcloneConfigID string) {
CDSRclone = schsdk.Rclone{
CDSRcloneID: rcloneID,
CDSRcloneConfigID: rcloneConfigID,
}
}
var InferencePlatform schsdk.InferencePlatform
func InitInferencePlatform(cfg schsdk.InferencePlatform) {
InferencePlatform = cfg
}