msglistService.go 23 KB

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