package grpcclient

import (
	"git.hlsq.asia/alex/service-common/discover"
	"git.hlsq.asia/alex/service-common/discover/common"
	"git.hlsq.asia/alex/service-common/net/grpc/resolver"
	"git.hlsq.asia/alex/service-common/proto/rs/rspb"
)

func SceneNewClient(sid ...int64) (rspb.SceneClient, error) {
	c, err := discover.FindServer(common.KeyDiscoverScene, sid...)
	if err != nil {
		return nil, err
	}
	return rspb.NewSceneClient(c), nil
}

func SceneNewClientLB() (rspb.SceneClient, error) {
	c, err := resolver.GetGrpcClientConn("etcd:///" + common.KeyDiscoverServiceNameScene)
	if err != nil {
		return nil, err
	}
	return rspb.NewSceneClient(c), nil
}

func SceneNewBroadcastClient() map[int64]rspb.SceneClient {
	clientM := make(map[int64]rspb.SceneClient)
	connM := discover.FindServerAll(common.KeyDiscoverScene)
	for sid, conn := range connM {
		clientM[sid] = rspb.NewSceneClient(conn)
	}
	return clientM
}