How to build a RESTful API and integrate Websockets using Golang?

Steps to build RESTful API and integrate Websockets in Golang: Install dependencies, write API processing function, create WebSocket upgrade function, use Gorilla Mux to register routing, start HTTP server, create a simple chat room in actual combat: define message structure, write API processing function processing WebSocket connection

How to build a RESTful API in Golang and integrate Websockets

To build a RESTful API and integrate Websockets in Golang, You can follow the following steps:

1. Install dependencies

go get github.com/gorilla/mux
go get github.com/gorilla/websocket

2. Write API processing function

func apiHandler(w http.ResponseWriter, r *http.Request) {
    // 处理 API 请求
}

3. Create WebSocket upgrade function

func wsUpgrader(w http.ResponseWriter, r *http.Request) (*websocket.Conn, error) {
    // 升级 HTTP 请求到 WebSocket 连接
}

4. Use Gorilla Mux to register routing

router := mux.NewRouter()
router.HandleFunc("/api", apiHandler)
router.HandleFunc("/ws", wsUpgrader)

5. Start HTTP server

http.ListenAndServe(":8080", router)

Practical case: Create a simple chat room

1. Define the message structure

type Message struct {
    Username string
    Message  string
}

2. Write API handling function

func chatApiHandler(w http.ResponseWriter, r *http.Request) {
    // 发送消息到 Websockets 连接
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(Message{Username: "Server", Message: "Hello!"})
}

3. Handling WebSocket connections

func wsHandler(ws *websocket.Conn) {
    // 循环读取和发送消息
    for {
        var msg Message
        if err := ws.ReadJSON(&msg); err != nil {
            break
        }
        fmt.Println("Received message:", msg.Message)
        if err := ws.WriteJSON(Message{Username: "Server", Message: "Echo: " + msg.Message}); err != nil {
            break
        }
    }
}

By following these steps, you can easily build RESTful APIs and integrate Websockets using Golang for Your application provides real-time communication capabilities.

The above is the detailed content of How to build a RESTful API and integrate Websockets using Golang?. For more information, please follow other related articles on the PHP Chinese website!