Javassm - item1-6 (菜品管理模块2)

package com.sky.dto;
import lombok.Data;
import java.io.Serializable;
@Data
public class DishPageQueryDTO implements Serializable {
    private int page;
    private int pageSize;
    private String name;
    private Integer categoryId; //分类id
    private Integer status; //状态 0表示禁用 1表示启用
}

package com.sky.vo;

import com.sky.entity.DishFlavor;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.io.Serializable;
import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class DishVO implements Serializable {

    private Long id;
    //菜品名称
    private String name;
    //菜品分类id
    private Long categoryId;
    //菜品价格
    private BigDecimal price;
    //图片
    private String image;
    //描述信息
    private String description;
    //0 停售 1 起售
    private Integer status;
    //更新时间
    private LocalDateTime updateTime;
    //分类名称
    private String categoryName;
    //菜品关联的口味
    private List<DishFlavor> flavors = new ArrayList<>();
}

  /**
     * 菜品分页查询
     *
     * @param dishPageQueryDTO
     * @return
     */
    @GetMapping("/page")
    @ApiOperation("菜品分页查询")
    public Result<PageResult> page(DishPageQueryDTO dishPageQueryDTO) {
        log.info("菜品分页查询:{}", dishPageQueryDTO);
        PageResult pageResult = dishService.pageQuery(dishPageQueryDTO);//后绪步骤定义
        return Result.success(pageResult);
    }

/**
  * 菜品分页查询
  *
  * @param dishPageQueryDTO
  * @return
  */
PageResult pageQuery(DishPageQueryDTO dishPageQueryDTO);

/**
* 菜品分页查询
*
* @param dishPageQueryDTO
* @return
*/
public PageResult pageQuery(DishPageQueryDTO dishPageQueryDTO) {
    PageHelper.startPage(dishPageQueryDTO.getPage(), dishPageQueryDTO.getPageSize());
    Page<DishVO> page = dishMapper.pageQuery(dishPageQueryDTO);//后绪步骤实现
    return new PageResult(page.getTotal(), page.getResult());
}

/**
  * 菜品分页查询
  *
  * @param dishPageQueryDTO
  * @return
  */
Page<DishVO> pageQuery(DishPageQueryDTO dishPageQueryDTO);

/**
 * 菜品批量删除
 *
 * @param ids
 * @return
 */
@DeleteMapping
@ApiOperation("菜品批量删除")
public Result delete(@RequestParam List<Long> ids) {
    log.info("菜品批量删除：{}", ids);
    dishService.deleteBatch(ids);//后绪步骤实现
    return Result.success();
}

/**
 * 菜品批量删除
 *
 * @param ids
 */
void deleteBatch(List<Long> ids);

@Autowired
private SetmealDishMapper setMealDishMapper;
/**
 * 菜品删除
 * @param ids
 */
@Override
@Transactional
public void deleteBatch(List<Long> ids) {
  //判断当前菜品是否能够删除---是否存在起售中的菜品
  for(Long id : ids) {
      Dish dish = dishMapper.getById(id);
      if(Objects.equals(dish.getStatus(), StatusConstant.ENABLE))
          throw new DeletionNotAllowedException(MessageConstant.DISH_ON_SALE);
  }

  //判断当前菜品是否能够删除---是否被套餐关联了
  List<Long> setMealIds = setMealDishMapper.getSetMealIdsByDishIds(ids);
  if(setMealIds != null && !setMealIds.isEmpty())
      throw new DeletionNotAllowedException(MessageConstant.DISH_BE_RELATED_BY_SETMEAL);

  //删除菜品表中的菜品数据
  dishMapper.deleteByIds(ids);
  dishFlavorMapper.deleteByDishIds(ids);
}

/**
 * 根据主键查询菜品
  *
  * @param id
  * @return
  */
@Select("select * from dish where id = #{id}")
Dish getById(Long id);

package com.sky.mapper;
import org.apache.ibatis.annotations.Mapper;
import java.util.List;
@Mapper
public interface SetMealDishMapper {
    // select id from setmeal_dish where dish_id in (x,x,x);
    List<Long> getSetMealIdsByDishIds(List<Long> ids);
}

/**
 * 根据id查询菜品
 *
 * @param id
 * @return
 */
@GetMapping("/{id}")
@ApiOperation("根据id查询菜品")
public Result<DishVO> getById(@PathVariable Long id) {
  log.info("根据id查询菜品：{}", id);
  DishVO dishVO = dishService.getByIdWithFlavor(id);//后绪步骤实现
  return Result.success(dishVO);
}

/**
 * 根据id查询菜品和对应的口味数据
 *
 * @param id
 * @return
 */
DishVO getByIdWithFlavor(Long id);

/**
 * 根据id查询菜品和对应的口味数据
 *
 * @param id
 * @return
 */
public DishVO getByIdWithFlavor(Long id) {
  //根据id查询菜品数据
  Dish dish = dishMapper.getById(id);

  //根据菜品id查询口味数据
  List<DishFlavor> dishFlavors = dishFlavorMapper.getByDishId(id);//后绪步骤实现

  //将查询到的数据封装到VO
  DishVO dishVO = new DishVO();
  BeanUtils.copyProperties(dish, dishVO);
  dishVO.setFlavors(dishFlavors);

  return dishVO;
}

/**
 * 根据菜品id查询对应的口味数据
 * @param dishId
 * @return
 */
@Select("select * from dish_flavor where dish_id = #{dishId}")
List<DishFlavor> getByDishId(Long dishId);

/**
 * 修改菜品
 * @param dishDTO
 * @return
 */
@PutMapping
@ApiOperation("修改菜品")
public Result<Void> update(@RequestBody DishDTO dishDTO) {
  log.info("修改菜品: {}", dishDTO);
  dishService.updateWithFlavor(dishDTO);
  return Result.success();
}

/**
 * 修改菜品
 * @param dishDTO
 */
void updateWithFlavor(DishDTO dishDTO);

@Override
@Transactional
public void updateWithFlavor(DishDTO dishDTO) {
    Dish dish = new Dish();
    BeanUtils.copyProperties(dishDTO, dish);

    // 修改菜品基本信息
    dishMapper.update(dish);

    // 删除原有口味
    dishFlavorMapper.deleteByDishIds(List.of(dish.getId()));

    // 重新插入口味数据
    List<DishFlavor> flavors = dishDTO.getFlavors();
    if (flavors != null && !flavors.isEmpty()) {
        flavors.forEach(dishFlavor -> {
            dishFlavor.setDishId(dishDTO.getId());
        });
        //向口味表插入n条数据
        dishFlavorMapper.insertBatch(flavors);
    }
}

/**
 * 根据id动态修改菜品数据
 *
 * @param dish
 */
@AutoFill(value = OperationType.UPDATE)
void update(Dish dish);