/
DynamoDBLoadItems.go
111 lines (92 loc) · 3.08 KB
/
DynamoDBLoadItems.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
// snippet-comment:[These are tags for the AWS doc team's sample catalog. Do not remove.]
// snippet-sourceauthor:[Doug-AWS]
// snippet-sourcedescription:[Adds items from a JSON file to an Amazon DynamoDB table.]
// snippet-keyword:[Amazon DynamoDB]
// snippet-keyword:[PutItem 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 (
"encoding/json"
"fmt"
"io/ioutil"
"os"
"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"`
}
// Get table items from JSON file
func getItems() []Item {
raw, err := ioutil.ReadFile("./movie_data.json")
if err != nil {
fmt.Println(err.Error())
os.Exit(1)
}
var items []Item
json.Unmarshal(raw, &items)
return items
}
const DDB_ENDPOINT = "http://127.0.0.1:8000"
func main() {
// Initialize a session in us-west-2 that the SDK will use to load
// credentials from the shared credentials file ~/.aws/credentials.
sess, err := session.NewSession(&aws.Config{
Region: aws.String("ap-northeast-2")},
)
// Create DynamoDB client
// svc := dynamodb.New(sess)
svc := dynamodb.New(sess, aws.NewConfig().WithEndpoint(DDB_ENDPOINT))
if err != nil {
fmt.Println("Error creating session:")
fmt.Println(err.Error())
os.Exit(1)
}
// Get table items from movie_data.json
items := getItems()
// Add each item to Movies table:
for _, item := range items {
av, err := dynamodbattribute.MarshalMap(item)
if err != nil {
fmt.Println("Got error marshalling map:")
fmt.Println(err.Error())
os.Exit(1)
}
// Create item in table Movies
input := &dynamodb.PutItemInput{
Item: av,
TableName: aws.String("Movies"),
}
_, err = svc.PutItem(input)
if err != nil {
fmt.Println("Got error calling PutItem:")
fmt.Println(err.Error())
os.Exit(1)
}
fmt.Println("Successfully added '",item.Title,"' (",item.Year,") to Movies table")
}
}