更改推理结构
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