采购-供应商报价看板：供应商报价项目看板已结束实时推送服务

@@ -3,6 +3,7 @@ package com.lframework.xingyun.sc.listeners.app;
 import com.lframework.xingyun.sc.events.quotation.CommonChangedEvent;
 import com.lframework.xingyun.sc.procurement.impl.quotation.SupplierQuotationProjectQuoteServiceImpl;
 import com.lframework.xingyun.sc.procurement.impl.quotation.SupplierQuotationProjectServiceImpl;
+import com.lframework.xingyun.sc.procurement.websocket.quotation.project.SupplierQuotationProjectFinishedRealtimePushService;
 import com.lframework.xingyun.sc.procurement.websocket.quotation.project.SupplierQuotationProjectRealtimePushService;
 import javax.annotation.Resource;
 import org.springframework.stereotype.Component;
@@ -17,6 +18,8 @@ public class SupplierQuotationProjectRealtimeListener {
 
     @Resource
     private SupplierQuotationProjectRealtimePushService realtimePushService;
+    @Resource
+    private SupplierQuotationProjectFinishedRealtimePushService finishedRealtimePushService;
 
     /**
      * 报价项目或供应商报价变更后，广播当日完整看板数据
@@ -30,5 +33,6 @@ public class SupplierQuotationProjectRealtimeListener {
             return;
         }
         realtimePushService.broadcastCurrentDayFullData(event.getEventType());
+        finishedRealtimePushService.broadcastCurrentDayFinishedData(event.getEventType());
     }
 }

@@ -14,6 +14,8 @@ import com.lframework.xingyun.sc.procurement.vo.quotation.UpdateSupplierQuotatio
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiOperation;
+import java.time.LocalDate;
+import java.util.List;
 import javax.annotation.Resource;
 import javax.validation.Valid;
 import javax.validation.constraints.NotBlank;
@@ -63,6 +65,14 @@ public class SupplierQuotationProjectController extends DefaultBaseController {
         return InvokeResultBuilder.success(data);
     }
 
+    @ApiOperation("当日已完成/已中止项目实时查询")
+    @HasPermission({"procure-manage:supplier-price:query"})
+    @GetMapping("/queryCurrentDayFinished")
+    public InvokeResult<List<SupplierQuotationProject>> queryCurrentDayFinished() {
+        return InvokeResultBuilder.success(
+                supplierQuotationProjectService.queryCurrentDayFinishedBoardList(LocalDate.now(), getCurrentUser().getId()));
+    }
+
     @ApiOperation("修改")
     @HasPermission({"procure-manage:supplier-price:modify"})
     @PutMapping

@@ -96,6 +96,20 @@ public class SupplierQuotationProjectServiceImpl
     }
 
     @Override
+    public List<SupplierQuotationProject> queryCurrentDayFinishedBoardList(LocalDate day, String userId) {
+        List<SupplierQuotationProject> datas = getBaseMapper().queryCurrentDayFinishedList(day);
+        fillComputedStatus(datas);
+        if (datas == null || datas.isEmpty()) {
+            return datas;
+        }
+        for (SupplierQuotationProject data : datas) {
+            List<SupplierQuotationProjectQuote> quoteList = supplierQuotationProjectQuoteService.listByProject(data.getId(), userId);
+            data.setSupplierQuotationProjectQuoteList(quoteList);
+        }
+        return datas;
+    }
+
+    @Override
     public SupplierQuotationProject findById(String id) {
         SupplierQuotationProject data = getBaseMapper().findById(id);
         if (data != null) {

@@ -16,5 +16,7 @@ public interface SupplierQuotationProjectMapper extends BaseMapper<SupplierQuota
     String getMaxCodeByDay(@Param("dayPrefix") String dayPrefix);
 
     List<SupplierQuotationProject> queryCurrentDayList(@Param("day") LocalDate day);
+
+    List<SupplierQuotationProject> queryCurrentDayFinishedList(@Param("day") LocalDate day);
 }
 

@@ -25,6 +25,8 @@ public interface SupplierQuotationProjectService extends BaseMpService<SupplierQ
 
     List<SupplierQuotationProject> queryCurrentDayBoardList(LocalDate day, String userId);
 
+    List<SupplierQuotationProject> queryCurrentDayFinishedBoardList(LocalDate day, String userId);
+
     /**
      * 更新项目状态
      *

+package com.lframework.xingyun.sc.procurement.websocket.quotation.project;
+
+import com.lframework.starter.web.core.utils.JsonUtil;
+import com.lframework.xingyun.sc.procurement.dto.quotation.SupplierQuotationProjectRealtimeMessage;
+import com.lframework.xingyun.sc.procurement.entity.SupplierQuotationProject;
+import com.lframework.xingyun.sc.procurement.service.quotation.SupplierQuotationProjectService;
+import java.io.IOException;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import javax.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.stereotype.Service;
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+
+@Slf4j
+@Service
+public class SupplierQuotationProjectFinishedRealtimePushService {
+
+    private final Map<String, WebSocketSession> sessionMap = new ConcurrentHashMap<>();
+
+    @Resource
+    private SupplierQuotationProjectService supplierQuotationProjectService;
+
+    public void register(WebSocketSession session) {
+        sessionMap.put(session.getId(), session);
+    }
+
+    public void unregister(WebSocketSession session) {
+        if (session != null) {
+            sessionMap.remove(session.getId());
+        }
+    }
+
+    public void broadcastCurrentDayFinishedData(String eventType) {
+        if (sessionMap.isEmpty()) {
+            return;
+        }
+
+        for (WebSocketSession session : sessionMap.values()) {
+            replayCurrentDayFinishedData(session, eventType);
+        }
+    }
+
+    public void replayCurrentDayFinishedData(WebSocketSession session, String eventType) {
+        if (session == null) {
+            return;
+        }
+        sendMessage(session, JsonUtil.toJsonString(buildCurrentDayMessage(session, eventType)));
+    }
+
+    private SupplierQuotationProjectRealtimeMessage buildCurrentDayMessage(WebSocketSession session, String eventType) {
+        LocalDate today = LocalDate.now();
+        String userId = getUserId(session);
+        List<SupplierQuotationProject> records = supplierQuotationProjectService.queryCurrentDayFinishedBoardList(today, userId);
+
+        SupplierQuotationProjectRealtimeMessage message = new SupplierQuotationProjectRealtimeMessage();
+        message.setMessageType("FULL_SYNC_FINISHED");
+        message.setEventType(eventType);
+        message.setBusinessDate(today);
+        message.setPushTime(LocalDateTime.now());
+        message.setRecords(records);
+        return message;
+    }
+
+    private void sendMessage(WebSocketSession session, String payload) {
+        if (session == null || !session.isOpen()) {
+            unregister(session);
+            return;
+        }
+
+        try {
+            synchronized (session) {
+                session.sendMessage(new TextMessage(payload));
+            }
+        } catch (IOException e) {
+            log.warn("供应商报价项目结束态实时推送失败，sessionId={}", session.getId(), e);
+            unregister(session);
+        }
+    }
+
+    private String getUserId(WebSocketSession session) {
+        Object userId = session.getAttributes().get(SupplierQuotationProjectWebSocketHandshakeInterceptor.ATTR_USER_ID);
+        return userId == null ? StringUtils.EMPTY : String.valueOf(userId);
+    }
+}
+

+package com.lframework.xingyun.sc.procurement.websocket.quotation.project;
+
+import javax.annotation.Resource;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.config.annotation.EnableWebSocket;
+import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
+import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;
+
+@Configuration
+@EnableWebSocket
+public class SupplierQuotationProjectFinishedWebSocketConfig implements WebSocketConfigurer {
+
+    public static final String ENDPOINT = "/ws/procurement/supplierQuotationProject/finished/realtime";
+
+    @Resource
+    private SupplierQuotationProjectFinishedWebSocketHandler webSocketHandler;
+    @Resource
+    private SupplierQuotationProjectWebSocketHandshakeInterceptor handshakeInterceptor;
+
+    @Override
+    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
+        registry.addHandler(webSocketHandler, ENDPOINT)
+                .addInterceptors(handshakeInterceptor)
+                .setAllowedOrigins("*");
+    }
+}
+

+package com.lframework.xingyun.sc.procurement.websocket.quotation.project;
+
+import javax.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.CloseStatus;
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+import org.springframework.web.socket.handler.TextWebSocketHandler;
+
+@Slf4j
+@Component
+public class SupplierQuotationProjectFinishedWebSocketHandler extends TextWebSocketHandler {
+
+    @Resource
+    private SupplierQuotationProjectFinishedRealtimePushService realtimePushService;
+
+    @Override
+    public void afterConnectionEstablished(WebSocketSession session) {
+        realtimePushService.register(session);
+        realtimePushService.replayCurrentDayFinishedData(session, "INIT");
+    }
+
+    @Override
+    protected void handleTextMessage(WebSocketSession session, TextMessage message) {
+        String payload = StringUtils.trimToEmpty(message.getPayload());
+        if ("SYNC_TODAY".equalsIgnoreCase(payload) || "RESYNC".equalsIgnoreCase(payload)) {
+            realtimePushService.replayCurrentDayFinishedData(session, "RESYNC");
+        }
+    }
+
+    @Override
+    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
+        realtimePushService.unregister(session);
+    }
+
+    @Override
+    public void handleTransportError(WebSocketSession session, Throwable exception) {
+        log.warn("供应商报价项目结束态 WebSocket 连接异常，sessionId={}", session.getId(), exception);
+        realtimePushService.unregister(session);
+    }
+}
+

@@ -96,5 +96,12 @@
         AND tb.status='PUBLISHED'
         ORDER BY tb.create_time DESC, tb.id DESC
     </select>
+
+    <select id="queryCurrentDayFinishedList" resultMap="SupplierQuotationProject">
+        <include refid="SupplierQuotationProjectSql"/>
+        WHERE DATE(tb.create_time) = #{day}
+        AND tb.status IN ('COMPLETED', 'ABORTED')
+        ORDER BY tb.update_time DESC, tb.id DESC
+    </select>
 </mapper>