Функции вынесены в утилиты

This commit is contained in:
set
2026-03-11 19:34:54 +02:00
parent 8d5d5df8b9
commit ebff80f5f1
2 changed files with 16 additions and 12 deletions

View File

@@ -2,7 +2,7 @@ package socket
import ( import (
"g365sfu/logger" "g365sfu/logger"
"math/rand" "g365sfu/utils"
"net/http" "net/http"
"os" "os"
@@ -18,6 +18,7 @@ var upgrader = websocket.Upgrader{
}, },
} }
// Получение секретного ключа из переменных окружения
func getSecret() string { func getSecret() string {
return os.Getenv("SECRET") return os.Getenv("SECRET")
} }
@@ -54,17 +55,7 @@ func HandleWebSocket(w http.ResponseWriter, r *http.Request) {
// Генерация случайного идентификатора для сокета // Генерация случайного идентификатора для сокета
func randomSocketIdentifier() string { func randomSocketIdentifier() string {
// Генерация случайного идентификатора для сокета // Генерация случайного идентификатора для сокета
return "sock_" + randomString(10) return "sock_" + utils.RandomString(10)
}
// Генерация случайной строки заданной длины
func randomString(n int) string {
const letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
b := make([]byte, n)
for i := range b {
b[i] = letters[rand.Intn(len(letters))]
}
return string(b)
} }
func processData(data <-chan []byte, connection *Connection) { func processData(data <-chan []byte, connection *Connection) {

13
utils/utils.go Normal file
View File

@@ -0,0 +1,13 @@
package utils
import "math/rand"
// Генерация случайной строки заданной длины
func RandomString(n int) string {
const letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
b := make([]byte, n)
for i := range b {
b[i] = letters[rand.Intn(len(letters))]
}
return string(b)
}