40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package models
|
||
|
||
import (
|
||
"pmail/db"
|
||
"pmail/utils/context"
|
||
"pmail/utils/errors"
|
||
)
|
||
|
||
type Rule struct {
|
||
Id int `xorm:"id int unsigned not null pk autoincr" json:"id"`
|
||
UserId int `xorm:"user_id notnull default(0) comment('用户id')" json:"user_id"`
|
||
Name string `xorm:"name notnull default('') comment('规则名称')" json:"name"`
|
||
Value string `xorm:"value text comment('规则内容')" json:"value"`
|
||
Action int `xorm:"action notnull default(0) comment('执行动作,1已读，2转发，3删除')" json:"action"`
|
||
Params string `xorm:"params notnull default('') comment('执行参数')" json:"params"`
|
||
Sort int `xorm:"sort notnull default(0) comment('排序，越大约优先')" json:"sort"`
|
||
}
|
||
|
||
func (p *Rule) TableName() string {
|
||
return "rule"
|
||
}
|
||
|
||
func (p *Rule) Save(ctx *context.Context) error {
|
||
|
||
if p.Id > 0 {
|
||
_, err := db.Instance.Exec(db.WithContext(ctx, "update rule set name=? ,value = ? ,action = ?,params = ?,sort = ? where id = ?"), p.Name, p.Value, p.Action, p.Params, p.Sort, p.Id)
|
||
if err != nil {
|
||
return errors.Wrap(err)
|
||
}
|
||
return nil
|
||
} else {
|
||
_, err := db.Instance.Exec(db.WithContext(ctx, "insert into rule (name,value,user_id,action,params,sort) values (?,?,?,?,?,?)"), p.Name, p.Value, ctx.UserID, p.Action, p.Params, p.Sort)
|
||
if err != nil {
|
||
return errors.Wrap(err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
}
|