修改功能
SetmealController
/**
* 根据id查询套餐信息和对应的菜品信息
* @param id
* @return
*/
@GetMapping("/{id}")
public R<SetmealDto> get(@PathVariable Long id){
SetmealDto setmealDto = setmealService.getByIdWithDishes(id);
return R.success(setmealDto);
}
/**
* 更新套餐信息,同时更新对应的菜品信息
* @param setmealDto
* @return
*/
@PutMapping
public R<String> update(@RequestBody SetmealDto setmealDto){
log.info(setmealDto.toString());
setmealService.updateWithDishes(setmealDto);
return R.success("修改菜品成功");
}
SetmealService
package com.itheima.reggie.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.itheima.reggie.dto.SetmealDto;
import com.itheima.reggie.entity.Setmeal;
import java.util.List;
/**
* @author snape
* @create 2022-05-24 18:20
*/
public interface SetmealService extends IService<Setmeal> {
//根据id查询套餐信息和对应的菜品信息
public SetmealDto getByIdWithDishes(Long id);
//更新套餐信息,同时更新对应的菜品信息
public void updateWithDishes(SetmealDto setmealDto);
}
SetmealServiceImpl
package com.itheima.reggie.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.reggie.common.CustomException;
import com.itheima.reggie.dto.SetmealDto;
import com.itheima.reggie.entity.Setmeal;
import com.itheima.reggie.entity.SetmealDish;
import com.itheima.reggie.mapper.SetmealMapper;
import com.itheima.reggie.service.SetmealDishService;
import com.itheima.reggie.service.SetmealService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.List;
import java.util.stream.Collectors;
/**
* @author snape
* @create 2022-05-24 18:25
*/
@Service
@Slf4j
public class SetmealServiceImpl extends ServiceImpl<SetmealMapper, Setmeal> implements SetmealService {
@Autowired
private SetmealDishService setmealDishService;
/**
* 根据id查询套餐信息和对应的菜品信息
* @param id
* @return
*/
@Override
public SetmealDto getByIdWithDishes(Long id) {
//查询套餐基本信息,从stemeal表查询
Setmeal setmeal = super.getById(id);
SetmealDto setmealDto = new SetmealDto();
BeanUtils.copyProperties(setmeal,setmealDto);
//查询当前套餐对应的菜品信息,从stemeal_dish表查询
LambdaQueryWrapper<SetmealDish> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(SetmealDish::getSetmealId,id);
List<SetmealDish> setmealDishes = setmealDishService.list(queryWrapper);
setmealDto.setSetmealDishes(setmealDishes);
return setmealDto;
}
/**
* 更新套餐信息,同时更新对应的菜品信息
* @param setmealDto
*/
@Override
@Transactional
public void updateWithDishes(SetmealDto setmealDto) {
//更新setmeal表基本信息
super.updateById(setmealDto);
//清理当前套餐对应菜品数据---setmeal_dish表的delete操作
LambdaQueryWrapper<SetmealDish> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(SetmealDish::getSetmealId,setmealDto.getId());
setmealDishService.remove(queryWrapper);
//添加当前提交过来的菜品数据---setmeal_dish表的insert操作
List<SetmealDish> setmealDishes = setmealDto.getSetmealDishes();
setmealDishes = setmealDishes.stream().map((item) -> {
item.setSetmealId(setmealDto.getId());
return item;
}).collect(Collectors.toList());
setmealDishService.saveBatch(setmealDishes);
}
}