Golang如何使用gRPC客户端Stream发送数据_Golang gRPC客户端Stream通信实践

service DataCollector {
  rpc UploadLogs(stream LogEntry) returns (UploadResult);
}

message LogEntry {
  string timestamp = 1;
  string level     = 2;
  string message   = 3;
}

message UploadResult {
  bool success = 1;
  int32 count  = 2;
}

protoc --go_out=. --go-grpc_out=. data_collector.proto

conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
if err != nil {
    log.Fatalf("无法连接: %v", err)
}
defer conn.Close()

client := NewDataCollectorClient(conn)
stream, err := client.UploadLogs(context.Background())
if err != nil {
    log.Fatalf("调用失败: %v", err)
}

logs := []LogEntry{
    {Timestamp: "2023-04-01T10:00:00", Level: "INFO", Message: "启动服务"},
    {Timestamp: "2023-04-01T10:01:00", Level: "ERROR", Message: "数据库连接失败"},
    {Timestamp: "2023-04-01T10:02:00", Level: "WARN", Message: "磁盘空间不足"},
}

for _, log := range logs {
    if err := stream.Send(&log); err != nil {
        log.Printf("发送日志失败: %v", err)
    }
}

// 关闭发送并接收最终结果
result, err := stream.CloseAndRecv()
if err != nil {
    log.Fatalf("接收响应失败: %v", err)
}

fmt.Printf("上传成功: %v, 条数: %d\n", result.Success, result.Count)

ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
defer cancel()

stream, err := client.UploadLogs(ctx)