更新凭证用于跨账户扮演角色所需的AWS stscreds SDK

今日不肯埋头，明日何以抬头！每日一句努力自己的话哈哈~哈喽，今天我将给大家带来一篇《更新凭证用于跨账户扮演角色所需的AWS stscreds SDK》，主要内容是讲解等等，感兴趣的朋友可以收藏或者有更好的建议在评论提出，我都会认真看的！大家一起进步，一起学习！

我已经设置了跨帐户读取 kinesis 流，但是当 kinesis 客户端读取记录时，我收到安全令牌过期错误。我使用 sts 承担角色来承担 accounta 中的角色a，然后使用角色a凭据来承担角色b，最后返回 kinesis 客户端，因此没有应用刷新功能，并且客户端默认将在 1 小时内过期。我查找了 stscreds assumeroleprovider，文档说它将刷新凭据。但我不知道如何刷新假定角色 a 的第一个凭据，然后刷新假定角色 b 的第二个凭据。还是调用该方法重新初始化kinesis客户端更好？

这是代码块。

cfg, err := config.LoadDefaultConfig(
        context.TODO(),
        config.WithRegion("us-west-2"),
    )

    if err != nil {
        return nil, err
    }

    stsclient := sts.NewFromConfig(cfg)

    assumingcnf, err := config.LoadDefaultConfig(
        context.TODO(),
        config.WithRegion("us-west-2"),
        config.WithCredentialsProvider(aws.NewCredentialsCache(
            stscreds.NewAssumeRoleProvider(
                stsclient,
                roleToAssumeArn1,
            )),
        ),
    )
    if err != nil {
        return nil, err
    }

    stsclient = sts.NewFromConfig(assumingcnf)

    cnf, err := config.LoadDefaultConfig(
        context.TODO(),
        config.WithRegion("us-west-2"),
        config.WithCredentialsProvider(aws.NewCredentialsCache(
            stscreds.NewAssumeRoleProvider(
                stsclient,
                roleToAssumeArn2,
            )),
        ),
    )
    if err != nil {
        return nil, err
    }
    kClient := kinesis.NewFromConfig(cnf)
    return kClient

正确答案

您应该能够通过 aws 提供的提供商来执行此操作。我假设您使用的是 aws-sdk-go-v2。

这将使生成的 credentialsprovider 返回缓存的凭据，直到它们过期；然后它会调用 provider2，它使用 sts2 来获取 roleb 的新凭证，并且 sts2 总是首先调用 provider1 来获取 rolea 的新凭证。

func createProvider(cfg aws.Config) aws.CredentialsProvider {
    sts1 := sts.NewFromConfig(cfg)
    provider1 := stscreds.NewAssumeRoleProvider(sts1, "roleA")
    sts2 := sts.NewFromConfig(cfg, func (o *sts.Options) { o.Credentials = provider1 })
    provider2 := stscreds.NewAssumeRoleProvider(sts2, "roleB")
    return aws.NewCredentialsCache(provider2)
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。