diff --git a/socket/socket.go b/socket/socket.go index b9196bc..09cdbad 100644 --- a/socket/socket.go +++ b/socket/socket.go @@ -2,7 +2,7 @@ package socket import ( "g365sfu/logger" - "math/rand" + "g365sfu/utils" "net/http" "os" @@ -18,6 +18,7 @@ var upgrader = websocket.Upgrader{ }, } +// Получение секретного ключа из переменных окружения func getSecret() string { return os.Getenv("SECRET") } @@ -54,17 +55,7 @@ func HandleWebSocket(w http.ResponseWriter, r *http.Request) { // Генерация случайного идентификатора для сокета func randomSocketIdentifier() string { // Генерация случайного идентификатора для сокета - return "sock_" + 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) + return "sock_" + utils.RandomString(10) } func processData(data <-chan []byte, connection *Connection) { diff --git a/utils/utils.go b/utils/utils.go new file mode 100644 index 0000000..787fc32 --- /dev/null +++ b/utils/utils.go @@ -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) +}