简单的go 静态文件服务器样例

最新推荐文章于 2024-08-26 08:42:10 发布

HOOKTTG

最新推荐文章于 2024-08-26 08:42:10 发布

阅读量2.1k

点赞数

分类专栏： Go web前端文章标签： golang 服务器静态文件 ServerFile

本文链接：https://blog.csdn.net/hookttg/article/details/52163172

版权

web前端同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

4 篇文章 0 订阅

订阅专栏

随学随记，留备查

1、核心是需要获取文件的绝对路径，os.Open()需要；

2、windows使用“\”表示路径，而go需要“/”表示路径，所以需要将“\”替换为“/”；

3、使用内置的http.ServerFile()函数提供后续的文件传输服务；

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"strings"
)

var StaticDir = "/static"

func getCurDir() string {
	dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
	if err != nil {
		log.Fatal(err)
	}
	return strings.Replace(dir, "\\", "/", -1)
}

func dealStaticFiles(w http.ResponseWriter, r *http.Request) {
	stDir := getCurDir()

	if strings.HasPrefix(r.URL.Path, StaticDir) {
		file := stDir + r.URL.Path
		fmt.Println(file)
		f, err := os.Open(file)
		defer f.Close()

		if err != nil && os.IsNotExist(err) {
			fmt.Fprintln(w, "File not exist")
			return
		}
		http.ServeFile(w, r, file)
		return
	} else {
		fmt.Fprintln(w, "Hello world")
	}
}
func main() {
	http.HandleFunc("/", dealStaticFiles)    //设置访问的路由
	err := http.ListenAndServe(":8080", nil) //设置监听的端口
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}