请使用golang实现websokect不断接受前端传输的不同数据且响应不同的数据
以下是一个使用Golang实现WebSocket的示例代码,可以接收前端传输的不同数据并响应不同的数据:
package main
import (
"fmt"
"log"
"net/http"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
func main() {
http.HandleFunc("/ws", websocketHandler)
log.Fatal(http.ListenAndServe(":8080", nil))
}
func websocketHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade error:", err)
return
}
defer conn.Close()
for {
messageType, message, err := conn.ReadMessage()
if err != nil {
log.Println("Read error:", err)
break
}
// 根据接收到的不同数据进行处理
switch string(message) {
case "hello":
// 响应不同的数据
response := "Hello, client!"
err = conn.WriteMessage(messageType, []byte(response))
if err != nil {
log.Println("Write error:", err)
break
}
case "time":
// 响应当前时间
response := fmt.Sprintf("Current time: %s", time.Now().Format("2006-01-02 15:04:05"))
err = conn.WriteMessage(messageType, []byte(response))
if err != nil {
log.Println("Write error:", err)
break
}
default:
log.Println("Invalid message:", string(message))
}
}
}
在上面的示例中,我们使用了Gorilla WebSocket库来处理WebSocket连接。首先,我们使用http.HandleFunc函数将/ws路径与websocketHandler函数绑定。在websocketHandler函数中,我们使用upgrader.Upgrade方法将HTTP连接升级为WebSocket连接,并进行一些初始化。然后,我们进入一个无限循环,通过conn.ReadMessage方法读取前端传输的消息。根据接收到的不同数据,我们可以编写不同的逻辑进行处理,并使用conn.WriteMessage方法向前端发送响应。
请注意,以上代码仅为示例,实际应用中可能需要根据具体需求进行更多的错误处理和逻辑扩展
原文地址: https://www.cveoy.top/t/topic/ijwG 著作权归作者所有。请勿转载和采集!