.

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,

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,
+	}
 }