/
read_ddb_client.go
111 lines (94 loc) · 3.19 KB
/
read_ddb_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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// https://github.com/awsdocs/aws-doc-sdk-examples/blob/master/go/example_code/dynamodb/read_item.go
// snippet-comment:[These are tags for the AWS doc team's sample catalog. Do not remove.]
// snippet-sourceauthor:[Doug-AWS]
// snippet-sourcedescription:[Gets an item from an Amazon DynamoDB table.]
// snippet-keyword:[Amazon DynamoDB]
// snippet-keyword:[GetItem function]
// snippet-keyword:[Go]
// snippet-sourcesyntax:[go]
// snippet-service:[dynamodb]
// snippet-keyword:[Code Sample]
// snippet-sourcetype:[full-example]
// snippet-sourcedate:[2018-03-16]
/*
Copyright 2010-2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
This file is licensed under the Apache License, Version 2.0 (the "License").
You may not use this file except in compliance with the License. A copy of
the License is located at
http://aws.amazon.com/apache2.0/
This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
CONDITIONS OF ANY KIND, either express or implied. See the License for the
specific language governing permissions and limitations under the License.
*/
package main
import (
"fmt"
"time"
"flag"
"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/aws/aws-sdk-go/service/dynamodb/dynamodbattribute"
)
// Create structs to hold info about new item
type ItemInfo struct {
Plot string `json:"plot"`
Rating float64 `json:"rating"`
}
type Item struct {
Year int `json:"year"`
Title string `json:"title"`
Info ItemInfo `json:"info"`
}
// const DDB_ENDPOINT = "dynamodb.ap-northeast-2.amazonaws.com"
var DDB_ENDPOINT string // "http://127.0.0.1:8000"
var elapsedTime = map[string]time.Time{}
var use_proxy = flag.Bool("use_proxy", true, "using proxy")
func main() {
// Initialize a session in us-west-2 that the SDK will use to load
// credentials from the shared credentials file ~/.aws/credentials.
elapsedTime["startTime"] = time.Now()
sess, err := session.NewSession(&aws.Config{
Region: aws.String("ap-northeast-2")},
)
// Create DynamoDB client
// svc := dynamodb.New(sess)
if *use_proxy {
DDB_ENDPOINT = "http://127.0.0.1:8000"
} else {
DDB_ENDPOINT = "dynamodb.ap-northeast-2.amazonaws.com"
}
svc := dynamodb.New(sess, aws.NewConfig().WithEndpoint(DDB_ENDPOINT))
result, err := svc.GetItem(&dynamodb.GetItemInput{
TableName: aws.String("Movies"),
Key: map[string]*dynamodb.AttributeValue{
"year": {
N: aws.String("2011"),
},
"title": {
S: aws.String("Samsara"),
},
},
})
if err != nil {
fmt.Println(err.Error())
return
}
item := Item{}
err = dynamodbattribute.UnmarshalMap(result.Item, &item)
if err != nil {
panic(fmt.Sprintf("Failed to unmarshal Record, %v", err))
}
if item.Title == "" {
fmt.Println("Could not find 'The Big New Movie' (2015)")
return
}
fmt.Println("Found item:")
fmt.Println("Year: ", item.Year)
fmt.Println("Title: ", item.Title)
fmt.Println("Plot: ", item.Info.Plot)
fmt.Println("Rating:", item.Info.Rating)
elapsedTime["endTime"] = time.Now()
etime := elapsedTime["endTime"].Sub(elapsedTime["startTime"]).Milliseconds() //unit: millisecond
fmt.Println("etime: ", etime)
}