-
Notifications
You must be signed in to change notification settings - Fork 3
/
google.go
102 lines (79 loc) · 1.93 KB
/
google.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
package google
import (
"context"
"cloud.google.com/go/datastore"
"google.golang.org/api/iterator"
"github.com/rfinochi/golang-workshop-todo/pkg/models"
)
const entityName string = "todoitem"
// ItemRepository godoc
type ItemRepository struct {
}
// GetItems godoc
func (ItemRepository) GetItems() (items []models.Item, err error) {
ctx, client, err := connnectToDatastore()
if err != nil {
return
}
query := datastore.NewQuery("todoitem").Order("id")
it := client.Run(ctx, query)
for {
var item models.Item
if _, err = it.Next(&item); err == iterator.Done {
err = nil
break
} else if err != nil {
return
}
items = append(items, item)
}
return
}
// GetItem godoc
func (ItemRepository) GetItem(id int) (item models.Item, err error) {
ctx, client, err := connnectToDatastore()
if err != nil {
return
}
key := datastore.IDKey(entityName, int64(id), nil)
err = client.Get(ctx, key, &item)
if err == datastore.ErrNoSuchEntity {
err = nil
}
return
}
// CreateItem godoc
func (ItemRepository) CreateItem(newItem models.Item) (err error) {
ctx, client, err := connnectToDatastore()
if err != nil {
return
}
key := datastore.IDKey(entityName, int64(newItem.ID), nil)
_, err = client.Put(ctx, key, &newItem)
return
}
// UpdateItem godoc
func (ItemRepository) UpdateItem(item models.Item) (err error) {
ctx, client, err := connnectToDatastore()
if err != nil {
return
}
key := datastore.IDKey(entityName, int64(item.ID), nil)
_, err = client.Put(ctx, key, &item)
return
}
// DeleteItem godoc
func (ItemRepository) DeleteItem(id int) (err error) {
ctx, client, err := connnectToDatastore()
if err != nil {
return
}
key := datastore.IDKey(entityName, int64(id), nil)
err = client.Delete(ctx, key)
return
}
func connnectToDatastore() (ctx context.Context, client *datastore.Client, err error) {
ctx = context.Background()
client, err = datastore.NewClient(ctx, "golang-workshop-todo")
return
}