107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
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)
|
|
}
|