Go语言WebSocket服务：解决403错误与Origin校验

package main

import (
    "fmt"
    "golang.org/x/net/websocket" // 注意：原先的code.google.com/p/go.net/websocket已迁移至此
    "net/http"
)

// webHandler 处理WebSocket连接，接收消息并打印
func webHandler(ws *websocket.Conn) {
    var s string
    // 从WebSocket连接中读取字符串
    if _, err := fmt.Fscan(ws, &s); err != nil {
        fmt.Printf("Error reading from websocket: %v\n", err)
        return
    }
    fmt.Println("Received: ", s)
    // 可以选择将消息回显给客户端
    // if _, err := fmt.Fprint(ws, "Echo: "+s); err != nil {
    //  fmt.Printf("Error writing to websocket: %v\n", err)
    // }
}

func main() {
    fmt.Println("Starting websocket server on :8080/echo")
    // 将webHandler包装成websocket.Handler并注册到HTTP路由
    http.Handle("/echo", websocket.Handler(webHandler))

    // 启动HTTP服务器监听8080端口
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        panic("ListenAndServe: " + err.Error())
    }
}

// 尝试连接到WebSocket服务
ws = new WebSocket("ws://localhost:8080/echo");

// 监听接收到的消息
ws.onmessage = function(e) {
    console.log("websock: " + e.data);
};

// 连接成功时发送消息
ws.onopen = function() {
    ws.send("Hello from client!");
};

// 连接关闭时
ws.onclose = function() {
    console.log("WebSocket connection closed.");
};

// 发生错误时
ws.onerror = function(err) {
    console.error("WebSocket error:", err);
};

package main

import (
    "fmt"
    "golang.org/x/net/websocket" // 注意：请确保导入路径正确
    "net/http"
)

// webHandler 处理WebSocket连接，接收消息并打印
func webHandler(ws *websocket.Conn) {
    var s string
    // 从WebSocket连接中读取字符串
    if _, err := fmt.Fscan(ws, &s); err != nil {
        // 忽略EOF错误，这通常表示客户端关闭了连接
        if err.Error() != "EOF" {
            fmt.Printf("Error reading from websocket: %v\n", err)
        }
        return
    }
    fmt.Println("Received: ", s)
    // 可以选择将消息回显给客户端
    if _, err := fmt.Fprint(ws, "Echo: "+s); err != nil {
        fmt.Printf("Error writing to websocket: %v\n", err)
    }
}

func main() {
    fmt.Println("Starting websocket server on :8080/echo")

    // 使用http.HandleFunc注册一个自定义处理函数
    http.HandleFunc("/echo",
        func(w http.ResponseWriter, req *http.Request) {
            // 创建websocket.Server实例
            // Handshake字段为nil时，默认的握手函数不会强制检查Origin
            s := websocket.Server{Handler: websocket.Handler(webHandler)}
            // 调用ServeHTTP方法来处理WebSocket连接
            s.ServeHTTP(w, req)
        })

    // 启动HTTP服务器监听8080端口
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        panic("ListenAndServe: " + err.Error())
    }
}

package main

import (
    "fmt"
    "golang.org/x/net/websocket"
    "net/http"
    "log" // 引入log包用于更专业的错误处理
)

// webHandler 处理WebSocket连接，接收消息并打印，并回显
func webHandler(ws *websocket.Conn) {
    defer ws.Close() // 确保连接关闭

    var msg string
    for {
        // 从WebSocket连接中读取字符串
        err := websocket.Message.Receive(ws, &msg) // 使用websocket.Message进行结构化读写
        if err != nil {
            if err.Error() != "EOF" { // 忽略EOF错误，通常表示客户端关闭了连接
                log.Printf("Error receiving message from websocket: %v\n", err)
            }
            break // 发生错误或客户端关闭时退出循环
        }
        log.Printf("Received from client %s: %s\n", ws.RemoteAddr().String(), msg)

        // 将消息回显给客户端
        err = websocket.Message.Send(ws, "Echo: "+msg)
        if err != nil {
            log.Printf("Error sending message to websocket: %v\n", err)
            break
        }
    }
    log.Printf("Client %s disconnected.\n", ws.RemoteAddr().String())
}

func main() {
    log.Println("Starting websocket server on :8080/echo")

    http.HandleFunc("/echo",
        func(w http.ResponseWriter, req *http.Request) {
            // 可以选择在此处手动检查Origin头部，以增加安全性
            // if req.Header.Get("Origin") != "http://localhost:8080" {
            //  http.Error(w, "Forbidden", http.StatusForbidden)
            //  return
            // }

            // 创建websocket.Server实例，Handler字段指向我们的业务逻辑
            // Handshake字段为nil时，默认的握手函数不会强制检查Origin
            s := websocket.Server{Handler: websocket.Handler(webHandler)}

            // 调用ServeHTTP方法来处理WebSocket连接
            s.ServeHTTP(w, req)
        })

    log.Fatal(http.ListenAndServe(":8080", nil)) // 使用log.Fatal来处理ListenAndServe错误
}