更改推理结构

This commit is contained in:
2023-06-28 14:29:44 +08:00
parent 9bcd9540ef
commit 723eca3353
7 changed files with 211 additions and 384 deletions

View File

@@ -1,106 +0,0 @@
package routers
import (
"encoding/json"
"io/ioutil"
"log"
"main/configs"
"main/models"
"main/utils"
"net/http"
"strconv"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
)
func TasksGet(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)
var task_list []models.Task
db := configs.ORMDB()
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&task_list)
for _, task := range task_list {
listview.List = append(listview.List, task)
}
db.Model(&models.Task{}).Count(&listview.Total)
listview.Next = listview.Page*listview.PageSize < int(listview.Total)
listview.WriteJSON(w)
}
func TasksPost(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
}
configs.ORMDB().Create(&task)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(task))
}
func TasksItemGet(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Upgrade") == "websocket" {
vars := mux.Vars(r)
id, _ := strconv.Atoi(vars["id"])
var task models.Task = models.Task{ID: id}
if err := configs.ORMDB().First(&task, id).Error; err != nil {
log.Println(err)
w.WriteHeader(http.StatusNotFound)
return
}
upgrader := websocket.Upgrader{}
ws, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(err)
return
}
defer ws.Close()
for {
_, message, err := ws.ReadMessage()
if err != nil {
log.Println(err)
break
}
task.Status = string(message)
configs.ORMDB().Model(&task).Update("status", task.Status)
}
return
}
task := models.Task{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
configs.ORMDB().First(&task)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(task))
}
func TasksItemPatch(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.ID = utils.ParamInt(mux.Vars(r)["id"], 0)
configs.ORMDB().Model(&task).Updates(task)
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.Write(utils.ToJSON(task))
}
func TasksItemDelete(w http.ResponseWriter, r *http.Request) {
task := models.Task{ID: utils.ParamInt(mux.Vars(r)["id"], 0)}
configs.ORMDB().Delete(&task)
w.WriteHeader(http.StatusNoContent)
}