biService/api/internal/svc/servicecontext.go

package svc

import (
	"app.yhyue.com/moapp/MessageCenter/rpc/messageclient"
	"bp.jydev.jianyu360.cn/BaseService/biService/api/internal/config"
	"bp.jydev.jianyu360.cn/BaseService/biService/rpc/biservice"
	"bp.jydev.jianyu360.cn/BaseService/resourceCenter/rpc/resource"
	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/usercenter"
	"github.com/zeromicro/go-zero/core/discov"
	"github.com/zeromicro/go-zero/zrpc"
)

type ServiceContext struct {
	Config            config.Config
	BiServiceRpc      biservice.BiService
	ResourceCenterRpc resource.Resource
	UserCenterRpc     usercenter.UserCenter
	MessageCenterRpc  messageclient.Message
}

func NewServiceContext(c config.Config) *ServiceContext {
	return &ServiceContext{
		Config:       c,
		BiServiceRpc: biservice.NewBiService(zrpc.MustNewClient(c.BiServiceRpc)),
		ResourceCenterRpc: resource.NewResource(zrpc.MustNewClient(zrpc.RpcClientConf{
			Etcd: discov.EtcdConf{
				Hosts: c.BiServiceRpc.Etcd.Hosts,
				//Hosts: []string{"172.31.31.205:2379"},
				Key: c.ResourceCenterKey,
			},
		})),
		UserCenterRpc: usercenter.NewUserCenter(zrpc.MustNewClient(zrpc.RpcClientConf{
			Etcd: discov.EtcdConf{
				Hosts: c.BiServiceRpc.Etcd.Hosts,
				//Hosts: []string{"172.31.31.205:2379"},
				Key: c.UserCenterKey,
			},
		})),
		MessageCenterRpc: messageclient.NewMessage(zrpc.MustNewClient(zrpc.RpcClientConf{
			Etcd: discov.EtcdConf{
				Hosts: c.BiServiceRpc.Etcd.Hosts,
				//Hosts: []string{"172.31.31.205:2379"},
				Key: c.MessageCenterKey,
			},
		})),
	}
}