以下是一个简单的示例代码,实现了服务端等待客户端连接后,读取客户端发送的消息:

服务端代码:

package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

func main() {
	http.HandleFunc("/ws", wsHandler)
	http.ListenAndServe(":8080", nil)
}

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

	fmt.Println("client connected")

	for {
		_, message, err := conn.ReadMessage()
		if err != nil {
			fmt.Println("read:", err)
			return
		}
		fmt.Printf("recv: %s\n", message)
	}
}

客户端代码:

package main

import (
	"fmt"
	"net/url"
	"time"

	"github.com/gorilla/websocket"
)

func main() {
	u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
	fmt.Println("connecting to", u.String())

	for {
		c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
		if err != nil {
			fmt.Println("dial:", err)
			time.Sleep(time.Second)
			continue
		}
		defer c.Close()

		fmt.Println("connected")

		for {
			err = c.WriteMessage(websocket.TextMessage, []byte("hello"))
			if err != nil {
				fmt.Println("write:", err)
				break
			}
			time.Sleep(time.Second)
		}
	}
}

在客户端启动前启动服务端,服务端会一直等待客户端连接并读取客户端发送的消息。当客户端启动后,会连接到服务端并发送消息。服务端会接收到客户端发送的消息并打印出来。如果客户端断开连接,服务端会退出循环。如果客户端重新连接,服务端会重新等待客户端连接并读取消息

请写一个golang websocket示例要求服务端先启动客户端还未启动时ReadMessage方法不能报错一直等待客户端启动后输入数据到服务端

原文地址: https://www.cveoy.top/t/topic/dHJO 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录