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

王林原创: 2023-12-17 10:23:21 823浏览

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>

五、运行测试