mirror of
https://github.com/hanzo-fi/stack.git
synced 2026-10-30 20:34:43 +00:00
20 lines
442 B
Go
20 lines
442 B
Go
package publish
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/IBM/sarama"
|
|
"github.com/aws/aws-msk-iam-sasl-signer-go/signer"
|
|
)
|
|
|
|
type MSKAccessTokenProvider struct {
|
|
region string
|
|
roleArn string
|
|
sessionName string
|
|
}
|
|
|
|
func (m *MSKAccessTokenProvider) Token() (*sarama.AccessToken, error) {
|
|
token, _, err := signer.GenerateAuthTokenFromRole(context.TODO(), m.region, m.roleArn, m.sessionName)
|
|
return &sarama.AccessToken{Token: token}, err
|
|
}
|