Merge branch 'test' into dependabot/go_modules/github.com/ollama/olla…

…ma-0.2.5
masa-finance · Jul 17, 2024 · ed9840b · ed9840b
2 parents 1efc7c9 + c226563
commit ed9840b
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 29 deletions.
diff --git a/pkg/pubsub/node_data.go b/pkg/pubsub/node_data.go
@@ -99,25 +99,25 @@ func NewNodeData(addr multiaddr.Multiaddr, peerId peer.ID, publicKey string, act
 	}
 }
 
-// CalculateCurrentUptime calculates the current uptime based on Unix timestamps.
-func (n *NodeData) CalculateCurrentUptime() {
-	if n.Activity == ActivityJoined {
-		n.CurrentUptime = time.Duration(n.LastUpdatedUnix-n.LastJoinedUnix) * time.Second
-	} else {
-		n.CurrentUptime = 0
-	}
-	n.CurrentUptimeStr = n.CurrentUptime.String()
-}
+//// CalculateCurrentUptime calculates the current uptime based on Unix timestamps.
+//func (n *NodeData) CalculateCurrentUptime() {
+//	if n.Activity == ActivityJoined {
+//		n.CurrentUptime = time.Duration(n.LastUpdatedUnix-n.LastJoinedUnix) * time.Second
+//	} else {
+//		n.CurrentUptime = 0
+//	}
+//	n.CurrentUptimeStr = n.CurrentUptime.String()
+//}
 
-// CalculateAccumulatedUptime calculates the accumulated uptime based on Unix timestamps.
-func (n *NodeData) CalculateAccumulatedUptime() {
-	if n.FirstJoinedUnix > 0 && n.LastLeftUnix > 0 {
-		n.AccumulatedUptime = time.Duration(n.LastLeftUnix-n.FirstJoinedUnix) * time.Second
-	} else {
-		n.AccumulatedUptime = 0
-	}
-	n.AccumulatedUptimeStr = n.AccumulatedUptime.String()
-}
+//// CalculateAccumulatedUptime calculates the accumulated uptime based on Unix timestamps.
+//func (n *NodeData) CalculateAccumulatedUptime() {
+//	if n.FirstJoinedUnix > 0 && n.LastLeftUnix > 0 {
+//		n.AccumulatedUptime = time.Duration(n.LastLeftUnix-n.FirstJoinedUnix) * time.Second
+//	} else {
+//		n.AccumulatedUptime = 0
+//	}
+//	n.AccumulatedUptimeStr = n.AccumulatedUptime.String()
+//}
 
 // Address returns a string representation of the NodeData's multiaddress
 // and peer ID in the format "/ip4/127.0.0.1/tcp/4001/p2p/QmcgpsyWgH8Y8ajJz1Cu72KnS5uo2Aa2LpzU7kinSupNKC".
@@ -159,9 +159,6 @@ func (n *NodeData) Joined() {
 	n.Activity = ActivityJoined
 	n.IsActive = true
 
-	n.CalculateCurrentUptime()
-	n.CalculateAccumulatedUptime()
-
 	n.Version = config.Version[1:]
 
 	logMessage := fmt.Sprintf("[+] %s node joined: %s", map[bool]string{true: "Staked", false: "Unstaked"}[n.IsStaked], n.Address())
@@ -181,13 +178,14 @@ func (n *NodeData) Left() {
 	}
 	n.LastLeftUnix = time.Now().Unix()
 	n.LastUpdatedUnix = n.LastLeftUnix
-	n.AccumulatedUptime += n.GetCurrentUptime()
 	n.CurrentUptime = 0
 	n.Activity = ActivityLeft
 	n.IsActive = false
+	// call this after setting activity flags
+	n.UpdateAccumulatedUptime()
 
-	n.CalculateCurrentUptime()
-	n.CalculateAccumulatedUptime()
+	//n.CalculateCurrentUptime()
+	//n.CalculateAccumulatedUptime()
 
 	logMessage := fmt.Sprintf("Node left: %s", n.Address())
 	if n.IsStaked {
@@ -220,10 +218,27 @@ func (n *NodeData) GetAccumulatedUptime() time.Duration {
 // Otherwise, it uses the time since the last joined event.
 func (n *NodeData) UpdateAccumulatedUptime() {
 	if n.Activity == ActivityLeft {
-		n.AccumulatedUptime += time.Since(time.Unix(n.LastLeftUnix, 0))
-		return
+		// Calculate the uptime for the most recent active period
+		recentUptime := n.LastLeftUnix - n.LastJoinedUnix
+		// Add this to the accumulated uptime
+		n.AccumulatedUptime += time.Duration(recentUptime) * time.Second
+	} else if n.Activity == ActivityJoined {
+		// If the node is currently active, calculate the uptime since it first joined
+		// This should only be done if the node is active and hasn't been updated yet
+		currentUptime := time.Now().Unix() - n.FirstJoinedUnix
+		// Update the accumulated uptime only if it's less than the current uptime
+		if currentUptime > int64(n.AccumulatedUptime.Seconds()) {
+			n.AccumulatedUptime = time.Duration(currentUptime) * time.Second
+		}
 	}
-	n.AccumulatedUptime += time.Since(time.Unix(n.LastJoinedUnix, 0))
+	// Ensure the accumulated uptime does not exceed the maximum possible uptime
+	if n.FirstJoinedUnix > 0 && n.LastLeftUnix > 0 {
+		maxAccumulatedUptime := time.Duration(n.LastLeftUnix-n.FirstJoinedUnix) * time.Second
+		if n.AccumulatedUptime > maxAccumulatedUptime {
+			n.AccumulatedUptime = maxAccumulatedUptime
+		}
+	}
+	n.AccumulatedUptimeStr = n.AccumulatedUptime.String()
 }
 
 // GetSelfNodeDataJson converts the local node's data into a JSON byte array.

diff --git a/pkg/pubsub/node_event_tracker.go b/pkg/pubsub/node_event_tracker.go
@@ -363,7 +363,6 @@ func (net *NodeEventTracker) AddOrUpdateNodeData(nodeData *NodeData, forceGossip
 		nd.Records = nodeData.Records
 		nd.Multiaddrs = nodeData.Multiaddrs
 		nd.EthAddress = nodeData.EthAddress
-		nd.IsActive = nodeData.IsActive
 
 		if nd.EthAddress == "" && nodeData.EthAddress != "" {
 			dataChanged = true
@@ -386,7 +385,7 @@ func (net *NodeEventTracker) AddOrUpdateNodeData(nodeData *NodeData, forceGossip
 		}
 
 		nd.LastUpdatedUnix = nodeData.LastUpdatedUnix
-		net.nodeData.Set(nodeData.PeerId.String(), nodeData)
+		net.nodeData.Set(nodeData.PeerId.String(), nd)
 	}
 	return nil
 }
@@ -401,6 +400,8 @@ func (net *NodeEventTracker) ClearExpiredBufferEntries() {
 		now := time.Now()
 		for peerID, entry := range net.ConnectBuffer {
 			if now.Sub(entry.ConnectTime) > time.Minute*1 {
+				// first force a leave event so that timestamps are updated properly
+				entry.NodeData.Left()
 				// Buffer period expired without a disconnect, process connect
 				entry.NodeData.Joined()
 				net.NodeDataChan <- entry.NodeData