-
Notifications
You must be signed in to change notification settings - Fork 9
/
factory.go
58 lines (46 loc) · 1.68 KB
/
factory.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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
package client
import (
"context"
"errors"
"fmt"
"os"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/config"
"go.opentelemetry.io/contrib/instrumentation/github.com/aws/aws-sdk-go-v2/otelaws"
"namespacelabs.dev/foundation/std/go/core"
"namespacelabs.dev/foundation/std/monitoring/tracing"
)
type ClientFactory struct {
openTelemetry tracing.DeferredTracerProvider
}
func (cf ClientFactory) NewWithCreds(ctx context.Context, optFns ...func(*config.LoadOptions) error) (aws.Config, error) {
tokenFile := os.Getenv("AWS_WEB_IDENTITY_TOKEN_FILE")
if tokenFile == "" {
return aws.Config{}, errors.New("AWS_WEB_IDENTITY_TOKEN_FILE is not set")
}
core.ZLog.Info().Msgf("[aws/client] using web identity credentials at %q", tokenFile)
return cf.New(ctx, optFns...)
}
func (cf ClientFactory) New(ctx context.Context, optFns ...func(*config.LoadOptions) error) (aws.Config, error) {
cfg, err := config.LoadDefaultConfig(ctx, optFns...)
if err != nil {
return aws.Config{}, fmt.Errorf("failed to create aws client config: %w", err)
}
if cf.openTelemetry != nil {
provider, err := cf.openTelemetry.GetTracerProvider()
if err != nil {
return aws.Config{}, err
}
if provider != nil {
otelaws.AppendMiddlewares(&cfg.APIOptions, otelaws.WithTracerProvider(provider))
}
}
return cfg, nil
}
func ProvideClientFactory(_ context.Context, _ *ClientFactoryArgs, deps ExtensionDeps) (ClientFactory, error) {
cf := ClientFactory{openTelemetry: deps.OpenTelemetry}
return cf, nil
}