2024年Go最全Golang中WebSocket和WSS的支持_golang wss(4)，2024年最新含泪狂刷Golang基础面试118题

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

}

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))

}

在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并使用`websocket.DefaultDialer.Dial`方法建立一个使用WSS的WebSocket连接。


### 安全性的考虑


在使用Golang构建WebSocket和WSS的过程中，保证数据的安全性是至关重要的。以下是几个在实现WebSocket和WSS时应考虑的安全性问题。


#### 1. SSL/TLS加密


WSS协议通过在HTTP协议基础上添加SSL/TLS加密层来确保数据的安全传输。在Golang中，可以使用`http.ListenAndServeTLS`函数来启动一个使用WSS的WebSocket服务器，通过提供证书和私钥来实现加密传输。


#### 2. 跨站脚本攻击（XSS）防护


跨站脚本攻击（XSS）是一种常见的网络攻击方式，攻击者通过在网页中插入恶意脚本来窃取用户的敏感信息或进行其他恶意行为。为了保护WebSocket应用程序免受XSS攻击，可以采取以下措施：


* 输入验证和过滤：确保所有用户输入的数据都进行验证和过滤，防止恶意脚本的注入。
* 安全的内容解析：使用安全的方式解析和渲染用户提供的内容，如使用HTML转义来防止注入攻击。


#### 3. 跨站请求伪造（CSRF）防护


跨站请求伪造（CSRF）是一种攻击方式，攻击者通过伪造合法用户的请求，以此来执行非法操作。为了防止WebSocket应用程序受到CSRF攻击，可以采取以下措施：




![img](https://img-blog.csdnimg.cn/img_convert/5a43823bbb8e2401add553e6be904afa.png)
![img](https://img-blog.csdnimg.cn/img_convert/4b0999e8e27ab94e761453a0d7caae1c.png)
![img](https://img-blog.csdnimg.cn/img_convert/dc2fdb54cd3ecdf39ee41e335862340e.png)

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

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

**[如果你需要这些资料，可以戳这里获取](https://bbs.csdn.net/topics/618658159)**

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

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

**[如果你需要这些资料，可以戳这里获取](https://bbs.csdn.net/topics/618658159)**