src/main/java/com/whyc/mapper/BOMFeedbakMapper.java | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 | |
src/main/java/com/whyc/service/BOMFeedbakService.java | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 | |
src/main/java/com/whyc/webSocket/WorksheetSocket.java | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
src/main/java/com/whyc/mapper/BOMFeedbakMapper.java
New file @@ -0,0 +1,6 @@ package com.whyc.mapper; import com.whyc.pojo.BOMFeedbak; public interface BOMFeedbakMapper extends CustomMapper<BOMFeedbak>{ } src/main/java/com/whyc/service/BOMFeedbakService.java
New file @@ -0,0 +1,29 @@ package com.whyc.service; import com.whyc.mapper.BOMFeedbakMapper; import com.whyc.pojo.DocUser; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.Map; @Service public class BOMFeedbakService { @Autowired(required = false) private BOMFeedbakMapper mapper; //统计 public int getFkStatistic(DocUser user) { int fk=0; switch (user.getRoleId()){ //普通员工 case "1001": break; //项目经理 case "1002":break; //总经理 case "1003":break; } return fk; } } src/main/java/com/whyc/webSocket/WorksheetSocket.java
@@ -3,6 +3,7 @@ import com.whyc.config.WebSocketConfig; import com.whyc.dto.Response; import com.whyc.pojo.DocUser; import com.whyc.service.BOMFeedbakService; import com.whyc.service.WorksheetMainService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; @@ -25,12 +26,16 @@ private Thread thread; private static WorksheetMainService mainService; private static BOMFeedbakService feedbakService; @Autowired public void setWorksheetMainService(WorksheetMainService mainService) { WorksheetSocket.mainService = mainService; } @Autowired public void setFeedbakService(BOMFeedbakService feedbakService) { WorksheetSocket.feedbakService = feedbakService; } @OnOpen public void onOpen(Session session, EndpointConfig config){ this.session = session; @@ -42,6 +47,7 @@ try { while (!currentThread().isInterrupted()) { Map<String, Integer> statisticMap = mainService.getStatusStatistic((DocUser) httpSession.getAttribute("user")); statisticMap.put("fk",1); session.getBasicRemote().sendObject(new Response().set(1, statisticMap)); sleep(2000); }