在上一篇文章中,我们详细探讨了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 界面实现更加友好的操作方式。