歸併

main.go

@@ -1,18 +1,158 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
+	"io/ioutil"
 	"log"
 	"net/http"
 	"runtime"
+	"strconv"
 	"time"
 
 	"main/models"
+
+	"github.com/gorilla/mux"
 )
 
+type Object interface {
+	ToJSON() []byte
+}
+
+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())
+	return
+}
+
+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())
+	return
+}
+
+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())
+	return
+}
+
+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())
+	return
+}
+
+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())
+	return
+}
+
 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")
+
+	http.ListenAndServe(":8080", r)
+
+	// 初始化數據庫
+	//db, err := configs.GetDB()
+	//if err != nil {
+	//	log.Fatal(err)
+	//}
+
+	/**
 	http.HandleFunc("/api", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
 		wecome := "Welcome to the home page!"
@@ -20,51 +160,274 @@ func main() {
 	})
 	http.HandleFunc("/api/images", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-		var listview models.ListView
-		w.Header().Set("Content-Type", "application/json; charset=utf-8")
-		w.Write(listview.ToJson())
+
+		// 處理GET請求
+		if r.Method == "GET" {
+			var listview models.ListView
+
+			// 獲取查詢參數: page, pageSize
+			listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+			listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+
+			// 拼接查詢條件
+			condition := ""
+			if user_id := ParamString(r.URL.Query().Get("user_id"), ""); user_id != "" {
+				condition += fmt.Sprintf(" AND user_id = '%s'", user_id)
+			}
+			if created_at := ParamString(r.URL.Query().Get("created_at"), ""); created_at != "" {
+				condition += fmt.Sprintf(" AND created_at = '%s'", created_at)
+			}
+			if updated_at := ParamString(r.URL.Query().Get("updated_at"), ""); updated_at != "" {
+				condition += fmt.Sprintf(" AND updated_at = '%s'", updated_at)
+			}
+
+			// 獲取總數
+			row := db.QueryRow("SELECT COUNT(*) FROM images WHERE 1 = 1" + condition)
+			err := row.Scan(&listview.Total)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 從數據庫查詢一組圖片
+			rows, err := db.Query("SELECT id, name, created_at, updated_at FROM images LIMIT ?, ?", (listview.Page-1)*listview.PageSize, listview.PageSize)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			defer rows.Close()
+
+			// 將查詢結果裝載到ListView中
+			for rows.Next() {
+				var image models.Image
+				err := rows.Scan(&image.ID, &image.Name, &image.CreatedAt, &image.UpdatedAt)
+				if err != nil {
+					log.Println(err)
+					return
+				}
+				listview.List = append(listview.List, image)
+			}
+
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(listview.ToJSON())
+			return
+		}
+
+		// 處理POST請求
+		if r.Method == "POST" {
+			// 從body中獲取收到的JSON數據
+			body, err := ioutil.ReadAll(r.Body)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 將JSON數據轉換成Image對象
+			image := models.Image{}
+			err = image.FromJSON(body)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 將Image對象存入數據庫
+			err = image.Create()
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(image.ToJSON())
+			return
+		}
+
+		// 處理DELETE請求
+		if r.Method == "DELETE" {
+			// 從body中獲取收到的JSON數據
+			body, err := ioutil.ReadAll(r.Body)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 將JSON數據轉換成Image對象
+			image := models.Image{}
+			err = image.FromJSON(body)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 將Image對象從數據庫中刪除
+			err = image.Delete()
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(image.ToJSON())
+			return
+		}
+
 	})
 	http.HandleFunc("/api/models", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
-		var listview models.ListView
-		w.Header().Set("Content-Type", "application/json; charset=utf-8")
-		w.Write(listview.ToJson())
+
+		// 創建一個模型
+		if r.Method == "POST" {
+			// 從body中解析出模型
+			body, err := ioutil.ReadAll(r.Body)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			defer r.Body.Close()
+
+			// 將JSON數據轉換成Model對象
+			var model models.Model
+			if err := json.Unmarshal(body, &model); err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 將Model對象存入數據庫
+			err = model.Create()
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// 返回創建的模型
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(model.ToJSON())
+			return
+		}
+
+		// 查詢一組模型
+		if r.Method == "GET" {
+			var listview models.ListView
+
+			// 獲取查詢參數: page, pageSize
+			listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+			listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+
+			// 拼接查詢條件
+			condition := ""
+			if user_id := ParamString(r.URL.Query().Get("user_id"), ""); user_id != "" {
+				condition += fmt.Sprintf(" AND user_id = '%s'", user_id)
+			}
+			if created_at := ParamString(r.URL.Query().Get("created_at"), ""); created_at != "" {
+				condition += fmt.Sprintf(" AND created_at = '%s'", created_at)
+			}
+			if updated_at := ParamString(r.URL.Query().Get("updated_at"), ""); updated_at != "" {
+				condition += fmt.Sprintf(" AND updated_at = '%s'", updated_at)
+			}
+
+			// 從數據庫查詢一組模型
+			rows, err := db.Query("SELECT id, name, created_at, updated_at FROM models LIMIT ?, ?", (listview.Page-1)*listview.PageSize, listview.PageSize)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			defer rows.Close()
+
+			// 將查詢結果裝載到ListView中
+			for rows.Next() {
+				var model models.Model
+				err := rows.Scan(&model.ID, &model.Name, &model.CreatedAt, &model.UpdatedAt)
+				if err != nil {
+					log.Println(err)
+					return
+				}
+				listview.List = append(listview.List, model)
+			}
+
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(listview.ToJSON())
+			return
+		}
 	})
 	http.HandleFunc("/api/tasks", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
 
-		// 查詢一組任務
-		// tasks, err := models.GetTasks()
-		// if err != nil {
-		// 	log.Println(err)
-		// 	return
-		// }
-		// fmt.Println(tasks)
+		// 創建一個任務
+		if r.Method == "POST" {
+			task := models.Task{}
+			task.Type = ParamString(r.URL.Query().Get("type"), "")
+			task.CreatedAt = ParamString(r.URL.Query().Get("created_at"), "")
+			task.UpdatedAt = ParamString(r.URL.Query().Get("updated_at"), "")
+			err := task.Create()
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(task.ToJSON())
+			return
+		}
 
-		// 將取得的任務裝載到ListView中
-		// listView := ListView{}
-		// for _, task := range tasks {
-		// 	listView.Tasks = append(listView.Tasks, task)
-		// }
-		var listview models.ListView
-		w.Header().Set("Content-Type", "application/json; charset=utf-8")
-		w.Write(listview.ToJson())
+		// 查詢一組任務
+		if r.Method == "GET" {
+			var listview models.ListView
+
+			// 獲取查詢參數: page, pageSize
+			listview.Page = ParamInt(r.URL.Query().Get("page"), 1)
+			listview.PageSize = ParamInt(r.URL.Query().Get("pageSize"), 10)
+
+			// 拼接查詢條件
+			condition := ""
+			if user_id := ParamString(r.URL.Query().Get("user_id"), ""); user_id != "" {
+				condition += fmt.Sprintf(" AND user_id = '%s'", user_id)
+			}
+			if created_at := ParamString(r.URL.Query().Get("created_at"), ""); created_at != "" {
+				condition += fmt.Sprintf(" AND created_at = '%s'", created_at)
+			}
+			if updated_at := ParamString(r.URL.Query().Get("updated_at"), ""); updated_at != "" {
+				condition += fmt.Sprintf(" AND updated_at = '%s'", updated_at)
+			}
+
+			// 從數據庫查詢一組任務
+			rows, err := db.Query("SELECT id, type, created_at, updated_at FROM tasks LIMIT ?, ?", (listview.Page-1)*listview.PageSize, listview.PageSize)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			defer rows.Close()
+
+			// 將查詢結果裝載到ListView中
+			for rows.Next() {
+				var task models.Task
+				err := rows.Scan(&task.ID, &task.Type, &task.CreatedAt, &task.UpdatedAt)
+				if err != nil {
+					log.Println(err)
+					return
+				}
+				listview.List = append(listview.List, task)
+			}
+
+			w.Header().Set("Content-Type", "application/json; charset=utf-8")
+			w.Write(listview.ToJSON())
+			return
+		}
 	})
 	http.HandleFunc("/api/users", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
 		var listview models.ListView
 		w.Header().Set("Content-Type", "application/json; charset=utf-8")
-		w.Write(listview.ToJson())
+		w.Write(listview.ToJSON())
 	})
 	http.HandleFunc("/api/tags", func(w http.ResponseWriter, r *http.Request) {
 		defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志
 		var listview models.ListView
 		w.Header().Set("Content-Type", "application/json; charset=utf-8")
-		w.Write(listview.ToJson())
+		w.Write(listview.ToJSON())
 	})
 
 	log.Println("Web Server is running on http://localhost:8080")
 	http.ListenAndServe(":8080", nil)
+	**/
 }
 
 func LogComponent(startTime int64, r *http.Request) {
@@ -88,3 +451,35 @@ func LogComponent(startTime int64, r *http.Request) {
 	url := fmt.Sprintf("\033[1;34m%s\033[0m", r.URL)       // 藍色加重
 	log.Println(method, url, endTime)
 }
+
+// 獲取查詢參數(int 類型)
+func ParamInt(value string, defaultValue int64) int64 {
+	if value == "" {
+		return defaultValue
+	}
+	result, err := strconv.ParseInt(value, 10, 64)
+	if err != nil {
+		return defaultValue
+	}
+	return result
+}
+
+// 獲取查詢參數(string 類型)
+func ParamString(value string, defaultValue string) string {
+	if value == "" {
+		return defaultValue
+	}
+	return value
+}
+
+// 獲取查詢參數(bool 類型)
+func ParamBool(value string, defaultValue bool) bool {
+	if value == "" {
+		return defaultValue
+	}
+	result, err := strconv.ParseBool(value)
+	if err != nil {
+		return defaultValue
+	}
+	return result
+}