Функции вынесены в утилиты
This commit is contained in:
@@ -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
13
utils/utils.go
Normal 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)
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user