This commit is contained in:
shaoyongjun 2024-05-20 10:42:53 +08:00
parent 47d60ce740
commit 917fa41e1b

View File

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