如何在 Golang 中使用 gRPC 实现文件上传？

如何使用 grpc 实现文件上传？创建配套服务定义，包括请求和响应消息。在客户端，打开要上传的文件并将其分成块，然后通过 grpc 流流式传输发送到服务端。在服务端，接收文件块并将其存储到文件中。服务端在文件上传完成后发送响应，指示上传是否成功。

如何在 Golang 中使用 gRPC 实现文件上传

gRPC（Google远程过程调用）是一个通用的、高性能的、开源的远程过程调用框架，可以在任何环境中运行。它提供了在不同的机器之间传输数据的有效方式，并且非常适合需要在分布式系统中传输大型文件的情况。

配套服务

立即学习“go语言免费学习笔记（深入）”；

在开始之前，你需要创建一个配套服务来处理文件上传。以下是一个示例服务定义：

网奇英文商城外贸系统

网奇Eshop是一个带有国际化语言支持的系统，可以同时在一个页面上显示全球任何一种语言而没有任何障碍、任何乱码。在本系统中您可以发现，后台可以用任意一种语言对前台进行管理、录入而没有阻碍。而任何一个国家的浏览者也可以用他们的本国语言在你的网站上下订单、留言。用户可以通过后台随意设定软件语言，也就是说你可以用本软件开设简体中文、繁体中文与英文或者其他语言的网上商店。网奇Eshop系统全部版本都使用模

下载

// 文件上传服务接口
syntax = "proto3";

package file_upload;

service FileUploadService {
  rpc Upload(stream FileChunk) returns (FileUploadResponse);
}

// 文件块信息
message FileChunk {
  bytes data = 1;
}

// 文件上传响应
message FileUploadResponse {
  bool success = 1;
}

gRPC 客户端实现

使用 gRPC 的客户端实现文件上传过程如下：

package main

import (
    "context"
    "io"
    "log"

    file_upload "github.com/example/file-upload"
    "google.golang.org/grpc"
)

func main() {
    // 建立与服务的连接
    conn, err := grpc.Dial("127.0.0.1:50051", grpc.WithInsecure())
    if err != nil {
        log.Fatalf("连接服务失败: %v", err)
    }
    defer conn.Close()

    // 创建文件上传客户端
    client := file_upload.NewFileUploadServiceClient(conn)

    // 打开要上传的文件
    file, err := os.Open("example.txt")
    if err != nil {
        log.Fatalf("打开文件失败: %v", err)
    }
    defer file.Close()

    // 将文件分成块进行流式传输
    stream, err := client.Upload(context.Background())
    if err != nil {
        log.Fatalf("创建流式传输失败: %v", err)
    }

    // 循环读取文件并发送块
    buf := make([]byte, 1024)
    for {
        n, err := file.Read(buf)
        if err == io.EOF {
            break
        }
        if err != nil {
            log.Fatalf("读取文件失败: %v", err)
        }

        // 发送块
        if err := stream.Send(&file_upload.FileChunk{Data: buf[:n]}); err != nil {
            log.Fatalf("发送块失败: %v", err)
        }
    }

    stream.CloseSend()

    // 接收上传响应
    resp, err := stream.Recv()
    if err != nil {
        log.Fatalf("接收上传响应失败: %v", err)
    }

    log.Printf("上传成功: %v", resp.Success)
}

配套服务实现

服务端实现接收和处理上传文件的逻辑：

package main

import (
    "context"
    "io"
    "log"

    file_upload "github.com/example/file-upload"
)

func main() {
    // 创建文件上传服务
    server := &FileUploadService{}

    // 创建 gRPC 服务器并注册服务
    s := grpc.NewServer()
    file_upload.RegisterFileUploadServiceServer(s, server)

    lis, err := net.Listen("tcp", "127.0.0.1:50051")
    if err != nil {
        log.Fatalf("监听端口失败: %v", err)
    }

    // 启动 gRPC 服务器
    if err := s.Serve(lis); err != nil {
        log.Fatalf("启动 gRPC 服务器失败: %v", err)
    }
}

// FileUploadService 服务的具体实现
type FileUploadService struct{}

func (s *FileUploadService) Upload(stream file_upload.FileUploadService_UploadServer) error {
    file, err := os.Create("received.txt")
    if err != nil {
        return err
    }

    for {
        chunk, err := stream.Recv()
        if err == io.EOF {
            break
        }
        if err != nil {
            return err
        }

        if _, err := file.Write(chunk.GetData()); err != nil {
            return err
        }
    }

    if err := stream.SendAndClose(&file_upload.FileUploadResponse{Success: true}); err != nil {
        return err
    }

    return nil
}

以上代码展示了如何在 Golang 中使用 gRPC 实现文件上传。客户端将文件分成块并通过流式传输发送到服务端，服务端则负责接收和处理上传的文件。