diff --git a/events/event_port.go b/events/event_port.go index db3ed8d..6d470f1 100644 --- a/events/event_port.go +++ b/events/event_port.go @@ -5,6 +5,7 @@ import "loafle.com/overflow/collector/discovery/types" //////////////////////////////////////////////////////////////////////// Port Event Start type PortStartEvent struct { Zone int64 + Host int64 PortCount uint16 } @@ -12,7 +13,7 @@ func (i PortStartEvent) GetUrl() string { return PORT_START } -func NewPortStartEvent(zone int64, portCount uint16) *PortStartEvent { +func NewPortStartEvent(zone int64, host int64, portCount uint16) *PortStartEvent { return &PortStartEvent{ Zone: zone, PortCount: portCount, diff --git a/events/event_service.go b/events/event_service.go index 590e63f..199a541 100644 --- a/events/event_service.go +++ b/events/event_service.go @@ -1,6 +1,9 @@ package events type ServiceEndEvent struct { + Zone int64 + Host int64 + Port uint16 ServiceName string } @@ -8,6 +11,11 @@ func (i ServiceEndEvent) GetUrl() string { return SERVICE_END } -func NewServiceEndEvent(s string) *ServiceEndEvent { - return &ServiceEndEvent{ServiceName: s} +func NewServiceEndEvent(z int64, h int64, p uint16, s string) *ServiceEndEvent { + return &ServiceEndEvent{ + Zone: z, + Host: h, + Port: p, + ServiceName: s, + } }