Golang中WebSocket和WSS的支持_golang wss(1)，2024年最新Golang高级工程师必备知识

机器学习之心AI
102 阅读4分钟 
WriteBufferSize: 1024,

}

func WebSocketHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println("Failed to upgrade to WebSocket:", err) return }

defer conn.Close()

for {
	// 读取客户端发送的消息
	\_, message, err := conn.ReadMessage()
	if err != nil {
		log.Println("Failed to read message:", err)
		break
	}

	fmt.Println("Received message:", string(message))

	// 向客户端发送消息
	err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
	if err != nil {
		log.Println("Failed to write message:", err)
		break
	}
}

}

func main() { http.HandleFunc("/websocket", WebSocketHandler) log.Println("WebSocket server is running on :8080") http.ListenAndServe(":8080", nil) }


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。在该函数中，我们使用`upgrader.Upgrade`方法将HTTP连接升级为WebSocket连接，并通过`conn.ReadMessage`方法读取客户端发送的消息，并使用`conn.WriteMessage`方法向客户端发送消息。最后，我们使用`http.HandleFunc`函数将`WebSocketHandler`函数注册为处理WebSocket请求的处理程序，并使用`http.ListenAndServe`函数启动WebSocket服务器。


#### 构建WebSocket客户端


要构建一个WebSocket客户端，我们可以使用`github.com/gorilla/websocket`包提供的`Dial`函数来建立与WebSocket服务器的连接。下面是一个简单的示例，演示了如何使用Golang构建一个WebSocket客户端：

package main

import ( "fmt" "log" "net/url" "github.com/gorilla/websocket" )

func main() { u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/websocket"} conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil) if err != nil { log.Println("Failed to connect to WebSocket server:", err) return }

defer conn.Close()

// 向服务器发送消息
err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

// 读取服务器发送的消息
\_, message, err := conn.ReadMessage()
if err != nil {
	log.Println("Failed to read message:", err)
	return
}

fmt.Println("Received message:", string(message))

}


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并使用`websocket.DefaultDialer.Dial`方法建立与WebSocket服务器的连接。然后，我们使用`conn.WriteMessage`方法向服务器发送消息，并使用`conn.ReadMessage`方法读取服务器发送的消息。最后，我们将接收到的消息打印出来。


### Golang中的WSS


#### 生成自签名证书


要在Golang中使用WSS，首先需要生成一个自签名证书。可以使用OpenSSL工具来生成自签名证书。下面是一个简单的示例，演示了如何生成自签名证书：

$ openssl req -x509 -nodes -days 365 -newkey rsa:2048 -keyout key.pem -out cert.pem


在上面的命令中，我们使用`openssl req`命令生成一个自签名证书，并将私钥保存到`key.pem`文件中，将公钥保存到`cert.pem`文件中。在实际使用中，你可能需要根据自己的需求来生成证书。


#### 使用WSS的WebSocket服务器和客户端


要在Golang中使用WSS，我们只需要在HTTP服务器和客户端中使用TLS配置即可。下面是一个简单的示例，演示了如何在Golang中构建一个使用WSS的WebSocket服务器和客户端：

package main

import ( "fmt" "log" "net/http" "github.com/gorilla/websocket" )

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

func WebSocketHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println("Failed to upgrade to WebSocket:", err) return }

defer conn.Close()

for {
	\_, message, err := conn.ReadMessage()
	if err != nil {
		log.Println("Failed to read message:", err)
		break
	}

	fmt.Println("Received message:", string(message))

	err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
	if err != nil {
		log.Println("Failed to write message:", err)
		break
	}
}

}

func main() { http.HandleFunc("/websocket", WebSocketHandler) log.Println("WebSocket server is running on :8080") err := http.ListenAndServeTLS(":8080", "cert.pem", "key.pem", nil) if err != nil { log.Fatal("Failed to start WebSocket server:", err) } }


在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。最后，我们使用`http.ListenAndServeTLS`函数启动一个使用WSS的WebSocket服务器，并提供证书文件`cert.pem`和私钥文件`key.pem`。


对于WebSocket客户端，只需要在建立连接时使用`DialTLS`方法，并提供证书的URL即可。下面是一个简单的示例：

package main

import ( "fmt" "log" "net/url" "github.com/gorilla/websocket" )

func main() { u := url.URL{Scheme: "wss", Host: "localhost:8080", Path: "/websocket"} conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil) if err != nil { log.Println("Failed to connect to WebSocket server:", err) return }

defer conn.Close()

err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

\_, message, err := conn.ReadMessage()
if err != nil {
	log.Println("Failed to read message:", err)
	return
}

fmt.Println("Received message:", string(message))

img img img

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以戳这里获取

avatar
文章
阅读
粉丝