add 'Since' to connection state

SENERGY-Platform · Feb 15, 2024 · 901ed32 · 901ed32
1 parent 399a0b3
commit 901ed32
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/controller/connectionstate.go b/lib/controller/connectionstate.go
@@ -19,6 +19,7 @@ package controller
 import (
 	"github.com/SENERGY-Platform/connection-log-worker/lib/model"
 	"gopkg.in/mgo.v2/bson"
+	"time"
 )
 
 func (this *Controller) setHubState(gatewayLog model.HubLog) (update bool, err error) {
@@ -30,7 +31,7 @@ func (this *Controller) setHubState(gatewayLog model.HubLog) (update bool, err e
 	}
 	update = count == 0
 	if update {
-		_, err = collection.Upsert(bson.M{"gateway": gatewayLog.Id}, HubState{Gateway: gatewayLog.Id, Online: gatewayLog.Connected})
+		_, err = collection.Upsert(bson.M{"gateway": gatewayLog.Id}, HubState{Gateway: gatewayLog.Id, Online: gatewayLog.Connected, Since: time.Now().Unix()})
 	}
 	return
 }
@@ -44,7 +45,7 @@ func (this *Controller) setDeviceState(deviceLog model.DeviceLog) (update bool,
 	}
 	update = count == 0
 	if update {
-		_, err = collection.Upsert(bson.M{"device": deviceLog.Id}, DeviceState{Device: deviceLog.Id, Online: deviceLog.Connected})
+		_, err = collection.Upsert(bson.M{"device": deviceLog.Id}, DeviceState{Device: deviceLog.Id, Online: deviceLog.Connected, Since: time.Now().Unix()})
 	}
 	return
 }

diff --git a/lib/controller/mongodb.go b/lib/controller/mongodb.go
@@ -57,9 +57,11 @@ func (this *Controller) getHubStateCollection() (session *mgo.Session, collectio
 type DeviceState struct {
 	Device string `json:"device,omitempty" bson:"device,omitempty"`
 	Online bool   `json:"online" bson:"online"`
+	Since  int64  `json:"since" bson:"since"`
 }
 
 type HubState struct {
 	Gateway string `json:"gateway,omitempty" bson:"gateway,omitempty"`
 	Online  bool   `json:"online" bson:"online"`
+	Since   int64  `json:"since" bson:"since"`
 }