0412更新

src/main/java/com/heai/modules/production/dao/DailyPlanMapper.java

@@ -5,6 +5,7 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.heai.modules.app.entity.ProductionWarehouseData;
 import com.heai.modules.app.entity.SoBomData;
 import com.heai.modules.production.entity.*;
+import com.heai.modules.sys.entity.SysRoleEntity;
 import com.heai.modules.sys.entity.VpPartData;
 import io.swagger.models.auth.In;
 import org.apache.ibatis.annotations.Mapper;
@@ -1039,4 +1040,7 @@ public interface DailyPlanMapper {
     List<SfdcTimeHistSOSData> getTimeHistListForSfdc(SfdcTimeHistSOSData inData);
 
     List<SoBomData> checkSPBOMIsIssure(SOScheduledRoutingOutData inData);
+
+    List<SysRoleEntity> checkUserRole(@Param("userId") Long userId,
+                                      @Param("typeName") String typeName);
 }

src/main/java/com/heai/modules/production/service/impl/ReworkRecordServiceImpl.java

@@ -9,7 +9,10 @@ import com.heai.modules.production.entity.ReworkRecord;
 import com.heai.modules.production.entity.ReworkRecordData;
 import com.heai.modules.production.entity.SOScheduledRoutingOutData;
 import com.heai.modules.production.service.ReworkRecordService;
+import com.heai.modules.sys.entity.SysRoleEntity;
+import com.heai.modules.sys.entity.SysUserEntity;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.shiro.SecurityUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -104,11 +107,22 @@ public class ReworkRecordServiceImpl extends ServiceImpl<ReworkRecordMapper,Rewo
 
     @Override
     public void removeRework(ReworkRecord reworkRecord) {
+        SysUserEntity currentUser = (SysUserEntity) SecurityUtils.getSubject().getPrincipal();
+        List<SysRoleEntity> checkUserRole=dailyPlanMapper.checkUserRole(currentUser.getUserId(),"管理员");
+        if(checkUserRole.size()==0){
+            throw new RuntimeException("该用户没有权限");
+        }
         lambdaUpdate().eq(ReworkRecord::getReworkNo,reworkRecord.getReworkNo()).remove();
     }
 
     @Override
     public void cancelRework(ReworkRecord reworkRecord) {
-        baseMapper.cancelRework(reworkRecord);
+        SysUserEntity currentUser = (SysUserEntity) SecurityUtils.getSubject().getPrincipal();
+        List<SysRoleEntity> checkUserRole=dailyPlanMapper.checkUserRole(currentUser.getUserId(),"管理员");
+        if(checkUserRole.size()==0){
+            throw new RuntimeException("该用户没有权限");
+        }
+        baseMapper.cancelRework(reworkRecord
+        );
     }
 }

src/main/resources/mapper/production/DailyPlanMapper.xml

@@ -1567,7 +1567,7 @@
         from WorkCenterResource where site=#{site} and ResourceID=#{resourceID} and WorkCenterNo=#{workCenterNo}
     </select>
 
-    <select id="checkFqcInspectForSeqNo" resultType="SOScheduledRoutingOutData">
+    <select id="checkFqcInspectForSeqNo" resultType="com.heai.modules.production.entity.SOOPInspectData">
         select a.Site,a.TransNo
         from SOOpsTransferHeader a left join SOOpsTransferDetail b on a.site=b.site and a.TransNo=b.TransNo
         where a.site=#{site} and b.seqNo=#{seqNo} and a.need_inspect_flag='Y' and inspected_flag='N' and ToTypeFlag_DB='I'
@@ -1619,4 +1619,10 @@
         where a.orderNo=#{orderNo} and a.site=#{site}
         group by a.ComponentPartNo
     </select>
+
+    <select id="checkUserRole" resultType="com.heai.modules.sys.entity.SysRoleEntity" >
+        select b.role_name
+        from sys_user_role a left join sys_role b on a.role_id=b.role_id
+        where user_id=#{userId} and role_name=#{typeName}
+    </select>
  </mapper>