分解

main.go

@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"runtime"
 	"strconv"
+	"text/template"
 	"time"
 
 	"main/models"
@@ -15,130 +16,244 @@ import (
 	"github.com/gorilla/mux"
 )
 
-func CreateObject(w http.ResponseWriter, r *http.Request) {
-	defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
-	// 從body中獲取收到的JSON數據
-	body, err := ioutil.ReadAll(r.Body)
-	if err != nil {
-		log.Println(err)
-		return
-	}
-	defer r.Body.Close()
-
-	// 將JSON數據轉換成目標對象
-	object := models.GetObject(mux.Vars(r)["name"], 0)
-	if err = json.Unmarshal(body, object); err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 將對象存入數據庫
-	err = object.Create()
-	if err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 返回結果
-	w.Header().Set("Content-Type", "application/json; charset=utf-8")
-	w.Write(object.ToJSON())
-}
-
-func GetObjetc(w http.ResponseWriter, r *http.Request) {
-	defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
-	// 獲取對象
-	object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
-	err := object.Get()
-	if err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 返回結果
-	w.Header().Set("Content-Type", "application/json; charset=utf-8")
-	w.Write(object.ToJSON())
-}
-
-func UpdateObject(w http.ResponseWriter, r *http.Request) {
-	defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
-	// 從body中獲取收到的JSON數據
-	body, err := ioutil.ReadAll(r.Body)
-	if err != nil {
-		log.Println(err)
-		return
-	}
-	defer r.Body.Close()
-
-	// 將JSON數據轉換成目標對象
-	object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
-	if err = json.Unmarshal(body, object); err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 更新對象
-	err = object.Update()
-	if err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 返回結果
-	w.Header().Set("Content-Type", "application/json; charset=utf-8")
-	w.Write(object.ToJSON())
-}
-
-func DeleteObject(w http.ResponseWriter, r *http.Request) {
-	defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
-	// 刪除對象
-	object := models.GetObject(mux.Vars(r)["name"], ParamInt(mux.Vars(r)["id"], 0))
-	err := object.Delete()
-	if err != nil {
-		log.Println(err)
-		return
-	}
-
-	// 返回結果
-	w.Header().Set("Content-Type", "application/json; charset=utf-8")
-	w.Write(object.ToJSON())
-}
-
-func GetObjects(w http.ResponseWriter, r *http.Request) {
-	defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-
-	var listview models.ListView
-
-	// 獲取查詢參數: page, pageSize
-	name := mux.Vars(r)["name"]
-	page := ParamInt(r.URL.Query().Get("page"), 1)
-	pagesize := ParamInt(r.URL.Query().Get("pageSize"), 10)
-
-	// 獲取對象列表
-	listview.Read(name, page, pagesize)
-
-	// 返回結果
-	w.Header().Set("Content-Type", "application/json; charset=utf-8")
-	w.Write(listview.ToJSON())
-}
-
 func main() {
 	runtime.GOMAXPROCS(runtime.NumCPU())
 
 	r := mux.NewRouter()
-	r.HandleFunc("/api/{name}", GetObjects).Methods("GET")
-	r.HandleFunc("/api/{name}", CreateObject).Methods("POST")
-	r.HandleFunc("/api/{name}/{id}", GetObjetc).Methods("GET")
-	r.HandleFunc("/api/{name}/{id}", UpdateObject).Methods("PATCH")
-	r.HandleFunc("/api/{name}/{id}", DeleteObject).Methods("DELETE")
+	r.Use(middleware)
+	r.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		t, _ := template.ParseFiles("templates/index.html")
+		t.Execute(w, nil)
+	})
+	r.HandleFunc("/api/models", models_get).Methods("GET")
+	r.HandleFunc("/api/models", models_post).Methods("POST")
+	r.HandleFunc("/api/models/{id}", models_item_get).Methods("GET")
+	r.HandleFunc("/api/models/{id}", models_item_patch).Methods("PATCH")
+	r.HandleFunc("/api/models/{id}", models_item_delete).Methods("DELETE")
+
+	r.HandleFunc("/api/images", images_get).Methods("GET")
+	r.HandleFunc("/api/images", images_post).Methods("POST")
+	r.HandleFunc("/api/images/{id}", images_item_get).Methods("GET")
+	r.HandleFunc("/api/images/{id}", images_item_patch).Methods("PATCH")
+	r.HandleFunc("/api/images/{id}", images_item_delete).Methods("DELETE")
+
+	r.HandleFunc("/api/tasks", tasks_get).Methods("GET")
+	r.HandleFunc("/api/tasks", tasks_post).Methods("POST")
+	r.HandleFunc("/api/tasks/{id}", tasks_item_get).Methods("GET")
+	r.HandleFunc("/api/tasks/{id}", tasks_item_patch).Methods("PATCH")
+	r.HandleFunc("/api/tasks/{id}", tasks_item_delete).Methods("DELETE")
 
 	log.Println("Web Server is running on http://localhost:8080")
 	http.ListenAndServe(":8080", r)
 }
 
+func models_get(w http.ResponseWriter, r *http.Request) {
+	var listview models.ListView
+	listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+	listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+	listview.List = models.QueryModels(listview.Page, listview.PageSize)
+	fmt.Println(listview.List)
+	listview.Total = models.CountModels()
+	listview.Next = listview.Page*listview.PageSize < listview.Total
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(listview.ToJSON())
+}
+
+func models_post(w http.ResponseWriter, r *http.Request) {
+	var model models.Model
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &model); err != nil {
+		log.Println(err)
+		return
+	}
+	model.Create()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(model))
+}
+
+func models_item_get(w http.ResponseWriter, r *http.Request) {
+	model := models.Model{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	model.Get()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(model))
+}
+
+func models_item_patch(w http.ResponseWriter, r *http.Request) {
+	model := models.Model{}
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &model); err != nil {
+		log.Println(err)
+		return
+	}
+	model.ID = ParamInt(mux.Vars(r)["id"], 0)
+	model.Update()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(model))
+}
+
+func models_item_delete(w http.ResponseWriter, r *http.Request) {
+	model := models.Model{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	model.Delete()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(model))
+}
+
+func images_get(w http.ResponseWriter, r *http.Request) {
+	var listview models.ListView
+	listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+	listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+	listview.List = models.QueryImages(listview.Page, listview.PageSize)
+	listview.Total = models.CountImages()
+	listview.Next = listview.Page*listview.PageSize < listview.Total
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(listview))
+}
+
+func images_post(w http.ResponseWriter, r *http.Request) {
+	var image models.Image
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &image); err != nil {
+		log.Println(err)
+		return
+	}
+	image.Create()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(image))
+}
+
+func images_item_get(w http.ResponseWriter, r *http.Request) {
+	image := models.Image{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	image.Get()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(image))
+}
+
+func images_item_patch(w http.ResponseWriter, r *http.Request) {
+	image := models.Image{}
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &image); err != nil {
+		log.Println(err)
+		return
+	}
+	image.ID = ParamInt(mux.Vars(r)["id"], 0)
+	image.Update()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(image))
+}
+
+func images_item_delete(w http.ResponseWriter, r *http.Request) {
+	image := models.Image{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	image.Delete()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(image))
+}
+
+func tasks_get(w http.ResponseWriter, r *http.Request) {
+	var listview models.ListView
+	listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+	listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+	listview.List = models.QueryTasks(listview.Page, listview.PageSize)
+	listview.Total = models.CountTasks()
+	listview.Next = listview.Page*listview.PageSize < listview.Total
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(listview))
+}
+
+func tasks_post(w http.ResponseWriter, r *http.Request) {
+	var task models.Task
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &task); err != nil {
+		log.Println(err)
+		return
+	}
+	task.Create()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(task))
+}
+
+func tasks_item_get(w http.ResponseWriter, r *http.Request) {
+	task := models.Task{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	task.Get()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(task))
+}
+
+func tasks_item_patch(w http.ResponseWriter, r *http.Request) {
+	task := models.Task{}
+	body, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	defer r.Body.Close()
+	if err = json.Unmarshal(body, &task); err != nil {
+		log.Println(err)
+		return
+	}
+	task.ID = ParamInt(mux.Vars(r)["id"], 0)
+	task.Update()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(task))
+}
+
+func tasks_item_delete(w http.ResponseWriter, r *http.Request) {
+	task := models.Task{ID: ParamInt(mux.Vars(r)["id"], 0)}
+	task.Delete()
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.Write(ToJSON(task))
+}
+
+// 中間件, 通用
+func middleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		defer 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")
+		// 處理OPTIONS請求
+		if r.Method == "OPTIONS" {
+			w.WriteHeader(http.StatusOK)
+			return
+		}
+		next.ServeHTTP(w, r)
+	})
+}
+
+func ToJSON(object interface{}) []byte {
+	json, err := json.MarshalIndent(object, "", "  ")
+	if err != nil {
+		log.Println(err)
+		return []byte{}
+	}
+	return json
+}
+
 func LogComponent(startTime int64, r *http.Request) {
 	ms := (time.Now().UnixNano() - startTime) / 1000000
 	color := "\033[1;32m%d\033[0m"
@@ -162,11 +277,11 @@ func LogComponent(startTime int64, r *http.Request) {
 }
 
 // 獲取查詢參數(int 類型)
-func ParamInt(value string, defaultValue int64) int64 {
+func ParamInt(value string, defaultValue int) int {
 	if value == "" {
 		return defaultValue
 	}
-	result, err := strconv.ParseInt(value, 10, 64)
+	result, err := strconv.Atoi(value)
 	if err != nil {
 		return defaultValue
 	}