Browse Source

优化代码

wuxw 2 years ago
parent
commit
819f02d016

+ 12 - 0
service-oa/src/main/java/com/java110/oa/cmd/work/DeleteWorkPoolCmd.java

@@ -22,6 +22,7 @@ import com.java110.core.context.CmdContextUtils;
 import com.java110.core.context.ICmdDataFlowContext;
 import com.java110.core.event.cmd.Cmd;
 import com.java110.core.event.cmd.CmdEvent;
+import com.java110.dto.workPool.WorkPoolDto;
 import com.java110.intf.oa.*;
 import com.java110.intf.user.IUserV1InnerServiceSMO;
 import com.java110.po.workCopy.WorkCopyPo;
@@ -76,6 +77,17 @@ public class DeleteWorkPoolCmd extends Cmd {
         Assert.hasKeyAndValue(reqJson, "workId", "workId不能为空");
         String storeId = CmdContextUtils.getStoreId(cmdDataFlowContext);
         reqJson.put("storeId",storeId);
+
+        String userId = CmdContextUtils.getUserId(cmdDataFlowContext);
+
+        WorkPoolDto workPoolDto = new WorkPoolDto();
+        workPoolDto.setWorkId(reqJson.getString("workId"));
+        workPoolDto.setStoreId(storeId);
+        workPoolDto.setCreateUserId(userId);
+        int count = workPoolV1InnerServiceSMOImpl.queryWorkPoolsCount(workPoolDto);
+        if(count < 1){
+            throw new CmdException("您没有权限删除");
+        }
     }
 
     @Override

+ 11 - 0
service-oa/src/main/java/com/java110/oa/cmd/work/UpdateWorkPoolCmd.java

@@ -96,10 +96,21 @@ public class UpdateWorkPoolCmd extends Cmd {
         String storeId = CmdContextUtils.getStoreId(cmdDataFlowContext);
         reqJson.put("storeId", storeId);
 
+        String userId = CmdContextUtils.getUserId(cmdDataFlowContext);
+
         if (!reqJson.containsKey("staffs")) {
             throw new CmdException("未包含处理人");
         }
 
+        WorkPoolDto workPoolDto = new WorkPoolDto();
+        workPoolDto.setWorkId(reqJson.getString("workId"));
+        workPoolDto.setStoreId(storeId);
+        workPoolDto.setCreateUserId(userId);
+        int count = workPoolV1InnerServiceSMOImpl.queryWorkPoolsCount(workPoolDto);
+        if(count < 1){
+            throw new CmdException("您没有权限修改");
+        }
+
         JSONArray staffs = reqJson.getJSONArray("staffs");
         if (staffs == null || staffs.isEmpty()) {
             throw new CmdException("未包含处理人");