msglistService.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596
  1. package common
  2. import (
  3. "app.yhyue.com/moapp/MessageCenter/entity"
  4. "app.yhyue.com/moapp/MessageCenter/rpc/type/message"
  5. "app.yhyue.com/moapp/MessageCenter/util"
  6. "app.yhyue.com/moapp/jybase/common"
  7. m "app.yhyue.com/moapp/jybase/mongodb"
  8. "app.yhyue.com/moapp/jybase/redis"
  9. "context"
  10. "errors"
  11. "fmt"
  12. "log"
  13. "strconv"
  14. "time"
  15. )
  16. func BitmapUserMsgList(this *message.UserMsgListReq) (*message.UserMsgList, *message.Messages) {
  17. var (
  18. unread, count int64
  19. data = new(message.UserMsgList)
  20. )
  21. //取出用户所有消息、已读消息id bitmap 查库
  22. userAllMsgArr, _, userUnreadArr, err := GetUserMsgSummary(this.UserId, this.IsfilterActive)
  23. if len(userAllMsgArr) == 0 && err != nil {
  24. fmt.Printf("此用户暂无消息 : %s", err.Error())
  25. }
  26. //用户分类消息
  27. _, userClassMsgMap := FindUserClassMsg(this.UserId)
  28. //用户分类未读消息
  29. classUnreadCountMap, classUnreadMsgMap := FindUserClassUnread(this.UserId)
  30. //fmt.Println("用户所有消息数:", userAllMsgArr, "已读消息数:", userReadArr, "未读消息数:", userUnreadArr)
  31. //导航未读消息总数
  32. if !this.IsMsgList && !this.IsColumnNewMsg && !this.IsColumn { //消息未读数统计
  33. _, unread = BitmapCountUnread(this.UserId, classUnreadCountMap, true)
  34. data.Unread = unread
  35. return data, nil
  36. }
  37. //获取栏目下的数据
  38. sData := make(map[string][]*message.Messages)
  39. //移动端消息列表首页
  40. if this.IsColumnNewMsg && this.SortSize > 0 {
  41. messageColumn := entity.MessageColumn
  42. if this.IsClassSearch {
  43. messageColumn = append(messageColumn, map[string]interface{}{
  44. "group_id": 12,
  45. })
  46. }
  47. for _, val := range messageColumn {
  48. if util.IntAll(val["group_id"]) > 0 && util.IntAll(val["group_id"]) < 999 {
  49. //用户分类未读 交集
  50. userClassUnreadMsgArr := classUnreadMsgMap[util.IntAll(val["group_id"])]
  51. var lastMsgId int
  52. if len(userClassUnreadMsgArr) > 0 {
  53. lastMsgId = userClassUnreadMsgArr[len(userClassUnreadMsgArr)-1]
  54. } else {
  55. //没有未读,分类展示最新一条消息
  56. classMsg := userClassMsgMap[util.IntAll(val["group_id"])]
  57. if len(classMsg) > 0 {
  58. lastMsgId = classMsg[len(classMsg)-1]
  59. } else {
  60. continue
  61. }
  62. }
  63. lastMsg := GlobMsgMap[lastMsgId]
  64. if lastMsg == nil || len(lastMsg) <= 0 {
  65. mess := entity.Mysql.FindOne("message_send_log", map[string]interface{}{"id": int(lastMsgId)}, "id,msg_type,title,content,send_time,link,menu_name,group_id,sign", "")
  66. if mess != nil && len(*mess) > 0 {
  67. lastMsg = *mess
  68. } else {
  69. continue
  70. }
  71. }
  72. if common.IntAll(lastMsg["sign"]) == 4 {
  73. //查询用户注册时间
  74. rData, _ := entity.MQFW.FindOneByField("user", map[string]interface{}{"_id": m.StringTOBsonId(this.UserId)}, `{"l_registedate":1}`)
  75. if rData != nil && len(*rData) > 0 {
  76. lastMsg["send_time"] = time.Unix(common.Int64All((*rData)["l_registedate"]), 0).Local().Format("2006-01-02 15:04:05")
  77. }
  78. }
  79. var msg = message.Messages{
  80. Id: strconv.FormatInt(util.Int64All(lastMsg["id"]), 10),
  81. Createtime: common.InterfaceToStr(lastMsg["send_time"]),
  82. Title: common.InterfaceToStr(lastMsg["title"]),
  83. MsgType: int64(util.IntAll(lastMsg["group_id"])),
  84. }
  85. if this.IsClassSearch && util.IntAll(val["group_id"]) == 12 {
  86. sData[common.InterfaceToStr(val["group_id"])] = []*message.Messages{&msg}
  87. continue
  88. }
  89. if sData[common.InterfaceToStr(lastMsg["group_id"])] == nil {
  90. sData[common.InterfaceToStr(lastMsg["group_id"])] = []*message.Messages{&msg}
  91. } else {
  92. sData[common.InterfaceToStr(lastMsg["group_id"])] = append(sData[common.InterfaceToStr(lastMsg["group_id"])], &msg)
  93. }
  94. }
  95. }
  96. }
  97. //pc消息列表消息分类
  98. var columnData []*message.AllSortData
  99. if this.IsColumn && this.MsgType > 0 && this.IsClassSearch {
  100. // 获取小分类下的未读数
  101. sortUnread, total := BitmapCountClassUnread(this.UserId, this.MsgType, classUnreadCountMap)
  102. columnArr := []entity.MsgClass{}
  103. if !this.IsColumnNewMsg { // 用于区分分类列表页和分类详情页 根据不同情况
  104. columnArr = append(columnArr, entity.ClassMap[this.MsgType])
  105. } else {
  106. columnArr = entity.ClassSearchMap[this.MsgType]
  107. }
  108. unread += total
  109. for i := 0; i < len(columnArr); i++ {
  110. tmp := columnArr[i]
  111. var column message.AllSortData
  112. column.Name = tmp.Name
  113. column.Img = fmt.Sprintf("/common-module/msgCenter/%s.png", tmp.Img)
  114. column.MsgType = tmp.MsgType
  115. // 消息未读数
  116. msgType := common.InterfaceToStr(tmp.MsgType)
  117. column.UnreadMessages = sortUnread[msgType]
  118. column.Data = sData[msgType]
  119. column.IsClassSearch = true
  120. columnData = append(columnData, &column)
  121. }
  122. } else if this.IsColumn || this.IsfilterActive {
  123. //获取所有分类未读数 不初始化
  124. sortUnread, total := BitmapCountUnread(this.UserId, classUnreadCountMap, false)
  125. //fmt.Println("*************", total)
  126. unread += total
  127. for _, v := range entity.MessageColumn {
  128. var column message.AllSortData
  129. column.Name = common.InterfaceToStr(v["name"])
  130. column.Img = fmt.Sprintf("/common-module/msgCenter/%s.png", common.InterfaceToStr(v["img"]))
  131. column.MsgType = common.Int64All(v["group_id"])
  132. if column.Name == "私信" {
  133. column.UnreadMessages = unreadMsg(this)
  134. } else if common.IntAll(v["group_id"]) > 0 {
  135. //消息未读数
  136. msgType := common.InterfaceToStr(v["group_id"])
  137. column.UnreadMessages = sortUnread[msgType]
  138. column.Data = sData[msgType]
  139. }
  140. if _, ok := entity.ClassSearchMap[column.MsgType]; ok {
  141. column.IsClassSearch = true
  142. }
  143. columnData = append(columnData, &column)
  144. }
  145. }
  146. data.SortData = columnData
  147. if len(userAllMsgArr) == 0 {
  148. fmt.Printf("此用户暂无消息 : %s", err.Error())
  149. return data, nil
  150. }
  151. count = int64(len(userAllMsgArr))
  152. //pc列表数据、移动端分类列表
  153. if this.IsMsgList {
  154. if count > 0 {
  155. var userMsgArr []int
  156. if this.OffSet <= 0 {
  157. this.OffSet = 1
  158. }
  159. if this.MsgType > 0 {
  160. var totalMsg []int
  161. //分类消息bitmap
  162. totalMsg = userClassMsgMap[int(this.MsgType)]
  163. if this.MsgType == 11 {
  164. if !this.IsClassSearch {
  165. totalMsg = append(totalMsg, userClassMsgMap[12]...)
  166. }
  167. }
  168. userMsgArr = totalMsg
  169. } else {
  170. userMsgArr = Uint32ArrToIntArr(userAllMsgArr)
  171. }
  172. if this.Read == 0 { //未读
  173. userMsgArr = Uint32ArrToIntArr(userUnreadArr)
  174. }
  175. if this.MsgType > 0 {
  176. count = int64(len(userMsgArr))
  177. }
  178. if len(userMsgArr) > 0 {
  179. userMsgArr = IntArrSort(userMsgArr)
  180. fmt.Println("列表消息:", userMsgArr)
  181. //总页数
  182. var pageCount = 0
  183. if this.PageSize != 0 {
  184. if len(userMsgArr)%int(this.PageSize) > 0 {
  185. pageCount = len(userMsgArr)/int(this.PageSize) + 1
  186. } else {
  187. pageCount = len(userMsgArr) / int(this.PageSize)
  188. }
  189. }
  190. start := len(userMsgArr) - int(this.PageSize*(this.OffSet-1))
  191. end := start - int(this.PageSize)
  192. if int(this.OffSet) == pageCount || this.PageSize == 0 {
  193. end = 0
  194. }
  195. //fmt.Println(end, start)
  196. if end < 0 || start < 0 {
  197. return data, nil
  198. }
  199. resArr := userMsgArr[end:start]
  200. resData := []*message.Messages{}
  201. if len(resArr) > 0 {
  202. //count = int64(len(resArr))
  203. unReadMap := map[int]bool{}
  204. for _, vv := range userUnreadArr {
  205. unReadMap[int(vv)] = true
  206. }
  207. for _, v := range resArr {
  208. msg := GlobMsgMap[v]
  209. if msg == nil || len(msg) <= 0 {
  210. ms := entity.Mysql.FindOne("message_send_log", map[string]interface{}{"id": v}, "id,msg_type,title,content,send_time,link,menu_name,group_id,sign", "")
  211. if ms != nil && len(*ms) > 0 {
  212. msg = *ms
  213. } else {
  214. continue
  215. }
  216. }
  217. if common.IntAll(msg["sign"]) == 4 {
  218. //查询用户注册时间
  219. rData, _ := entity.MQFW.FindOneByField("user", map[string]interface{}{"_id": m.StringTOBsonId(this.UserId)}, `{"l_registedate":1}`)
  220. if rData != nil && len(*rData) > 0 {
  221. msg["send_time"] = time.Unix(common.Int64All((*rData)["l_registedate"]), 0).Local().Format("2006-01-02 15:04:05")
  222. }
  223. }
  224. resData = append(resData, &message.Messages{
  225. Id: common.InterfaceToStr(v),
  226. Createtime: common.InterfaceToStr(msg["send_time"]),
  227. Title: common.InterfaceToStr(msg["title"]),
  228. MsgType: int64(util.IntAll(msg["group_id"])),
  229. Link: common.InterfaceToStr(msg["link"]),
  230. Content: common.InterfaceToStr(msg["content"]),
  231. IsRead: common.Int64All(common.If(unReadMap[v], 0, 1)),
  232. MsgLogId: int64(v),
  233. })
  234. }
  235. for i := len(resData) - 1; i >= 0; i-- {
  236. data.Data = append(data.Data, resData[i])
  237. }
  238. }
  239. }
  240. }
  241. }
  242. data.Count = count
  243. if this.Read == 0 {
  244. //unread = count
  245. if this.IsContainLetter { //是否需要统计私信未读数
  246. unread += unreadMsg(this)
  247. }
  248. }
  249. data.Unread = unread
  250. //置顶消息
  251. if len(classUnreadMsgMap[1]) > 0 {
  252. activeMsgArr := classUnreadMsgMap[1]
  253. res2 := BitmapMessageGetLast(this, activeMsgArr[len(activeMsgArr)-1])
  254. return data, res2
  255. }
  256. return data, nil
  257. }
  258. // BitmapMessageGetLast 获取优惠活动的最新一条消息
  259. func BitmapMessageGetLast(this *message.UserMsgListReq, msgId int) *message.Messages {
  260. if !this.IsMsgList && !this.IsColumnNewMsg && !this.IsColumn {
  261. return nil
  262. }
  263. lastMsg := GlobMsgMap[msgId]
  264. if lastMsg == nil && len(lastMsg) <= 0 {
  265. mess := entity.Mysql.FindOne("message_send_log", map[string]interface{}{"id": msgId}, "id,msg_type,title,content,send_time,link,menu_name,group_id,sign", "")
  266. if mess != nil && len(*mess) > 0 {
  267. lastMsg = *mess
  268. }
  269. }
  270. if lastMsg != nil && len(lastMsg) > 0 {
  271. if common.IntAll(lastMsg["sign"]) == 4 {
  272. //查询用户注册时间
  273. rData, _ := entity.MQFW.FindOneByField("user", map[string]interface{}{"_id": m.StringTOBsonId(this.UserId)}, `{"l_registedate":1}`)
  274. if rData != nil && len(*rData) > 0 {
  275. lastMsg["send_time"] = time.Unix(common.Int64All((*rData)["l_registedate"]), 0).Local().Format("2006-01-02 15:04:05")
  276. }
  277. }
  278. msg := message.Messages{
  279. Id: common.InterfaceToStr(lastMsg["id"]),
  280. Createtime: common.InterfaceToStr(lastMsg["send_time"]),
  281. Title: common.InterfaceToStr(lastMsg["title"]),
  282. MsgType: common.Int64All(lastMsg["group_id"]),
  283. Link: common.InterfaceToStr(lastMsg["link"]),
  284. Content: common.InterfaceToStr(lastMsg["content"]),
  285. IsRead: 0,
  286. MsgLogId: common.Int64All(lastMsg["id"]),
  287. }
  288. return &msg
  289. }
  290. return nil
  291. }
  292. type UserClassUnread struct {
  293. GroupId int32 `ch:"group_id"`
  294. UnreadArr []uint32 `ch:"unreadArr"`
  295. }
  296. // BitmapCountUnread 未读消息合计 isRedis 是否需要初始化redis
  297. func BitmapCountUnread(userId string, classUnreadCountMap map[int]int, isRedis bool) (map[string]int64, int64) {
  298. var (
  299. count int64
  300. msgTypes, groupIds []string
  301. )
  302. data := make(map[string]int64)
  303. for _, v := range entity.MessageColumn {
  304. if util.IntAll(v["group_id"]) > 0 && util.IntAll(v["group_id"]) < 999 {
  305. //去除全部与私信
  306. msgTypes = append(msgTypes, fmt.Sprintf(`"%s"`, common.InterfaceToStr(v["group_id"])))
  307. key := fmt.Sprintf(MsgCountKey, userId, util.IntAll(v["group_id"]))
  308. groupIds = append(groupIds, common.InterfaceToStr(v["group_id"]))
  309. if exists, _ := redis.Exists(redisModule, key); exists {
  310. ct := util.Int64All(redis.GetInt(redisModule, key))
  311. data[common.InterfaceToStr(v["group_id"])] = ct
  312. count += ct
  313. }
  314. }
  315. }
  316. //classUnreadCountMap, _ := FindUserClassUnread(userId)
  317. if len(msgTypes) > 0 && len(msgTypes) != len(data) {
  318. for _, val := range entity.MessageColumn {
  319. if util.IntAll(val["group_id"]) > 0 && util.IntAll(val["group_id"]) < 999 {
  320. oneClassUnread := classUnreadCountMap[util.IntAll(val["group_id"])]
  321. if util.IntAll(val["group_id"]) == 11 {
  322. oneClassUnread += classUnreadCountMap[12]
  323. }
  324. if isRedis {
  325. //用户分类未读
  326. if oneClassUnread >= 0 {
  327. key := fmt.Sprintf(MsgCountKey, userId, util.IntAll(val["group_id"]))
  328. redis.Put(redisModule, key, oneClassUnread, 60*60)
  329. }
  330. }
  331. data[common.InterfaceToStr(val["group_id"])] = int64(oneClassUnread)
  332. count += int64(oneClassUnread)
  333. }
  334. }
  335. }
  336. return data, count //分类未读、未读总量
  337. }
  338. //BitmapCountClassUnread userUnreadBitmap 用户未读
  339. func BitmapCountClassUnread(userId string, groupId int64, classUnreadCountMap map[int]int) (classCount map[string]int64, total int64) {
  340. var (
  341. count int64
  342. )
  343. data := make(map[string]int64)
  344. if _, ok := entity.ClassSearchMap[groupId]; !ok {
  345. return
  346. }
  347. for i := 0; i < len(entity.ClassSearchMap[groupId]); i++ {
  348. msgClass := entity.ClassSearchMap[groupId][i]
  349. //fmt.Println("msgClass", msgClass)
  350. key := fmt.Sprintf(MsgClassCountKey, userId, msgClass.MsgType)
  351. if exists, _ := redis.Exists(redisModule, key); exists {
  352. ct := util.Int64All(redis.GetInt(redisModule, key))
  353. data[fmt.Sprintf("%d", msgClass.MsgType)] = ct
  354. count += ct
  355. } else {
  356. //分类所有消息
  357. classUnreadCount := classUnreadCountMap[util.IntAll(msgClass.MsgType)]
  358. if classUnreadCount >= 0 {
  359. redis.Put(redisModule, key, classUnreadCount, -1)
  360. data[fmt.Sprintf("%d", msgClass.MsgType)] = int64(classUnreadCount)
  361. count += int64(classUnreadCount)
  362. } else {
  363. log.Println("查询classUnreadCount失败,msgType:", msgClass.MsgType)
  364. }
  365. }
  366. }
  367. return data, count
  368. }
  369. //FindUserClassUnread 查询用户分类未读消息id
  370. func FindUserClassUnread(userId string) (map[int]int, map[int][]int) {
  371. sql := fmt.Sprintf(`select a2.group_id,bitmapToArray(bitmapAnd(a1.unreadArr,a2.msg_bitmap)) as unreadArr from
  372. (SELECT bitmapAndnot(mus.allMsg,mus.readMsg) unreadArr FROM message_user_summary mus WHERE mus.userId = '%s') a1,
  373. (select msg_bitmap,group_id from message_summary ms where 1=1 limit 10) a2`, userId)
  374. fmt.Println("FindUserClassUnread", sql)
  375. rows, err := entity.ClickhouseConn.Query(context.Background(), sql)
  376. if err != nil {
  377. log.Println("获取各分类未读消息数组出错:", err)
  378. return nil, nil
  379. }
  380. classUnreadCountMap := map[int]int{}
  381. classUnreadMsgMap := map[int][]int{}
  382. for rows.Next() {
  383. group := UserClassUnread{}
  384. err = rows.ScanStruct(&group)
  385. if err != nil {
  386. log.Println("获取各分类读取分类数据出错:", err)
  387. return nil, nil
  388. }
  389. classUnreadCountMap[int(group.GroupId)] = len(group.UnreadArr)
  390. classUnreadMsgMap[int(group.GroupId)] = Uint32ArrToIntArr(group.UnreadArr)
  391. }
  392. return classUnreadCountMap, classUnreadMsgMap
  393. }
  394. //FindUserClassMsg 查询用户分类所有消息id
  395. func FindUserClassMsg(userId string) (map[int]int, map[int][]int) {
  396. sql := fmt.Sprintf(`select a2.group_id,bitmapToArray(bitmapAnd(a1.allMsg,a2.msg_bitmap)) as unreadArr from
  397. (SELECT mus.allMsg as allMsg FROM message_user_summary mus WHERE mus.userId = '%s') a1,
  398. (select msg_bitmap,group_id from message_summary ms where 1=1 limit 10) a2`, userId)
  399. fmt.Println("FindUserClassMsg", sql)
  400. rows, err := entity.ClickhouseConn.Query(context.Background(), sql)
  401. if err != nil {
  402. log.Println("获取各分类未读消息数组出错:", err)
  403. return nil, nil
  404. }
  405. classCountMap := map[int]int{}
  406. classMsgMap := map[int][]int{}
  407. for rows.Next() {
  408. group := UserClassUnread{}
  409. err = rows.ScanStruct(&group)
  410. if err != nil {
  411. log.Println("获取各分类读取分类数据出错:", err)
  412. return nil, nil
  413. }
  414. //fmt.Println(int(group.GroupId), len(group.UnreadArr))
  415. classCountMap[int(group.GroupId)] = len(group.UnreadArr)
  416. classMsgMap[int(group.GroupId)] = Uint32ArrToIntArr(group.UnreadArr)
  417. }
  418. return classCountMap, classMsgMap
  419. }
  420. // GetUserMsgSummary 从用户消息汇总表取数据
  421. func GetUserMsgSummary(userId string, isfilterActive bool) (userAllMsg, userReadMsg, userUnreadMsg []uint32, err error) {
  422. var count uint64
  423. sqlc := ""
  424. if isfilterActive {
  425. sqlc = fmt.Sprintf("select a2.group_id,bitmapToArray(bitmapAndnot(a1.unreadMsg,a2.msg_bitmap)) as msgArr from (SELECT bitmapAndnot(mus.allMsg,mus.readMsg) as unreadMsg FROM messageCenter.message_user_summary mus WHERE mus.userId = '%s') a1,(select msg_bitmap,group_id from messageCenter.message_summary ms where group_id = 1 ) a2", userId)
  426. } else {
  427. sqlc = fmt.Sprintf("SELECT COUNT(*) as count from message_user_summary WHERE userId = '%s'", userId)
  428. }
  429. log.Println("GetUserMsgSummary selcet ", sqlc)
  430. row1 := entity.ClickhouseConn.QueryRow(context.Background(), sqlc)
  431. err = row1.Scan(&count)
  432. if count == 0 {
  433. err = errors.New("用户暂无数据")
  434. return
  435. }
  436. sql := fmt.Sprintf("SELECT bitmapToArray(allMsg) as userAllMsg,bitmapToArray(readMsg) as userReadMsg,bitmapToArray(bitmapAndnot(allMsg,readMsg)) as userunRead from message_user_summary where userId ='%s'", userId)
  437. log.Println("GetUserMsgSummary selcet2 ", sql)
  438. row := entity.ClickhouseConn.QueryRow(context.Background(), sql)
  439. err = row.Scan(&userAllMsg, &userReadMsg, &userUnreadMsg)
  440. if err != nil {
  441. log.Println("此用户暂无数据:", err)
  442. }
  443. return
  444. }
  445. // Uint32ArrToIntArr []uint32转[]int
  446. func Uint32ArrToIntArr(arr []uint32) []int {
  447. tmp := make([]int, 0)
  448. for _, v := range arr {
  449. tmp = append(tmp, int(v))
  450. }
  451. return tmp
  452. }
  453. func IntArrSort(arr []int) []int {
  454. if len(arr) <= 1 {
  455. return arr
  456. }
  457. splitData := arr[0] //第一个数据为基准
  458. low := make([]int, 0, 0) //比我小的数据
  459. height := make([]int, 0, 0) //比我大的数据
  460. mid := make([]int, 0, 0) //与我一样大的数据
  461. mid = append(mid, splitData) //加入一个
  462. for i := 1; i < len(arr); i++ {
  463. if arr[i] < splitData {
  464. low = append(low, arr[i])
  465. } else if arr[i] > splitData {
  466. height = append(height, arr[i])
  467. } else {
  468. mid = append(mid, arr[i])
  469. }
  470. }
  471. low, height = IntArrSort(low), IntArrSort(height) //切割递归处理
  472. resArr := append(append(low, mid...), height...)
  473. return resArr
  474. }
  475. func WorkDeskList(in *message.WorkingDesktopReq) (res1, res2 []*message.Messages, err error) {
  476. //待办查询
  477. sqlStr := ""
  478. needDo := map[int][]uint32{} //msgType对应的消息id
  479. if in.NeedDealtWithCount {
  480. sqlStr = fmt.Sprintf(" or p_group_id = 11")
  481. }
  482. sql := fmt.Sprintf(`select a2.group_id,bitmapToArray(bitmapAnd(a1.allMsg,a2.msg_bitmap)) as unreadArr from
  483. (SELECT mus.allMsg as allMsg FROM messageCenter.message_user_summary mus WHERE mus.userId = '%s') a1,
  484. (select msg_bitmap,group_id from messageCenter.message_summary ms where group_id = %d %s) a2`, in.UserId, in.MsgType, sqlStr)
  485. log.Println("WorkDeskList", sql)
  486. rows, err := entity.ClickhouseConn.Query(context.Background(), sql)
  487. if err != nil {
  488. return nil, nil, err
  489. }
  490. for rows.Next() {
  491. group := UserClassUnread{}
  492. err = rows.ScanStruct(&group)
  493. if err != nil {
  494. log.Println("获取各分类读取分类数据出错:", err)
  495. return nil, nil, err
  496. }
  497. if group.GroupId == int32(in.MsgType) {
  498. res1 = GetMsgList(Uint32ArrToIntArr(group.UnreadArr), int(in.PageSize), in.UserId)
  499. } else {
  500. if len(group.UnreadArr) > 0 {
  501. needDo[int(group.GroupId)] = group.UnreadArr
  502. }
  503. }
  504. }
  505. if len(needDo) > 0 {
  506. totalMsg := append(needDo[11], needDo[12]...)
  507. needMsgArr := IntArrSort(Uint32ArrToIntArr(totalMsg))
  508. res2 = GetMsgList(needMsgArr, int(in.PageSize), in.UserId)
  509. }
  510. return res1, res2, nil
  511. }
  512. func GetMsgList(arr []int, size int, userId string) []*message.Messages {
  513. newMsg := []int{}
  514. if len(arr) > 0 {
  515. if len(arr) >= size {
  516. newMsg = arr[len(arr)-size:]
  517. } else {
  518. newMsg = arr
  519. }
  520. }
  521. resData := []*message.Messages{}
  522. for _, v := range newMsg {
  523. msg := GetMsgInfo(int(v), userId)
  524. links4 := common.InterfaceToStr(msg["link"])
  525. link4, androidUrl4, iosUrl4, weChatUrl4 := util.LinkSplit(links4)
  526. resData = append(resData, &message.Messages{
  527. Id: common.InterfaceToStr(v),
  528. Createtime: common.InterfaceToStr(msg["send_time"]),
  529. Title: common.InterfaceToStr(msg["title"]),
  530. MsgType: int64(util.IntAll(msg["group_id"])),
  531. Link: link4,
  532. Content: common.InterfaceToStr(msg["content"]),
  533. MsgLogId: int64(v),
  534. Url: map[string]string{
  535. "androidUrl": androidUrl4,
  536. "iosUrl": iosUrl4,
  537. "weChatUrl": weChatUrl4,
  538. },
  539. })
  540. }
  541. return resData
  542. }
  543. func GetMsgInfo(msgId int, userId string) map[string]interface{} {
  544. msg := GlobMsgMap[msgId]
  545. if msg == nil || len(msg) <= 0 {
  546. ms := entity.Mysql.FindOne("message_send_log", map[string]interface{}{"id": msgId}, "id,msg_type,title,content,send_time,link,menu_name,group_id,sign", "")
  547. if ms != nil && len(*ms) > 0 {
  548. msg = *ms
  549. }
  550. }
  551. if common.IntAll(msg["sign"]) == 4 {
  552. //查询用户注册时间
  553. rData, _ := entity.MQFW.FindOneByField("user", map[string]interface{}{"_id": m.StringTOBsonId(userId)}, `{"l_registedate":1}`)
  554. if rData != nil && len(*rData) > 0 {
  555. msg["send_time"] = time.Unix(common.Int64All((*rData)["l_registedate"]), 0).Local().Format("2006-01-02 15:04:05")
  556. }
  557. }
  558. return msg
  559. }
  560. // PutRedisUserAllMsg 加载用户的所有消息
  561. /*func PutRedisUserAllMsg(userId string, allMsgBitmap *roaring.Bitmap) (rData []map[string]interface{}) {
  562. str := ""
  563. for i, v := range allMsgBitmap.ToArray() {
  564. if i < len(allMsgBitmap.ToArray())-1 {
  565. str += common.InterfaceToStr(v) + ","
  566. } else {
  567. str += common.InterfaceToStr(v)
  568. }
  569. }
  570. //查询用户所有消息
  571. key := fmt.Sprintf(BitmapUserMsg, userId)
  572. m := entity.Mysql.SelectBySql("SELECT id,msg_type,title,content,send_time,link,menu_name,group_id FROM message_send_log WHERE send_status = 4 AND isdel = 1 AND id in (?) ORDER BY send_time DESC", str)
  573. if m != nil && len(*m) > 0 {
  574. rData = *m
  575. redis.Put(redisModule, key, rData, -1)
  576. }
  577. return rData
  578. }*/