Added phased startup string function for bringing services back online gradually

SnowcloakUtils.go

@@ -1,26 +0,0 @@
-package SnowcloakUtils
-
-import (
-	"crypto/rand" // Probably better than math/rand for this
-	"log"
-)
-
-func GenerateRandomString(length int) string {
-	// C# had optional parameters that allowed lowercase for chardata and gpose lobbies, Go doesn't.
-	// We can probably get away with just uppercase.
-	allowedArray := []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
-	random := make([]byte, length)
-	_, err := rand.Read(random)
-	if err != nil {
-		log.Fatalf("Couldn't generate secure slice: %s", err)
-	}
-	allowedLength := len(allowedArray)
-	i := 0
-	charArray := make([]rune, length)
-	for i < length {
-		charArray[i] = allowedArray[int(random[i])%allowedLength] // Casting random[i] due to type mismatch
-		i++
-	}
-
-	return string(charArray)
-}

files/files.go

@@ -0,0 +1,48 @@
+package files
+
+import (
+	"github.com/showwin/speedtest-go/speedtest"
+)
+
+type ConnectionStatistics struct {
+	downloadBytes uint64 // Server download speed in bytes
+	uploadBytes   uint64 // Upload speed in bytes
+	downloadSlots uint16 // How fany files can be downloaded by the server at once before queueing kicks in
+	uploadSlots   uint16 // How many uploads we can do at once
+
+	// Slots are calculated with a minimum acceptable speed of 128KiB/s per slot
+	// Using uint16 for this MIGHT get a bit munted if someone tries to use something faster than a 25gbit connection
+	// or something but I really, really doubt anyone is going to do that. Problem for later me if it turns into an issue
+}
+
+func CalculateConnectionStatistics() ConnectionStatistics {
+	var speedtestClient = speedtest.New()
+	serverList, _ := speedtestClient.FetchServers()
+	// Picks server with the lowest ping
+	targets, _ := serverList.FindServer([]int{})
+	for _, s := range targets {
+		s.PingTest(nil)
+		s.DownloadTest()
+		s.UploadTest()
+		connStats := ConnectionStatistics{
+			downloadBytes: uint64(s.DLSpeed), // Casting since speedtest-go dev implemented this as float64
+			uploadBytes:   uint64(s.ULSpeed),
+			downloadSlots: uint16(uint64(s.DLSpeed) / 131072), // 128KiB per slot
+			uploadSlots:   uint16(uint64(s.ULSpeed) / 131072), // Casting to avoid float
+		}
+
+		if connStats.downloadSlots > 1000 {
+			connStats.downloadSlots = 1000 // Max slot count should be 1000 to avoid DDoS-like behaviour
+		}
+		if connStats.uploadSlots > 1000 {
+			connStats.uploadSlots = 1000
+		}
+		s.Context.Reset()
+		return connStats
+
+	}
+	// If all else fails... fuckin' yolo it
+	connStats := ConnectionStatistics{0, 0, 1000, 1000}
+	return connStats
+
+}

files/files_test.go

@@ -0,0 +1,13 @@
+package files
+
+import (
+	"fmt"
+	"strconv"
+	"testing"
+)
+
+func TestCalculateConnectionStatistics(t *testing.T) {
+	connStats := CalculateConnectionStatistics()
+	fmt.Printf(" Download: %s bytes/s\n Upload: %s bytes/s\n Estimated Client Slots: %s\n Estimated Internal Slots: %s\n", strconv.FormatUint(connStats.downloadBytes, 10), strconv.FormatUint(connStats.uploadBytes, 10), strconv.Itoa(int(connStats.uploadSlots)), strconv.Itoa(int(connStats.downloadSlots)))
+
+}

global/global.go

@@ -0,0 +1,38 @@
+package global
+
+import (
+	crand "crypto/rand" // Probably better than math/rand for this
+	"log"
+	"math/rand"
+)
+
+func GenerateRandomString(length int) string {
+	// C# had optional parameters that allowed lowercase for chardata and gpose lobbies, Go doesn't.
+	// We can probably get away with just uppercase.
+	allowedArray := []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
+	random := make([]byte, length)
+	_, err := crand.Read(random)
+	if err != nil {
+		log.Fatalf("Couldn't generate secure slice: %s", err)
+	}
+	allowedLength := len(allowedArray)
+	i := 0
+	charArray := make([]rune, length)
+	for i < length {
+		charArray[i] = allowedArray[int(random[i])%allowedLength] // Casting random[i] due to type mismatch
+		i++
+	}
+	return string(charArray)
+}
+
+func PhasedStartupString() string {
+	// Redis is used for a lot of stuff, which means the Postgres DB gets HAMMERED on a fresh startup. Use
+	// this to generate a shuffled string that can be used to only allow e.g. UIDs starting with [x] to connect
+	// for the first however many seconds and incrementally expand access so that data doesn't need to be loaded all at
+	// once
+	chars := []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
+	rand.Shuffle(len(chars), func(i, j int) {
+		chars[i], chars[j] = chars[j], chars[i]
+	})
+	return string(chars)
+}

global/global_test.go

@@ -1,4 +1,4 @@
-package SnowcloakUtils
+package global
 
 import (
 	"fmt"
@@ -15,3 +15,7 @@ func TestGenerateRandomString(t *testing.T) {
 	fmt.Println("Test: 30 chars")
 	fmt.Println(GenerateRandomString(30))
 }
+
+func TestPhasedStartup(t *testing.T) {
+	fmt.Printf("%c", PhasedStartupString()[1])
+}

go.mod

@@ -1,3 +1,5 @@
 module SnowcloakUtils
 
 go 1.25
+
+require github.com/showwin/speedtest-go v1.7.10

go.sum

@@ -0,0 +1,2 @@
+github.com/showwin/speedtest-go v1.7.10 h1:9o5zb7KsuzZKn+IE2//z5btLKJ870JwO6ETayUkqRFw=
+github.com/showwin/speedtest-go v1.7.10/go.mod h1:Ei7OCTmNPdWofMadzcfgq1rUO7mvJy9Jycj//G7vyfA=