diff --git a/internal/api/nats.go b/internal/api/nats.go index bbbd151f..c0a8c174 100644 --- a/internal/api/nats.go +++ b/internal/api/nats.go @@ -324,11 +324,12 @@ func (api *NatsAPI) processNodestateEvent(msg lp.CCMessage) { } repo := repository.GetNodeRepository() + requestReceived := time.Now().Unix() for _, node := range req.Nodes { state := determineState(node.States) nodeState := schema.NodeStateDB{ - TimeStamp: time.Now().Unix(), + TimeStamp: requestReceived, NodeState: state, CpusAllocated: node.CpusAllocated, MemoryAllocated: node.MemoryAllocated, diff --git a/internal/api/node.go b/internal/api/node.go index 4ad5337a..c3fe8492 100644 --- a/internal/api/node.go +++ b/internal/api/node.go @@ -63,11 +63,13 @@ func (api *RestAPI) updateNodeStates(rw http.ResponseWriter, r *http.Request) { return } repo := repository.GetNodeRepository() + requestReceived := time.Now().Unix() for _, node := range req.Nodes { state := determineState(node.States) nodeState := schema.NodeStateDB{ - TimeStamp: time.Now().Unix(), NodeState: state, + TimeStamp: requestReceived, + NodeState: state, CpusAllocated: node.CpusAllocated, MemoryAllocated: node.MemoryAllocated, GpusAllocated: node.GpusAllocated,