From c9a8fc944c4823109fedcce5956d3d6ab49e3611 Mon Sep 17 00:00:00 2001
From: "review512jwy@163.com" <“review512jwy@163.com”>
Date: Tue, 9 Dec 2025 14:22:25 +0800
Subject: [PATCH] =?UTF-8?q?bug=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/techsor/datacenter/sender/dao/DashboardAlertDao.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/techsor/datacenter/sender/dao/DashboardAlertDao.java b/src/main/java/com/techsor/datacenter/sender/dao/DashboardAlertDao.java
index 3ba09d5..37de8da 100644
--- a/src/main/java/com/techsor/datacenter/sender/dao/DashboardAlertDao.java
+++ b/src/main/java/com/techsor/datacenter/sender/dao/DashboardAlertDao.java
@@ -29,6 +29,8 @@ public class DashboardAlertDao {
 
     @Resource
     private JdbcTemplate jdbcTemplate;
+	@Resource
+	private CompanyInfoDao companyInfoDao;
 
 	public void upsertDeviceRawData(DynamodbEntity entity) {
 		if (StringUtils.isEmpty(entity.getDeviceId())) {
@@ -72,7 +74,8 @@ public class DashboardAlertDao {
 				.collect(Collectors.groupingBy(e -> e.getEntity().getCompanyId()));
 
 		grouped.forEach((companyId, subList) -> {
-			String dsKey = "dataSourceForCompany_" + companyId;
+			long topCompanyId = companyInfoDao.getTopCompanyId(String.valueOf(companyId));
+			String dsKey = "dataSourceForCompany_" + topCompanyId;
 			try {
 				DataSourceContextHolder.setCurrentDataSourceKey(dsKey);