111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
package reminders
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/jinzhu/copier"
|
||
"github.com/samber/lo"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
|
||
)
|
||
|
||
type Reminder struct {
|
||
Content string `copier:"ViolationType"` // 内容
|
||
Title string // 标题
|
||
ReceiverID string `copier:"UserId"` // 消息接收者的 User_id
|
||
Type int `copier:"ReminderType"` // 类型
|
||
Status int // 状态
|
||
ProjectID int `copier:"ProjectId"` // 项目 ID
|
||
TargetID int `copier:"OrderId"` // 对应跳转的主键ID
|
||
|
||
//TargetID int `copier:"OrderId"` // 是否需要转换
|
||
}
|
||
|
||
// UserIDOrderIDDel 根据userID与orderID 删除提醒
|
||
func UserIDOrderIDDel(userId string, orderID int64) error {
|
||
columns := dao.Reminders.Columns()
|
||
_, err := dao.Reminders.Ctx(context.Background()).Where(columns.UserId, userId).Where(columns.OrderId, orderID).Delete()
|
||
return err
|
||
}
|
||
|
||
// PublishReminder 发布提醒
|
||
func PublishReminder(message Reminder, convertOpenID bool) error {
|
||
if convertOpenID && HasLetter(message.ReceiverID) {
|
||
message.ReceiverID = GetUserIDByOpenID(message.ReceiverID)
|
||
if message.ReceiverID == "" {
|
||
return fmt.Errorf("未找到对应的用户")
|
||
}
|
||
}
|
||
|
||
var reminder do.Reminders
|
||
copier.Copy(&reminder, &message)
|
||
|
||
// 保存到数据库
|
||
_, err := dao.Reminders.Ctx(context.Background()).Insert(&reminder)
|
||
return err
|
||
}
|
||
|
||
// CheckInReminder 打卡提醒
|
||
// Reminders.CheckIn 表示上班打卡
|
||
// Reminders.CheckOut 表示下班打卡
|
||
func CheckInReminder(checkType ReminderType, projectID int, userID, time string) error {
|
||
// 1. 构建消息模板
|
||
message := Reminder{
|
||
Type: checkType, // 打卡类型
|
||
ProjectID: projectID, // 项目 ID
|
||
ReceiverID: userID, // 消息接收者的 sys_user.Id
|
||
Title: time, // 时间，只需要多少分钟，如 "10"，"20"
|
||
}
|
||
|
||
return PublishReminder(message, false)
|
||
}
|
||
|
||
// hasLetter 判断字符串中是否包含字母
|
||
func HasLetter(s string) bool {
|
||
for _, c := range s {
|
||
if (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// GetUserIDByOpenID 通过 OpenID 获取用户对应的 sys_user.Id
|
||
func GetUserIDByOpenID(openID string) string {
|
||
openid, err := dao.BusConstructionUser.Ctx(context.Background()).As("bu").
|
||
Fields("su.id").
|
||
InnerJoin("sys_user AS su", "bu.phone = su.mobile").
|
||
Where("bu.openid", openID).Value()
|
||
if err != nil {
|
||
return ""
|
||
}
|
||
|
||
return openid.String()
|
||
}
|
||
|
||
type userOpenID struct {
|
||
OpenID string
|
||
ID int
|
||
}
|
||
|
||
// GetUserIDByOpenIDs 传入多个 OpenID，返回对应的 User.ID
|
||
func GetUserIDByOpenIDs(openIDs []string) map[string]string {
|
||
// SELECT su.id,bu.openid
|
||
// FROM bus_construction_user AS bu
|
||
// INNER JOIN sys_user AS su ON bu.phone = su.mobile
|
||
// WHERE bu.openid in ("oLYsI4x-L6ZVKCinQ4Rw1D9XtFZ0","oLYsI49aMJHTYorhmvVuRrasPTbU")
|
||
|
||
var userOpenIDList []userOpenID
|
||
err := dao.BusConstructionUser.Ctx(context.Background()).As("bu").
|
||
Fields("su.id", "bu.openid").
|
||
InnerJoin("sys_user AS su", "bu.phone = su.mobile").
|
||
WhereIn("bu.openid", openIDs).Scan(&userOpenIDList)
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
|
||
return lo.SliceToMap(userOpenIDList, func(item userOpenID) (string, string) {
|
||
return item.OpenID, fmt.Sprintf("%d", item.ID)
|
||
})
|
||
}
|