util.go

package main

import (
	"crypto/rand"
	"fmt"
	"sync"
)

func CreateClusters(shards, clusters int) {
	shardIds := make([]int, 0, shards)
	for i := 0; i < shards; i++ {
		shardIds = append(shardIds, i)
	}
	avgShardsPerCluster := shards / clusters
	for i := 0; i < len(shardIds); i += avgShardsPerCluster {
		Server.Clients = append(Server.Clients, &Cluster{
			ID:         len(Server.Clients),
			Client:     nil,
			PingRecv:   false,
			Block:      ClusterBlock{Shards: shardIds[i : i+avgShardsPerCluster], Total: Config.Shards},
			State:      ClusterWaiting,
			pingTicker: nil,
			mutex:      &sync.Mutex{},
			statsChan:  make(chan map[string]interface{}),
		})
	}
}

func RandomID() string {
	bytes := make([]byte, 4)
	_, err := rand.Read(bytes)
	if err != nil {
		return ""
	}
	return fmt.Sprintf("%x", bytes)
}

func NextClusterID() int {
	for index, cluster := range Server.Clients {
		if cluster.State == ClusterWaiting {
			return index
		}
	}
	return -1
}

func GetHealthyClusters() int {
	healthy := 0
	for _, cluster := range Server.Clients {
		if cluster.State == ClusterReady {
			healthy++
		}
	}
	return healthy
}