/
ssm.go
61 lines (53 loc) · 1.36 KB
/
ssm.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
59
60
61
package aws
import (
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/aws/external"
"github.com/aws/aws-sdk-go-v2/service/ssm"
"github.com/kahgeh/devenv/utils/ctx"
)
type SsmSession struct {
api *ssm.Client
}
func NewSession() (*Session, error) {
cfg, err := external.LoadDefaultAWSConfig()
if err != nil {
panic("unable to load SDK config, " + err.Error())
}
session := &Session{config: cfg}
return session, nil
}
func (session *Session) NewSsmSession() *SsmSession {
api := ssm.New(session.config)
return &SsmSession{
api: api,
}
}
func (session *SsmSession) GetParameterValue(name string) (value *string, err error) {
api := session.api
request := api.GetParameterRequest(&ssm.GetParameterInput{
Name: aws.String(name),
WithDecryption: aws.Bool(true),
})
response, err := request.Send(ctx.GetContext())
if err != nil {
value = nil
return
}
value = response.GetParameterOutput.Parameter.Value
return
}
func (session *SsmSession) SaveParameter(name string, value string) (version *int64, err error) {
api := session.api
request := api.PutParameterRequest(&ssm.PutParameterInput{
Name: aws.String(name),
Value: &value,
Type: ssm.ParameterTypeString,
Overwrite: aws.Bool(true),
})
response, err := request.Send(ctx.GetContext())
if err != nil {
return
}
version = response.Version
return
}