因上下文取消导致GCP SecretIterator异常，求无传参解决方案

问题重现 #

使用GCP Secrets Manager获取密钥列表时，通过辅助函数直接返回SecretIterator会触发错误：rpc error: code = Canceled desc = context canceled。错误代码如下：

func listSecrets(client *secretmanager.Client, req *secretmanagerpb.ListSecretsRequest) *secretmanager.SecretIterator {
    ctx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancelFunc()
    return client.ListSecrets(ctx, req)
}

func main() {
    client, err := secretmanager.NewClient(context.Background())
    // ... 省略错误处理与变量定义
    req := &secretmanagerpb.ListSecretsRequest{
      Parent: parent,
      Filter: filter,
    }

    result := listSecrets(client, req)
    // 遍历result时触发上下文取消错误
}

问题原因 #

辅助函数内创建的上下文会在函数返回前被defer cancelFunc()取消，而SecretIterator是延迟加载的——它的Next()方法才会实际发起RPC请求。当在main中调用Next()时，依赖的上下文已经失效，因此报错。

可行解决方案（无需传递上下文参数） #

方案1：在辅助函数内完成遍历，返回密钥列表 #

将迭代器的遍历逻辑放在辅助函数内部，利用函数内的上下文完成所有RPC请求，最终返回收集好的密钥列表。这样上下文的生命周期覆盖整个请求过程，不会提前失效。

代码示例：

import "google.golang.org/api/iterator"

func listSecrets(client *secretmanager.Client, req *secretmanagerpb.ListSecretsRequest) ([]*secretmanagerpb.Secret, error) {
    ctx, cancelFunc := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancelFunc()
    
    iter := client.ListSecrets(ctx, req)
    var secrets []*secretmanagerpb.Secret
    for {
        secret, err := iter.Next()
        if err == iterator.Done {
            break
        }
        if err != nil {
            return nil, err
        }
        secrets = append(secrets, secret)
    }
    return secrets, nil
}

func main() {
    client, err := secretmanager.NewClient(context.Background())
    if err != nil {
        // 错误处理逻辑
    }
    defer client.Close()

    parent := "projects/your-project-id"
    filter := ""
    req := &secretmanagerpb.ListSecretsRequest{
        Parent: parent,
        Filter: filter,
    }

    secrets, err := listSecrets(client, req)
    if err != nil {
        // 错误处理逻辑
    }

    // 遍历已获取的密钥列表
    for _, secret := range secrets {
        // 处理每个密钥
    }
}

方案2：封装自定义迭代器（进阶） #

如果必须返回迭代器形式，可以封装一个自定义迭代器，将上下文和原SecretIterator绑定，在自定义迭代器被完全遍历后再取消上下文。不过这种方式需要额外的代码维护，不如方案1简洁直接。

关键说明 #

GCP Secrets Manager的SecretIterator与创建它的上下文强绑定，所有后续的Next()调用都会复用该上下文。因此不能在上下文被取消后再使用迭代器，必须确保上下文的生命周期覆盖迭代器的整个使用过程。

内容的提问来源于stack exchange，提问作者ArigatoManga