JCS-pub/client/internal/config/config.go

package config

import (
	"gitlink.org.cn/cloudream/common/pkgs/distlock"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/common/pkgs/mq"
	cdssdk "gitlink.org.cn/cloudream/common/sdks/storage"
	"gitlink.org.cn/cloudream/common/utils/config"
	stgmodels "gitlink.org.cn/cloudream/storage2/common/models"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/connectivity"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/downloader"
	"gitlink.org.cn/cloudream/storage2/common/pkgs/downloader/strategy"
	agtrpc "gitlink.org.cn/cloudream/storage2/common/pkgs/grpc/agent"
)

type Config struct {
	Local            stgmodels.LocalMachineInfo `json:"local"`
	AgentGRPC        agtrpc.PoolConfig          `json:"agentGRPC"`
	Logger           logger.Config              `json:"logger"`
	RabbitMQ         mq.Config                  `json:"rabbitMQ"`
	DistLock         distlock.Config            `json:"distlock"`
	Connectivity     connectivity.Config        `json:"connectivity"`
	Downloader       downloader.Config          `json:"downloader"`
	DownloadStrategy strategy.Config            `json:"downloadStrategy"`
	StorageID        cdssdk.StorageID           `json:"storageID"`     // TODO 进行访问量统计时，当前客户端所属的存储ID。临时解决方案。
	AuthAccessKey    string                     `json:"authAccessKey"` // TODO 临时办法
	AuthSecretKey    string                     `json:"authSecretKey"`
	MaxHTTPBodySize  int64                      `json:"maxHttpBodySize"`
}

var cfg Config

// Init 初始化client
// TODO 这里的modeulName参数弄成可配置的更好
func Init() error {
	return config.DefaultLoad("client", &cfg)
}

func Cfg() *Config {
	return &cfg
}