Added command to load test SSM parameters

This commit is contained in:
Leon Mika 2022-04-15 06:03:55 +10:00
parent 71acd75685
commit d48263e43f
5 changed files with 53 additions and 6 deletions

View file

@ -34,7 +34,7 @@ func main() {
var dynamoClient *dynamodb.Client
if *flagLocal {
dynamoClient = dynamodb.NewFromConfig(cfg,
dynamodb.WithEndpointResolver(dynamodb.EndpointResolverFromURL("http://localhost:8000")))
dynamodb.WithEndpointResolver(dynamodb.EndpointResolverFromURL("http://localhost:4566")))
} else {
dynamoClient = dynamodb.NewFromConfig(cfg)
}

View file

@ -2,6 +2,7 @@ package main
import (
"context"
"flag"
"fmt"
"github.com/aws/aws-sdk-go-v2/config"
"github.com/aws/aws-sdk-go-v2/service/ssm"
@ -18,6 +19,9 @@ import (
)
func main() {
var flagLocal = flag.Bool("local", false, "local endpoint")
flag.Parse()
// Pre-determine if layout has dark background. This prevents calls for creating a list to hang.
lipgloss.HasDarkBackground()
@ -28,7 +32,14 @@ func main() {
if err != nil {
cli.Fatalf("cannot load AWS config: %v", err)
}
ssmClient := ssm.NewFromConfig(cfg)
var ssmClient *ssm.Client
if *flagLocal {
ssmClient = ssm.NewFromConfig(cfg,
ssm.WithEndpointResolver(ssm.EndpointResolverFromURL("http://localhost:4566")))
} else {
ssmClient = ssm.NewFromConfig(cfg)
}
provider := awsssm.NewProvider(ssmClient)
service := ssmparameters.NewService(provider)