/
update.go
108 lines (90 loc) · 2.7 KB
/
update.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
package graphql
import (
"context"
"strings"
"github.com/graphql-go/graphql/language/ast"
"github.com/spaceuptech/space-cloud/model"
"github.com/spaceuptech/space-cloud/utils"
)
func (graph *Module) execUpdateRequest(ctx context.Context, field *ast.Field, token string, store utils.M) (map[string]interface{}, error) {
dbType, err := graph.GetDBAlias(field)
if err != nil {
return nil, err
}
col := strings.TrimPrefix(field.Name.Value, "update_")
req, err := generateUpdateRequest(field, store)
if err != nil {
return nil, err
}
status, err := graph.auth.IsUpdateOpAuthorised(ctx, graph.project, dbType, col, token, req)
if err != nil {
return nil, err
}
return utils.M{"status": status}, graph.crud.Update(ctx, dbType, graph.project, col, req)
}
func (graph *Module) genrateUpdateReq(ctx context.Context, field *ast.Field, token string, store map[string]interface{}) (*model.AllRequest, error) {
dbType, err := graph.GetDBAlias(field)
if err != nil {
return nil, err
}
col := strings.TrimPrefix(field.Name.Value, "update_")
req, err := generateUpdateRequest(field, store)
if err != nil {
return nil, err
}
_, err = graph.auth.IsUpdateOpAuthorised(ctx, graph.project, dbType, col, token, req)
if err != nil {
return nil, err
}
return generateUpdateAllRequest(req), nil
}
func generateUpdateAllRequest(req *model.UpdateRequest) *model.AllRequest {
return &model.AllRequest{Operation: req.Operation, Find: req.Find, Update: req.Update}
}
func extractUpdateOperation(args []*ast.Argument, store utils.M) (string, error) {
for _, v := range args {
switch v.Name.Value {
case "op":
temp, err := ParseValue(v.Value, store)
if err != nil {
return "", err
}
if temp.(string) == "upsert" {
return utils.Upsert, nil
}
return utils.All, nil
}
}
return utils.All, nil
}
func generateUpdateRequest(field *ast.Field, store utils.M) (*model.UpdateRequest, error) {
var err error
var updateRequest model.UpdateRequest
updateRequest.Operation, err = extractUpdateOperation(field.Arguments, store)
if err != nil {
return nil, err
}
updateRequest.Find, err = ExtractWhereClause(field.Arguments, store)
if err != nil {
return nil, err
}
updateRequest.Update, err = extractUpdateArgs(field.Arguments, store)
if err != nil {
return nil, err
}
return &updateRequest, nil
}
func extractUpdateArgs(args []*ast.Argument, store utils.M) (utils.M, error) {
t := map[string]interface{}{}
for _, v := range args {
switch v.Name.Value {
case "set", "inc", "mul", "max", "min", "currentTimestamp", "currentDate", "push", "rename", "remove":
temp, err := ParseValue(v.Value, store)
if err != nil {
return nil, err
}
t["$"+v.Name.Value] = temp
}
}
return t, nil
}