package com.kwan.springbootkwan.controller; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.kwan.springbootkwan.entity.CsdnUserInfo; import com.kwan.springbootkwan.entity.Result; import com.kwan.springbootkwan.entity.dto.CsdnUserInfoDTO; import com.kwan.springbootkwan.entity.query.CsdnUserInfoQuery; import com.kwan.springbootkwan.service.CsdnUserInfoService; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; import java.util.Objects; @Slf4j @RestController @RequestMapping("/csdn/user") public class CsdnUserController { @Autowired private CsdnUserInfoService csdnUserInfoService; /** * 分页查询所有数据 * * @return 所有数据 */ @PostMapping("/page") public Result selectAll(@RequestBody CsdnUserInfoQuery query) { final Integer userWeight = query.getUserWeight(); final String nickName = query.getNickName(); final String userName = query.getUserName(); Page pageParm = new Page<>(); pageParm.setCurrent(query.getPage()); pageParm.setSize(query.getPageSize()); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.orderByDesc("user_weight"); wrapper.eq("is_delete", 0); if (StringUtils.isNotEmpty(userName)) { wrapper.eq("user_name", userName); } if (Objects.nonNull(userWeight)) { wrapper.eq("user_weight", userWeight); } if (StringUtils.isNotEmpty(nickName)) { wrapper.like("nick_name", nickName); } return Result.ok(CsdnUserInfoDTO.Converter.INSTANCE.from(this.csdnUserInfoService.page(pageParm, wrapper))); } /** * 新增用户 * * @return 所有数据 */ @PostMapping("/add") public Result add(@RequestBody CsdnUserInfoQuery addInfo) { final String userName = addInfo.getUserName(); final Integer addType = addInfo.getAddType(); if (StringUtils.isEmpty(userName)) { return Result.error("名字不能为空"); } //批量添加 if (addType == 1) { final String[] split = userName.split("\n"); for (String str : split) { str = str.trim(); if (StringUtils.isEmpty(str)) { continue; } CsdnUserInfo csdnUserInfo = new CsdnUserInfo(); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("user_name", str); wrapper.eq("is_delete", 0); final CsdnUserInfo one = this.csdnUserInfoService.getOne(wrapper); if (one == null) { BeanUtils.copyProperties(addInfo, csdnUserInfo); csdnUserInfo.setUserName(str); this.csdnUserInfoService.save(csdnUserInfo); } } } else { CsdnUserInfo csdnUserInfo = new CsdnUserInfo(); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("user_name", userName); wrapper.eq("is_delete", 0); final CsdnUserInfo one = this.csdnUserInfoService.getOne(wrapper); if (one == null) { BeanUtils.copyProperties(addInfo, csdnUserInfo); this.csdnUserInfoService.save(csdnUserInfo); return Result.ok(); } else { return Result.error("该用户已存在"); } } return Result.ok(); } /** * 更新用户 * * @param query * @return */ @PostMapping("/update") public Result update(@RequestBody CsdnUserInfoQuery query) { CsdnUserInfo csdnUserInfo = new CsdnUserInfo(); csdnUserInfo.setId(query.getId()); csdnUserInfo.setUserName(query.getUserName()); csdnUserInfo.setNickName(query.getNickName()); csdnUserInfo.setUserWeight(query.getUserWeight()); return Result.ok(this.csdnUserInfoService.updateById(csdnUserInfo)); } /** * 删除用户 * * @param id * @return */ @GetMapping("/delete") public Result delete(@RequestParam("id") Integer id) { CsdnUserInfo csdnUserInfo = new CsdnUserInfo(); csdnUserInfo.setIsDelete(1); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("id", id); return Result.ok(this.csdnUserInfoService.update(csdnUserInfo, wrapper)); } }