94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package schglb
|
|
|
|
import (
|
|
"gitlink.org.cn/cloudream/common/sdks/blockchain"
|
|
pcmsdk "gitlink.org.cn/cloudream/common/sdks/pcm"
|
|
pcmsch "gitlink.org.cn/cloudream/common/sdks/pcmscheduler"
|
|
schsdk "gitlink.org.cn/cloudream/common/sdks/scheduler"
|
|
"gitlink.org.cn/cloudream/common/sdks/storage/cdsapi"
|
|
uopsdk "gitlink.org.cn/cloudream/common/sdks/unifyops"
|
|
schmod "gitlink.org.cn/cloudream/scheduler/common/models"
|
|
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
|
|
|
|
var PCMSchePool pcmsch.Pool
|
|
|
|
//var UploaderPool uploadersdk.Pool
|
|
|
|
var BlockChainPool blockchain.Pool
|
|
|
|
var BlockChainConfig *blockchain.Config
|
|
|
|
var UploadCache map[string][]schmod.UploadStatus
|
|
|
|
func InitUploadCache() {
|
|
UploadCache = make(map[string][]schmod.UploadStatus)
|
|
}
|
|
|
|
func InitBlockChainPool(cfg *blockchain.Config) {
|
|
BlockChainPool = blockchain.NewPool(cfg)
|
|
BlockChainConfig = cfg
|
|
}
|
|
|
|
//func InitUploaderPool(cfg *uploadersdk.Config) {
|
|
// UploaderPool = uploadersdk.NewPool(cfg)
|
|
//}
|
|
|
|
func InitPCMSchePool(cfg *pcmsch.Config) {
|
|
PCMSchePool = pcmsch.NewPool(cfg)
|
|
}
|
|
|
|
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
|
|
}
|