Files
ai/main.go
2023-04-28 18:10:03 +08:00

72 lines
2.5 KiB
Go

package main
import (
"io/ioutil"
"log"
"net/http"
"runtime"
"time"
"main/routers"
"main/utils"
"github.com/gorilla/mux"
"github.com/russross/blackfriday"
)
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
r := mux.NewRouter()
// 設定中間件
r.Use(func(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
defer utils.LogComponent(time.Now().UnixNano(), r) // 最后打印日志
w.Header().Set("Access-Control-Allow-Origin", "*") // 處理跨域請求
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, Authorization, X-Requested-With")
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, PATCH, DELETE, OPTIONS")
if r.Method == "OPTIONS" {
w.WriteHeader(http.StatusOK) // 處理OPTIONS請求
return
}
next.ServeHTTP(w, r)
})
})
// 設定路由
r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
// 從本地讀取Markdown文件
input, err := ioutil.ReadFile("./README.md")
if err != nil {
log.Println(err)
return
}
//output := blackfriday.MarkdownBasic(input)
output := blackfriday.Markdown(input, blackfriday.HtmlRenderer(0, "", ""), blackfriday.EXTENSION_TABLES|blackfriday.EXTENSION_FENCED_CODE|blackfriday.EXTENSION_AUTOLINK)
w.Write(output)
})
r.HandleFunc("/api/models", routers.ModelsGet).Methods("GET")
r.HandleFunc("/api/models", routers.ModelsPost).Methods("POST")
r.HandleFunc("/api/models/{id}", routers.ModelItemGet).Methods("GET")
r.HandleFunc("/api/models/{id}", routers.ModelItemPatch).Methods("PATCH")
r.HandleFunc("/api/models/{id}", routers.ModelItemDelete).Methods("DELETE")
r.HandleFunc("/api/images", routers.ImagesGet).Methods("GET")
r.HandleFunc("/api/images", routers.ImagesPost).Methods("POST")
r.HandleFunc("/api/images/{id}", routers.ImagesItemGet).Methods("GET")
r.HandleFunc("/api/images/{id}", routers.ImagesItemPatch).Methods("PATCH")
r.HandleFunc("/api/images/{id}", routers.ImagesItemDelete).Methods("DELETE")
r.HandleFunc("/api/tasks", routers.TasksGet).Methods("GET")
r.HandleFunc("/api/tasks", routers.TasksPost).Methods("POST")
r.HandleFunc("/api/tasks/{id}", routers.TasksItemGet).Methods("GET")
r.HandleFunc("/api/tasks/{id}", routers.TasksItemPatch).Methods("PATCH")
r.HandleFunc("/api/tasks/{id}", routers.TasksItemDelete).Methods("DELETE")
r.HandleFunc("/api/params/model", routers.ParamsModelsGet).Methods("GET")
log.Println("Web Server is running on http://localhost:8080")
http.ListenAndServe(":8080", r)
}