初版

定时任务管理

models/task_group.go

@@ -0,0 +1,71 @@
+/*
+* @Author: haodaquan
+* @Date:   2017-06-21 12:22:37
+* @Last Modified by:   haodaquan
+* @Last Modified time: 2017-06-21 12:22:55
+ */
+
+package models
+
+import (
+	"fmt"
+	"github.com/astaxie/beego/orm"
+)
+
+type TaskGroup struct {
+	Id          int
+	UserId      int
+	GroupName   string
+	Description string
+	CreateTime  int64
+}
+
+func (t *TaskGroup) TableName() string {
+	return TableName("task_group")
+}
+
+func (t *TaskGroup) Update(fields ...string) error {
+	if t.GroupName == "" {
+		return fmt.Errorf("组名不能为空")
+	}
+	if _, err := orm.NewOrm().Update(t, fields...); err != nil {
+		return err
+	}
+	return nil
+}
+
+func TaskGroupAdd(obj *TaskGroup) (int64, error) {
+	if obj.GroupName == "" {
+		return 0, fmt.Errorf("组名不能为空")
+	}
+	return orm.NewOrm().Insert(obj)
+}
+
+func TaskGroupGetById(id int) (*TaskGroup, error) {
+	obj := &TaskGroup{
+		Id: id,
+	}
+
+	err := orm.NewOrm().Read(obj)
+	if err != nil {
+		return nil, err
+	}
+	return obj, nil
+}
+
+func TaskGroupDelById(id int) error {
+	_, err := orm.NewOrm().QueryTable(TableName("task_group")).Filter("id", id).Delete()
+	return err
+}
+
+func TaskGroupGetList(page, pageSize int) ([]*TaskGroup, int64) {
+	offset := (page - 1) * pageSize
+
+	list := make([]*TaskGroup, 0)
+
+	query := orm.NewOrm().QueryTable(TableName("task_group"))
+	total, _ := query.Count()
+	query.OrderBy("-id").Limit(pageSize, offset).All(&list)
+
+	return list, total
+}