This repository has been archived by the owner on Jun 19, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 75
/
storage.go
71 lines (60 loc) · 2.19 KB
/
storage.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
/*
Copyright 2019 Google LLC
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package converters
import (
"context"
"errors"
"fmt"
"cloud.google.com/go/pubsub"
cev2 "github.com/cloudevents/sdk-go/v2"
schemasv1 "github.com/google/knative-gcp/pkg/schemas/v1"
)
var (
// Mapping of GCS eventTypes to CloudEvent types.
storageEventTypes = map[string]string{
"OBJECT_FINALIZE": schemasv1.CloudStorageObjectFinalizedEventType,
"OBJECT_ARCHIVE": schemasv1.CloudStorageObjectArchivedEventType,
"OBJECT_DELETE": schemasv1.CloudStorageObjectDeletedEventType,
"OBJECT_METADATA_UPDATE": schemasv1.CloudStorageObjectMetadataUpdatedEventType,
}
)
func convertCloudStorage(ctx context.Context, msg *pubsub.Message) (*cev2.Event, error) {
event := cev2.NewEvent(cev2.VersionV1)
event.SetID(msg.ID)
event.SetTime(msg.PublishTime)
event.SetDataSchema(schemasv1.CloudStorageEventDataSchema)
// TODO: figure out if we want to continue to add these as extensions.
if val, ok := msg.Attributes["bucketId"]; ok {
event.SetSource(schemasv1.CloudStorageEventSource(val))
} else {
return nil, errors.New("received event did not have bucketId")
}
if val, ok := msg.Attributes["objectId"]; ok {
event.SetSubject(schemasv1.CloudStorageEventSubject(val))
} else {
return nil, errors.New("received event did not have objectId")
}
if val, ok := msg.Attributes["eventType"]; ok {
if eventType, ok := storageEventTypes[val]; ok {
event.SetType(eventType)
} else {
return nil, fmt.Errorf("unknown event type %s", val)
}
} else {
return nil, errors.New("received event did not have eventType")
}
if err := event.SetData(cev2.ApplicationJSON, msg.Data); err != nil {
return nil, err
}
return &event, nil
}