mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-24 06:46:39 +00:00
50 lines
967 B
Go
50 lines
967 B
Go
|
package p2p
|
||
|
|
||
|
import (
|
||
|
"sync"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
const defaultServicesID = "services_localai"
|
||
|
|
||
|
type NodeData struct {
|
||
|
Name string
|
||
|
ID string
|
||
|
TunnelAddress string
|
||
|
LastSeen time.Time
|
||
|
}
|
||
|
|
||
|
func (d NodeData) IsOnline() bool {
|
||
|
now := time.Now()
|
||
|
// if the node was seen in the last 40 seconds, it's online
|
||
|
return now.Sub(d.LastSeen) < 40*time.Second
|
||
|
}
|
||
|
|
||
|
var mu sync.Mutex
|
||
|
var nodes = map[string]map[string]NodeData{}
|
||
|
|
||
|
func GetAvailableNodes(serviceID string) []NodeData {
|
||
|
if serviceID == "" {
|
||
|
serviceID = defaultServicesID
|
||
|
}
|
||
|
mu.Lock()
|
||
|
defer mu.Unlock()
|
||
|
var availableNodes = []NodeData{}
|
||
|
for _, v := range nodes[serviceID] {
|
||
|
availableNodes = append(availableNodes, v)
|
||
|
}
|
||
|
return availableNodes
|
||
|
}
|
||
|
|
||
|
func AddNode(serviceID string, node NodeData) {
|
||
|
if serviceID == "" {
|
||
|
serviceID = defaultServicesID
|
||
|
}
|
||
|
mu.Lock()
|
||
|
defer mu.Unlock()
|
||
|
if nodes[serviceID] == nil {
|
||
|
nodes[serviceID] = map[string]NodeData{}
|
||
|
}
|
||
|
nodes[serviceID][node.ID] = node
|
||
|
}
|