-
Notifications
You must be signed in to change notification settings - Fork 163
/
aws.go
65 lines (54 loc) · 1.29 KB
/
aws.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
62
63
64
65
package datasource
import (
"context"
"fmt"
"net/url"
"os"
"strings"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/dynamodb"
"github.com/k1LoW/tbls/drivers/dynamo"
"github.com/k1LoW/tbls/schema"
)
// AnalizeDynamodb analyze `dynamodb://`
func AnalyzeDynamodb(urlstr string) (*schema.Schema, error) {
s := &schema.Schema{}
u, err := url.Parse(urlstr)
if err != nil {
return s, err
}
values := u.Query()
err = setEnvAWSCredentials(values)
if err != nil {
return s, err
}
region := u.Host
sess := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
config := aws.NewConfig().WithRegion(region)
if os.Getenv("AWS_ENDPOINT_URL") != "" {
config = config.WithEndpoint(os.Getenv("AWS_ENDPOINT_URL"))
}
client := dynamodb.New(sess, config)
ctx := context.Background()
driver, err := dynamo.New(ctx, client)
if err != nil {
return s, err
}
s.Name = fmt.Sprintf("Amazon DynamoDB (%s)", region)
err = driver.Analyze(s)
if err != nil {
return s, err
}
return s, nil
}
func setEnvAWSCredentials(values url.Values) error {
for k := range values {
if strings.HasPrefix(k, "aws_") {
return os.Setenv(strings.ToUpper(k), values.Get(k))
}
}
return nil
}