三表一册数据来源

bonus-modules/bonus-job/src/main/java/com/bonus/job/domain/AttProMonthPo.java

@@ -4,6 +4,7 @@ import lombok.Data;
 
 @Data
 public class AttProMonthPo {
+    private Integer id;
     private Integer proId;
     private String month;
     private Integer rosterNum;

bonus-modules/bonus-job/src/main/java/com/bonus/job/mapper/WorkerJobMapper.java

@@ -92,9 +92,11 @@ public interface WorkerJobMapper {
 
     List<AttProMonthPo> getProMonthData(String lastMonth);
 
-    boolean getProMonthDataByTable(AttProMonthPo attProMonthPo);
+    Integer getProMonthDataByTable(AttProMonthPo attProMonthPo);
 
     void insertProMonthData(AttProMonthPo attProMonthPo);
 
     void updateProMonthData(AttProMonthPo attProMonthPo);
+
+    void updateMonthId(AttProMonthPo attProMonthPo);
 }

bonus-modules/bonus-job/src/main/java/com/bonus/job/task/ThreeTableOneRosterTask.java

@@ -137,13 +137,16 @@ public class ThreeTableOneRosterTask {
                 //获取整合数据
                 List<AttProMonthPo> listRosterMonthAll = mapper.getProMonthData(lastMonth.toString());
                 for (AttProMonthPo attProMonthPo : listRosterMonthAll) {
-                    boolean isExistMonth =mapper.getProMonthDataByTable(attProMonthPo);
+                    Integer id =mapper.getProMonthDataByTable(attProMonthPo);
-                    if(isExistMonth){
+                    if(id>0){
+                        attProMonthPo.setId(id);
                         mapper.updateProMonthData(attProMonthPo);
                     }else{
                         mapper.insertProMonthData(attProMonthPo);
                     }
+                    mapper.updateMonthId(attProMonthPo);
                 }
+
             }
         }catch (Exception e){
             logger.error("人员入场更新表失败，{}",e.getMessage());

bonus-modules/bonus-job/src/main/resources/mapper/job/WorkerJobMapper.xml

@@ -321,16 +321,17 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 			pro_id
 	</select>
 
-	<select id="getProMonthDataByTable" resultType="boolean">
+	<select id="getProMonthDataByTable" resultType="Integer">
 		SELECT
-		COUNT(1) AS count
+			id
 		FROM
 		tb_pro_month_table
 		WHERE
 		table_month = #{month} and pro_id = #{proId}
+		limit 1
 	</select>
 
-	<insert id="insertProMonthData">
+	<insert id="insertProMonthData" keyProperty="id" keyColumn="id" useGeneratedKeys="true">
 		insert into tb_pro_month_table(pro_id,table_month,roster_num,attendance_num,repair_num,salary_num,sub_num,team_num,gross_salary,net_salary,pay_num)
 		values(#{proId},#{month},#{rosterNum},#{attendanceNum},#{repairNum},#{salaryNum},#{subNum},#{teamNum},#{grossSalary},#{netSalary},#{payNum})
 	</insert>
@@ -340,4 +341,8 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		sub_num=#{subNum},team_num=#{teamNum},gross_salary=#{grossSalary},net_salary=#{netSalary},pay_num=#{payNum}
 		where pro_id=#{proId} and table_month=#{month}
 	</update>
+
+	<update id="updateMonthId">
+		update tb_pro_month_table_roster set month_id=#{id} where pro_id=#{proId} and month=#{month}
+	</update>
 </mapper>