gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gorm

Browse Source
This commit is contained in:
桜華
2023-05-14 04:45:52 +08:00
parent 9870c903ca
commit 2423213e9a
6 changed files with 75 additions and 158 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
routers/models.go
View File

@@ -5,6 +5,7 @@ import (
"fmt"
"io/ioutil"
"log"
"main/configs"
"main/models"
"main/utils"
"net/http"
@@ -17,11 +18,21 @@ import (
var manager = models.NewWebSocketManager()
func ModelsGet(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.QueryModels(listview.Page, listview.PageSize)
listview.Total = models.CountModels()
// 獲取模型列表
var model_list []models.Model
db := configs.ORMDB()
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&model_list)
for _, model := range model_list {
listview.List = append(listview.List, model)
}
// 獲取總數
var total int64
db.Model(&models.Model{}).Count(&total)
listview.Total = int(total)
listview.Next = listview.Page*listview.PageSize < listview.Total
listview.WriteJSON(w)
}
@@ -43,7 +54,10 @@ func ModelsPost(w http.ResponseWriter, r *http.Request) {
log.Println(err)
return
}
model.Create()
if err := configs.ORMDB().Create(&model).Error; err != nil {
log.Println(err)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(model))
})
@@ -53,11 +67,13 @@ func ModelItemGet(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Upgrade") == "websocket" {
vars := mux.Vars(r)
id, _ := strconv.Atoi(vars["id"])
model := models.QueryModel(id)
if model.ID == 0 {
var model = models.Model{ID: id}
if err := configs.ORMDB().Take(&model, id).Error; err != nil {
w.WriteHeader(http.StatusNotFound)
return
}
upgrader := websocket.Upgrader{}
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
@@ -80,16 +96,20 @@ func ModelItemGet(w http.ResponseWriter, r *http.Request) {
}
return
}
model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
model.Get()
var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)); err != nil {
w.WriteHeader(http.StatusNotFound)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(model))
}
func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
// 取出原始数据
model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
if err := model.Get(); err != nil {
var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)); err != nil {
w.WriteHeader(http.StatusNotFound)
return
}
@@ -125,13 +145,21 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
model.Image = model_new.Image
}
model.Update()
if err := configs.ORMDB().Save(&model).Error; err != nil {
log.Println(err)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(model))
}
func ModelItemDelete(w http.ResponseWriter, r *http.Request) {
model := models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
model.Delete()
w.WriteHeader(http.StatusNoContent)
var model = models.Model{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
if err := configs.ORMDB().Take(&model, utils.ParamInt(mux.Vars(r)["id"], 0)); err != nil {
w.WriteHeader(http.StatusNotFound)
return
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(model))
}