-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.go
103 lines (84 loc) · 2.14 KB
/
db.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
package controllers
import (
"fmt"
"strconv"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/dynamodb"
"github.com/aws/aws-sdk-go/service/dynamodb/dynamodbattribute"
"github.com/spur-dev/api/models"
)
func (sc SessionController) GetVideoMetadata(vid string) (models.MetaData, error) {
result, err := sc.db.GetItem(&dynamodb.GetItemInput{
Key: map[string]*dynamodb.AttributeValue{
"VID": {
S: aws.String(vid),
},
},
TableName: aws.String(TableName),
})
if err != nil {
// log.Fatalln(err)
fmt.Println("Expecting error here")
fmt.Println(err)
return models.MetaData{}, err
}
v := models.MetaData{}
err = dynamodbattribute.UnmarshalMap(result.Item, &v)
return v, err
}
func (sc SessionController) CreateVideoMetadata(v models.MetaData) error {
// TODO: check if entry exists
fmt.Println("Putting video metadata")
fmt.Println(v)
_, err := sc.db.PutItem(&dynamodb.PutItemInput{ // This will override existing entry
Item: map[string]*dynamodb.AttributeValue{
"VID": {
S: aws.String(v.VID),
},
"Timestamp": {
N: aws.String(strconv.Itoa(v.Timestamp)),
},
"UID": {
S: aws.String(v.UID),
},
"State": {
S: aws.String(v.State),
},
},
TableName: &TableName,
})
return err
}
func (sc SessionController) UpdateVideoMetadataStatus(vid string, s string) error {
// TODO: check if entry exists
_, err := sc.db.UpdateItem(&dynamodb.UpdateItemInput{
ExpressionAttributeNames: map[string]*string{
"#S": aws.String("Status"),
},
ExpressionAttributeValues: map[string]*dynamodb.AttributeValue{
":Status": {
S: aws.String(s),
},
},
Key: map[string]*dynamodb.AttributeValue{
"VID": {
S: aws.String(vid),
},
},
TableName: &TableName,
UpdateExpression: aws.String("SET #S = :Status"),
})
return err
}
func (sc SessionController) DeleteVideoMetadata(id string) error {
// TODO: This needs to do further processing to also delete from raw bucket and final bucket
_, err := sc.db.DeleteItem(&dynamodb.DeleteItemInput{
Key: map[string]*dynamodb.AttributeValue{
"VID": {
S: aws.String(id),
},
},
TableName: &TableName,
})
return err
}