Golang은 웹 개발에도 널리 사용되는 빠르게 컴파일되고 효율적이며 안전한 프로그래밍 언어입니다. 그중 Golang을 사용하여 Websocket을 구축하는 것은 매우 인기 있는 애플리케이션 시나리오입니다. 이 기사에서는 Golang을 사용하여 Websocket 애플리케이션을 구축하는 방법을 소개합니다.
1. Websocket이란
Websocket은 브라우저와 서버 간의 양방향 통신 연결을 허용하는 프로토콜입니다. 전통적인 HTTP 프로토콜은 브라우저를 통해 서버에 요청을 보내고 서버로부터 응답을 받을 수만 있습니다. 그러나 Websocket을 사용하면 서버가 브라우저에 적극적으로 메시지를 보내 서버와 브라우저 간의 전이중 통신을 실현할 수 있습니다.
2. Golang이 Websocket을 구현하는 방법
Golang은 표준 라이브러리 net/http
와 함께 제공됩니다. 이 패키지는 웹 구축에 사용할 수 있는 http.Server
구조를 제공합니다. 섬기는 사람. 또한 net/http
는 WebSocket 서버를 구축하는 데 필요한 모든 기능을 제공하는 websocket
하위 패키지도 제공합니다. net/http
,这个包提供了一个http.Server
结构体,它可以用来构建Web服务器。除此之外,net/http
也提供了一个websocket
子包,它为我们提供了构建WebSocket服务器所需的全部功能。
要建立WebSocket连接,客户端需要向服务器发送一个HTTP请求,并携带Upgrade头部,指示请求升级为WebSocket协议。服务器端在收到此请求后,需要检查Upgrade头部,并根据是否为WebSocket请求来进行相应的处理。
服务器端可以使用websocket.IsWebSocketUpgrade
函数来检查请求是否为WebSocket请求,并使用websocket.Upgrade
函数升级请求。升级成功后,服务器就可以通过*websocket.Conn
类型的连接对象与客户端进行通信了。
func handler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println(err) return } defer conn.Close() // 使用conn进行通信 }
其中,upgrader
是一个websocket.Upgrader
类型的变量,通过设置它的属性可以控制WebSocket连接的处理方式。Upgrade
函数在升级请求时会阻塞,因此要将其放在goroutine中运行。
一旦WebSocket连接建立成功,客户端和服务器都可以通过*websocket.Conn
对象进行通信。使用这个对象可以发送和接收WebSocket消息。
发送消息很简单,只需要调用WriteMessage
方法即可:
err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))
接收消息的方法有三种:
ReadMessage
:阻塞读取,直到有消息到来NextReader
:阻塞等待下一个消息到来NextWriter
:返回一个io.WriteCloser
对象,可以使用这个对象发送消息一般使用ReadMessage
方法接收WebSocket消息:
_, message, err := conn.ReadMessage() if err != nil { log.Println(err) return } log.Printf("recv: %s", message)
三、一个简单的WebSocket应用
在这个例子中,我们将创建一个简单的聊天室应用,客户端发送消息之后,服务器将消息广播给所有连接的客户端。
客户端代码:(使用JavaScript)
<!DOCTYPE html> <html> <head> <title>Chatroom</title> </head> <body> <input type="text" id="input-message"/> <button onclick="sendMessage()">发送</button> <ul id="message-list"></ul> <script> var ws = new WebSocket("ws://localhost:8080/ws"); ws.onmessage = function(event) { var message = event.data; var li = document.createElement("li"); li.appendChild(document.createTextNode(message)); document.getElementById("message-list").appendChild(li); }; function sendMessage() { var message = document.getElementById("input-message").value; ws.send(message); } </script> </body> </html>
服务器端代码:
package main import ( "fmt" "log" "net/http" "github.com/gorilla/websocket" ) var upgrader = websocket.Upgrader{} func main() { http.HandleFunc("/ws", wsHandler) err := http.ListenAndServe(":8080", nil) if err != nil { log.Fatal("ListenAndServe: ", err) } } func wsHandler(w http.ResponseWriter, r *http.Request) { conn, err := upgrader.Upgrade(w, r, nil) if err != nil { log.Println(err) return } defer conn.Close() for { _, message, err := conn.ReadMessage() if err != nil { log.Println(err) break } fmt.Printf("recv: %s ", message) // 广播给所有客户端 for _, conn := range conns { err := conn.WriteMessage(websocket.TextMessage, message) if err != nil { log.Println(err) break } } } }
在上面的代码中,我们将所有连接的websocket.Conn
对象存储在conns
变量中。每当有一个新消息到来时,服务器会将消息广播给所有客户端。
总结
使用Golang搭建WebSocket应用程序非常简单,只需要导入net/http
和websocket
websocket.IsWebSocketUpgrade
함수를 사용하여 해당 요청이 WebSocket 요청인지 확인하고, websocket.Upgrade
함수를 사용하여 요청을 업그레이드할 수 있습니다. 업그레이드가 성공한 후 서버는 *websocket.Conn
유형의 연결 개체를 통해 클라이언트와 통신할 수 있습니다. 🎜rrreee🎜 그 중 upgrader
는 websocket.Upgrader
유형의 변수로 해당 속성을 설정하여 WebSocket 연결 처리 방법을 제어할 수 있습니다. 업그레이드
기능은 요청을 업그레이드할 때 차단되므로 고루틴에서 실행해야 합니다. 🎜*websocket.Conn 개체가 통신합니다. 이 개체를 사용하여 WebSocket 메시지를 보내고 받습니다. 🎜🎜메시지 보내기는 매우 간단합니다. <code>WriteMessage
메서드를 호출하면 됩니다. 🎜rrreee🎜메시지를 받는 방법에는 세 가지가 있습니다. 🎜ReadMessage
: 차단 메시지가 도착할 때까지 Get 읽기NextReader
: 다음 메시지가 도착할 때까지 기다리는 블록NextWriter
: io .WriteCloser
객체, 이 객체를 사용하여 메시지를 보낼 수 있습니다.ReadMessage
메서드를 사용하세요: 🎜rrreee🎜 3. A 간단한 WebSocket 애플리케이션🎜🎜 이 예에서는 클라이언트가 메시지를 보낸 후 서버가 연결된 모든 클라이언트에게 메시지를 브로드캐스트합니다. 🎜🎜클라이언트 측 코드: (JavaScript 사용) 🎜rrreee🎜서버 측 코드: 🎜rrreee🎜위 코드에서는 연결된 모든 websocket.Conn
개체를 conns 코드에 저장합니다. > 변수. 새 메시지가 도착할 때마다 서버는 모든 클라이언트에게 메시지를 브로드캐스트합니다. 🎜🎜요약🎜🎜Golang을 사용하여 WebSocket 애플리케이션을 구축하는 것은 매우 간단합니다. <code>net/http
및 websocket
패키지를 가져오고 제공된 API를 직접 사용하면 됩니다. WebSocket 애플리케이션 개발을 완료하기 위해 표준 라이브러리를 사용합니다. 이 글이 Golang Websocket 개발에 도움이 되기를 바랍니다. 🎜위 내용은 Golang은 웹소켓을 구축합니다.의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!