dao.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933
  1. package roomAdmin
  2. import (
  3. "bytes"
  4. "context"
  5. "encoding/json"
  6. "fmt"
  7. "github.com/jinzhu/gorm"
  8. banned "go-common/app/service/live/banned_service/api/liverpc/v1"
  9. v12 "go-common/app/service/live/fans_medal/api/liverpc/v2"
  10. "go-common/app/service/live/room/api/liverpc/v1"
  11. "go-common/app/service/live/room/api/liverpc/v2"
  12. v1pb "go-common/app/service/live/xuser/api/grpc/v1"
  13. "go-common/app/service/live/xuser/conf"
  14. "go-common/app/service/live/xuser/dao"
  15. "go-common/app/service/live/xuser/model"
  16. account "go-common/app/service/main/account/model"
  17. accrpc "go-common/app/service/main/account/rpc/client"
  18. "go-common/library/cache/memcache"
  19. "go-common/library/cache/redis"
  20. "go-common/library/database/orm"
  21. xsql "go-common/library/database/sql"
  22. "go-common/library/ecode"
  23. "go-common/library/log"
  24. bm "go-common/library/net/http/blademaster"
  25. "io/ioutil"
  26. "math"
  27. "net/http"
  28. "sort"
  29. "strconv"
  30. "time"
  31. )
  32. // Dao dao
  33. type Dao struct {
  34. c *conf.Config
  35. mc *memcache.Pool
  36. redis *redis.Pool
  37. db *xsql.DB
  38. orm *gorm.DB
  39. RoomAdminExpire int32
  40. // acc rpc
  41. acc *accrpc.Service3
  42. client *bm.Client
  43. }
  44. const (
  45. userPrefix = "up_v1_%d"
  46. roomPrefix = "rp_v1_%d"
  47. mcExpire = 3600
  48. maxAdminsNum = 100
  49. )
  50. // KeyUser return the mc key by user mid.
  51. func KeyUser(uid int64) string {
  52. return fmt.Sprintf(userPrefix, uid)
  53. }
  54. // KeyRoom return the mc key by anchor mid.
  55. func KeyRoom(uid int64) string {
  56. return fmt.Sprintf(roomPrefix, uid)
  57. }
  58. //go:generate $GOPATH/src/go-common/app/tool/cache/mc
  59. type _mc interface {
  60. // 获取主播的房管列表
  61. // mc: -key=KeyRoom
  62. CacheRoomAdminRoom(c context.Context, anchor int64) ([]*model.RoomAdmin, error)
  63. // 获取用户的房管列表
  64. // mc: -key=KeyUser
  65. CacheRoomAdminUser(c context.Context, user int64) ([]*model.RoomAdmin, error)
  66. // mc: -key=KeyRoom -expire=d.RoomAdminExpire -encode=json|gzip
  67. AddCacheKeyAnchorRoom(c context.Context, anchor int64, value []*model.RoomAdmin) error
  68. // mc: -key=KeyUser -expire=d.RoomAdminExpire -encode=gob
  69. AddCacheRoomAdminUser(c context.Context, user int64, value []*model.RoomAdmin) error
  70. // mc: -key=KeyRoom
  71. DelCacheKeyAnchorRoom(c context.Context, anchor int64) error
  72. // mc: -key=KeyUser
  73. DelCacheRoomAdminUser(c context.Context, user int64) error
  74. }
  75. // New init mysql db
  76. func New(c *conf.Config) (dao *Dao) {
  77. dao = &Dao{
  78. c: c,
  79. mc: memcache.NewPool(c.Memcache),
  80. redis: redis.NewPool(c.Redis),
  81. db: xsql.NewMySQL(c.LiveAppMySQL),
  82. orm: orm.NewMySQL(c.LiveAppORM),
  83. RoomAdminExpire: mcExpire,
  84. acc: accrpc.New3(c.AccountRPC),
  85. client: bm.NewClient(c.BMClient),
  86. }
  87. return
  88. }
  89. // Close close the resource.
  90. func (d *Dao) Close() {
  91. d.mc.Close()
  92. d.redis.Close()
  93. d.orm.Close()
  94. d.db.Close()
  95. }
  96. // Ping dao ping
  97. func (d *Dao) Ping(c context.Context) error {
  98. // TODO: if you need use mc,redis, please add
  99. return d.db.Ping(c)
  100. }
  101. // HasAnyAdmin whether he has any admin in any room.
  102. func (d *Dao) HasAnyAdmin(c context.Context, uid int64) (int64, error) {
  103. noAdmin := int64(0)
  104. hasAdmin := int64(1)
  105. rst, err := d.GetAllByUid(c, uid)
  106. if nil == rst {
  107. return noAdmin, err
  108. }
  109. return hasAdmin, err
  110. }
  111. // GetByUidPage get admins by uid and page.
  112. func (d *Dao) GetByUidPage(c context.Context, uid int64, page int64, pageSize int64) (resp *v1pb.RoomAdminGetByUidResp, err error) {
  113. resp = &v1pb.RoomAdminGetByUidResp{}
  114. resp.Page = &v1pb.RoomAdminGetByUidResp_Page{
  115. Page: page,
  116. PageSize: pageSize,
  117. TotalPage: 1,
  118. TotalCount: 0,
  119. }
  120. rst, err := d.GetAllByUid(c, uid)
  121. //spew.Dump("GetAllByUid", rst, err)
  122. if err != nil {
  123. return
  124. }
  125. if rst == nil {
  126. return
  127. }
  128. sort.Sort(sort.Reverse(model.RoomAdmins(rst)))
  129. //spew.Dump(rst)
  130. resp.Page.TotalCount = int64(len(rst))
  131. resp.Page.PageSize = pageSize
  132. resp.Page.TotalPage = int64(math.Ceil(float64(len(rst)) / float64(resp.Page.PageSize)))
  133. begin := (page - 1) * pageSize
  134. end := page * pageSize
  135. if page*pageSize > int64(len(rst)) {
  136. end = int64(len(rst))
  137. }
  138. roomUid, mids, err := d.getAnchorUidsFromAdmins(c, rst)
  139. // 没获取到房间信息
  140. if err != nil {
  141. return resp, err
  142. }
  143. // 可能从room获取主播信息不全
  144. if int64(len(mids)) < end {
  145. end = int64(len(mids))
  146. }
  147. if begin > end {
  148. begin = end
  149. }
  150. if err != nil {
  151. return resp, err
  152. }
  153. args := &account.ArgMids{Mids: mids[begin:end]}
  154. accData, err := d.acc.Infos3(c, args)
  155. //spew.Dump("d.acc.Infos3", accData, err)
  156. if err != nil {
  157. log.Error("call account.Infos3(%v) error(%v)", args, err)
  158. return resp, err
  159. }
  160. for _, v := range rst[begin:end] {
  161. item := &v1pb.RoomAdminGetByUidResp_Data{
  162. Uid: v.Uid,
  163. Roomid: v.Roomid,
  164. Ctime: v.Ctime.Time().Format("2006-01-02 15:04:05"),
  165. }
  166. if _, ok := roomUid[v.Roomid]; ok {
  167. item.AnchorId = roomUid[v.Roomid]
  168. if _, ok := accData[item.AnchorId]; ok {
  169. item.AnchorCover = accData[item.AnchorId].Face
  170. item.Uname = accData[item.AnchorId].Name
  171. } else {
  172. log.Error("没有这个人的用户信息 uid(%v) data(%v)", item.AnchorId, accData)
  173. }
  174. } else {
  175. log.Error("没有这个人的房间信息 room (%v) data(%v)", v.Roomid, roomUid)
  176. }
  177. resp.Data = append(resp.Data, item)
  178. }
  179. //spew.Dump("resp.Data", resp.Data)
  180. return
  181. }
  182. // GetAllByUid get by uid.
  183. func (d *Dao) GetAllByUid(c context.Context, uid int64) ([]*model.RoomAdmin, error) {
  184. rstMc, err := d.CacheRoomAdminUser(c, uid)
  185. //spew.Dump("HasAnyAdmin1", rstMc, err)
  186. //spew.Dump("lenMc", len(rstMc))
  187. if err != nil {
  188. return nil, err
  189. }
  190. // 空缓存标识
  191. if rstMc != nil {
  192. if rstMc[0].Id == -1 {
  193. return nil, err
  194. }
  195. return rstMc, err
  196. }
  197. rstDb, err := d.GetByUserMysql(c, uid)
  198. if err != nil {
  199. return nil, err
  200. }
  201. if len(rstDb) == 0 {
  202. d.AddCacheNoneUser(c, uid)
  203. return nil, err
  204. }
  205. d.AddCacheRoomAdminUser(c, uid, rstDb)
  206. return rstDb, err
  207. }
  208. // getAnchorUidsFromAdmins .
  209. // 根据批量房管获取对应主播的房间号和UID
  210. func (d *Dao) getAnchorUidsFromAdmins(c context.Context, admins []*model.RoomAdmin) (roomUid map[int64]int64, uids []int64, err error) {
  211. var roomIds []int64
  212. roomUid = make(map[int64]int64)
  213. for _, a := range admins {
  214. roomIds = append(roomIds, a.Roomid)
  215. }
  216. if len(roomIds) == 0 {
  217. return
  218. }
  219. reply, err := dao.RoomAPI.V2Room.GetByIds(c, &v2.RoomGetByIdsReq{Ids: roomIds})
  220. if err != nil {
  221. log.Error("dao.RoomAPI.V2Room.GetByIds (%v) error(%v)", roomIds, err)
  222. return roomUid, uids, err
  223. }
  224. if reply.GetCode() != 0 {
  225. err = ecode.Int(int(reply.GetCode()))
  226. log.Error("dao.RoomAPI.V2Room.GetByIds (%v) error code(%v)", roomIds, err)
  227. return roomUid, uids, err
  228. }
  229. for aRoomId, r := range reply.Data {
  230. roomUid[aRoomId] = r.Uid
  231. uids = append(uids, r.Uid)
  232. }
  233. return
  234. }
  235. // Del delete a roomadmin
  236. func (d *Dao) Del(c context.Context, uid int64, roomId int64) (err error) {
  237. if err = d.DelAllCache(c, uid, roomId); err != nil {
  238. log.Error("DelAllCache(%v) uid (%v) roomid (%v) error(%v)", uid, roomId, err)
  239. return
  240. }
  241. admin, err := d.GetByRoomIdUidMysql(c, uid, roomId)
  242. if len(admin) == 0 {
  243. log.Error("GetByRoomIdUidMysql empty uid(%v) roomId (%v)error(%v)", uid, roomId, err)
  244. return
  245. }
  246. //spew.Dump("GetByRoomIdUidMysql", admin)
  247. if err = d.DelDbAdminMysql(c, admin[0].Id); err != nil {
  248. log.Error("DelCacheRoomAdminUser uid(%v) roomId (%v) error(%v)", uid, roomId, err)
  249. return err
  250. }
  251. return
  252. }
  253. func (d *Dao) getInfoByName(c context.Context, name string) (userInfo *account.Info, err error) {
  254. userInfo = &account.Info{}
  255. infosByName, err := d.acc.InfosByName3(c, &account.ArgNames{
  256. Names: []string{name},
  257. })
  258. if err != nil {
  259. log.Error("d.acc.InfosByName3(%v) error(%v)", name, err)
  260. return userInfo, err
  261. }
  262. log.Info("d.acc.InfosByName3(%v) return (%v)", name, infosByName)
  263. if len(infosByName) != 0 {
  264. for _, info := range infosByName {
  265. return info, err
  266. }
  267. }
  268. return
  269. }
  270. // SearchForAdmin search user list by keyword.
  271. func (d *Dao) SearchForAdmin(c context.Context, keyword string, anchorId int64) (resp []*v1pb.RoomAdminSearchForAdminResp_Data, err error) {
  272. isUid := 0
  273. matchUid, _ := strconv.ParseInt(keyword, 10, 64)
  274. if matchUid != 0 && keyword != "0" {
  275. isUid = 1
  276. }
  277. // get by name
  278. infoMatchName, err := d.getInfoByName(c, keyword)
  279. if err != nil {
  280. return resp, err
  281. }
  282. if nil != infoMatchName && infoMatchName.Mid != 0 {
  283. log.Info("SearchForAdmin infoMatchName keyword (%v) ret (%v)", keyword, infoMatchName)
  284. isAdminName, _ := d.isAdminByUid(c, infoMatchName.Mid, anchorId)
  285. medalData, errMedal := d.getMedalInfoByUids(c, []int64{infoMatchName.Mid})
  286. if errMedal != nil {
  287. return resp, errMedal
  288. }
  289. itemName := &v1pb.RoomAdminSearchForAdminResp_Data{
  290. Uid: infoMatchName.Mid,
  291. IsAdmin: isAdminName,
  292. Uname: infoMatchName.Name,
  293. Face: infoMatchName.Face,
  294. }
  295. if _, ok := medalData[infoMatchName.Mid]; ok {
  296. itemName.MedalName = medalData[infoMatchName.Mid].MedalName
  297. itemName.Level = medalData[infoMatchName.Mid].Level
  298. } else {
  299. log.Info("没有这个人的勋章信息 uid(%v) data(%v)", infoMatchName.Mid, medalData)
  300. }
  301. resp = append(resp, itemName)
  302. }
  303. //spew.Dump("searchForadmin2", resp)
  304. // just name
  305. if 0 == isUid {
  306. return resp, nil
  307. }
  308. // get by uid
  309. infoMatchUid, err := d.getInfoByUid(c, matchUid)
  310. if err != nil {
  311. return resp, err
  312. }
  313. if infoMatchUid == nil {
  314. return resp, nil
  315. }
  316. isAdminUid, _ := d.isAdminByUid(c, matchUid, anchorId)
  317. medalDataUid, err := d.getMedalInfoByUids(c, []int64{infoMatchUid.Mid})
  318. if err != nil {
  319. return resp, err
  320. }
  321. itemUid := &v1pb.RoomAdminSearchForAdminResp_Data{
  322. Uid: infoMatchUid.Mid,
  323. IsAdmin: isAdminUid,
  324. Uname: infoMatchUid.Name,
  325. Face: infoMatchUid.Face,
  326. }
  327. if _, ok := medalDataUid[infoMatchUid.Mid]; ok {
  328. itemUid.MedalName = medalDataUid[infoMatchUid.Mid].MedalName
  329. itemUid.Level = medalDataUid[infoMatchUid.Mid].Level
  330. }
  331. resp = append(resp, itemUid)
  332. //spew.Dump("searchForadmin2", resp)
  333. return
  334. }
  335. func (d *Dao) isAdminByUid(c context.Context, uid int64, anchorId int64) (rst int64, err error) {
  336. roomId, err := d.getRoomIdByUid(c, anchorId)
  337. if err != nil {
  338. return rst, err
  339. }
  340. if 0 == roomId {
  341. return rst, nil
  342. }
  343. return d.IsAdminByRoomId(c, uid, roomId)
  344. }
  345. func (d *Dao) getInfoByUid(c context.Context, uid int64) (info *account.Info, err error) {
  346. info, err = d.acc.Info3(c, &account.ArgMid{
  347. Mid: uid,
  348. })
  349. if err != nil {
  350. log.Error("d.acc.Info3(%v) error(%v)", uid, err)
  351. return info, err
  352. }
  353. log.Info("d.acc.Info3(%v) return (%v)", uid, info)
  354. return
  355. }
  356. func (d *Dao) getRoomInfoByUid(c context.Context, uid int64) (roomInfo *v1.RoomGetStatusInfoByUidsResp_RoomInfo, err error) {
  357. roomInfo = &v1.RoomGetStatusInfoByUidsResp_RoomInfo{}
  358. reply, err := dao.RoomAPI.V1Room.GetStatusInfoByUids(c, &v1.RoomGetStatusInfoByUidsReq{
  359. Uids: []int64{uid},
  360. ShowHidden: 1,
  361. })
  362. if err != nil {
  363. log.Error("dao.RoomAPI.V1Room.GetStatusInfoByUids (%v) error(%v)", uid, err)
  364. return roomInfo, err
  365. }
  366. if reply.GetCode() != 0 {
  367. err = ecode.Int(int(reply.GetCode()))
  368. log.Error("dao.RoomAPI.V2Room.GetByIds (%v) error code(%v)", uid, err)
  369. return roomInfo, err
  370. }
  371. if len(reply.Data) == 0 {
  372. return
  373. }
  374. for aUid, aInfo := range reply.Data {
  375. if aUid == uid {
  376. return aInfo, nil
  377. }
  378. }
  379. return
  380. }
  381. func (d *Dao) getMedalInfoByUids(c context.Context, uids []int64) (medalInfo map[int64]*v12.AnchorQueryLiveWearingResp_Medal, err error) {
  382. medalInfo = make(map[int64]*v12.AnchorQueryLiveWearingResp_Medal)
  383. reply, err := dao.FansMedalAPI.V2Anchor.QueryLiveWearing(c, &v12.AnchorQueryLiveWearingReq{
  384. UidList: uids,
  385. })
  386. log.Info("call dao.FansMedalAPI.V2Anchor.QueryLiveWearing (%v) rst (%v)", uids, reply)
  387. if err != nil {
  388. log.Error("dao.FansMedalAPI.V2Anchor.QueryLiveWearing (%v) error(%v)", uids, err)
  389. return medalInfo, err
  390. }
  391. if reply.GetCode() != 0 {
  392. err = ecode.Int(int(reply.GetCode()))
  393. log.Error("dao.RoomAPI.V2Room.GetByIds (%v) error code(%v)", uids, err)
  394. return medalInfo, err
  395. }
  396. if len(reply.Data) == 0 {
  397. return
  398. }
  399. return reply.Data, err
  400. }
  401. // IsAdminByRoomId ...
  402. func (d *Dao) IsAdminByRoomId(c context.Context, uid int64, roomId int64) (rst int64, err error) {
  403. rst = 0
  404. admins, err := d.GetAllByRoomId(c, roomId)
  405. if err != nil {
  406. log.Error("GetAllByRoomId(%v) error(%v)", roomId, err)
  407. return rst, err
  408. }
  409. if len(admins) == 0 {
  410. return rst, nil
  411. }
  412. for _, v := range admins {
  413. if v.Uid == uid {
  414. rst = 1
  415. return rst, nil
  416. }
  417. }
  418. return
  419. }
  420. // GetByAnchorIdPage get by anchor id and page .
  421. func (d *Dao) GetByAnchorIdPage(c context.Context, anchorId int64, page int64, pageSize int64) (resp *v1pb.RoomAdminGetByAnchorResp, err error) {
  422. resp = &v1pb.RoomAdminGetByAnchorResp{}
  423. resp.Page = &v1pb.RoomAdminGetByAnchorResp_Page{
  424. Page: page,
  425. PageSize: pageSize,
  426. TotalPage: 1,
  427. TotalCount: 0,
  428. }
  429. roomId, err := d.getRoomIdByUid(c, anchorId)
  430. if err != nil {
  431. return resp, err
  432. }
  433. if 0 == roomId {
  434. return resp, nil
  435. }
  436. allAdmins, err := d.GetAllByRoomId(c, roomId)
  437. //spew.Dump("GetAllByUid", allAdmins, err)
  438. if err != nil {
  439. return resp, err
  440. }
  441. if allAdmins == nil {
  442. return resp, nil
  443. }
  444. sort.Sort(sort.Reverse(model.RoomAdmins(allAdmins)))
  445. //spew.Dump(allAdmins)
  446. resp.Page.TotalCount = int64(len(allAdmins))
  447. resp.Page.PageSize = pageSize
  448. resp.Page.TotalPage = int64(math.Ceil(float64(len(allAdmins)) / float64(resp.Page.PageSize)))
  449. begin := (page - 1) * pageSize
  450. end := page * pageSize
  451. if page*pageSize > int64(len(allAdmins)) {
  452. end = int64(len(allAdmins))
  453. }
  454. uids, _ := d.getUidsFromAdmins(c, allAdmins)
  455. if int64(len(uids)) < end {
  456. end = int64(len(uids))
  457. }
  458. if begin > end {
  459. begin = end
  460. }
  461. //spew.Dump("getAnchorUidsFromAdmins", uids, err)
  462. accArgs := &account.ArgMids{Mids: uids[begin:end]}
  463. accData, err := d.acc.Infos3(c, accArgs)
  464. if err != nil {
  465. log.Error("d.acc.Infos3(%v) error(%v)", accArgs, err)
  466. }
  467. medalData, err := d.getMedalInfoByUids(c, uids)
  468. if err != nil {
  469. log.Error("d.getMedalInfoByUids(%v) error(%v)", uids, err)
  470. return resp, err
  471. }
  472. if err != nil {
  473. log.Error("call account.Infos3(%v) error(%v)", accArgs, err)
  474. return resp, err
  475. }
  476. for _, v := range allAdmins[begin:end] {
  477. item := &v1pb.RoomAdminGetByAnchorResp_Data{
  478. Uid: v.Uid,
  479. Ctime: v.Ctime.Time().Format("2006-01-02 15:04:05"),
  480. Roomid: v.Roomid,
  481. }
  482. if _, ok := accData[v.Uid]; ok {
  483. item.Uname = accData[v.Uid].Name
  484. item.Face = accData[v.Uid].Face
  485. } else {
  486. log.Error("没有这个人的用户信息 uid(%v) data(%v)", v.Uid, accData)
  487. }
  488. if _, ok := medalData[v.Uid]; ok {
  489. item.Level = medalData[v.Uid].Level
  490. item.MedalName = medalData[v.Uid].MedalName
  491. } else {
  492. log.Info("没有这个人的勋章信息 uid(%v) data(%v)", v.Uid, medalData)
  493. }
  494. resp.Data = append(resp.Data, item)
  495. }
  496. //spew.Dump("resp.Data", resp.Data)
  497. return
  498. }
  499. // GetAllByRoomId get by uid.
  500. func (d *Dao) GetAllByRoomId(c context.Context, roomId int64) ([]*model.RoomAdmin, error) {
  501. rstMc, err := d.CacheRoomAdminRoom(c, roomId)
  502. //spew.Dump("HasAnyAdmin1", rstMc, err)
  503. //spew.Dump("lenMc", len(rstMc))
  504. if err != nil {
  505. return nil, err
  506. }
  507. // 空缓存标识
  508. if rstMc != nil {
  509. if rstMc[0].Id == -1 {
  510. return nil, err
  511. }
  512. return rstMc, err
  513. }
  514. rstDb, err := d.GetByRoomIdMysql(c, roomId)
  515. if err != nil {
  516. return nil, err
  517. }
  518. if len(rstDb) == 0 {
  519. d.AddCacheNoneRoom(c, roomId)
  520. return nil, err
  521. }
  522. d.AddCacheKeyAnchorRoom(c, roomId, rstDb)
  523. return rstDb, err
  524. }
  525. // getUidsFromAdmins .
  526. // 返回房管列表中的uid
  527. func (d *Dao) getUidsFromAdmins(c context.Context, admins []*model.RoomAdmin) (uids []int64, err error) {
  528. for _, a := range admins {
  529. uids = append(uids, a.Uid)
  530. }
  531. return
  532. }
  533. // DismissAnchor del a admin
  534. func (d *Dao) DismissAnchor(c context.Context, uid int64, anchorId int64) (resp *v1pb.RoomAdminDismissAdminResp, err error) {
  535. resp = &v1pb.RoomAdminDismissAdminResp{}
  536. roomId, err := d.getRoomIdByUid(c, anchorId)
  537. if err != nil {
  538. return resp, err
  539. }
  540. if 0 == roomId {
  541. return resp, nil
  542. }
  543. isAdmin, err := d.IsAdminByRoomId(c, uid, roomId)
  544. if err != nil {
  545. log.Error("IsAdminByRoomId uid(%v) roomid (%v) error(%v)", uid, roomId, err)
  546. return resp, err
  547. }
  548. if 0 == isAdmin {
  549. err = ecode.Error(ecode.XUserAddRoomAdminNotAdminError, "该用户已经不是房管啦")
  550. return
  551. }
  552. err = d.Del(c, uid, roomId)
  553. if err != nil {
  554. log.Error("getRoomInfoByUid uid (%v) roomid (%v) error(%v)", uid, roomId, err)
  555. return
  556. }
  557. return
  558. }
  559. // DelAllCache delete cache .
  560. func (d *Dao) DelAllCache(c context.Context, uid int64, roomId int64) (err error) {
  561. if err = d.DelCacheKeyAnchorRoom(c, roomId); err != nil {
  562. log.Error("DelCacheKeyAnchorRoom(%v) error(%v)", roomId, err)
  563. return err
  564. }
  565. if err = d.DelCacheRoomAdminUser(c, uid); err != nil {
  566. log.Error("DelCacheRoomAdminUser(%v) error(%v)", uid, err)
  567. return err
  568. }
  569. return
  570. }
  571. // Add add a admin
  572. func (d *Dao) Add(c context.Context, uid int64, anchorId int64) (resp *v1pb.RoomAdminAddResp, err error) {
  573. resp = &v1pb.RoomAdminAddResp{}
  574. roomId, err := d.getRoomIdByUid(c, anchorId)
  575. if err != nil {
  576. return resp, err
  577. }
  578. if 0 == roomId {
  579. return resp, nil
  580. }
  581. allRoomAdmin, err := d.GetAllByRoomId(c, roomId)
  582. //spew.Dump("Add", roomId, allRoomAdmin)
  583. if err != nil {
  584. return resp, err
  585. }
  586. if len(allRoomAdmin) >= maxAdminsNum {
  587. err = ecode.Error(ecode.XUserAddRoomAdminOverLimitError, "最多设置100个房间管理员")
  588. //err = &pb.Error{
  589. // ErrCode: 2,
  590. // ErrMessage: "最多设置100个房间管理员",
  591. //}
  592. return
  593. }
  594. isAdmin, err := d.IsAdminByRoomId(c, uid, roomId)
  595. if err != nil {
  596. log.Error("IsAdminByRoomId uid(%v) roomid (%v) error(%v)", uid, roomId, err)
  597. return resp, err
  598. }
  599. if 1 == isAdmin {
  600. err = ecode.Error(ecode.XUserAddRoomAdminIsAdminError, "该用户已经是你的房管啦")
  601. //err = &pb.Error{
  602. // ErrCode: 1,
  603. // ErrMessage: "他已经是房管",
  604. //}
  605. return
  606. }
  607. banArg := &banned.SilentMngIsBlockUserReq{
  608. Uid: uid,
  609. Roomid: roomId,
  610. Type: 1,
  611. }
  612. retBan, err := dao.BannedAPI.V1SilentMng.IsBlockUser(c, banArg)
  613. if err != nil {
  614. log.Error("call dao.BannedAPI.V1SilentMng.IsBlockUser(%v) error(%v)", banArg, err)
  615. return
  616. }
  617. if retBan.Code != 0 || nil == retBan.Data {
  618. log.Error("call dao.BannedAPI.V1SilentMng.IsBlockUser(%v) error return (%v)", banArg, retBan)
  619. }
  620. if retBan.Data.GetIsBlockUser() {
  621. err = ecode.Error(ecode.XUserAddRoomAdminIsSilentError, "他已经被禁言,无法添加房管")
  622. //err = &pb.Error{
  623. // ErrCode: 3,
  624. // ErrMessage: "他已经被禁言,无法添加房管",
  625. //}
  626. return
  627. }
  628. if err = d.DelAllCache(c, uid, roomId); err != nil {
  629. log.Error("DelCacheKeyAnchorRoom(%v) error(%v)", roomId, err)
  630. return resp, err
  631. }
  632. if err = d.AddAdminMysql(c, uid, roomId); err != nil {
  633. log.Error("DelCacheKeyAnchorRoom(%v) error(%v)", roomId, err)
  634. return
  635. }
  636. resp.Uid = uid
  637. resp.Roomid = roomId
  638. resp.Userinfo = &v1pb.RoomAdminAddResp_UI{}
  639. resp.Userinfo.Uid = uid
  640. info, _ := d.getInfoByUid(c, uid)
  641. if info != nil {
  642. resp.Userinfo.Uname = info.Name
  643. }
  644. d.adminChange(c, uid, roomId)
  645. return
  646. }
  647. func (d *Dao) getRoomIdByUid(c context.Context, anchorId int64) (roomId int64, err error) {
  648. roomInfo, err := d.getRoomInfoByUid(c, anchorId)
  649. if err != nil {
  650. log.Error("getRoomInfoByUid (%v) error(%v)", anchorId, err)
  651. return roomId, err
  652. }
  653. if roomInfo == nil {
  654. return roomId, nil
  655. }
  656. return roomInfo.RoomId, nil
  657. }
  658. // adminChange send broadcast
  659. func (d *Dao) adminChange(c context.Context, uid int64, roomId int64) (err error) {
  660. roomInfo, err := dao.RoomAPI.V2Room.GetByIds(c, &v2.RoomGetByIdsReq{
  661. Ids: []int64{roomId},
  662. NeedUinfo: 1,
  663. })
  664. if err != nil {
  665. log.Error("dao.RoomAPI.V2Room.GetByIds(%v) error(%v)", roomId, err)
  666. return
  667. }
  668. if roomInfo.Code != 0 || 0 == len(roomInfo.Data) {
  669. log.Error("dao.RoomAPI.V1Room.GetInfoById(%v) error code (%v) data (%v)", roomId, roomInfo.Code, roomInfo.Data)
  670. return
  671. }
  672. postJson := make(map[string]interface{})
  673. postJson["cmd"] = "room_admin_entrance"
  674. postJson["uid"] = uid
  675. postJson["msg"] = "系统提示:你已被主播设为房管"
  676. if err = d.sendBroadcastRoom(roomId, postJson); err != nil {
  677. return err
  678. }
  679. admins, err := d.GetAllByRoomId(c, roomId)
  680. if err != nil {
  681. return err
  682. }
  683. var adminUids []int64
  684. // adminUids := make([]int64, 100)
  685. for _, v := range admins {
  686. adminUids = append(adminUids, v.Uid)
  687. }
  688. postJson2 := make(map[string]interface{})
  689. postJson2["cmd"] = "ROOM_ADMINS"
  690. postJson2["uids"] = adminUids
  691. if err = d.sendBroadcastRoom(roomId, postJson2); err != nil {
  692. return err
  693. }
  694. return
  695. }
  696. // sendBroadcastRoom .
  697. func (d *Dao) sendBroadcastRoom(roomid int64, postJson map[string]interface{}) (err error) {
  698. log.Info("send reward broadcast begin:%d", roomid)
  699. var endPoint = fmt.Sprintf("http://live-dm.bilibili.co/dm/1/push?cid=%d&ensure=1", roomid)
  700. bytesData, err := json.Marshal(postJson)
  701. if err != nil {
  702. log.Error("json.Marshal(%v) error(%v)", postJson, err)
  703. return
  704. }
  705. req, err := http.NewRequest("POST", endPoint, bytes.NewReader(bytesData))
  706. req.Header.Add("Content-Type", "application/json")
  707. if err != nil {
  708. log.Error("http.NewRequest(%v) url(%v) error(%v)", postJson, endPoint, err)
  709. return
  710. }
  711. client := http.Client{
  712. Timeout: time.Second,
  713. }
  714. // use httpClient to send request
  715. response, err := client.Do(req)
  716. if err != nil {
  717. log.Error("sending request to API endpoint(%v) error(%v)", req, err)
  718. return
  719. }
  720. body, err := ioutil.ReadAll(response.Body)
  721. if err != nil {
  722. log.Error("parse resp body(%v) error(%v)", body, err)
  723. }
  724. log.Info("send reward broadcast end:%d", roomid)
  725. return
  726. }
  727. // IsAdmin return whether a user is admin.
  728. func (d *Dao) IsAdmin(c context.Context, uid int64, anchorId int64, roomId int64) (resp *v1pb.RoomAdminIsAdminResp, err error) {
  729. resp = &v1pb.RoomAdminIsAdminResp{}
  730. isAdmin, err := d.IsAdminByRoomId(c, uid, roomId)
  731. if err != nil {
  732. log.Error("IsAdminByRoomId uid(%v) roomid (%v) error(%v)", uid, roomId, err)
  733. return resp, err
  734. }
  735. if 0 == isAdmin {
  736. if uid != anchorId {
  737. err = ecode.Error(120014, "Ta不是该主播的房管")
  738. //err = ecode.Error{
  739. // ErrCode: 120014, // 接口迁移, code 为老业务error code
  740. // ErrMessage: "Ta不是该主播的房管",
  741. //}
  742. return
  743. }
  744. }
  745. userInfo, err := d.getInfoByUid(c, uid)
  746. if err != nil {
  747. return resp, err
  748. }
  749. if userInfo != nil {
  750. resp.Userinfo = &v1pb.RoomAdminIsAdminResp_UI{}
  751. resp.Roomid = roomId
  752. resp.Uid = uid
  753. resp.Userinfo.Uid = uid
  754. resp.Userinfo.Uname = userInfo.Name
  755. return
  756. }
  757. return
  758. }