Gorilla websocket очень медленный, когда его используют более одного клиента - PullRequest
Новая
       33

Gorilla websocket очень медленный, когда его используют более одного клиента

0 голосов
/ 19 апреля 2020

Я сделал небольшое приложение для чата, используя gorilla websocket. Когда есть только один клиент, это быстро, как и ожидалось. Но когда я подключаюсь к другому клиенту, обновление чата может занять 3 секунды, даже если оно размещено на моем локальном компьютере.

Код для веб-сокета: 

var clients = make(map[*websocket.Conn]bool)
var broadcast = make(chan Message)
var upgrader = websocket.Upgrader{}
func main() {
    router.HandleFunc("/ws", handleConnections)
    err := http.ListenAndServe(":80", router)
}
func handleConnections(w http.ResponseWriter, r *http.Request) {
    ws, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Fatal(err)
    }
    defer ws.Close()
    clients[ws] = true
    for {
        var msg Message
        err := ws.ReadJSON(&msg)
        if err != nil {
            log.Printf("error: %v", err)
            delete(clients, ws)
            break
        }
        broadcast <- msg
    }
}

func handleMessages() {
    for {
        msg := <-broadcast
        log.Println(msg)
        database, _ := sql.Open("sqlite3", "./database.db")
        statement, _ := database.Prepare("CREATE TABLE IF NOT EXISTS messages (id INTEGER PRIMARY KEY, username TEXT, message TEXT, timestamp INTEGER)")
        statement.Exec()
        statement, _ = database.Prepare("INSERT INTO messages (username, message, timestamp) VALUES (?, ?, ?)")
        statement.Exec(&msg.Username, &msg.Message, &msg.Timestamp)
        for client := range clients {
            err := client.WriteJSON(msg)
            if err != nil {
                log.Printf("error: %v", err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

Полный код: https://github.com/mismaah-abdulla/Chatapp-Backend/blob/de8cc0eb88fa7f18b293bb0f2229368887c53adf/src/main.go

1 Ответ

0 голосов
/ 29 апреля 2020

Проблема заключалась в том, что база данных была заблокирована при попытке записи. Исправить это просто, добавив отсрочку rows.Close () при каждом запуске запроса. Например: 

func messages(w http.ResponseWriter, r *http.Request) {
    rows, _ := database.Query("SELECT username, message, timestamp FROM messages")
    defer rows.Close()
    var messages []Message
    for rows.Next() {
        var msg Message
        rows.Scan(&msg.Username, &msg.Message, &msg.Timestamp)
        messages = append(messages, msg)
    }
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(messages)
}
...