Golang builds websocket-Golang-php.cn

Home

Backend Development

Golang

Golang builds websocket

WBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWB

May 22, 2023 pm 07:21 PM

Golang is a fast-compilation, efficient, and safe programming language that is also widely used in Web development. Among them, using Golang to build Websocket is a very popular application scenario. This article will introduce how to use Golang to build a Websocket application.

1. What is Websocket

Websocket is a protocol that allows the establishment of a two-way communication connection between the browser and the server. The traditional HTTP protocol can only send requests to the server by the browser and receive responses from the server. However, using Websocket allows the server to actively send messages to the browser, realizing full-duplex communication between the server and the browser.

2. How Golang implements Websocket

Golang comes with a standard library net/http. This package provides a http.Server structure body, which can be used to build web servers. In addition, net/http also provides a websocket sub-package, which provides us with all the functions needed to build a WebSocket server.

Establishing a WebSocket connection

To establish a WebSocket connection, the client needs to send an HTTP request to the server and carry the Upgrade header to indicate that the request is upgraded to the WebSocket protocol. After receiving this request, the server needs to check the Upgrade header and process it accordingly depending on whether it is a WebSocket request.

The server can use the websocket.IsWebSocketUpgrade function to check whether the request is a WebSocket request, and use the websocket.Upgrade function to upgrade the request. After the upgrade is successful, the server can communicate with the client through the connection object of type *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进行通信
}

Among them, upgrader is a websocket.Upgrader type variable. By setting its properties, you can control the processing method of WebSocket connection. UpgradeThe function will block when upgrading the request, so it must be run in a goroutine.

Sending and receiving WebSocket messages

Once the WebSocket connection is successfully established, both the client and the server can communicate through the *websocket.Conn object. Use this object to send and receive WebSocket messages.

Sending a message is very simple, just call the WriteMessage method:

err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))

There are three ways to receive messages:

ReadMessage: Blocks reading until a message arrives
NextReader: Blocks and waits for the next message to arrive
NextWriter: Returns a io.WriteCloser object, you can use this object to send messages

Generally use the ReadMessage method to receive WebSocket messages:

_, message, err := conn.ReadMessage()
if err != nil {
    log.Println(err)
    return
}
log.Printf("recv: %s", message)

3. One Simple WebSocket application

In this example, we will create a simple chat room application. After the client sends a message, the server broadcasts the message to all connected clients.

Client side code: (using 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>

Server side code:

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

In the above code, we connect all connected websocket.ConnObjects are stored in conns variables. Whenever a new message arrives, the server broadcasts the message to all clients.

Summary

It is very simple to build a WebSocket application using Golang. You only need to import the net/http and websocket packages and use the standard library directly. The provided API can complete the development of WebSocket applications. I hope this article can be helpful to Golang Websocket development.

The above is the detailed content of Golang builds websocket. For more information, please follow other related articles on the PHP Chinese website!

Statement

The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

Go Binary Encoding/Decoding: A Practical Guide with ExamplesMay 07, 2025 pm 05:37 PM

Go's encoding/binary package is a tool for processing binary data. 1) It supports small-endian and large-endian endian byte order and can be used in network protocols and file formats. 2) The encoding and decoding of complex structures can be handled through Read and Write functions. 3) Pay attention to the consistency of byte order and data type when using it, especially when data is transmitted between different systems. This package is suitable for efficient processing of binary data, but requires careful management of byte slices and lengths.

Go 'bytes' Package: Compare, Join, Split & MoreMay 07, 2025 pm 05:29 PM

The"bytes"packageinGoisessentialbecauseitoffersefficientoperationsonbyteslices,crucialforbinarydatahandling,textprocessing,andnetworkcommunications.Byteslicesaremutable,allowingforperformance-enhancingin-placemodifications,makingthispackage

Go Strings Package: Essential Functions You Need to KnowMay 07, 2025 pm 04:57 PM

Go'sstringspackageincludesessentialfunctionslikeContains,TrimSpace,Split,andReplaceAll.1)Containsefficientlychecksforsubstrings.2)TrimSpaceremoveswhitespacetoensuredataintegrity.3)SplitparsesstructuredtextlikeCSV.4)ReplaceAlltransformstextaccordingto

Mastering String Manipulation with Go's 'strings' Package: a practical guideMay 07, 2025 pm 03:57 PM

ThestringspackageinGoiscrucialforefficientstringmanipulationduetoitsoptimizedfunctionsandUnicodesupport.1)ItsimplifiesoperationswithfunctionslikeContains,Join,Split,andReplaceAll.2)IthandlesUTF-8encoding,ensuringcorrectmanipulationofUnicodecharacters

Mastering Go Binary Data: A Deep Dive into the 'encoding/binary' PackageMay 07, 2025 pm 03:49 PM

The"encoding/binary"packageinGoiscrucialforefficientbinarydatamanipulation,offeringperformancebenefitsinnetworkprogramming,fileI/O,andsystemoperations.Itsupportsendiannessflexibility,handlesvariousdatatypes,andisessentialforcustomprotocolsa

Implementing Mutexes and Locks in Go for Thread SafetyMay 05, 2025 am 12:18 AM

In Go, using mutexes and locks is the key to ensuring thread safety. 1) Use sync.Mutex for mutually exclusive access, 2) Use sync.RWMutex for read and write operations, 3) Use atomic operations for performance optimization. Mastering these tools and their usage skills is essential to writing efficient and reliable concurrent programs.

Benchmarking and Profiling Concurrent Go CodeMay 05, 2025 am 12:18 AM

How to optimize the performance of concurrent Go code? Use Go's built-in tools such as getest, gobench, and pprof for benchmarking and performance analysis. 1) Use the testing package to write benchmarks to evaluate the execution speed of concurrent functions. 2) Use the pprof tool to perform performance analysis and identify bottlenecks in the program. 3) Adjust the garbage collection settings to reduce its impact on performance. 4) Optimize channel operation and limit the number of goroutines to improve efficiency. Through continuous benchmarking and performance analysis, the performance of concurrent Go code can be effectively improved.

Error Handling in Concurrent Go Programs: Avoiding Common PitfallsMay 05, 2025 am 12:17 AM

The common pitfalls of error handling in concurrent Go programs include: 1. Ensure error propagation, 2. Processing timeout, 3. Aggregation errors, 4. Use context management, 5. Error wrapping, 6. Logging, 7. Testing. These strategies help to effectively handle errors in concurrent environments.

See all articles