golang Websocket教學：如何開發線上投稿功能-Golang-PHP中文網

golang Websocket教學：如何開發線上投稿功能

王林

發布： 2023-12-17 10:23:21

原創

1110 人瀏覽過

golang Websocket教程：如何开发在线投稿功能

Golang Websocket教學：如何開發線上投稿功能

#引言：
隨著網路的飛速發展，越來越多的應用程式需要實現即時通信功能。在網路應用中，Websocket技術已成為一種常用的通訊協定。本文旨在教導讀者如何使用Golang語言開發一個線上投稿功能，借助Websocket實現即時的互動。

一、Websocket簡介
Websocket是一種全雙工的通訊協議，可以在瀏覽器和伺服器之間建立持久的連接，實現即時通訊。相較於傳統的HTTP協議，Websocket具有更低的延遲和更高的吞吐量，適合處理即時資料的應用場景。

二、開發環境搭建
在開始開發之前，需要先搭建Golang的開發環境。

安裝Golang：造訪官方網站 https://golang.org/doc/install 下載並安裝適合自己作業系統的Golang版本。
安裝相關函式庫：
```
go get github.com/gorilla/websocket
```
登入後複製

三、專案結構設計
在開發線上投稿功能之前，需要先規劃好專案的目錄結構。

- main.go
- handler.go
- template
  - index.html

登入後複製

四、程式碼實作

main.go

package main

import (
    "html/template"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var (
    upgrader = websocket.Upgrader{}
    clients  = make(map[*websocket.Conn]bool)
    broadcast = make(chan []byte)
    homeTemplate *template.Template
)

func homeHandler(w http.ResponseWriter, r *http.Request) {
    homeTemplate.Execute(w, nil)
}

func websocketHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    clients[conn] = true

    for {
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            delete(clients, conn)
            return
        }
        broadcast <- msg
    }
}

func handleMessages() {
    for {
        msg := <-broadcast
        for client := range clients {
            err := client.WriteMessage(websocket.TextMessage, msg)
            if err != nil {
                log.Println(err)
                client.Close()
                delete(clients, client)
            }
        }
    }
}

func main() {
    // 配置文件路径
    homeTemplate = template.Must(template.ParseFiles("template/index.html"))

    http.HandleFunc("/", homeHandler)
    http.HandleFunc("/ws", websocketHandler)

    go handleMessages()

    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal(err)
    }
}

登入後複製

handler.go

package main

import (
    "net/http"
    "path/filepath"
)

func homeHandler(w http.ResponseWriter, r *http.Request) {
    filePath, _ := filepath.Abs("template/index.html")
    http.ServeFile(w, r, filePath)
}

登入後複製

template/index.html

<!DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8">
    <title>在线投稿功能</title>
</head>
<body>
    <h1>在线投稿功能</h1>
    <textarea id="message-input" rows="5" cols="30"></textarea>
    <button id="send-btn">发送</button>
    <br>
    <ul id="message-list"></ul>
</body>
<script>
    var socket = new WebSocket("ws://localhost:8080/ws");
    socket.onmessage = function (event) {
        var messageList = document.getElementById("message-list");
        var messageItem = document.createElement("li");
        messageItem.innerText = event.data;
        messageList.appendChild(messageItem);
    };

    document.getElementById("send-btn").addEventListener("click", function () {
        var messageInput = document.getElementById("message-input");
        var message = messageInput.value;
        socket.send(message);
    });
</script>
</html>

登入後複製

五、執行測試