小编点评

**gRPC 文件传输示例** **服务端 (.proto)** ```proto3 service FileService { rpc UploadFile(stream Chunk) returns (UploadStatus) {} } message Chunk { bytes content = 1; } message UploadStatus { bool success = 1; string message = 2; } ``` **服务器端 (.go)** ```go func (s *server) UploadFile(stream pb.FileService_UploadFileServer) error { // 创建文件写入对象 file, err := os.Create("uploaded.file") if err != nil { return err } defer file.Close() // 从流中读取数据 for { chunk, err := stream.Recv() if err == io.EOF { return stream.SendAndClose(&pb.UploadStatus{ Success: true, Message: "文件上传成功", }) } if err != nil { return err } _, writeErr := file.Write(chunk.Content) if writeErr != nil { return writeErr } } } ``` **客户端端 (.go)** ```go func uploadFile(client pb.FileServiceClient, filename string) error { // 获取流 stream, err := client.UploadFile(context.Background()) if err != nil { return err } // 读取文件并将其分成小块 buffer := make([]byte, 1024) for { n, err := file.Read(buffer) if err == io.EOF { break } if err != nil { return err } err := stream.Send(&pb.Chunk{Content: buffer[:n]}) if err != nil { return err } } // 关闭流 stream.CloseAndRecv() // 检查状态 status, err := stream.CloseAndRecv() if err != nil { return err } if status.Success { fmt.Println("文件上传成功") } else { fmt.Println("失败：", status.Message) } return nil } ```

正文

在gRPC中，可以通过将文件分割成多个小块，然后使用流式RPC将这些小块发送到服务器来传输文件。以下是一个简单的示例，展示了如何在gRPC中实现文件传输。

首先，我们需要定义一个服务来处理文件传输。在.proto文件中，我们可以定义一个UploadFile服务，它接收一个流式的Chunk消息，并返回一个UploadStatus消息。

syntax = "proto3";

service FileService {
  rpc UploadFile(stream Chunk) returns (UploadStatus) {}
}

message Chunk {
  bytes content = 1;
}

message UploadStatus {
  bool success = 1;
  string message = 2;
}

然后，我们可以在服务器端实现这个服务。在UploadFile方法中，我们从客户端接收Chunk消息，然后将它们写入到一个文件中。

type server struct{}

func (s *server) UploadFile(stream pb.FileService_UploadFileServer) error {
  file, err := os.Create("uploaded.file")
  if err != nil {
    return err
  }
  defer file.Close()

  for {
    chunk, err := stream.Recv()
    if err == io.EOF {
      return stream.SendAndClose(&pb.UploadStatus{
        Success: true,
        Message: "File uploaded successfully",
      })
    }
    if err != nil {
      return err
    }

    _, writeErr := file.Write(chunk.Content)
    if writeErr != nil {
      return writeErr
    }
  }
}

在客户端，我们可以读取一个文件，将它分割成多个小块，然后使用UploadFile方法将这些小块发送到服务器。

func uploadFile(client pb.FileServiceClient, filename string) error {
  stream, err := client.UploadFile(context.Background())
  if err != nil {
    return err
  }

  file, err := os.Open(filename)
  if err != nil {
    return err
  }
  defer file.Close()

  buffer := make([]byte, 1024)
  for {
    n, err := file.Read(buffer)
    if err == io.EOF {
      break
    }
    if err != nil {
      return err
    }

    err = stream.Send(&pb.Chunk{
      Content: buffer[:n],
    })
    if err != nil {
      return err
    }
  }

  status, err := stream.CloseAndRecv()
  if err != nil {
    return err
  }

  if status.Success {
    fmt.Println("File uploaded successfully")
  } else {
    fmt.Println("Failed to upload file:", status.Message)
  }

  return nil
}

以上就是在gRPC中实现文件传输的基本步骤，实际的实现可能需要处理更多的细节，比如错误处理和文件的并发访问。

---

声明：本作品采用署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0)进行许可，使用时请注明出处。
Author: mengbin
blog: mengbin
Github: mengbin92
cnblogs: 恋水无意

---