openbilibili/app/interface/main/app-resource/dao/broadcast/broadcast.go

package broadcast

import (
	"context"
	"fmt"

	"go-common/app/interface/main/app-resource/conf"
	pb "go-common/app/service/main/broadcast/api/grpc/v1"
	warden "go-common/app/service/main/broadcast/api/grpc/v1"
	wardenclient "go-common/app/service/main/broadcast/api/grpc/v1"
	"go-common/library/log"
)

type Dao struct {
	c *conf.Config
	// grpc
	rpcClient pb.ZergClient
}

func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		c: c,
	}
	var err error
	if d.rpcClient, err = wardenclient.NewClient(c.BroadcastRPC); err != nil {
		panic(fmt.Sprintf("BroadcastRPC warden.NewClient error (%+v)", err))
	}
	return
}

// ServerList warden server list
func (d *Dao) ServerList(ctx context.Context, platform string) (res *warden.ServerListReply, err error) {
	arg := &warden.ServerListReq{
		Platform: platform,
	}
	if res, err = d.rpcClient.ServerList(ctx, arg); err != nil {
		log.Error("d.rpcClient.ServerList error(%v)", err)
		return
	}
	return
}