40 lines
736 B
Go
40 lines
736 B
Go
package supervisor
|
|
|
|
type StatsEvent struct {
|
|
s *Supervisor
|
|
}
|
|
|
|
type UnsubscribeStatsEvent struct {
|
|
s *Supervisor
|
|
}
|
|
|
|
type StopStatsEvent struct {
|
|
s *Supervisor
|
|
}
|
|
|
|
func (h *StatsEvent) Handle(e *Event) error {
|
|
i, ok := h.s.containers[e.ID]
|
|
if !ok {
|
|
return ErrContainerNotFound
|
|
}
|
|
e.Stats = h.s.statsCollector.collect(i.container)
|
|
return nil
|
|
}
|
|
|
|
func (h *UnsubscribeStatsEvent) Handle(e *Event) error {
|
|
i, ok := h.s.containers[e.ID]
|
|
if !ok {
|
|
return ErrContainerNotFound
|
|
}
|
|
h.s.statsCollector.unsubscribe(i.container, e.Stats)
|
|
return nil
|
|
}
|
|
|
|
func (h *StopStatsEvent) Handle(e *Event) error {
|
|
i, ok := h.s.containers[e.ID]
|
|
if !ok {
|
|
return ErrContainerNotFound
|
|
}
|
|
h.s.statsCollector.stopCollection(i.container)
|
|
return nil
|
|
}
|