153 lines
3.7 KiB
Go
153 lines
3.7 KiB
Go
/**
|
|
* @Author: Echo
|
|
* @Author:1711788888@qq.com
|
|
* @Date: 2021/9/3 18:15
|
|
* @Desc: 推送消息实体
|
|
*/
|
|
|
|
package push
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"git.echol.cn/loser/tencent-im/internal/entity"
|
|
)
|
|
|
|
var errInvalidPushCondition = errors.New("attrs and tags condition cannot be set at the same time")
|
|
|
|
type Message struct {
|
|
entity.Message
|
|
condition *condition
|
|
}
|
|
|
|
func NewMessage() *Message {
|
|
return &Message{}
|
|
}
|
|
|
|
// SetConditionTagsOr 设置标签的或条件（设置会冲掉之前的标签或条件）
|
|
func (m *Message) SetConditionTagsOr(tags ...string) {
|
|
if m.condition != nil && m.condition.TagsOr != nil {
|
|
m.condition.TagsOr = m.condition.TagsOr[0:0]
|
|
}
|
|
|
|
m.AddConditionTagsOr(tags...)
|
|
}
|
|
|
|
// AddConditionTagsOr 添加标签的或条件（添加会累加之前的条件或条件）
|
|
func (m *Message) AddConditionTagsOr(tags ...string) {
|
|
if m.condition == nil {
|
|
m.condition = &condition{}
|
|
}
|
|
if m.condition.TagsOr == nil {
|
|
m.condition.TagsOr = make([]string, 0)
|
|
}
|
|
m.condition.TagsOr = append(m.condition.TagsOr, tags...)
|
|
}
|
|
|
|
// SetConditionTagsAnd 设置标签的与条件（设置会冲掉之前的标签与条件）
|
|
func (m *Message) SetConditionTagsAnd(tags ...string) {
|
|
if m.condition != nil && m.condition.TagsAnd != nil {
|
|
m.condition.TagsAnd = m.condition.TagsAnd[0:0]
|
|
}
|
|
|
|
m.AddConditionTagsAnd(tags...)
|
|
}
|
|
|
|
// AddConditionTagsAnd 添加标签的与条件（添加会累加之前的标签与条件）
|
|
func (m *Message) AddConditionTagsAnd(tags ...string) {
|
|
if m.condition == nil {
|
|
m.condition = &condition{}
|
|
}
|
|
if m.condition.TagsAnd == nil {
|
|
m.condition.TagsAnd = make([]string, 0)
|
|
}
|
|
m.condition.TagsAnd = append(m.condition.TagsAnd, tags...)
|
|
}
|
|
|
|
// SetConditionAttrsOr 设置属性的或条件（设置会冲掉之前的属性或条件）
|
|
func (m *Message) SetConditionAttrsOr(attrs map[string]interface{}) {
|
|
if m.condition != nil && m.condition.AttrsOr != nil {
|
|
m.condition.AttrsOr = make(map[string]interface{})
|
|
}
|
|
|
|
m.AddConditionAttrsOr(attrs)
|
|
}
|
|
|
|
// AddConditionAttrsOr 添加属性的或条件（添加会累加之前的属性或条件）
|
|
func (m *Message) AddConditionAttrsOr(attrs map[string]interface{}) {
|
|
if m.condition == nil {
|
|
m.condition = &condition{}
|
|
}
|
|
if m.condition.AttrsOr == nil {
|
|
m.condition.AttrsOr = make(map[string]interface{})
|
|
}
|
|
for k, v := range attrs {
|
|
m.condition.AttrsOr[k] = v
|
|
}
|
|
}
|
|
|
|
// SetConditionAttrsAnd 设置属性的与条件（设置会冲掉之前的属性与条件）
|
|
func (m *Message) SetConditionAttrsAnd(attrs map[string]interface{}) {
|
|
if m.condition != nil && m.condition.AttrsAnd != nil {
|
|
m.condition.AttrsAnd = make(map[string]interface{})
|
|
}
|
|
|
|
m.AddConditionAttrsAnd(attrs)
|
|
}
|
|
|
|
// AddConditionAttrsAnd 添加属性的与条件（添加会累加之前的属性与条件）
|
|
func (m *Message) AddConditionAttrsAnd(attrs map[string]interface{}) {
|
|
if m.condition == nil {
|
|
m.condition = &condition{}
|
|
}
|
|
if m.condition.AttrsAnd == nil {
|
|
m.condition.AttrsAnd = make(map[string]interface{})
|
|
}
|
|
for k, v := range attrs {
|
|
m.condition.AttrsAnd[k] = v
|
|
}
|
|
}
|
|
|
|
// GetCondition 获取推送条件
|
|
func (m *Message) GetCondition() *condition {
|
|
return m.condition
|
|
}
|
|
|
|
// checkError 检测错误
|
|
func (m *Message) checkError() (err error) {
|
|
if err = m.CheckLifeTimeArgError(); err != nil {
|
|
return
|
|
}
|
|
|
|
if err = m.CheckBodyArgError(); err != nil {
|
|
return
|
|
}
|
|
|
|
if err = m.checkConditionArgError(); err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// checkConditionArgError 检测条件参数错误
|
|
func (m *Message) checkConditionArgError() error {
|
|
hasAttrs, hasTags := false, false
|
|
|
|
if m.condition != nil {
|
|
if m.condition.AttrsAnd != nil || m.condition.AttrsOr != nil {
|
|
hasAttrs = true
|
|
}
|
|
|
|
if m.condition.TagsAnd != nil || m.condition.TagsOr != nil {
|
|
hasTags = true
|
|
}
|
|
}
|
|
|
|
if hasAttrs && hasTags {
|
|
return errInvalidPushCondition
|
|
}
|
|
|
|
return nil
|
|
}
|