forked from AliyunContainerService/pouch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
events.go
111 lines (91 loc) · 3.72 KB
/
events.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
package mgr
import (
"context"
"strings"
"github.com/alibaba/pouch/apis/types"
"github.com/docker/libnetwork"
)
// LogContainerEvent generates an event related to a container with only the default attributes.
func (mgr *ContainerManager) LogContainerEvent(ctx context.Context, container *Container, action string) {
mgr.LogContainerEventWithAttributes(ctx, container, action, map[string]string{})
}
// LogContainerEventWithAttributes generates an event related to a container with specific given attributes.
func (mgr *ContainerManager) LogContainerEventWithAttributes(ctx context.Context, container *Container, action string, attributes map[string]string) {
copyAttributes(attributes, container.Config.Labels)
if container.Config.Image != "" {
attributes["image"] = container.Config.Image
}
attributes["name"] = strings.TrimLeft(container.Name, "/")
actor := &types.EventsActor{
ID: container.ID,
Attributes: attributes,
}
_ = mgr.eventsService.Publish(ctx, action, types.EventTypeContainer, actor)
}
// LogVolumeEvent generates an event related to a volume
func (vm *VolumeManager) LogVolumeEvent(ctx context.Context, volumeID, action string, attributes map[string]string) {
actor := &types.EventsActor{
ID: volumeID,
Attributes: attributes,
}
_ = vm.eventsService.Publish(ctx, action, types.EventTypeVolume, actor)
}
// LogNetworkEvent generates an event related to a network with only the default attributes
func (nm *NetworkManager) LogNetworkEvent(ctx context.Context, nw libnetwork.Network, action string) {
attributes := map[string]string{}
attributes["name"] = nw.Name()
attributes["type"] = nw.Type()
actor := &types.EventsActor{
ID: nw.ID(),
Attributes: attributes,
}
_ = nm.eventsService.Publish(ctx, action, types.EventTypeNetwork, actor)
}
// LogNetworkEventWithAttributes generates an event related to a network with specific given attributes
// Use ContainerManager to publish network event may be a little bit ugly now
func (mgr *ContainerManager) LogNetworkEventWithAttributes(ctx context.Context, nw libnetwork.Network, action string, attributes map[string]string) {
attributes["name"] = nw.Name()
attributes["type"] = nw.Type()
actor := &types.EventsActor{
ID: nw.ID(),
Attributes: attributes,
}
_ = mgr.eventsService.Publish(ctx, action, types.EventTypeNetwork, actor)
}
// LogImageEvent generates an event related to an image with only the default attributes
func (mgr *ImageManager) LogImageEvent(ctx context.Context, imageID, refName, action string) {
mgr.LogImageEventWithAttributes(ctx, imageID, refName, action, map[string]string{})
}
// LogImageEventWithAttributes generates an event related to an image with specific given attributes
func (mgr *ImageManager) LogImageEventWithAttributes(ctx context.Context, imageID, refName, action string, attributes map[string]string) {
img, err := mgr.GetImage(ctx, imageID)
if err == nil && img.Config != nil {
copyAttributes(attributes, img.Config.Labels)
}
if refName != "" {
attributes["Name"] = refName
}
actor := &types.EventsActor{
ID: imageID,
Attributes: attributes,
}
_ = mgr.eventsService.Publish(ctx, action, types.EventTypeImage, actor)
}
// copyAttributes guarantees that labels are not mutated by event triggers.
func copyAttributes(attributes, labels map[string]string) {
if labels == nil {
return
}
for k, v := range labels {
attributes[k] = v
}
}
// publishContainerdEvent sends containerd events to pouchd event service.
func (mgr *ContainerManager) publishContainerdEvent(ctx context.Context, id, action string, attributes map[string]string) error {
c, err := mgr.container(id)
if err != nil {
return err
}
mgr.LogContainerEventWithAttributes(ctx, c, action, attributes)
return nil
}