100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/samber/lo"
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/model/entity"
|
|
|
|
wxDao "github.com/tiger1103/gfast/v3/internal/app/wxApplet/dao"
|
|
wxApplet "github.com/tiger1103/gfast/v3/internal/app/wxApplet/model/do"
|
|
)
|
|
|
|
// 获取所有的 sys_user 的用户，如果该用户在 bus_construction_user 中的手机号相同
|
|
// 则将该用户的在 bus_construction_user 中的项目绑定到 App 中
|
|
func TestUserProjectTans(t *testing.T) {
|
|
// 获取所有的 sys_user 的用户
|
|
sysUsers := GetSysUsers()
|
|
|
|
// 提取出所有用户的手机号
|
|
userMobiles := lo.FilterMap(sysUsers, func(user entity.SysUser, _ int) (string, bool) {
|
|
if user.Mobile == "" {
|
|
return "", false
|
|
}
|
|
return user.Mobile, true
|
|
})
|
|
|
|
// 获取所有的 bus_construction_user 的用户
|
|
busUsers := GetBusUsers(userMobiles)
|
|
|
|
// 1. 获取所有的 sys_user 的用户
|
|
// 2. 获取所有的 bus_construction_user 的用户
|
|
// 3. 如果该用户在 bus_construction_user 中的手机号相同
|
|
// 4. 则将该用户的在 bus_construction_user 中的项目绑定到PC中
|
|
|
|
// 批量插入列表
|
|
var insertList []wxApplet.SysUserProjectRelevancy
|
|
|
|
for _, sysUser := range sysUsers {
|
|
if busUser, ok := busUsers[sysUser.Mobile]; ok {
|
|
// 如果该数据在 sys_user_project_relevancy 中不存在，则插入
|
|
|
|
exist, err := wxDao.SysUserProjectRelevancy.Ctx(context.Background()).
|
|
Where(wxDao.SysUserProjectRelevancy.Columns().UserId, sysUser.Id).
|
|
Where(wxDao.SysUserProjectRelevancy.Columns().ProjectId, busUser.ProjectId).
|
|
Where("source", 0).
|
|
Count()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if exist == 0 {
|
|
insertList = append(insertList, wxApplet.SysUserProjectRelevancy{
|
|
UserId: sysUser.Id,
|
|
ProjectId: busUser.ProjectId,
|
|
Source: 0,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(insertList) > 0 {
|
|
if _, err := wxDao.SysUserProjectRelevancy.Ctx(context.Background()).Insert(insertList); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 获取所有的 sys_user 的用户
|
|
func GetSysUsers() []entity.SysUser {
|
|
List := []entity.SysUser{}
|
|
|
|
ctx := context.Background()
|
|
// 获取所有的 sys_user 的用户
|
|
if err := dao.SysUser.Ctx(ctx).Scan(&List); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return List
|
|
}
|
|
|
|
// 获取所有的 bus_construction_user 的用户
|
|
func GetBusUsers(mobiles []string) map[string]entity.BusConstructionUser {
|
|
ctx := context.Background()
|
|
|
|
List := []entity.BusConstructionUser{}
|
|
// 获取所有的 bus_construction_user 的用户
|
|
if err := dao.BusConstructionUser.Ctx(ctx).
|
|
WhereIn(dao.BusConstructionUser.Columns().Phone, mobiles).
|
|
WhereNotNull(dao.BusConstructionUser.Columns().ProjectId).
|
|
Scan(&List); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return lo.Associate(List, func(item entity.BusConstructionUser) (string, entity.BusConstructionUser) {
|
|
return item.Phone, item
|
|
})
|
|
}
|