培训班管理

src/main/java/com/peanut/modules/master/controller/TrainingClassController.java

@@ -0,0 +1,95 @@
+package com.peanut.modules.master.controller;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.github.yulichang.wrapper.MPJLambdaWrapper;
+import com.peanut.common.utils.R;
+import com.peanut.modules.common.entity.MyUserEntity;
+import com.peanut.modules.common.entity.TrainingClass;
+import com.peanut.modules.common.entity.TrainingToUser;
+import com.peanut.modules.common.service.MyUserService;
+import com.peanut.modules.common.service.TrainingClassService;
+import com.peanut.modules.common.service.TrainingToUserService;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 培训班管理
+ */
+@Slf4j
+@RestController("masterTrainingClass")
+@RequestMapping("master/trainingClass")
+public class TrainingClassController {
+
+    @Autowired
+    private TrainingClassService trainingClassService;
+    @Autowired
+    private TrainingToUserService trainingToUserService;
+    @Autowired
+    private MyUserService userService;
+
+    //新建培训班
+    @RequestMapping("/addTrainingClass")
+    public R addTrainingClass(@RequestBody TrainingClass trainingClass) {
+        trainingClassService.save(trainingClass);
+        return R.ok();
+    }
+
+    //修改培训班
+    @RequestMapping("/editTrainingClass")
+    public R editTrainingClass(@RequestBody TrainingClass trainingClass) {
+        trainingClassService.updateById(trainingClass);
+        return R.ok();
+    }
+
+    //培训班列表
+    @RequestMapping("/trainingClassList")
+    public R trainingClassList(@RequestBody Map<String,Object> params) {
+        Page<TrainingClass> trainingClassList = trainingClassService.page(new Page<>(
+                Long.parseLong(params.get("current").toString()), Long.parseLong(params.get("limit").toString())),new LambdaQueryWrapper<TrainingClass>()
+                .like(StringUtils.isNotEmpty(params.get("title").toString()),TrainingClass::getTitle,params.get("title"))
+                .eq(StringUtils.isNotEmpty(params.get("type").toString()),TrainingClass::getType,params.get("type"))
+                .eq(StringUtils.isNotEmpty(params.get("year").toString()),TrainingClass::getYear,params.get("year"))
+                .eq(StringUtils.isNotEmpty(params.get("singupFlag").toString()),TrainingClass::getSingupFlag,params.get("singupFlag"))
+                .orderByAsc(TrainingClass::getSort));
+        return R.ok().put("trainingClassList",trainingClassList);
+    }
+
+    //培训班下用户列表
+    @RequestMapping("/trainingClassUserList")
+    public R trainingClassUserList(@RequestBody Map<String,Object> params) {
+        Page<TrainingToUser> trainingClassUserList = trainingToUserService.page(new Page<>(
+                Long.parseLong(params.get("current").toString()), Long.parseLong(params.get("limit").toString())),new MPJLambdaWrapper<TrainingToUser>()
+                .eq(TrainingToUser::getTrainingId,params.get("trainingId"))
+                .orderByDesc(TrainingToUser::getCreateTime));
+        for (TrainingToUser toUser:trainingClassUserList.getRecords()){
+            toUser.setUser(userService.getById(toUser.getUserId()));
+        }
+        return R.ok().put("trainingClassUserList",trainingClassUserList);
+    }
+
+    //培训班添加用户
+    @RequestMapping("/addUserToTrainingClass")
+    public R addUserToTrainingClass(@RequestBody TrainingToUser trainingToUser) {
+        List list = trainingToUserService.list(new LambdaQueryWrapper<TrainingToUser>()
+                .eq(TrainingToUser::getTrainingId,trainingToUser.getTrainingId())
+                .eq(TrainingToUser::getUserId,trainingToUser.getUserId()));
+        if (list.size()>0){
+            return R.error("已存在");
+        }
+        trainingToUserService.save(trainingToUser);
+        return R.ok();
+    }
+
+
+
+
+
+}