文件传输工具FTransferor＜优化篇＞

在上一篇文章中，我们详细探讨了FTransferor文件传输工具的设计与实现，并展示了它在局域网文件传输方面的高效性。然而，随着互联网应用场景的不断丰富，传统的基于 TCP/UDP 的传输方式已经无法满足部分开发者的需求。特别是在跨平台、跨网络传输场景中，对 HTTP 协议的支持变得尤为重要。因此，本篇文章将围绕 FTransferor 的优化改造，为其增加 HTTP 协议支持，并提升其在复杂网络环境下的适用性。

为什么要支持 HTTP 协议？

HTTP 是一种通用性极强的传输协议，具备以下特点：

1）广泛的兼容性：几乎所有平台都支持 HTTP 协议，可以无缝集成到现有的系统中。

2）易用性：基于 HTTP 的文件传输可以通过浏览器直接访问，无需额外安装客户端工具。

通过为 FTransferor 增加 HTTP 协议支持，我们可以大幅提升工具的实用性，使其不仅适用于局域网场景，还能满足跨网络的传输需求。

功能上的补充

在上一篇中我们对文件传输工具的功能仅支持上传文件，在增加HTTP协议支持的同时，还对下载、查看文件列表 等功能进行了丰富，并且利用密码支持了一定程度的安全性。

具体实现

增加HTTP Server并实现一个简单的拦截器

首先是HTTP Server，这个实现相对简单，主要就是文件下载和文件列表查看两个HTTP处理器。

go 复制代码

const (
    PathList      = "/files"
    PathDownload  = "/download/"
    QueryParamKey = "secret"
)

func runHttpServer(port int) {
    fn := "runHttpServer"
    http.HandleFunc(PathList, secretFilterHandler(fileListHandler))
    http.HandleFunc(PathDownload, secretFilterHandler(fileDownloadHandler))

    fmt.Printf("%s is listening on port %d\n", fn, port)
    if err := http.ListenAndServe(fmt.Sprintf(":%d", port), nil); err != nil {
        fmt.Println("Error starting file server:", err)
    }
}

拦截器实现，主要就是拦截query参数中的secret key，需要和启动HTTP Server时输入的参数一致。

go 复制代码

func secretFilterHandler(next http.HandlerFunc) func(w http.ResponseWriter, r *http.Request) {
    return func(w http.ResponseWriter, r *http.Request) {
        getSecret := r.URL.Query().Get(QueryParamKey)
        if getSecret != secret {
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }
        next(w, r)
    }
}

运行以上代码后，用户可以通过浏览器和客户端命令中查看和下载当前目录中的文件，客户端的实现请继续往下看。

HTTP Server端具体实现

首先是文件列表查看功能的实现：

go 复制代码

func fileListHandler(w http.ResponseWriter, r *http.Request) {
    files, err := os.ReadDir(path)
    if err != nil {
        http.Error(w, "Unable to list files", http.StatusInternalServerError)
        return
	}

    result := make([]string, 0)
    w.Header().Set("Content-Type", "application/json")
    for _, f := range files {
        if !f.IsDir() {
            result = append(result, f.Name())
        }
    }

    marshal, err := json.Marshal(map[string]interface{}{
        "data": result,
    })
    if err != nil {
        http.Error(w, "Unable to list files", http.StatusInternalServerError)
        return
    }
    _, _ = w.Write(marshal)
}

其次是文件下载功能的实现：

go 复制代码

func fileDownloadHandler(w http.ResponseWriter, r *http.Request) {
    fileName := strings.TrimPrefix(r.URL.Path, PathDownload)
    filePath := filepath.Join(path, fileName)
    if _, err := os.Stat(filePath); os.IsNotExist(err) {
        http.Error(w, "File not found", http.StatusNotFound)
        return
    }
    http.ServeFile(w, r, filePath)
}

客户端实现

客户端我们可以直接使用浏览器进行操作，但是考虑到在Linux服务器的情况下可能只有终端命令行，因此也提供命令的方式进行操作，主要的思路就是增加命令参数，根据参数去访问服务端不同的HTTP接口。

代码实现：

go 复制代码

func httpClient(action Action, f string) {
    httpServerAddr := fmt.Sprintf("%s%s", Scheme, server)

    client := http.Client{}
    switch action {
    case ActionGet:
        url := fmt.Sprintf("%s%s%s?%s=%s", httpServerAddr, PathDownload, f, QueryParamKey, passwd)
        resp, err := client.Get(url)
        if err != nil {
            fmt.Println("Error downloading file:", err)
        }

        defer func() {
            _ = resp.Body.Close()
        }()

        // 创建文件以保存下载的内容
        file, err := os.Create(f)
        if err != nil {
            fmt.Println("Error creating file:", err)
            return
        }
        defer func() {
            _ = file.Close()
        }()

        // 将响应的内容写入文件
        if _, err = io.Copy(file, resp.Body); err != nil {
            fmt.Println("Error writing to file:", err)
            return
        }
        fmt.Println("File downloaded successfully!")
    case ActionList:
        url := fmt.Sprintf("%s%s?%s=%s", httpServerAddr, PathList, QueryParamKey, passwd)
        resp, err := client.Get(url)
        if err != nil {
            fmt.Println("Error downloading file:", err)
        }

        defer func() {
            _ = resp.Body.Close()
        }()

        // 读取响应的body
        body, err := io.ReadAll(resp.Body)
        if err != nil {
            fmt.Println("Error reading response body:", err)
            return
        }

        // 将body转换为map
        var data map[string]interface{}
        if err = json.Unmarshal(body, &data); err != nil {
            fmt.Println("Error unmarshaling JSON:", err)
            return
        }
        // 打印map
        fmt.Println("Data as map:", data)
    default:
    }
}

新功能演示

首先启动服务端，TCP和HTTP共同启动，但是注意不能使用同一个端口号：

shell 复制代码

./FTransferor server --path filepath --port 8081 --webport 8082 --secret D&J$HE23

然后使用客户端命令，查看文件列表：

shell 复制代码

./FTransferor.exe cli --server localhost:8082 --action list --passwd D&J$HE23

下载文件：

shell 复制代码

./FTransferor.exe cli --server localhost:8082 --action get --passwd D&J$HE23 --file t.zip

总结

通过本次优化，我们为 FTransferor 增加了对 HTTP 协议的支持，显著提升了其在不同网络环境下的适用性。未来，我们计划进一步优化工具的交互体验，例如通过 Web 界面实现更加友好的操作方式。