diff --git a/static/init.go b/static/init.go index de43650..9209e28 100644 --- a/static/init.go +++ b/static/init.go @@ -2,6 +2,8 @@ package static import ( "embed" + "fmt" + "os" ) import _ "embed" @@ -19,21 +21,19 @@ var imgMap = initImgMap var cssMap = initCssMap() func initPageMap() map[string][]byte { - var pageMap = make(map[string][]byte, 8) - - //login.html - data, err := pageList.ReadFile("login.html") - if err == nil { - pageMap["login.html"] = data + list, err := pageList.ReadDir(".") + //list, err := fs.ReadDir(multiCf, ".") + if err != nil { + os.Exit(-1) + return nil } - //ai-chat.html - data, err = pageList.ReadFile("ai-chat.html") - if err == nil { - pageMap["ai-chat.html"] = data + var dataMap = make(map[string][]byte, len(list)) + for _, file := range list { + dataMap[file.Name()] = initHtml(file.Name()) } - return pageMap + return dataMap } func initImgMap() map[string][]byte { @@ -58,3 +58,13 @@ func initCssMap() map[string][]byte { return dataMap } + +func initHtml(fileName string) []byte { + //读取配置文件 + data, err := pageList.ReadFile(fileName) + if err != nil { + fmt.Println(err) + } + + return data +}