pw-protobuf/client/chat.go

package client

import (
	"git.ikuban.com/server/kratos-nacos/registry"
	"git.ikuban.com/server/pw-protobuf/api/chat"
	"git.ikuban.com/server/pw-protobuf/api/scrip"
	"git.ikuban.com/server/pw-protobuf/api/websocket"
	"github.com/go-kratos/kratos/v2/log"
)

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

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

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