userCenter/rpc/internal/config/config.go

package config

import (
	"userCenter/entity"

	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/resource"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/zrpc"
)

type Config struct {
	zrpc.RpcServerConf
	FileSystemConf   zrpc.RpcClientConf
	CalleeId         string             // 服务名字
	ResourceEtcdConf zrpc.RpcClientConf //资源中台
	Node             int                // 节点
	Mysql            Mysql
	BaseMysql        Mysql
	Logx             logx.LogConf
	IsRun            bool   //定时任务是否开启
	CheckEntIsExpire string //
	DoMain           string
	RedisAddrees     []string
	RedisOutTime     int64
	InternalTime     int64
	Mongo            struct {
		Main *MongoStruct
	}
	BigMemberOff   bool
	CommonlySize   int64 //常用功能设置数量
	ManagerUserIds []string
	UserRolePower  []string
}
type MongoStruct struct {
	Address        string `json:"address"`
	Size           int    `json:"size"`
	DbName         string `json:"dbName"`
	UserName       string `json:"userName,optional"`
	Password       string `json:"password,optional"`
	Collection     string `json:"collection,optional"`
	CollectionBack string `json:"collectionBack,optional"`
	MaxOpenConns   int    `json:"maxOpenConns,optional"`
	MaxIdleConns   int    `json:"maxIdleConns,optional"`
}

var (
	ConfigJson        Config
	ManagerUserIdsMap = map[string]bool{}
)

type Mongodb struct {
	DbName  string `json:"dbName"`
	Size    int    `json:"size"`
	Address string `json:"address"`
}

type Mysql struct {
	DbName       string `json:"dbName"`
	Address      string `json:"address"`
	UserName     string `json:"userName"`
	PassWord     string `json:"passWord"`
	MaxOpenConns int    `json:"maxOpenConns"`
	MaxIdleConns int    `json:"maxIdleConns"`
}

func init() {
	conf.MustLoad("etc/usercenter.yaml", &ConfigJson)
	//初始化资源中台相关
	resourceClient := zrpc.MustNewClient(zrpc.RpcClientConf{
		Etcd: discov.EtcdConf{
			Hosts: ConfigJson.ResourceEtcdConf.Etcd.Hosts,
			Key:   ConfigJson.ResourceEtcdConf.Etcd.Key,
		},
	})
	entity.ResourceLib = resource.NewResource(resourceClient)
	//
	if len(ConfigJson.ManagerUserIds) > 0 {
		for _, v := range ConfigJson.ManagerUserIds {
			ManagerUserIdsMap[v] = true
		}
	}
}