/
instance.go
46 lines (37 loc) · 975 Bytes
/
instance.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
package secretmanager
import (
"context"
"reflect"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/config"
"github.com/aws/aws-sdk-go-v2/service/secretsmanager"
)
type secretManager struct {
client *secretsmanager.Client
}
type SecretManager interface {
GetSecretString(secretName string) (*string, error)
}
func NewInstance(region string, awsConfig aws.Config) (sm SecretManager, err error) {
if reflect.DeepEqual(awsConfig, aws.Config{}) {
if awsConfig, err = config.LoadDefaultConfig(
context.TODO(),
config.WithRegion(region),
); err != nil {
return nil, err
}
}
return secretManager{
client: secretsmanager.NewFromConfig(awsConfig),
}, err
}
func (s secretManager) GetSecretString(secretName string) (*string, error) {
secret, err := s.client.GetSecretValue(
context.TODO(),
&secretsmanager.GetSecretValueInput{SecretId: &secretName},
)
if err != nil {
return nil, err
}
return secret.SecretString, nil
}