-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
38 lines (31 loc) · 1.04 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
package main
import (
ddlambda "github.com/DataDog/datadog-lambda-go"
"github.com/SKF/go-rest-utility/client/auth"
"github.com/SKF/go-utility/v2/env"
"github.com/aws/aws-lambda-go/lambda"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/secretsmanager"
"github.com/SKF/jwt-rotator/pkg/jwtrotator"
)
var (
UsersCredentialsSecretID = env.MustGetAsString("USERS_CREDENTIALS_SECRETS_MANAGER_ARN")
)
func main() {
sess := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
credentialsTokenProvider := auth.SecretCredentialsTokenProvider{
SecretID: UsersCredentialsSecretID,
SecretsClient: auth.SecretsManagerV1Client{},
}
jwtRotator := jwtrotator.JWTRotator{
SecretsManager: secretsmanager.New(sess),
TokenProvider: &credentialsTokenProvider,
}
lambda.Start(ddlambda.WrapFunction(jwtRotator.Rotate, &ddlambda.Config{
DDTraceEnabled: true,
// ShouldUseLogForwarder flushes traces and metrics to CloudWatch
ShouldUseLogForwarder: true,
}))
}