-
Notifications
You must be signed in to change notification settings - Fork 0
/
schema.resolvers.go
128 lines (108 loc) · 3.52 KB
/
schema.resolvers.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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package graph
// This file will be automatically regenerated based on the schema, any resolver implementations
// will be copied through when generating and any unknown code will be moved to the end.
import (
"context"
"errors"
"sync"
"github.com/jorgerasillo/spamhouse/graph/generated"
"github.com/jorgerasillo/spamhouse/graph/model"
"github.com/jorgerasillo/spamhouse/pkg/spamhous"
"github.com/jorgerasillo/spamhouse/repo/db"
"github.com/sirupsen/logrus"
)
func (r *mutationResolver) Enqueue(ctx context.Context, input []string) (*model.IPAddressResult, error) {
var wg sync.WaitGroup
qChan := make(chan *db.IPAddress)
for _, i := range input {
ip, err := db.NewIP(i)
r.Logger.WithFields(logrus.Fields{"ip": i, "uuid": ip.UUID}).Debug("processing input")
if err != nil {
r.Logger.WithFields(logrus.Fields{
"err": err,
"ip": ip,
}).Error("unable to create new ip address")
}
wg.Add(1)
go spamhous.Query(&ip, qChan, &wg)
}
go func() {
wg.Wait()
close(qChan)
}()
addresses := make([]*model.IPAddress, 0)
for val := range qChan {
logEntry := r.Logger.WithFields(logrus.Fields{"ip": val.IP, "uuid": val.UUID})
logEntry.Debug("ip received")
// query if it exists
oldIP, err := r.Repository.GetIP(val.String())
if err != nil {
// check if not found
if errors.Is(err, db.ErrRecordNotFound) {
logEntry.Debug("record was not found, creating new entry")
ipAddress, err := r.Repository.AddIP(*val)
if err != nil {
r.Logger.WithFields(logrus.Fields{
"err": err,
"ip": val,
}).Error("unable to add new ip address")
continue
}
r.Logger.WithFields(logrus.Fields{
"ip": ipAddress.IP,
"uuid": ipAddress.UUID}).Debug("new record added")
addresses = append(addresses, modelToResponse(ipAddress))
}
logEntry.Debug("db error attempting to look for record")
continue
}
oldIP.ResponseCode = val.ResponseCode
r.Logger.WithFields(logrus.Fields{
"ip": oldIP.IP,
"uuid": oldIP.UUID}).Debug("record existed, updating entry")
// save if it does
ipAddress, err := r.Repository.UpdateIP(oldIP)
if err != nil {
r.Logger.WithFields(logrus.Fields{
"err": err,
"ip": val,
}).Error("unable to update ip address")
continue
}
r.Logger.WithFields(logrus.Fields{
"ip": ipAddress.IP,
"uuid": ipAddress.UUID}).Debug("updated record")
addresses = append(addresses, modelToResponse(ipAddress))
}
return &model.IPAddressResult{
Node: addresses,
}, nil
}
func (r *queryResolver) GetIPDetails(ctx context.Context, input string) (*model.IPAddressResult, error) {
ipAddress, err := r.Repository.GetIP(input)
if err != nil {
r.Logger.WithField("err", err).Debug("Error while retrieving ip address")
return &model.IPAddressResult{
Message: "ip not found",
}, err
}
responses := make([]*model.IPAddress, 0)
modelToResponse := model.IPAddress{
UUID: ipAddress.UUID,
CreatedAt: ipAddress.CreatedAt,
UpdatedAt: ipAddress.UpdatedAt,
ResponseCode: ipAddress.ResponseCode,
IPAddress: ipAddress.IP,
}
responses = append(responses, &modelToResponse)
return &model.IPAddressResult{
Message: "success",
Node: responses,
}, nil
}
// Mutation returns generated.MutationResolver implementation.
func (r *Resolver) Mutation() generated.MutationResolver { return &mutationResolver{r} }
// Query returns generated.QueryResolver implementation.
func (r *Resolver) Query() generated.QueryResolver { return &queryResolver{r} }
type mutationResolver struct{ *Resolver }
type queryResolver struct{ *Resolver }