zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-13 01:40:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: support embed file system to serve files in rest (#4253)

Browse Source
This commit is contained in:
Kevin Wan
2024-07-17 16:21:08 +08:00
committed by GitHub
parent a00c956776
commit 5dd6f2a43a
4 changed files with 33 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rest/internal/fileserver/filehandler.go
View File

@@ -5,8 +5,9 @@ import (
"strings"
)
func Middleware(path, dir string) func(http.HandlerFunc) http.HandlerFunc {
fileServer := http.FileServer(http.Dir(dir))
// Middleware returns a middleware that serves files from the given file system.
func Middleware(path string, fs http.FileSystem) func(http.HandlerFunc) http.HandlerFunc {
fileServer := http.FileServer(fs)
pathWithTrailSlash := ensureTrailingSlash(path)
pathWithoutTrailSlash := ensureNoTrailingSlash(path)

2
rest/internal/fileserver/filehandler_test.go
View File

@@ -44,7 +44,7 @@ func TestMiddleware(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
middleware := Middleware(tt.path, tt.dir)
middleware := Middleware(tt.path, http.Dir(tt.dir))
nextHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusNotFound)
})