msglistService.go 24 KB

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