30 lines
721 B
Go
30 lines
721 B
Go
package main
|
|
|
|
import (
|
|
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"math/rand"
|
|
"time"
|
|
)
|
|
|
|
// RandomNums 生成一定数量的随机数-服务端流式
|
|
func (s *Server) RandomNums(req *pb.RandomNumsRequest, stream pb.CalculateService_RandomNumsServer) error {
|
|
if req.Num <= 0 {
|
|
return status.Error(codes.InvalidArgument, "num must be greater than 0")
|
|
}
|
|
|
|
for i := 0; i < int(req.Num); i++ {
|
|
// 生成一个随机整数
|
|
s := rand.NewSource(time.Now().UnixNano())
|
|
|
|
if err := stream.Send(&pb.RandomNumsResponse{Num: s.Int63() % 1000}); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 间隔20ms
|
|
time.Sleep(20 * time.Millisecond)
|
|
}
|
|
return nil
|
|
}
|