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);