calculator-grpc流模式。

This commit is contained in:
guoqiang 2023-12-03 14:23:37 +08:00
parent a9ed403506
commit e3af30df41
13 changed files with 1070 additions and 1 deletions

View File

@ -14,5 +14,15 @@ greet-evans:
greet-evans-reflect: greet-evans-reflect:
evans -r repl --port 5632 evans -r repl --port 5632
.PHONY: tidy greet-gen greet-build greet-evans greet-evans-reflect calculator-gen:
protoc -Icalculator/proto/ --go_out=./calculator --go_opt=module=git.gqnotes.com/guoqiang/grpcexercises/calculator --go-grpc_out=./calculator --go-grpc_opt=module=git.gqnotes.com/guoqiang/grpcexercises/calculator calculator/proto/*.proto
calculator-build:
go build -o ./calculator/bin/server-osx ./calculator/server
go build -o ./calculator/bin/client-osx ./calculator/client
calculator-evans-reflect:
evans -r repl --port 5633
.PHONY: tidy greet-gen greet-build greet-evans greet-evans-reflect calculator-gen calculator-build calculator-evans-reflect

View File

@ -0,0 +1,46 @@
package main
import (
"context"
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"log"
)
// Average 计算平均数-双向流式
func Average(client pb.CalculateServiceClient) {
// 构造数据
nums := []int64{1, 2, 3, 4, 5}
avClient, err := client.Average(context.Background())
if err != nil {
return
}
// 发送数据
for _, num := range nums {
err = avClient.Send(&pb.AverageRequest{
Num: num,
})
if err != nil {
return
}
recv, err := avClient.Recv()
if err != nil {
return
}
log.Printf("num:%+v\trecv:%+v\n", num, recv)
}
// 关闭连接
err = avClient.CloseSend()
if err != nil {
log.Fatalf("failed to close: %v", err)
return
}
}

46
calculator/client/Sum.go Normal file
View File

@ -0,0 +1,46 @@
package main
import (
"context"
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"io"
"log"
)
// Sum 计算和-客户端流式
func Sum(client pb.CalculateServiceClient) {
// 构造一个切片
nums := []int64{1, 2, 3, 4, 5}
stream, err := client.Sum(context.Background())
if err != nil {
log.Fatalf("failed to call: %v", err)
return
}
// 发送数据
for i := 0; i < len(nums); i++ {
err = stream.Send(&pb.SumRequest{
Num: nums[i],
})
if err == io.EOF {
break
}
if err != nil {
log.Fatalf("failed to send: %v", err)
return
}
}
// 接收数据
resp, err := stream.CloseAndRecv()
if err != nil {
log.Fatalf("failed to recv: %v", err)
}
log.Printf("nums:%+v\tresp:%+v\n", nums, resp)
}

26
calculator/client/main.go Normal file
View File

@ -0,0 +1,26 @@
package main
import (
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"log"
)
func main() {
conn, err := grpc.Dial(":5633", grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
log.Fatalf("failed to dial: %v", err)
return
}
defer conn.Close()
// 实例化客户端
client := pb.NewCalculateServiceClient(conn)
Sum(client)
Average(client)
}

View File

@ -0,0 +1,475 @@
// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// protoc-gen-go v1.31.0
// protoc v4.25.1
// source: calculator.proto
package pb
import (
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
reflect "reflect"
sync "sync"
)
const (
// Verify that this generated code is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
// Verify that runtime/protoimpl is sufficiently up-to-date.
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)
type SumRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Num int64 `protobuf:"varint,1,opt,name=num,proto3" json:"num,omitempty"`
}
func (x *SumRequest) Reset() {
*x = SumRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[0]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *SumRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*SumRequest) ProtoMessage() {}
func (x *SumRequest) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[0]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use SumRequest.ProtoReflect.Descriptor instead.
func (*SumRequest) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{0}
}
func (x *SumRequest) GetNum() int64 {
if x != nil {
return x.Num
}
return 0
}
type SumResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Total int64 `protobuf:"varint,1,opt,name=total,proto3" json:"total,omitempty"`
}
func (x *SumResponse) Reset() {
*x = SumResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[1]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *SumResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*SumResponse) ProtoMessage() {}
func (x *SumResponse) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[1]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use SumResponse.ProtoReflect.Descriptor instead.
func (*SumResponse) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{1}
}
func (x *SumResponse) GetTotal() int64 {
if x != nil {
return x.Total
}
return 0
}
type RandomNumsRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Num int64 `protobuf:"varint,1,opt,name=num,proto3" json:"num,omitempty"`
}
func (x *RandomNumsRequest) Reset() {
*x = RandomNumsRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[2]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *RandomNumsRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*RandomNumsRequest) ProtoMessage() {}
func (x *RandomNumsRequest) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[2]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use RandomNumsRequest.ProtoReflect.Descriptor instead.
func (*RandomNumsRequest) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{2}
}
func (x *RandomNumsRequest) GetNum() int64 {
if x != nil {
return x.Num
}
return 0
}
type RandomNumsResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Num int64 `protobuf:"varint,1,opt,name=num,proto3" json:"num,omitempty"`
}
func (x *RandomNumsResponse) Reset() {
*x = RandomNumsResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[3]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *RandomNumsResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*RandomNumsResponse) ProtoMessage() {}
func (x *RandomNumsResponse) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[3]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use RandomNumsResponse.ProtoReflect.Descriptor instead.
func (*RandomNumsResponse) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{3}
}
func (x *RandomNumsResponse) GetNum() int64 {
if x != nil {
return x.Num
}
return 0
}
type AverageRequest struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Num int64 `protobuf:"varint,1,opt,name=num,proto3" json:"num,omitempty"`
}
func (x *AverageRequest) Reset() {
*x = AverageRequest{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[4]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *AverageRequest) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*AverageRequest) ProtoMessage() {}
func (x *AverageRequest) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[4]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use AverageRequest.ProtoReflect.Descriptor instead.
func (*AverageRequest) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{4}
}
func (x *AverageRequest) GetNum() int64 {
if x != nil {
return x.Num
}
return 0
}
type AverageResponse struct {
state protoimpl.MessageState
sizeCache protoimpl.SizeCache
unknownFields protoimpl.UnknownFields
Average float32 `protobuf:"fixed32,1,opt,name=average,proto3" json:"average,omitempty"`
}
func (x *AverageResponse) Reset() {
*x = AverageResponse{}
if protoimpl.UnsafeEnabled {
mi := &file_calculator_proto_msgTypes[5]
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
ms.StoreMessageInfo(mi)
}
}
func (x *AverageResponse) String() string {
return protoimpl.X.MessageStringOf(x)
}
func (*AverageResponse) ProtoMessage() {}
func (x *AverageResponse) ProtoReflect() protoreflect.Message {
mi := &file_calculator_proto_msgTypes[5]
if protoimpl.UnsafeEnabled && x != nil {
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
if ms.LoadMessageInfo() == nil {
ms.StoreMessageInfo(mi)
}
return ms
}
return mi.MessageOf(x)
}
// Deprecated: Use AverageResponse.ProtoReflect.Descriptor instead.
func (*AverageResponse) Descriptor() ([]byte, []int) {
return file_calculator_proto_rawDescGZIP(), []int{5}
}
func (x *AverageResponse) GetAverage() float32 {
if x != nil {
return x.Average
}
return 0
}
var File_calculator_proto protoreflect.FileDescriptor
var file_calculator_proto_rawDesc = []byte{
0x0a, 0x10, 0x63, 0x61, 0x6c, 0x63, 0x75, 0x6c, 0x61, 0x74, 0x6f, 0x72, 0x2e, 0x70, 0x72, 0x6f,
0x74, 0x6f, 0x12, 0x05, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x1e, 0x0a, 0x0a, 0x53, 0x75, 0x6d,
0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x6e, 0x75, 0x6d, 0x18, 0x01,
0x20, 0x01, 0x28, 0x03, 0x52, 0x03, 0x6e, 0x75, 0x6d, 0x22, 0x23, 0x0a, 0x0b, 0x53, 0x75, 0x6d,
0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x14, 0x0a, 0x05, 0x74, 0x6f, 0x74, 0x61,
0x6c, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03, 0x52, 0x05, 0x74, 0x6f, 0x74, 0x61, 0x6c, 0x22, 0x25,
0x0a, 0x11, 0x52, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x4e, 0x75, 0x6d, 0x73, 0x52, 0x65, 0x71, 0x75,
0x65, 0x73, 0x74, 0x12, 0x10, 0x0a, 0x03, 0x6e, 0x75, 0x6d, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03,
0x52, 0x03, 0x6e, 0x75, 0x6d, 0x22, 0x26, 0x0a, 0x12, 0x52, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x4e,
0x75, 0x6d, 0x73, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x10, 0x0a, 0x03, 0x6e,
0x75, 0x6d, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03, 0x52, 0x03, 0x6e, 0x75, 0x6d, 0x22, 0x22, 0x0a,
0x0e, 0x41, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12,
0x10, 0x0a, 0x03, 0x6e, 0x75, 0x6d, 0x18, 0x01, 0x20, 0x01, 0x28, 0x03, 0x52, 0x03, 0x6e, 0x75,
0x6d, 0x22, 0x2b, 0x0a, 0x0f, 0x41, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x52, 0x65, 0x73, 0x70,
0x6f, 0x6e, 0x73, 0x65, 0x12, 0x18, 0x0a, 0x07, 0x61, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x18,
0x01, 0x20, 0x01, 0x28, 0x02, 0x52, 0x07, 0x61, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x32, 0xcb,
0x01, 0x0a, 0x10, 0x43, 0x61, 0x6c, 0x63, 0x75, 0x6c, 0x61, 0x74, 0x65, 0x53, 0x65, 0x72, 0x76,
0x69, 0x63, 0x65, 0x12, 0x30, 0x0a, 0x03, 0x53, 0x75, 0x6d, 0x12, 0x11, 0x2e, 0x70, 0x72, 0x6f,
0x74, 0x6f, 0x2e, 0x53, 0x75, 0x6d, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x12, 0x2e,
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x53, 0x75, 0x6d, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73,
0x65, 0x22, 0x00, 0x28, 0x01, 0x12, 0x45, 0x0a, 0x0a, 0x52, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x4e,
0x75, 0x6d, 0x73, 0x12, 0x18, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x52, 0x61, 0x6e, 0x64,
0x6f, 0x6d, 0x4e, 0x75, 0x6d, 0x73, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x19, 0x2e,
0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x52, 0x61, 0x6e, 0x64, 0x6f, 0x6d, 0x4e, 0x75, 0x6d, 0x73,
0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x30, 0x01, 0x12, 0x3e, 0x0a, 0x07,
0x41, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x12, 0x15, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e,
0x41, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x16,
0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x41, 0x76, 0x65, 0x72, 0x61, 0x67, 0x65, 0x52, 0x65,
0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x28, 0x01, 0x30, 0x01, 0x42, 0x36, 0x5a, 0x34,
0x67, 0x69, 0x74, 0x2e, 0x67, 0x71, 0x6e, 0x6f, 0x74, 0x65, 0x73, 0x2e, 0x63, 0x6f, 0x6d, 0x2f,
0x67, 0x75, 0x6f, 0x71, 0x69, 0x61, 0x6e, 0x67, 0x2f, 0x67, 0x72, 0x70, 0x63, 0x65, 0x78, 0x65,
0x72, 0x63, 0x69, 0x73, 0x65, 0x73, 0x2f, 0x63, 0x61, 0x6c, 0x63, 0x75, 0x6c, 0x61, 0x74, 0x6f,
0x72, 0x2f, 0x70, 0x62, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
}
var (
file_calculator_proto_rawDescOnce sync.Once
file_calculator_proto_rawDescData = file_calculator_proto_rawDesc
)
func file_calculator_proto_rawDescGZIP() []byte {
file_calculator_proto_rawDescOnce.Do(func() {
file_calculator_proto_rawDescData = protoimpl.X.CompressGZIP(file_calculator_proto_rawDescData)
})
return file_calculator_proto_rawDescData
}
var file_calculator_proto_msgTypes = make([]protoimpl.MessageInfo, 6)
var file_calculator_proto_goTypes = []interface{}{
(*SumRequest)(nil), // 0: proto.SumRequest
(*SumResponse)(nil), // 1: proto.SumResponse
(*RandomNumsRequest)(nil), // 2: proto.RandomNumsRequest
(*RandomNumsResponse)(nil), // 3: proto.RandomNumsResponse
(*AverageRequest)(nil), // 4: proto.AverageRequest
(*AverageResponse)(nil), // 5: proto.AverageResponse
}
var file_calculator_proto_depIdxs = []int32{
0, // 0: proto.CalculateService.Sum:input_type -> proto.SumRequest
2, // 1: proto.CalculateService.RandomNums:input_type -> proto.RandomNumsRequest
4, // 2: proto.CalculateService.Average:input_type -> proto.AverageRequest
1, // 3: proto.CalculateService.Sum:output_type -> proto.SumResponse
3, // 4: proto.CalculateService.RandomNums:output_type -> proto.RandomNumsResponse
5, // 5: proto.CalculateService.Average:output_type -> proto.AverageResponse
3, // [3:6] is the sub-list for method output_type
0, // [0:3] is the sub-list for method input_type
0, // [0:0] is the sub-list for extension type_name
0, // [0:0] is the sub-list for extension extendee
0, // [0:0] is the sub-list for field type_name
}
func init() { file_calculator_proto_init() }
func file_calculator_proto_init() {
if File_calculator_proto != nil {
return
}
if !protoimpl.UnsafeEnabled {
file_calculator_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*SumRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_calculator_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*SumResponse); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_calculator_proto_msgTypes[2].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*RandomNumsRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_calculator_proto_msgTypes[3].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*RandomNumsResponse); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_calculator_proto_msgTypes[4].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*AverageRequest); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
file_calculator_proto_msgTypes[5].Exporter = func(v interface{}, i int) interface{} {
switch v := v.(*AverageResponse); i {
case 0:
return &v.state
case 1:
return &v.sizeCache
case 2:
return &v.unknownFields
default:
return nil
}
}
}
type x struct{}
out := protoimpl.TypeBuilder{
File: protoimpl.DescBuilder{
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
RawDescriptor: file_calculator_proto_rawDesc,
NumEnums: 0,
NumMessages: 6,
NumExtensions: 0,
NumServices: 1,
},
GoTypes: file_calculator_proto_goTypes,
DependencyIndexes: file_calculator_proto_depIdxs,
MessageInfos: file_calculator_proto_msgTypes,
}.Build()
File_calculator_proto = out.File
file_calculator_proto_rawDesc = nil
file_calculator_proto_goTypes = nil
file_calculator_proto_depIdxs = nil
}

View File

@ -0,0 +1,282 @@
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
// versions:
// - protoc-gen-go-grpc v1.3.0
// - protoc v4.25.1
// source: calculator.proto
package pb
import (
context "context"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
status "google.golang.org/grpc/status"
)
// This is a compile-time assertion to ensure that this generated file
// is compatible with the grpc package it is being compiled against.
// Requires gRPC-Go v1.32.0 or later.
const _ = grpc.SupportPackageIsVersion7
const (
CalculateService_Sum_FullMethodName = "/proto.CalculateService/Sum"
CalculateService_RandomNums_FullMethodName = "/proto.CalculateService/RandomNums"
CalculateService_Average_FullMethodName = "/proto.CalculateService/Average"
)
// CalculateServiceClient is the client API for CalculateService service.
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type CalculateServiceClient interface {
// 求和-客户端流式
Sum(ctx context.Context, opts ...grpc.CallOption) (CalculateService_SumClient, error)
// 生成一定数量的随机数-服务端流式
RandomNums(ctx context.Context, in *RandomNumsRequest, opts ...grpc.CallOption) (CalculateService_RandomNumsClient, error)
// 双向流式求平均值
Average(ctx context.Context, opts ...grpc.CallOption) (CalculateService_AverageClient, error)
}
type calculateServiceClient struct {
cc grpc.ClientConnInterface
}
func NewCalculateServiceClient(cc grpc.ClientConnInterface) CalculateServiceClient {
return &calculateServiceClient{cc}
}
func (c *calculateServiceClient) Sum(ctx context.Context, opts ...grpc.CallOption) (CalculateService_SumClient, error) {
stream, err := c.cc.NewStream(ctx, &CalculateService_ServiceDesc.Streams[0], CalculateService_Sum_FullMethodName, opts...)
if err != nil {
return nil, err
}
x := &calculateServiceSumClient{stream}
return x, nil
}
type CalculateService_SumClient interface {
Send(*SumRequest) error
CloseAndRecv() (*SumResponse, error)
grpc.ClientStream
}
type calculateServiceSumClient struct {
grpc.ClientStream
}
func (x *calculateServiceSumClient) Send(m *SumRequest) error {
return x.ClientStream.SendMsg(m)
}
func (x *calculateServiceSumClient) CloseAndRecv() (*SumResponse, error) {
if err := x.ClientStream.CloseSend(); err != nil {
return nil, err
}
m := new(SumResponse)
if err := x.ClientStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
func (c *calculateServiceClient) RandomNums(ctx context.Context, in *RandomNumsRequest, opts ...grpc.CallOption) (CalculateService_RandomNumsClient, error) {
stream, err := c.cc.NewStream(ctx, &CalculateService_ServiceDesc.Streams[1], CalculateService_RandomNums_FullMethodName, opts...)
if err != nil {
return nil, err
}
x := &calculateServiceRandomNumsClient{stream}
if err := x.ClientStream.SendMsg(in); err != nil {
return nil, err
}
if err := x.ClientStream.CloseSend(); err != nil {
return nil, err
}
return x, nil
}
type CalculateService_RandomNumsClient interface {
Recv() (*RandomNumsResponse, error)
grpc.ClientStream
}
type calculateServiceRandomNumsClient struct {
grpc.ClientStream
}
func (x *calculateServiceRandomNumsClient) Recv() (*RandomNumsResponse, error) {
m := new(RandomNumsResponse)
if err := x.ClientStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
func (c *calculateServiceClient) Average(ctx context.Context, opts ...grpc.CallOption) (CalculateService_AverageClient, error) {
stream, err := c.cc.NewStream(ctx, &CalculateService_ServiceDesc.Streams[2], CalculateService_Average_FullMethodName, opts...)
if err != nil {
return nil, err
}
x := &calculateServiceAverageClient{stream}
return x, nil
}
type CalculateService_AverageClient interface {
Send(*AverageRequest) error
Recv() (*AverageResponse, error)
grpc.ClientStream
}
type calculateServiceAverageClient struct {
grpc.ClientStream
}
func (x *calculateServiceAverageClient) Send(m *AverageRequest) error {
return x.ClientStream.SendMsg(m)
}
func (x *calculateServiceAverageClient) Recv() (*AverageResponse, error) {
m := new(AverageResponse)
if err := x.ClientStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
// CalculateServiceServer is the server API for CalculateService service.
// All implementations must embed UnimplementedCalculateServiceServer
// for forward compatibility
type CalculateServiceServer interface {
// 求和-客户端流式
Sum(CalculateService_SumServer) error
// 生成一定数量的随机数-服务端流式
RandomNums(*RandomNumsRequest, CalculateService_RandomNumsServer) error
// 双向流式求平均值
Average(CalculateService_AverageServer) error
mustEmbedUnimplementedCalculateServiceServer()
}
// UnimplementedCalculateServiceServer must be embedded to have forward compatible implementations.
type UnimplementedCalculateServiceServer struct {
}
func (UnimplementedCalculateServiceServer) Sum(CalculateService_SumServer) error {
return status.Errorf(codes.Unimplemented, "method Sum not implemented")
}
func (UnimplementedCalculateServiceServer) RandomNums(*RandomNumsRequest, CalculateService_RandomNumsServer) error {
return status.Errorf(codes.Unimplemented, "method RandomNums not implemented")
}
func (UnimplementedCalculateServiceServer) Average(CalculateService_AverageServer) error {
return status.Errorf(codes.Unimplemented, "method Average not implemented")
}
func (UnimplementedCalculateServiceServer) mustEmbedUnimplementedCalculateServiceServer() {}
// UnsafeCalculateServiceServer may be embedded to opt out of forward compatibility for this service.
// Use of this interface is not recommended, as added methods to CalculateServiceServer will
// result in compilation errors.
type UnsafeCalculateServiceServer interface {
mustEmbedUnimplementedCalculateServiceServer()
}
func RegisterCalculateServiceServer(s grpc.ServiceRegistrar, srv CalculateServiceServer) {
s.RegisterService(&CalculateService_ServiceDesc, srv)
}
func _CalculateService_Sum_Handler(srv interface{}, stream grpc.ServerStream) error {
return srv.(CalculateServiceServer).Sum(&calculateServiceSumServer{stream})
}
type CalculateService_SumServer interface {
SendAndClose(*SumResponse) error
Recv() (*SumRequest, error)
grpc.ServerStream
}
type calculateServiceSumServer struct {
grpc.ServerStream
}
func (x *calculateServiceSumServer) SendAndClose(m *SumResponse) error {
return x.ServerStream.SendMsg(m)
}
func (x *calculateServiceSumServer) Recv() (*SumRequest, error) {
m := new(SumRequest)
if err := x.ServerStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
func _CalculateService_RandomNums_Handler(srv interface{}, stream grpc.ServerStream) error {
m := new(RandomNumsRequest)
if err := stream.RecvMsg(m); err != nil {
return err
}
return srv.(CalculateServiceServer).RandomNums(m, &calculateServiceRandomNumsServer{stream})
}
type CalculateService_RandomNumsServer interface {
Send(*RandomNumsResponse) error
grpc.ServerStream
}
type calculateServiceRandomNumsServer struct {
grpc.ServerStream
}
func (x *calculateServiceRandomNumsServer) Send(m *RandomNumsResponse) error {
return x.ServerStream.SendMsg(m)
}
func _CalculateService_Average_Handler(srv interface{}, stream grpc.ServerStream) error {
return srv.(CalculateServiceServer).Average(&calculateServiceAverageServer{stream})
}
type CalculateService_AverageServer interface {
Send(*AverageResponse) error
Recv() (*AverageRequest, error)
grpc.ServerStream
}
type calculateServiceAverageServer struct {
grpc.ServerStream
}
func (x *calculateServiceAverageServer) Send(m *AverageResponse) error {
return x.ServerStream.SendMsg(m)
}
func (x *calculateServiceAverageServer) Recv() (*AverageRequest, error) {
m := new(AverageRequest)
if err := x.ServerStream.RecvMsg(m); err != nil {
return nil, err
}
return m, nil
}
// CalculateService_ServiceDesc is the grpc.ServiceDesc for CalculateService service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
var CalculateService_ServiceDesc = grpc.ServiceDesc{
ServiceName: "proto.CalculateService",
HandlerType: (*CalculateServiceServer)(nil),
Methods: []grpc.MethodDesc{},
Streams: []grpc.StreamDesc{
{
StreamName: "Sum",
Handler: _CalculateService_Sum_Handler,
ClientStreams: true,
},
{
StreamName: "RandomNums",
Handler: _CalculateService_RandomNums_Handler,
ServerStreams: true,
},
{
StreamName: "Average",
Handler: _CalculateService_Average_Handler,
ServerStreams: true,
ClientStreams: true,
},
},
Metadata: "calculator.proto",
}

View File

@ -0,0 +1,40 @@
syntax = "proto3";
package proto;
option go_package = "git.gqnotes.com/guoqiang/grpcexercises/calculator/pb";
service CalculateService {
// -
rpc Sum(stream SumRequest) returns(SumResponse) {}
// -
rpc RandomNums(RandomNumsRequest) returns(stream RandomNumsResponse) {}
//
rpc Average(stream AverageRequest) returns(stream AverageResponse) {}
}
message SumRequest {
int64 num =1;
}
message SumResponse {
int64 total = 1;
}
message RandomNumsRequest {
int64 num =1;
}
message RandomNumsResponse {
int64 num = 1;
}
message AverageRequest {
int64 num =1;
}
message AverageResponse {
float average = 1;
}

View File

@ -0,0 +1,34 @@
package main
import (
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"io"
)
// Average 计算平均数-双向流式
func (s *Server) Average(stream pb.CalculateService_AverageServer) (err error) {
var sum int64
var count int64
for {
req, err := stream.Recv()
if err == io.EOF {
break
}
if err != nil {
return err
}
sum += req.Num
count++
if err := stream.Send(&pb.AverageResponse{Average: float32(sum) / float32(count)}); err != nil {
return err
}
}
return
}

View File

@ -0,0 +1,29 @@
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
}

33
calculator/server/Sum.go Normal file
View File

@ -0,0 +1,33 @@
package main
import (
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"io"
"log"
)
// Sum 计算和-客户端流式
func (s *Server) Sum(stream pb.CalculateService_SumServer) error {
var sum int64
for {
req, err := stream.Recv()
if err == io.EOF {
log.Printf("req.Num: %v", req)
break
}
if err != nil {
return err
}
sum += req.Num
}
if err := stream.SendAndClose(&pb.SumResponse{Total: sum}); err != nil {
return err
}
return nil
}

38
calculator/server/main.go Normal file
View File

@ -0,0 +1,38 @@
package main
import (
"git.gqnotes.com/guoqiang/grpcexercises/calculator/pb"
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
"log"
"net"
)
func main() {
// 实现服务端逻辑监听5633端口
lis, err := net.Listen("tcp", ":5633")
if err != nil {
log.Fatalf("failed to listen: %v", err)
return
}
s := grpc.NewServer()
// 反射服务
reflection.Register(s)
// 注册服务
pb.RegisterCalculateServiceServer(s, &Server{})
log.Println("grpc server start")
if err = s.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err)
return
}
}
type Server struct {
pb.UnimplementedCalculateServiceServer
}

1
go.mod
View File

@ -3,6 +3,7 @@ module git.gqnotes.com/guoqiang/grpcexercises
go 1.21.4 go 1.21.4
require ( require (
google.golang.org/appengine v1.6.7
google.golang.org/grpc v1.59.0 google.golang.org/grpc v1.59.0
google.golang.org/protobuf v1.31.0 google.golang.org/protobuf v1.31.0
) )

9
go.sum
View File

@ -1,16 +1,25 @@
github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg= github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38= github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
golang.org/x/net v0.14.0 h1:BONx9s002vGdD9umnlX1Po8vOZmrgH34qlHcD1MfK14= golang.org/x/net v0.14.0 h1:BONx9s002vGdD9umnlX1Po8vOZmrgH34qlHcD1MfK14=
golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI= golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM= golang.org/x/sys v0.11.0 h1:eG7RXZHdqOJ1i+0lgLgCpSXAp6M3LYlAo6osgSi0xOM=
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc= golang.org/x/text v0.12.0 h1:k+n5B8goJNdU7hSvEtMUz3d1Q6D/XW4COJSJR6fN0mc=
golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
google.golang.org/appengine v1.6.7 h1:FZR1q0exgwxzPzp/aF+VccGrSfxfPpkBqjIIEq3ru6c=
google.golang.org/appengine v1.6.7/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d h1:uvYuEyMHKNt+lT4K3bN6fGswmK8qSvcreM3BwjDh+y4= google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d h1:uvYuEyMHKNt+lT4K3bN6fGswmK8qSvcreM3BwjDh+y4=
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d/go.mod h1:+Bk1OCOj40wS2hwAMA+aCW9ypzm63QTBBHp6lQ3p+9M= google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d/go.mod h1:+Bk1OCOj40wS2hwAMA+aCW9ypzm63QTBBHp6lQ3p+9M=
google.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk= google.golang.org/grpc v1.59.0 h1:Z5Iec2pjwb+LEOqzpB2MR12/eKFhDPhuqW91O+4bwUk=