go中的lambda多部分/文件上传

Question

我需要使用 Go 对通过 AWS Lambda 上传的上传文件的内容进行一些简单的操作，但不确定如何解析接收内容，因为我是 Go 新手。到目前为止我找到的解决方案与http包和MultiPart表单函数有关。

type Request events.APIGatewayProxyRequest

func Handler(ctx context.Context, req Request) (Response, error) {
fmt.Println(req.Body)
....
}

这就是我的请求正文的样子

------WebKitFormBoundaryx0SVDelfa90Fi5Uo
Content-Disposition: form-data; name="file"; filename="upload.txt"
Content-Type: text/plain

this is content

------WebKitFormBoundaryx0SVDelfa90Fi5Uo--

我的请求是 的实例APIGatewayProxyRequest。

我想知道是否可以获得一个自定义结构，我可以从中访问 fe 等数据

customStruct.content => "this is content"
customStruct.fileName => upload.txt
customStruct.fileExtension => txt

Answer 1

这有 3 个部分：

1. 创建一个multipart.Reader来自events.APIGatewayProxyRequest
2. 获取 MIME 部分
3. 提取 MIME 部分值

第 1 步：创建一个multipart.Reader

它multipart.NewReader采用一个io.Readerandboundary字符串，如签名所示：

func NewReader(r io.Reader, boundary string) *Reader

为此，您需要从Content-TypeHTTP 请求标头中提取边界字符串，这可以使用mime.ParseMediaType.

NewReaderMultipart执行此操作的一个简单方法是从go-awslambda具有以下签名的包中调用：

func NewReaderMultipart(req events.APIGatewayProxyRequest) (*multipart.Reader, error)

第 2 步：获取 MIME 部分

获得 后mime.Reader，导航 MIME 消息，直到找到所需的 MIME 部分。

在此示例中，只有一部分，因此您只需调用：

part, err := reader.NextPart()

步骤 3：提取 MIME 部分值

一旦获得 MIME 部分，就可以提取所需的值。

步骤3.1：内容

content, err := io.ReadAll(part)

步骤3.2：文件名

从MIME部分获取文件名如下：

filename := part.FileName()

步骤3.3：文件扩展名

称呼path/filepath.Ext。.这将在扩展中添加前导句点，但这可以轻松删除。

ext := filepath.Ext(part.FileName())

概括

您可以按如下方式组合：

import (
    "context"
    "encoding/json"
    "io"

    "github.com/aws/aws-lambda-go/events"
    "github.com/grokify/go-awslambda"
)

type customStruct struct {
    Content       string
    FileName      string
    FileExtension string
}

func handleRequest(ctx context.Context, req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
    res := events.APIGatewayProxyResponse{}
    r, err := awslambda.NewReaderMultipart(req)
    if err != nil {
        return res, err
    }
    part, err := r.NextPart()
    if err != nil {
        return res, err
    }
    content, err := io.ReadAll(part)
    if err != nil {
        return res, err
    }
    custom := customStruct{
        Content:       string(content),
        FileName:      part.FileName(),
        FileExtension: filepath.Ext(part.FileName())}

    customBytes, err := json.Marshal(custom)
    if err != nil {
        return res, err
    }

    res = events.APIGatewayProxyResponse{
        StatusCode: 200,
        Headers: map[string]string{
            "Content-Type": "application/json"},
        Body: string(customBytes)}
    return res, nil
}