bl/logic/rpc.go

package main

import (
	"blazing/common/api"
	"context"
	"fmt"
	"io"
	"log"
	"net"
	"time"

	"google.golang.org/grpc"
)

const rpcaddr = "127.0.0.1:9999"

// Server
type Server struct {
	api.UnimplementedBothWayStreamServerServer
}

// DemoMethod 实现 proto 的方法
func (s *Server) Call(server api.BothWayStreamServer_CallServer) error {
	var count int
	// 启一个携程监听读事件
	go func() {
		for {
			p, err := server.Recv()
			if err != nil && err == io.EOF {
				return
			}
			if err != nil {
				fmt.Println("服务端接收错误", err)
				break
			}
			fmt.Println("服务端接收数据", p.GetKickResponse().UserId)
		}
	}()
	// 持续写事件
	for {
		count++

		err := server.Send(&api.GenericMessage{
			Payload: &api.GenericMessage_KickResponse{
				KickResponse: &api.KickRequest{
					UserId: int32(count),
				},
			},
		})
		if err != nil {
			break
		}
		time.Sleep(time.Second)
	}
	return nil
}

func rpcserver() {
	// 实例化一个 grpc 服务
	g := grpc.NewServer()
	s := new(Server)
	// 绑定
	api.RegisterBothWayStreamServerServer(g, s)

	// grpc 监听在 8888 端口
	l, err := net.Listen("tcp", rpcaddr)
	if err != nil {
		log.Fatal(err)
	}
	// 服务启动
	err = g.Serve(l)
	if err != nil {
		panic(err)
	}
}

func rpcclient() {
	connect, err := grpc.Dial(rpcaddr, grpc.WithInsecure())
	if err != nil {
		panic(err)
	}
	defer connect.Close()
	client := api.NewBothWayStreamServerClient(connect)
	stream, err := client.Call(context.Background())
	if err != nil {
		panic(err)
	}
	// 接收数据
	go func() {
		for {
			reply, err := stream.Recv()
			if err != nil {
				break
			}
			fmt.Println("客户端接收到的数据是:", reply.Payload)
		}
	}()

	// 发送数据
	var count int
	for {
		count++
		err := stream.Send(&api.GenericMessage{
			Payload: &api.GenericMessage_KickResponse{
				KickResponse: &api.KickRequest{
					UserId: int32(count),
				},
			},
		})
		if err != nil {
			log.Printf("发送失败: %v", err)
			break
		}
		time.Sleep(time.Second)
		// // 10 次后发送关闭
		// if count == 10 {
		// 	err := stream.CloseSend()
		// 	if err != nil {
		// 		break
		// 	}
		// }
	}
}