60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
|
package hello
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
|
|||
|
|
"github.com/infraboard/mcube/v2/ioc"
|
|||
|
|
grpc_server "github.com/infraboard/mcube/v2/ioc/config/grpc"
|
|||
|
|
"gitlab.com/go-course-project/go17/skills/grpc/service"
|
|||
|
|
"google.golang.org/grpc"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
ioc.Controller().Registry(&HelloServiceServer{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type HelloServiceServer struct {
|
|||
|
|
service.UnimplementedHelloServiceServer
|
|||
|
|
ioc.ObjectImpl
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *HelloServiceServer) Name() string {
|
|||
|
|
return "hello"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *HelloServiceServer) Init() error {
|
|||
|
|
// 需要拿到全局的 grpc server对象, ioc
|
|||
|
|
service.RegisterHelloServiceServer(grpc_server.Get().Server(), &HelloServiceServer{})
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (HelloServiceServer) Hello(ctx context.Context, req *service.HelloRequest) (*service.HelloResponse, error) {
|
|||
|
|
return &service.HelloResponse{
|
|||
|
|
Message: "hello, " + req.MyName,
|
|||
|
|
}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (HelloServiceServer) Chat(stream grpc.BidiStreamingServer[service.ChatRequest, service.ChatResponse]) error {
|
|||
|
|
// 1. 获取用户的请求，然后处理, 处理完成后，通过stream 返回给客户端
|
|||
|
|
for {
|
|||
|
|
// 接收一个请求
|
|||
|
|
req, err := stream.Recv()
|
|||
|
|
if err != nil {
|
|||
|
|
// 如果遇到io.EOF表示客户端流被关闭
|
|||
|
|
if err == io.EOF {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 处理请求
|
|||
|
|
fmt.Println(req.Message)
|
|||
|
|
stream.Send(&service.ChatResponse{
|
|||
|
|
Id: req.Id,
|
|||
|
|
IsSuccess: true,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|