gorm

routers/sessions.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
+	"main/configs"
 	"main/models"
 	"main/utils"
 	"net/http"
@@ -17,9 +18,14 @@ func SessionsGet(w http.ResponseWriter, r *http.Request) {
 	var listview models.ListView
 	listview.Page = utils.ParamInt(r.URL.Query().Get("page"), 1)
 	listview.PageSize = utils.ParamInt(r.URL.Query().Get("pageSize"), 10)
-	listview.List = models.QuerySessions(listview.Page, listview.PageSize)
-	listview.Total = models.CountSessions()
-	listview.Next = listview.Page*listview.PageSize < listview.Total
+	var session_list []models.Session
+	db := configs.ORMDB()
+	db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&session_list)
+	for _, session := range session_list {
+		listview.List = append(listview.List, session)
+	}
+	db.Model(&models.Session{}).Count(&listview.Total)
+	listview.Next = listview.Page*listview.PageSize < int(listview.Total)
 	listview.WriteJSON(w)
 }
 
@@ -55,8 +61,8 @@ func SessionsPost(w http.ResponseWriter, r *http.Request) {
 	}
 
 	// 使用Email獲取用戶
-	user, err := models.GetUserByEmail(form.Email)
-	if err != nil {
+	var user models.User
+	if err := configs.ORMDB().Where("email = ?", form.Email).First(&user).Error; err != nil {
 		w.WriteHeader(http.StatusUnauthorized)
 		w.Write([]byte("404 - User Not Found"))
 		return
@@ -71,7 +77,11 @@ func SessionsPost(w http.ResponseWriter, r *http.Request) {
 
 	// 創建會話(生成一個不重複的 uuid 作爲 sid)
 	session := &models.Session{ID: uuid.New().String(), UserID: user.ID}
-	session.Create()
+	if err := configs.ORMDB().Create(session).Error; err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte("500 - Internal Server Error"))
+		return
+	}
 
 	// 寫入Cookie
 	cookie := http.Cookie{Name: "session_id", Value: session.ID, Path: "/", HttpOnly: true}
@@ -85,7 +95,11 @@ func SessionsPost(w http.ResponseWriter, r *http.Request) {
 // 獲取會話
 func SessionsItemGet(w http.ResponseWriter, r *http.Request) {
 	session := models.Session{ID: mux.Vars(r)["session_id"]}
-	session.Get()
+	if err := configs.ORMDB().Find(&session).Error; err != nil {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte("404 - Not Found"))
+		return
+	}
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(session))
 }
@@ -93,8 +107,11 @@ func SessionsItemGet(w http.ResponseWriter, r *http.Request) {
 // 更新會話
 func SessionsItemPatch(w http.ResponseWriter, r *http.Request) {
 	session := models.Session{ID: mux.Vars(r)["session_id"]}
-	session.Get()
-	session.Update()
+	if err := configs.ORMDB().Model(&session).Updates(GetForm(r)); err != nil {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte("404 - Not Found"))
+		return
+	}
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(session))
 }
@@ -113,14 +130,22 @@ func SessionsItemDelete(w http.ResponseWriter, r *http.Request) {
 
 	// 獲取當前session
 	session := models.Session{ID: cookie.Value}
-	session.Get()
+	if err := configs.ORMDB().Find(&session).Error; err != nil {
+		w.WriteHeader(http.StatusUnauthorized)
+		w.Write([]byte("401 - 會話已過期"))
+		return
+	}
 
 	// 獲取當前用戶
 	user := models.User{ID: session.UserID}
-	user.Get()
+	configs.ORMDB().Find(&user)
 
 	sessionx := models.Session{ID: mux.Vars(r)["session_id"]}
-	sessionx.Get()
+	if err := configs.ORMDB().Find(&sessionx).Error; err != nil {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte("404 - Not Found"))
+		return
+	}
 
 	if user.ID != sessionx.UserID {
 		w.WriteHeader(http.StatusUnauthorized)
@@ -128,7 +153,11 @@ func SessionsItemDelete(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	sessionx.Delete()
+	if err := configs.ORMDB().Delete(&sessionx).Error; err != nil {
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte("404 - Not Found"))
+		return
+	}
 	w.Header().Set("Content-Type", "application/json; charset=utf-8")
 	w.Write(utils.ToJSON(sessionx))
 }