mirror of https://github.com/portainer/portainer
36 lines
615 B
Go
36 lines
615 B
Go
|
package status
|
||
|
|
|
||
|
|
import (
|
||
|
|
portainer "github.com/portainer/portainer/api"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NodesCount returns the total node number of all environments
|
||
|
|
func NodesCount(endpoints []portainer.Endpoint) int {
|
||
|
|
nodes := 0
|
||
|
|
for _, env := range endpoints {
|
||
|
|
nodes += countNodes(&env)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nodes
|
||
|
|
}
|
||
|
|
|
||
|
|
func countNodes(endpoint *portainer.Endpoint) int {
|
||
|
|
if len(endpoint.Snapshots) == 1 {
|
||
|
|
return max(endpoint.Snapshots[0].NodeCount, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(endpoint.Kubernetes.Snapshots) == 1 {
|
||
|
|
return max(endpoint.Kubernetes.Snapshots[0].NodeCount, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1
|
||
|
|
}
|
||
|
|
|
||
|
|
func max(a, b int) int {
|
||
|
|
if a > b {
|
||
|
|
return a
|
||
|
|
}
|
||
|
|
|
||
|
|
return b
|
||
|
|
}
|