JavaWeb_瑞吉外卖_业务开发Day5-套餐管理, 短信发送, 手机验证码登录

套餐管理

新增套餐

@Override
@Transactional
public void saveWithDish(SetmealDto setmealDto) {
    this.save(setmealDto);

    // 获取套餐id
    Long setmealId = setmealDto.getId();

    // 套餐所包含的菜品
    List<SetmealDish> dishes = setmealDto.getSetmealDishes();
    dishes.stream().map((item)->{
        item.setSetmealId(setmealId);
        return item;
    }).collect(Collectors.toList());

    // 保存套餐的菜品数据到套餐菜品表setmeal_dish
    setmealDishService.saveBatch(dishes);
}

套餐分页查询

@GetMapping("/page")
public R<Page> page(int page, int pageSize, String name)
{
    log.info("page={}, pageSize={}, name={}", page, pageSize, name);

    Page pageInfo = new Page(page, pageSize);
    Page setmealDtoPage = new Page();
    LambdaQueryWrapper<Setmeal> queryWrapper = new LambdaQueryWrapper<>();
    queryWrapper.like(name!=null, Setmeal::getName, name);
    queryWrapper.orderByDesc(Setmeal::getUpdateTime);
    setmealService.page(pageInfo, queryWrapper);

    BeanUtils.copyProperties(pageInfo, setmealDtoPage, "records");

    List<Setmeal> records = pageInfo.getRecords();
    List<SetmealDto> setmealDtoList = records.stream().map((item) -> {

        SetmealDto setmealDto = new SetmealDto();
        BeanUtils.copyProperties(item, setmealDto);

        // 获取套餐分类id
        Long categoryId = item.getCategoryId();
        // 获取套餐分类对象
        Category category = categoryService.getById(categoryId);
        if(category != null) {
            // 获取套餐分类名称
            String categoryName = category.getName();
            setmealDto.setCategoryName(categoryName);
        }
        return setmealDto;
    }).collect(Collectors.toList());
    setmealDtoPage.setRecords(setmealDtoList);
    return R.success(setmealDtoPage);
}

修改套餐

// Service 数据回显
public SetmealDto getByIdWithDish(Long id) {
    SetmealDto setmealDto = new SetmealDto();
    // 获取套餐信息
    Setmeal setmeal = this.getById(id);
    // 获取对应的菜品信息
    LambdaQueryWrapper<SetmealDish> queryWrapper = new LambdaQueryWrapper<>();
    queryWrapper.eq(SetmealDish::getSetmealId, id);
    List<SetmealDish> setmealDishList = setmealDishService.list(queryWrapper);

    BeanUtils.copyProperties(setmeal, setmealDto);
    setmealDto.setSetmealDishes(setmealDishList);
    return setmealDto;
}
// Service 修改套餐
@Transactional
public void updateWithDish(SetmealDto setmealDto) {
    // 更新套餐信息
    this.updateById(setmealDto);
    // 先删除对应的菜品
    //   获取套餐的id
    Long setmealDtoId = setmealDto.getId();
    //   删除对应的菜品
    LambdaQueryWrapper<SetmealDish> queryWrapper = new LambdaQueryWrapper<>();
    queryWrapper.eq(SetmealDish::getSetmealId, setmealDtoId);
    setmealDishService.remove(queryWrapper);

    // 再添加提交的菜品
    List<SetmealDish> setmealDishList = setmealDto.getSetmealDishes();
    setmealDishList.stream().map((item)->{
        item.setSetmealId(setmealDtoId);
        return item;
    }).collect(Collectors.toList());
    setmealDishService.saveBatch(setmealDishList);

}

删除套餐

// Controller
@DeleteMapping
public R<String> remove(@RequestParam List<Long> ids)
{
    log.info("ids: {}", ids);
    setmealService.removeWithDish(ids);
    return R.success("删除套餐成功");
}
// Service
@Transactional
public void removeWithDish(List<Long> ids) {

    // 查询套餐状态, 确定是否可用删除
    LambdaQueryWrapper<Setmeal> queryWrapper = new LambdaQueryWrapper();
    queryWrapper.in(Setmeal::getId, ids);
    queryWrapper.eq(Setmeal::getStatus, 1);
    int count = this.count(queryWrapper);
    if(count > 0){
        throw new CustomException("套餐正在售卖中, 不能删除");
    }
    // 删除套餐表数据
    this.removeByIds(ids);
    // 删除套餐菜品表数据
    LambdaQueryWrapper<SetmealDish> dishLambdaQueryWrapper = new LambdaQueryWrapper<>();
    dishLambdaQueryWrapper.in(SetmealDish::getSetmealId, ids);
    setmealDishService.remove(dishLambdaQueryWrapper);
}

总结

1. 接收List数据时, 需要加上@RequestParam注解
2. 操作2个及2个以上表, 加上@Transactional事务注解, 保持数据的一致性.

修改套餐售卖状态

@PostMapping("/status/{status}")
public R<String> updateStatus(@PathVariable int status, @RequestParam List<Long> ids)
{
    LambdaUpdateWrapper<Setmeal> updateWrapper = new LambdaUpdateWrapper<>();
    updateWrapper.in(Setmeal::getId, ids);
    updateWrapper.set(Setmeal::getStatus, status);
    setmealService.update(updateWrapper);
    return R.success("修改售卖状态成功");
}

短信发送

工具类

发送短信

/**
 * 短信发送工具类
 */
public class SMSUtils {

	/**
	 * 发送短信
	 * @param signName 签名
	 * @param templateCode 模板
	 * @param phoneNumbers 手机号
	 * @param param 参数
	 */
	public static void sendMessage(String signName, String templateCode,String phoneNumbers,String param){
		DefaultProfile profile = DefaultProfile.getProfile("cn-hangzhou", "xxx", "xxx");
		IAcsClient client = new DefaultAcsClient(profile);

		SendSmsRequest request = new SendSmsRequest();
		request.setSysRegionId("cn-hangzhou");
		request.setPhoneNumbers(phoneNumbers);
		request.setSignName(signName);
		request.setTemplateCode(templateCode);
		request.setTemplateParam("{\"code\":\""+param+"\"}");
		try {
			SendSmsResponse response = client.getAcsResponse(request);
			System.out.println("短信发送成功");
		}catch (ClientException e) {
			e.printStackTrace();
		}
	}

}

生成验证码

/**
 * 随机生成验证码工具类
 */
public class ValidateCodeUtils {
    /**
     * 随机生成验证码
     * @param length 长度为4位或者6位
     * @return
     */
    public static Integer generateValidateCode(int length){
        Integer code =null;
        if(length == 4){
            code = new Random().nextInt(9999);//生成随机数，最大为9999
            if(code < 1000){
                code = code + 1000;//保证随机数为4位数字
            }
        }else if(length == 6){
            code = new Random().nextInt(999999);//生成随机数，最大为999999
            if(code < 100000){
                code = code + 100000;//保证随机数为6位数字
            }
        }else{
            throw new RuntimeException("只能生成4位或6位数字验证码");
        }
        return code;
    }

    /**
     * 随机生成指定长度字符串验证码
     * @param length 长度
     * @return
     */
    public static String generateValidateCode4String(int length){
        Random rdm = new Random();
        String hash1 = Integer.toHexString(rdm.nextInt());
        String capstr = hash1.substring(0, length);
        return capstr;
    }
}

发送短信验证码

@PostMapping("/sendMsg")
public R<String> sendMsg(@RequestBody User user, HttpSession session){
    // 获取手机号
    String phone = user.getPhone();

    if(StringUtils.isNotEmpty(phone)){
        // 生成随机的4位验证码
        String code = ValidateCodeUtils.generateValidateCode(4).toString();
        log.info("code: {}", code);
        // 调用阿里云提供的短信服务API完成发送短信
        // SMSUtils.sendMessage("xxx", "xxx", phone, code);
        // 需要将生成的验证码保存到Session
        session.setAttribute(phone, code);

        return R.success("手机验证码短信发送成功");
    }
    return R.error("短信发送失败");
}

手机验证码登录

@PostMapping("/login")
public R<User> login(@RequestBody Map map, HttpSession session)
{
    // 获取手机号
    String phone = map.get("phone").toString();
    // 获取验证码
    String code = map.get("code").toString();

    // 从Session中获取保存的验证码
    Object codeInSession = session.getAttribute(phone);

    // 验证码比对
    if(codeInSession != null && codeInSession.equals(code)){
        // 登录成功
        // 判断当前手机号对应的用户是否位新用户, 如果是新用户就自动完成注册
        LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(User::getPhone, phone);

        User user = userService.getOne(queryWrapper);
        if(user == null){
            // 新用户注册
            user = new User();
            user.setPhone(phone);
            userService.save(user);
        }
        session.setAttribute("user", user.getId());
        return R.success(user);
    }
    return R.error("登录失败");
}

来源

黑马程序员. 瑞吉外卖项目