Go Web 程式設計之 靜態檔案
概述
在 Web 開發中,需要處理很多靜態資原始檔,如 css/js 和圖片檔案等。本文將介紹在 Go 語言中如何處理檔案請求。
接下來,我們將介紹兩種處理檔案請求的方式:原始方式和http.FileServer
方法。
原始方式
原始方式比較簡單粗暴,直接讀取檔案,然後返回給客戶端。
func main() { mux := http.NewServeMux() mux.HandleFunc("/static/", fileHandler) server := &http.Server { Addr: ":8080", Handler: mux, } if err := server.ListenAndServe(); err != nil { log.Fatal(err) } }
上面我們建立了一個檔案處理器,將它掛載到路徑/static/
上。一般地,靜態檔案的路徑有一個共同的字首,以便與其它路徑區分。如這裡的/static/
,還有一些常用的,例如/public/
等。
程式碼的其它部分與程式模板沒什麼不同,這裡就不贅述了。
另外需要注意的是,這裡的註冊路徑/static/
最後的/
不能省略。我們在前面的文章程式結構中介紹過,如果請求的路徑沒有精確匹配的處理,會逐步去掉路徑最後部分再次查詢。
靜態檔案的請求路徑一般為/static/hello.html
這種形式。沒有精確匹配的路徑,繼而查詢/static/
,這個路徑與/static
是不能匹配的。
接下來,我們看看檔案處理器的實現:
func fileHandler(w http.ResponseWriter, r *http.Request) { path := "." + r.URL.Path fmt.Println(path) f, err := os.Open(path) if err != nil { Error(w, toHTTPError(err)) return } defer f.Close() d, err := f.Stat() if err != nil { Error(w, toHTTPError(err)) return } if d.IsDir() { DirList(w, r, f) return } data, err := ioutil.ReadAll(f) if err != nil { Error(w, toHTTPError(err)) return } ext := filepath.Ext(path) if contentType := extensionToContentType[ext]; contentType != "" { w.Header().Set("Content-Type", contentType) } w.Header().Set("Content-Length", strconv.FormatInt(d.Size(), 10)) w.Write(data) }
首先我們讀出請求路徑,再加上相對可執行檔案的路徑。一般地,static
目錄與可執行檔案在同一個目錄下。然後開啟該路徑,檢視資訊。
如果該路徑表示的是一個檔案,那麼根據檔案的字尾設定Content-Type
,讀取檔案的內容並返回。程式碼中簡單列舉了幾個字尾對應的Content-Type
:
var extensionToContentType = map[string]string { ".html": "text/html; charset=utf-8", ".css": "text/css; charset=utf-8", ".js": "application/javascript", ".xml": "text/xml; charset=utf-8", ".jpg": "image/jpeg", }
如果該路徑表示的是一個目錄,那麼返回目錄下所有檔案與目錄的列表:
func DirList(w http.ResponseWriter, r *http.Request, f http.File) {
dirs, err := f.Readdir(-1)
if err != nil {
Error(w, http.StatusInternalServerError)
return
}
sort.Slice(dirs, func(i, j int) bool { return dirs[i].Name() < dirs[j].Name() })
w.Header().Set("Content-Type", "text/html; charset=utf-8")
fmt.Fprintf(w, "<pre>\n")
for _, d := range dirs {
name := d.Name()
if d.IsDir() {
name += "/"
}
url := url.URL{Path: name}
fmt.Fprintf(w, "<a href=\"%s\">%s</a>\n", url.String(), name)
}
fmt.Fprintf(w, "</pre>\n")
}
上面的函式先讀取目錄下第一層的檔案和目錄,然後按照名字排序。最後拼裝成包含超連結的 HTML 返回。使用者可以點選超連結訪問對應的檔案或目錄。
如何上述過程中出現錯誤,我們使用toHTTPError
函式將錯誤轉成對應的響應碼,然後通過Error
回覆給客戶端。
func toHTTPError(err error) int {
if os.IsNotExist(err) {
return http.StatusNotFound
}
if os.IsPermission(err) {
return http.StatusForbidden
}
return http.StatusInternalServerError
}
func Error(w http.ResponseWriter, code int) {
w.WriteHeader(code)
}
同級目錄下static
目錄內容:
static
├── folder
│ ├── file1.txt
│ └── file2.txt
│ └── file3.txt
├── hello.css
├── hello.html
├── hello.js
└── hello.txt
執行程式看看效果:
$ go run main.go
開啟瀏覽器,請求localhost:8080/static/hello.html
:
可以看到頁面hello.html
已經呈現了:
<!-- hello.html -->
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>Go Web 程式設計之 靜態檔案</title>
<link rel="stylesheet" href="/static/hello.css">
</head>
<body>
<p class="greeting">Hello World!</p>
<script src="/static/hello.js"></script>
</body>
</html>
html 使用的 css 和 js 檔案也是通過/static/
路徑請求的,兩個檔案都比較簡單:
.greeting {
font-family: sans-serif;
font-size: 15px;
font-style: italic;
font-weight: bold;
}
console.log("Hello World!")
"Hello World!"字型顯示為 css 設定的樣式,通過觀察控制檯也能看到 js 列印的資訊。
再來看看檔案目錄瀏覽,在瀏覽器中請求localhost:8080/static/
:
可以依次點選列表中的檔案檢視其內容。
點選hello.css
:
點選hello.js
:
依次點選folder
和file1.txt
:
靜態檔案的請求路徑也會輸出到執行伺服器的控制檯中:
$ go run main.go
./static/
./static/hello.css
./static/hello.js
./static/folder/
./static/folder/file1.txt
原始方式的實現有一個缺點,實現邏輯複雜。上面的程式碼儘管我們已經忽略很多情況的處理了,程式碼量還是不小。自己編寫很繁瑣,而且容易產生 BUG。
靜態檔案服務的邏輯其實比較一致,應該通過庫的形式來提供。為此,Go 語言提供了http.FileServer
方法。
http.FileServer
先來看看如何使用:
package main
import (
"log"
"net/http"
)
func main() {
mux := http.NewServeMux()
mux.Handle("/static/", http.FileServer(http.Dir("")))
server := &http.Server {
Addr: ":8080",
Handler: mux,
}
if err := server.ListenAndServe(); err != nil {
log.Fatal(err)
}
}
上面的程式碼使用http.Server
方法,幾行程式碼就實現了與原始方式相同的效果,是不是很簡單?這就是使用庫的好處!
http.FileServer
接受一個http.FileSystem
介面型別的變數:
// src/net/http/fs.go
type FileSystem interface {
Open(name string) (File, error)
}
傳入http.Dir
型別變數,注意http.Dir
是一個型別,其底層型別為string
,並不是方法。因而http.Dir("")
只是一個型別轉換,而非方法呼叫:
// src/net/http/fs.go
type Dir string
http.Dir
表示檔案的起始路徑,空即為當前路徑。呼叫Open
方法時,傳入的引數需要在前面拼接上該起始路徑得到實際檔案路徑。
http.FileServer
的返回值型別是http.Handler
,所以需要使用Handle
方法註冊處理器。http.FileServer
將收到的請求路徑傳給http.Dir
的Open
方法開啟對應的檔案或目錄進行處理。
在上面的程式中,如果請求路徑為/static/hello.html
,那麼拼接http.Dir
的起始路徑.
,最終會讀取路徑為./static/hello.html
的檔案。
有時候,我們想要處理器的註冊路徑和http.Dir
的起始路徑不相同。有些工具在打包時會將靜態檔案輸出到public
目錄中。
這時需要使用http.StripPrefix
方法,該方法會將請求路徑中特定的字首去掉,然後再進行處理:
package main
import (
"log"
"net/http"
)
func main() {
mux := http.NewServeMux()
mux.Handle("/static/", http.StripPrefix("/static", http.FileServer(http.Dir("./public"))))
server := &http.Server {
Addr: ":8080",
Handler: mux,
}
if err := server.ListenAndServe(); err != nil {
log.Fatal(err)
}
}
這時,請求localhost:8080/static/hello.html
將會返回./public/hello.html
檔案。
路徑/static/index.html
經過處理器http.StripPrefix
去掉了字首/static
得到/index.html
,然後又加上了http.Dir
的起始目錄./public
得到檔案最終路徑./public/hello.html
。
除此之外,http.FileServer
還會根據請求檔案的字尾推斷內容型別,更全面:
// src/mime/type.go
var builtinTypesLower = map[string]string{
".css": "text/css; charset=utf-8",
".gif": "image/gif",
".htm": "text/html; charset=utf-8",
".html": "text/html; charset=utf-8",
".jpeg": "image/jpeg",
".jpg": "image/jpeg",
".js": "application/javascript",
".mjs": "application/javascript",
".pdf": "application/pdf",
".png": "image/png",
".svg": "image/svg+xml",
".wasm": "application/wasm",
".webp": "image/webp",
".xml": "text/xml; charset=utf-8",
}
如果檔案字尾無法推斷,http.FileServer
將讀取檔案的前 512 個位元組,根據內容來推斷內容型別。感興趣可以看一下原始碼src/net/http/sniff.go
。
http.ServeContent
除了直接使用http.FileServer
之外,net/http
庫還暴露了ServeContent
方法。這個方法可以用在處理器需要返回一個檔案內容的時候,非常易用。
例如下面的程式,根據 URL 中的file
引數返回對應的檔案內容:
package main
import (
"fmt"
"log"
"net/http"
"os"
"time"
)
func ServeFileContent(w http.ResponseWriter, r *http.Request, name string, modTime time.Time) {
f, err := os.Open(name)
if err != nil {
w.WriteHeader(500)
fmt.Fprint(w, "open file error:", err)
return
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
w.WriteHeader(500)
fmt.Fprint(w, "call stat error:", err)
return
}
if fi.IsDir() {
w.WriteHeader(400)
fmt.Fprint(w, "no such file:", name)
return
}
http.ServeContent(w, r, name, fi.ModTime(), f)
}
func fileHandler(w http.ResponseWriter, r *http.Request) {
query := r.URL.Query()
filename := query.Get("file")
if filename == "" {
w.WriteHeader(400)
fmt.Fprint(w, "filename is empty")
return
}
ServeFileContent(w, r, filename, time.Time{})
}
func main() {
mux := http.NewServeMux()
mux.HandleFunc("/show", fileHandler)
server := &http.Server {
Addr: ":8080",
Handler: mux,
}
if err := server.ListenAndServe(); err != nil {
log.Fatal(err)
}
}
http.ServeContent
除了接受引數http.ResponseWriter
和http.Request
,還需要檔名name
,修改時間modTime
和io.ReadSeeker
介面型別的引數。
modTime
引數是為了設定響應的Last-Modified
首部。如果請求中攜帶了If-Modified-Since
首部,ServeContent
方法會根據modTime
判斷是否需要傳送內容。
如果需要傳送內容,ServeContent
方法從io.ReadSeeker
介面重讀取內容。*os.File
實現了介面io.ReadSeeker
。
使用場景
Web 開發中的靜態資源都可以使用http.FileServer
來處理。除此之外,http.FileServer
還可以用於實現一個簡單的檔案伺服器,瀏覽或下載檔案:
package main
import (
"flag"
"log"
"net/http"
)
var (
ServeDir string
)
func init() {
flag.StringVar(&ServeDir, "sd", "./", "the directory to serve")
}
func main() {
flag.Parse()
mux := http.NewServeMux()
mux.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir(ServeDir))))
server := &http.Server {
Addr: ":8080",
Handler: mux,
}
if err := server.ListenAndServe(); err != nil {
log.Fatal(err)
}
}
在上面的程式碼中,我們構建了一個簡單的檔案伺服器。編譯之後,將想瀏覽的目錄作為引數傳給命令列選項,就可以瀏覽和下載該目錄下的檔案了:
$ ./main.exe -sd D:/code/golang
可以將埠也作為命令列選項,這樣做出一個通用的檔案伺服器,編譯之後就可以在其它機器上使用了