/
task.go
44 lines (37 loc) · 893 Bytes
/
task.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package models
import (
"log"
"github.com/hellojukay/todo-server/utils"
"gorm.io/gorm"
)
// Task 任务实体
type Task struct {
ID int64 `gorm:"primaryKey" json:"id"`
Title string `gorm:"title" json:"title"`
Desc string `gorm:"desc" json:"desc"`
Finished bool `gorm:"finished" json:"finished"`
}
func ListAllTasks(_finished bool) []Task {
finish := 0
if _finished {
finish = 1
}
var tasks []Task
if err := Store.Where(map[string]interface{}{"finished": finish}).Find(&tasks).Error; err != nil && err != gorm.ErrRecordNotFound {
log.Fatal(err)
}
return tasks
}
func SaveTask(task Task) (Task, error) {
if task.ID == 0 {
task.ID = utils.NextID()
}
var err = Store.Save(&task).Error
return task, err
}
func RemoveTask(id int64) error {
return Store.Delete(Task{ID: id}).Error
}
func UpdateTask(task Task) error {
return Store.Save(&task).Error
}