业务编写

src/main/java/com/sztzjy/forex/trading_trading/controller/TrainingController.java

@@ -30,9 +30,7 @@ import org.w3c.dom.stylesheets.LinkStyle;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
-import java.util.ArrayList;
+import java.util.*;
-import java.util.List;
-import java.util.Map;
 import java.util.stream.Collectors;
 
 @Api(tags = "实训管理")
@@ -60,6 +58,9 @@ public class TrainingController {
                 .map(String::valueOf)
                 .collect(Collectors.joining(","));
         List<Map<String, Object>> studentInfos = TzApi.GetStudentInfoByClassIdForForeignExchangeTrading(classIds, currentUser);
+        if (bo.getAddUserInfos() != null && bo.getAddUserInfos().size() > 0) {
+            studentInfos.addAll(bo.getAddUserInfos());
+        }
         List<Member> members = buildMembers(studentInfos, training);
         if (members != null && members.size() > 0) memberService.insertAll(members);
         training.setPeopleCount(members.size());
@@ -191,6 +192,7 @@ public class TrainingController {
             memberService.insert(member);
             buildPracticeMember(member);
         });
+        trainingService.updatePeopleNum(true,trainingId);
         return new ResultEntity(HttpStatus.OK);
     }
 
@@ -202,14 +204,19 @@ public class TrainingController {
         Assert.notNull(trainingId, "实训记录不存在");
         Assert.notNull(memberId, "参赛用户不存在");
         memberService.deleteById(memberId);
+        trainingService.updatePeopleNum(false,trainingId);
         return new ResultEntity(HttpStatus.OK);
     }
 
 
     private List<Member> buildMembers(List<Map<String, Object>> list, Training training) {
         List<Member> members = new ArrayList<>();
+        Set<String> userNos = new HashSet<>();
         if (list != null && list.size() > 0) {
             for (Map<String, Object> map : list) {
+                if (!userNos.add(String.valueOf(map.get("studentNo")))) {
+                    continue;
+                }
                 Member member = new Member();
                 member.setMemberId(IdUtil.simpleUUID());
                 member.setName(String.valueOf(map.get("name")));

src/main/java/com/sztzjy/forex/trading_trading/dto/TrainingBO.java

@@ -11,6 +11,7 @@ import org.springframework.data.annotation.Transient;
 
 import java.util.Date;
 import java.util.List;
+import java.util.Map;
 
 @Data
 @Getter
@@ -21,10 +22,8 @@ public class TrainingBO {
 
     @ApiModelProperty("实训开始时间")
     private Long startTime;
-
     @ApiModelProperty("实训结束时间")
     private Long endTime;
-
     @ApiModelProperty("实训名称")
     private String trainingName;
     @ApiModelProperty("参加实训人数")
@@ -32,6 +31,7 @@ public class TrainingBO {
     private Integer peopleCount;
     @ApiModelProperty("实训班级")
     private List<Integer> classId;
-
+    @ApiModelProperty("单个新增的用户信息")
+    private List<Map<String,Object>> addUserInfos;
 
 }

src/main/java/com/sztzjy/forex/trading_trading/mappers/TrainingMapper.java

@@ -2,6 +2,7 @@ package com.sztzjy.forex.trading_trading.mappers;
 
 import com.sztzjy.forex.trading_trading.entity.Training;
 import com.sztzjy.forex.trading_trading.entity.TrainingExample;
+
 import java.util.List;
 import java.util.Map;
 
@@ -135,4 +136,11 @@ public interface TrainingMapper {
 
     @Select("select training_id from sys_training where status= 'FINISHED'")
     List<String> selectTrainingIdByStatus();
+
+    @Update("update sys_training set people_num=people_num+1 where training_id=#{trainingId}")
+    void addPeopleNum(@Param("trainingId") String trainingId);
+
+    @Update("update sys_training set people_num=people_num-1 where training_id=#{trainingId}")
+    void deletePeopleNum(@Param("trainingId") String trainingId);
+
 }

src/main/java/com/sztzjy/forex/trading_trading/service/MemberService.java

@@ -128,7 +128,6 @@ public class MemberService {
             orCriteria.andStudentNumberEqualTo("%"+keyword+"%");
             example.or(orCriteria);
         }
-
         example.setOrderByClause("stu_rank ASC");
         PageHelper.startPage(index, size);
         return new PageInfo<>(memberMapper.selectByExample(example));

src/main/java/com/sztzjy/forex/trading_trading/service/TrainingService.java

@@ -138,4 +138,13 @@ public class TrainingService {
         List<String> list = trainingMapper.selectProceedTraining();
         return list;
     }
+
+    public void updatePeopleNum(Boolean mark, String trainingId) {
+        if (!mark) {
+            trainingMapper.deletePeopleNum(trainingId);
+        } else {
+            trainingMapper.addPeopleNum(trainingId);
+        }
+    }
+
 }