76 lines
2.5 KiB
Go
76 lines
2.5 KiB
Go
package test
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/gogf/gf/v2/container/gvar"
|
||
"github.com/gogf/gf/v2/util/gconv"
|
||
"github.com/samber/lo"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
|
||
"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
|
||
wxDao "github.com/tiger1103/gfast/v3/internal/app/wxApplet/dao"
|
||
"github.com/tiger1103/gfast/v3/library/liberr"
|
||
richtext "github.com/tiger1103/gfast/v3/third/richText"
|
||
)
|
||
|
||
// 获取所有的后台项目公告，同步一份给App
|
||
func TestSyncIntro(t *testing.T) {
|
||
intructID := []entity.SysProjectIntroduce{}
|
||
if err := dao.SysProjectIntroduce.Ctx(context.Background()).Scan(&intructID); err != nil {
|
||
t.Error(err)
|
||
}
|
||
|
||
ctx := context.Background()
|
||
|
||
var notificationsList []do.NotificationRecipients
|
||
|
||
for _, v := range intructID {
|
||
// imags := richtext.ExtractImageURLs(v.RichText)
|
||
data := entity.Notifications{
|
||
Title: v.Headline,
|
||
NotificationText: richtext.RemoveRichText(v.RichText),
|
||
NotificationTime: v.CreatedAt.Format("2006-01-02 15:04:05"),
|
||
IsApp: 2,
|
||
Initiator: gconv.Int(v.CreatedBy),
|
||
ProjectId: int(v.ProjectId),
|
||
IntroduceId: int(v.ID),
|
||
}
|
||
notifiID, err := dao.Notifications.Ctx(ctx).InsertAndGetId(data)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 获取项目下的所有用户
|
||
projectUserIDs, err := wxDao.SysUserProjectRelevancy.Ctx(ctx).
|
||
Where(wxDao.SysUserProjectRelevancy.Columns().ProjectId, int(v.ProjectId)).
|
||
Fields(wxDao.SysUserProjectRelevancy.Columns().UserId).Distinct().Array()
|
||
liberr.ErrIsNil(ctx, err)
|
||
|
||
constructionUserIDs, err := dao.BusConstructionUser.Ctx(ctx).As("bcs").
|
||
InnerJoin("sys_user as su", "bcs.phone = su.mobile").
|
||
Where("bcs.project_id", int(v.ProjectId)).
|
||
Fields("su.id").Distinct().Array()
|
||
liberr.ErrIsNil(ctx, err)
|
||
|
||
// 将两个用户ID列表合并并去重
|
||
Users := lo.Uniq(lo.Map(append(projectUserIDs, constructionUserIDs...), func(item *gvar.Var, _ int) int {
|
||
return item.Int()
|
||
}))
|
||
|
||
lo.ForEach(Users, func(item int, _ int) {
|
||
notificationsList = append(notificationsList, do.NotificationRecipients{
|
||
NotificationId: notifiID, // 通知ID
|
||
RecipientId: item, // 接收者ID
|
||
NotificationStatus: 0, // 0 未读
|
||
IntroduceId: int(v.ID), // 关联的PC 端项目新闻ID
|
||
})
|
||
})
|
||
}
|
||
|
||
if _, err := dao.NotificationRecipients.Ctx(ctx).Insert(notificationsList); err != nil {
|
||
t.Error(err)
|
||
}
|
||
}
|