golang Websocket教學：如何開發線上文件傳輸功能-Golang-PHP中文網

golang Websocket教學：如何開發線上文件傳輸功能

WBOY

發布： 2023-12-17 14:46:10

原創

1118 人瀏覽過

golang Websocket教程：如何开发在线文件传输功能

Golang Websocket教學：如何開發線上檔案傳輸功能

#引言：
隨著網路的快速發展，檔案傳輸成了人們在網路上常常面臨的需求。而利用Websocket技術，我們可以輕鬆實現線上文件傳輸功能。本文將介紹如何使用Golang開發一個基於Websocket的線上文件傳輸應用，並提供具體的程式碼範例。

一、Websocket簡介
Websocket是一種基於TCP連接實現的全雙工通訊協議，它可以在瀏覽器和伺服器之間建立持久性的連接，實現即時資料傳輸。相較於傳統的HTTP協議，Websocket具有更低的延遲，更高的並發性能和更好的數據傳輸能力，使得它成為了開發即時應用和線上互動功能的理想選擇。

二、開發環境準備
為了完成本教程，你需要先安裝Golang的開發環境，並具備一定的Golang程式設計基礎。同時，我們也需要引進第三方的Websocket庫，以便快速開發。

安裝第三方Websocket函式庫：

go get github.com/gorilla/websocket

登入後複製

三、專案結構及程式碼說明
我們的專案將包含兩個主要檔案：main.go和index.html。其中，main.go是服務端程式碼，index.html是客戶端程式碼。下面，我們將逐一介紹這兩個檔案的程式碼實作。

main.go：

匯入所需的套件:

package main

import (
 "log"
 "net/http"

 "github.com/gorilla/websocket"
)

登入後複製

定義WebSocket連線升級器：

var upgrader = websocket.Upgrader{
 CheckOrigin: func(r *http.Request) bool {
     return true
 },
}

登入後複製

定義WebSocket處理器函數：

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

 // 处理文件传输逻辑
}

登入後複製

啟動HTTP伺服器並註冊WebSocket處理器：

func main() {
 http.HandleFunc("/ws", websocketHandler)
 http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
     http.ServeFile(w, r, "index.html")
 })

 log.Println("Server started on localhost:8000")
 log.Fatal(http.ListenAndServe(":8000", nil))
}

登入後複製

#index.html：

<!DOCTYPE html>
<html>
<head>
    <meta charset="UTF-8">
    <title>文件传输</title>
</head>
<body>
    <input type="file" id="file" name="file" />
    <button onclick="sendFile()">传输文件</button>

    <script>
        var ws = new WebSocket("ws://localhost:8000/ws");

        ws.onopen = function(){
            console.log("已连接");
        }

        ws.onmessage = function(e){
            console.log("接收到消息");
            // 处理接收到的文件数据
        }

        ws.onclose = function(){
            console.log("连接已关闭");
        }

        function sendFile(){
            var fileInput = document.getElementById("file");
            var file = fileInput.files[0];

            if(file){
                var reader = new FileReader();
                reader.readAsBinaryString(file);

                reader.onload = function(e){
                    var fileData = e.target.result;
                    ws.send(fileData);
                }
            }
        }
    </script>
</body>
</html>

登入後複製

四、檔案傳輸實作
在main.go中的websocketHandler函數中，我們可以寫一個特定的檔案傳輸邏輯。這裡以簡單的檔案讀取和傳送為例：

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

    // 文件读取
    filePath := "example.txt"
    file, err := os.Open(filePath)
    if err != nil {
        log.Println(err)
        return
    }
    defer file.Close()

    fileData, err := ioutil.ReadAll(file)
    if err != nil {
        log.Println(err)
        return
    }

    // 文件发送
    if err := conn.WriteMessage(websocket.BinaryMessage, fileData); err != nil {
        log.Println(err)
        return
    }
}

登入後複製

在index.html中，我們使用了JavaScript的FileReader對象，來讀取待傳輸的檔案數據，並透過WebSocket傳送給伺服器：

function sendFile(){
    var fileInput = document.getElementById("file");
    var file = fileInput.files[0];

    if(file){
        var reader = new FileReader();
        reader.readAsBinaryString(file);

        reader.onload = function(e){
            var fileData = e.target.result;
            ws.send(fileData);
        }
    }
}

登入後複製

五、執行項目
在命令列中，進入專案目錄，執行以下命令啟動伺服器：