更改推理结构
This commit is contained in:
106
routers/tasks.go
106
routers/tasks.go
@@ -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)
|
||||
}
|
Reference in New Issue
Block a user