新增自定义通知模板

2019-02-16 20:43:33 +08:00
parent 374b305d96
commit 7bd18fdb64
16 changed files with 830 additions and 112 deletions
--- a/models/notify_tpl.go
+++ b/models/notify_tpl.go
@@ -0,0 +1,102 @@
+/************************************************************
+** @Description: models
+** @Author: Bee
+** @Date:   2019-02-15 20:21
+** @Last Modified by:   Bee
+** @Last Modified time: 2019-02-15 20:21
+*************************************************************/
+package models
+
+import (
+	"fmt"
+	"github.com/astaxie/beego/orm"
+	"time"
+)
+
+type NotifyTpl struct {
+	Id         int
+	Type       string
+	TplName    string
+	TplType    int
+	Title      string
+	Content    string
+	Status     int
+	CreateId   int
+	UpdateId   int
+	CreateTime int64
+	UpdateTime int64
+}
+
+func (t *NotifyTpl) TableName() string {
+	return TableName("notify_tpl")
+}
+
+func (t *NotifyTpl) Update(fields ...string) error {
+	if t.TplName == "" {
+		return fmt.Errorf("模板名称不能为空")
+	}
+
+	if t.Content == "" {
+		return fmt.Errorf("模板内容不能为空")
+	}
+
+	if t.CreateTime == 0 {
+		t.CreateTime = time.Now().Unix()
+	}
+
+	if _, err := orm.NewOrm().Update(t, fields...); err != nil {
+		return err
+	}
+	return nil
+}
+
+func NotifyTplAdd(obj *NotifyTpl) (int64, error) {
+	if obj.TplName == "" {
+		return 0, fmt.Errorf("模板名称不能为空")
+	}
+	if obj.Content == "" {
+		return 0, fmt.Errorf("模板内容不能为空")
+	}
+	if obj.CreateTime == 0 {
+		obj.CreateTime = time.Now().Unix()
+	}
+	return orm.NewOrm().Insert(obj)
+}
+
+func NotifyTplGetById(id int) (*NotifyTpl, error) {
+	obj := &NotifyTpl{
+		Id: id,
+	}
+	err := orm.NewOrm().Read(obj)
+	if err != nil {
+		return nil, err
+	}
+	return obj, nil
+}
+
+func NotifyTplGetByTplTypeList(tpl_type int) ([]*NotifyTpl, int64, error) {
+	list := make([]*NotifyTpl, 0)
+	total, err := orm.NewOrm().QueryTable(TableName("notify_tpl")).Filter("tpl_type", tpl_type).Filter("status", 1).All(&list)
+	return list, total, err
+}
+
+func NotifyTplDelById(id int) error {
+	_, err := orm.NewOrm().QueryTable(TableName("notify_tpl")).Filter("id", id).Delete()
+	return err
+}
+
+func NotifyTplGetList(page, pageSize int, filters ...interface{}) ([]*NotifyTpl, int64) {
+
+	offset := (page - 1) * pageSize
+	list := make([]*NotifyTpl, 0)
+	query := orm.NewOrm().QueryTable(TableName("notify_tpl"))
+	if len(filters) > 0 {
+		l := len(filters)
+		for k := 0; k < l; k += 2 {
+			query = query.Filter(filters[k].(string), filters[k+1])
+		}
+	}
+	total, _ := query.Count()
+	query.OrderBy("-id").Limit(pageSize, offset).All(&list)
+	return list, total
+}