pw-protobuf/client/base.go

package client

import (
	"git.ikuban.com/server/kratos-nacos/registry"
	"git.ikuban.com/server/pw-protobuf/api/basics"
	"git.ikuban.com/server/pw-protobuf/api/check"
	"git.ikuban.com/server/pw-protobuf/api/statistics"
	"github.com/go-kratos/kratos/v2/log"
)

func NewStatisticsClient(r *registry.Registry, logger log.Logger) statistics.StatisticsClient {
	conn, err := getDialInsecure(r, logger, "discovery://pw/pw-statistics")
	if err != nil {
		panic(err)
	}
	if conn == nil {
		return nil
	}
	client := statistics.NewStatisticsClient(conn)
	return client
}

func NewBasicsClient(r *registry.Registry, logger log.Logger) basics.BasicsClient {
	conn, err := getDialInsecure(r, logger, "discovery://pw/pw-basics")
	if err != nil {
		panic(err)
	}
	if conn == nil {
		return nil
	}
	client := basics.NewBasicsClient(conn)
	return client
}

func NewCheckClient(r *registry.Registry, logger log.Logger) check.CheckClient {
	conn, err := getDialInsecure(r, logger, "discovery://pw/pw-check")
	if err != nil {
		panic(err)
	}
	if conn == nil {
		return nil
	}
	client := check.NewCheckClient(conn)
	return client
}