golang实现弹幕

go get -u GitHub.com/gorilla/websocket

package main

import (
    "log"
    "net/Http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Println("WebSocket server listen at: 8080")
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Upgrade:", err)
        return
    }
    defer conn.Close()

    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println("ReadMessage:", err)
            break
        }
        log.Printf("Received message: %s", message)

        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println("WriteMessage:", err)
            break
        }
    }
}

<script>
    var socket = new WebSocket("ws://localhost:8080/ws");

    socket.onopen = function() {
        console.log("WebSocket connection established.");
    };

    socket.onmessage = function(event) {
        // 收到服务端返回的消息
        console.log("Received message: " + event.data);
    };

    function send() {
        // 发送弹幕消息
        var message = document.getElementById("input").value;
        socket.send(message);
        console.log("Sent message: " + message);
    }
</script>

<input type="text" id="input" />
<button onclick="send()">Send</button>