/
client.go
76 lines (55 loc) · 1.54 KB
/
client.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
66
67
68
69
70
71
72
73
74
75
76
package dynamodb
// Move this in to aaronland/go-aws-dynamodb
import (
"context"
"fmt"
aa_session "github.com/aaronland/go-aws-session"
"github.com/aws/aws-sdk-go/aws"
aws_session "github.com/aws/aws-sdk-go/aws/session"
aws_dynamodb "github.com/aws/aws-sdk-go/service/dynamodb"
"net/url"
"os"
"strconv"
)
func NewSessionWithURI(ctx context.Context, uri string) (*aws_session.Session, error) {
u, err := url.Parse(uri)
if err != nil {
return nil, fmt.Errorf("Failed to parse URI, %w", err)
}
q := u.Query()
region := q.Get("region")
credentials := q.Get("credentials")
local := q.Get("local")
is_local := false
if local != "" {
l, err := strconv.ParseBool(local)
if err != nil {
return nil, fmt.Errorf("Failed to parse ?local parameter, %w", err)
}
is_local = l
}
if is_local {
os.Setenv("AWS_ACCESS_KEY_ID", "DUMMYIDEXAMPLE")
os.Setenv("AWS_SECRET_ACCESS_KEY", "DUMMYEXAMPLEKEY")
credentials = "env:"
region = "us-east-1"
}
dsn := fmt.Sprintf("credentials=%s region=%s", credentials, region)
sess, err := aa_session.NewSessionWithDSN(dsn)
if err != nil {
return nil, fmt.Errorf("Failed to create new session, %w", err)
}
if is_local {
endpoint := "http://localhost:8000"
sess.Config.Endpoint = aws.String(endpoint)
}
return sess, nil
}
func NewClientWithURI(ctx context.Context, uri string) (*aws_dynamodb.DynamoDB, error) {
sess, err := NewSessionWithURI(ctx, uri)
if err != nil {
return nil, fmt.Errorf("Failed to create session, %w", err)
}
client := aws_dynamodb.New(sess)
return client, nil
}