From 7d42712db2fa490b8763183fecb1c0e291b80f7a Mon Sep 17 00:00:00 2001
From: otto <965391514@qq.com>
Date: Mon, 15 Apr 2024 17:46:54 +0800
Subject: [PATCH] up

---
 go.mod       |  2 +-
 task/task.go | 26 ++++++++++++++++++++++++--
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/go.mod b/go.mod
index a16c51b..5048276 100644
--- a/go.mod
+++ b/go.mod
@@ -3,7 +3,7 @@ module git.gz.internal.jumaiyx.cn/job/hotspot-clear
 go 1.20
 
 require (
-	git.gz.internal.jumaiyx.cn/jm/jmproto v0.0.0-20240321070012-9ad2cc15ec1c
+	git.gz.internal.jumaiyx.cn/jm/jmproto v0.0.0-20240415093446-0004fca23b7b
 	git.gz.internal.jumaiyx.cn/pkg/client v0.0.0-20231123090823-91017d7a1a1e
 	git.gz.internal.jumaiyx.cn/pkg/config/v2 v2.0.0-20240312130125-0f8bbcc94011
 	git.gz.internal.jumaiyx.cn/pkg/k8s-client/v2 v2.0.3-0.20240229083126-a1abe7ddc19f
diff --git a/task/task.go b/task/task.go
index 0c76cc5..720bd4d 100644
--- a/task/task.go
+++ b/task/task.go
@@ -2,6 +2,7 @@ package task
 
 import (
 	"context"
+	livev1 "git.gz.internal.jumaiyx.cn/jm/jmproto/live/v1"
 	v2 "git.gz.internal.jumaiyx.cn/jm/jmproto/room/v2"
 	"git.gz.internal.jumaiyx.cn/pkg/client"
 	configv2 "git.gz.internal.jumaiyx.cn/pkg/config/v2"
@@ -79,14 +80,35 @@ func task(start, end int64) error {
 		if err != nil {
 			return err
 		}
+
+		liveClient, liveClientClose, err := client.GetLiveClient(context.Background())
+		if err != nil {
+			return err
+		}
+		defer func() { _ = liveClientClose() }()
+
+		onlineTotalResp, err := liveClient.GetLiveOnlineTotal(context.Background(), &livev1.GetLiveOnlineTotalReq{
+			RoomIdList: ids,
+		})
+		if err != nil {
+			return err
+		}
 		roomClient, roomClientClose, err := client.GetRoomClientV2(context.Background())
 		if err != nil {
 			return err
 		}
 		defer func() { _ = roomClientClose() }()
+		onlineMap := onlineTotalResp.Items
+		var items []*v2.IncrRoomHotspotReq_Item
+		for _, id := range ids {
+			online := onlineMap[id]
+			items = append(items, &v2.IncrRoomHotspotReq_Item{
+				RoomId:   id,
+				HotValue: int32(online) * 10,
+			})
+		}
 		_, err = roomClient.IncrRoomHotspot(context.Background(), &v2.IncrRoomHotspotReq{
-			RoomIdList: ids,
-			HotValue:   0,
+			Items: items,
 		})
 		if err != nil {
 			return err