-
Notifications
You must be signed in to change notification settings - Fork 0
/
modify.go
115 lines (103 loc) · 2.69 KB
/
modify.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
package modify
import (
"cloud.google.com/go/logging"
"cloud.google.com/go/storage"
"context"
"fmt"
cloudevents "github.com/cloudevents/sdk-go/v2"
"log"
"strconv"
"strings"
)
const secondsInYear = 31536000
type App struct {
logger *logging.Logger
storageClient *storage.Client
}
func NewApp(logger *logging.Logger, bucket *storage.Client) *App {
return &App{
logger: logger,
storageClient: bucket,
}
}
func GcsUploadEvent(a *App) func(context.Context, cloudevents.Event) error {
return func(ctx context.Context, event cloudevents.Event) error {
defer func() {
err := a.logger.Flush()
if err != nil {
log.Printf("failed to flush logger %v", err)
}
}()
eventData, err := extractEventData(event)
if err != nil {
a.logger.Log(logging.Entry{
Severity: logging.Error,
Payload: fmt.Sprintf("failed to extract event data - %v", err),
})
return err
}
err = validateEventData(eventData)
if err != nil {
a.logger.Log(logging.Entry{
Severity: logging.Error,
Payload: fmt.Sprintf("invalid event data - %v", err),
})
return err
}
a.logger.Log(logging.Entry{
Severity: logging.Debug,
Payload: fmt.Sprintf("processing object %s/%s", eventData.Bucket, eventData.Name),
})
if !strings.Contains(eventData.Name, "images/") {
a.logger.Log(logging.Entry{
Severity: logging.Info,
Payload: fmt.Sprintf("skipping object %s/%s", eventData.Bucket, eventData.Name),
})
return nil
}
bucket := a.storageClient.Bucket(eventData.Bucket)
object := bucket.Object(eventData.Name)
object.BucketName()
_, err = object.Attrs(ctx)
if err != nil {
if err.Error() == "storage: object doesn't exist" {
a.logger.Log(logging.Entry{
Severity: logging.Info,
Payload: fmt.Sprintf("object %s/%s doesn't exist", eventData.Bucket, eventData.Name),
})
// Object was probably deleted, we don't need to worry about it anymore.
return nil
}
a.logger.Log(logging.Entry{
Severity: logging.Error,
Payload: fmt.Sprintf(
"failed reading object %s/%s attributes - %v",
eventData.Bucket,
eventData.Name,
err,
),
})
return err
}
_, err = object.Update(ctx, storage.ObjectAttrsToUpdate{
CacheControl: fmt.Sprintf("public, max-age=%s, immutable", strconv.Itoa(secondsInYear)),
})
if err != nil {
a.logger.Log(logging.Entry{
Severity: logging.Error,
Payload: fmt.Sprintf(
"failed updating object %s/%s - %v",
eventData.Bucket,
eventData.Name,
err,
),
})
return err
}
a.logger.Log(logging.Entry{
Severity: logging.Info,
Payload: fmt.Sprintf("metadata set for object %s/%s", eventData.Bucket, eventData.Name),
})
return nil
}
}