-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.go
executable file
·86 lines (62 loc) · 1.74 KB
/
main.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
package main
import (
"context"
"sync"
"time"
"github.com/aws/aws-lambda-go/events"
"github.com/aws/aws-lambda-go/lambda"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/dynamodb"
"github.com/aws/aws-xray-sdk-go/xray"
"github.com/hashicorp/go-retryablehttp"
"github.com/kazimanzurrashid/aws-scheduler-go/worker/services"
)
var (
httpClient services.Client
database services.Storage
)
func handler(ctx context.Context, e events.DynamoDBEvent) error {
var queuedRecords []events.DynamoDBEventRecord
for _, record := range e.Records {
if record.EventName != "MODIFY" {
continue
}
if status := record.Change.NewImage["status"]; status.String() != services.ScheduleStatusQueued {
continue
}
queuedRecords = append(queuedRecords, record)
}
if len(queuedRecords) == 0 {
return nil
}
var wg sync.WaitGroup
uis := make([]*services.UpdateInput, len(queuedRecords))
for i, record := range queuedRecords {
wg.Add(1)
go func(attrs map[string]events.DynamoDBAttributeValue, index int) {
defer wg.Done()
ri := services.CreateRequestInput(attrs)
ui := services.CreateUpdateInput(attrs)
ui.StartedAt = aws.Int64(time.Now().Unix())
ro := httpClient.Request(ctx, ri)
ui.Status = ro.Status
ui.Result = aws.String(ro.Result)
ui.CompletedAt = aws.Int64(time.Now().Unix())
uis[index] = ui
}(record.Change.NewImage, i)
}
wg.Wait()
return database.Update(ctx, uis)
}
func init() {
ses := session.Must(session.NewSession())
ddbc := dynamodb.New(ses)
xray.AWS(ddbc.Client)
database = services.NewDatabase(ddbc)
httpClient = services.NewHttpClient(
xray.Client(retryablehttp.NewClient().StandardClient()))
}
func main() {
lambda.Start(handler)
}