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
}