diff --git a/main.go b/main.go index 3107884..7b3c0c5 100644 --- a/main.go +++ b/main.go @@ -15,10 +15,6 @@ import ( "github.com/gorilla/mux" ) -type Object interface { - ToJSON() []byte -} - func CreateObject(w http.ResponseWriter, r *http.Request) { defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志 @@ -47,7 +43,6 @@ func CreateObject(w http.ResponseWriter, r *http.Request) { // 返回結果 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(object.ToJSON()) - return } func GetObjetc(w http.ResponseWriter, r *http.Request) { @@ -64,7 +59,6 @@ func GetObjetc(w http.ResponseWriter, r *http.Request) { // 返回結果 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(object.ToJSON()) - return } func UpdateObject(w http.ResponseWriter, r *http.Request) { @@ -95,7 +89,6 @@ func UpdateObject(w http.ResponseWriter, r *http.Request) { // 返回結果 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(object.ToJSON()) - return } func DeleteObject(w http.ResponseWriter, r *http.Request) { @@ -112,7 +105,6 @@ func DeleteObject(w http.ResponseWriter, r *http.Request) { // 返回結果 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(object.ToJSON()) - return } func GetObjects(w http.ResponseWriter, r *http.Request) { @@ -131,7 +123,6 @@ func GetObjects(w http.ResponseWriter, r *http.Request) { // 返回結果 w.Header().Set("Content-Type", "application/json; charset=utf-8") w.Write(listview.ToJSON()) - return } func main() { @@ -144,290 +135,8 @@ func main() { 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!" - w.Write([]byte(wecome)) - }) - http.HandleFunc("/api/images", func(w http.ResponseWriter, r *http.Request) { - defer LogComponent(time.Now().UnixNano(), r) // 最后打印日志 - - // 處理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) // 最后打印日志 - - // 創建一個模型 - 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) // 最后打印日志 - - // 創建一個任務 - 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 - } - - // 查詢一組任務 - 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()) - }) - 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()) - }) - log.Println("Web Server is running on http://localhost:8080") - http.ListenAndServe(":8080", nil) - **/ + http.ListenAndServe(":8080", r) } func LogComponent(startTime int64, r *http.Request) {