diff --git a/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java b/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java index 3e7118c..0b7c431 100644 --- a/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java +++ b/src/main/java/com/yupi/springbootinit/service/impl/HostInfoServiceImpl.java @@ -18,12 +18,14 @@ import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.util.StopWatch; +import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import javax.annotation.Resource; import java.util.ArrayList; import java.util.List; import java.util.concurrent.CompletableFuture; + /* * @author: ziin * @date: 2025/6/10 19:04 @@ -56,10 +58,14 @@ public class HostInfoServiceImpl extends ServiceImpl i StopWatch stopWatch = new StopWatch(); stopWatch.start(); saveBatch(newHosts); - newHosts.forEach(newHost -> { - SseEmitterUtil.sendMessage(newHost.getTenantId().toString()+"-"+newHost.getUserId().toString(), - JsonUtils.toJsonString(newHost)); - }); + NewHosts sseConnectConfirmInfo = newHosts.get(0); + SseEmitter sseEmitter = SseEmitterUtil.getSseEmitter(sseConnectConfirmInfo.getTenantId().toString() + "-" + sseConnectConfirmInfo.getUserId().toString()); + if (sseEmitter != null) { + newHosts.forEach(newHost -> { + SseEmitterUtil.sendMessage(newHost.getTenantId().toString()+"-"+newHost.getUserId().toString(), + JsonUtils.toJsonString(newHost)); + }); + } stopWatch.stop(); long totalTimeMillis = stopWatch.getTotalTimeMillis(); log.info("当前存储数据量大小 {}, 存储花费: {}ms",newHosts.size(),totalTimeMillis);