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
|
package dynamodb
import (
"errors"
"fmt"
simplejson "github.com/bitly/go-simplejson"
)
func (t *Table) Query(attributeComparisons []AttributeComparison) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
return runQuery(q, t)
}
func (t *Table) QueryOnIndex(attributeComparisons []AttributeComparison, indexName string) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddIndex(indexName)
return runQuery(q, t)
}
func (t *Table) QueryOnIndexDescending(attributeComparisons []AttributeComparison, indexName string) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddIndex(indexName)
q.ScanIndexDescending()
return runQuery(q, t)
}
func (t *Table) LimitedQuery(attributeComparisons []AttributeComparison, limit int64) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddLimit(limit)
return runQuery(q, t)
}
func (t *Table) LimitedQueryOnIndex(attributeComparisons []AttributeComparison, indexName string, limit int64) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddIndex(indexName)
q.AddLimit(limit)
return runQuery(q, t)
}
func (t *Table) LimitedQueryDescending(attributeComparisons []AttributeComparison, limit int64) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddLimit(limit)
q.ScanIndexDescending()
return runQuery(q, t)
}
func (t *Table) LimitedQueryOnIndexDescending(attributeComparisons []AttributeComparison, indexName string, limit int64) ([]map[string]*Attribute, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddIndex(indexName)
q.AddLimit(limit)
q.ScanIndexDescending()
return runQuery(q, t)
}
func (t *Table) CountQuery(attributeComparisons []AttributeComparison) (int64, error) {
q := NewQuery(t)
q.AddKeyConditions(attributeComparisons)
q.AddSelect("COUNT")
jsonResponse, err := t.Server.queryServer("DynamoDB_20120810.Query", q)
if err != nil {
return 0, err
}
json, err := simplejson.NewJson(jsonResponse)
if err != nil {
return 0, err
}
itemCount, err := json.Get("Count").Int64()
if err != nil {
return 0, err
}
return itemCount, nil
}
func runQuery(q *Query, t *Table) ([]map[string]*Attribute, error) {
jsonResponse, err := t.Server.queryServer("DynamoDB_20120810.Query", q)
if err != nil {
return nil, err
}
json, err := simplejson.NewJson(jsonResponse)
if err != nil {
return nil, err
}
itemCount, err := json.Get("Count").Int()
if err != nil {
message := fmt.Sprintf("Unexpected response %s", jsonResponse)
return nil, errors.New(message)
}
results := make([]map[string]*Attribute, itemCount)
for i, _ := range results {
item, err := json.Get("Items").GetIndex(i).Map()
if err != nil {
message := fmt.Sprintf("Unexpected response %s", jsonResponse)
return nil, errors.New(message)
}
results[i] = parseAttributes(item)
}
return results, nil
}
|
