openbilibili/app/service/live/live_data/api/liverpc/client.go

// Code generated by liverpcgen, DO NOT EDIT.
// source: *.proto files under this directory
// If you want to change this file, Please see README in go-common/app/tool/liverpc/protoc-gen-liverpc/
package liverpc

import (
	"go-common/app/service/live/live_data/api/liverpc/v0"
	"go-common/app/service/live/live_data/api/liverpc/v1"
	"go-common/library/net/rpc/liverpc"
)

// Client that represents a liverpc live_data service api
type Client struct {
	cli *liverpc.Client
	// V0Message presents the controller in liverpc
	V0Message v0.Message
	// V1Record presents the controller in liverpc
	V1Record v1.Record
}

// DiscoveryAppId the discovery id is not the tree name
var DiscoveryAppId = "live.livedata"

// New a Client that represents a liverpc live.livedata service api
// conf can be empty, and it will use discovery to find service by default
// conf.AppID will be overwrite by a fixed value DiscoveryAppId
// therefore is no need to set
func New(conf *liverpc.ClientConfig) *Client {
	if conf == nil {
		conf = &liverpc.ClientConfig{}
	}
	conf.AppID = DiscoveryAppId
	var realCli = liverpc.NewClient(conf)
	cli := &Client{cli: realCli}
	cli.clientInit(realCli)
	return cli
}

func (cli *Client) GetRawCli() *liverpc.Client {
	return cli.cli
}

func (cli *Client) clientInit(realCli *liverpc.Client) {
	cli.V0Message = v0.NewMessageRpcClient(realCli)
	cli.V1Record = v1.NewRecordRpcClient(realCli)
}