1、公共字段自动填充
问题分析
我们已经完成了后台系统的员工管理功能开发，在新增员工时需要设置创建时间、创建人、修改时间、修改人等字段，在编辑员工时需要设置修改时间和修改人等字段。这些字段属于公共字段，也就是很多表中都有这些字段。可以使用Mybatis Plus提供的公共字段自动填充功能，对这些公共字段在某个地方统一处理。
代码实现
Mybatis Plus公共字段自动填充，也就是在插入或者更新的时候为指定字段赋予指定的值，使用它的好处就是可以统一对这些字段进行处理，避免了重复代码。
实现步骤:
1、在实体类的属性上加入@TableField注解，指定自动填充的策略
@TableField(fill = FieldFill.INSERT)
private Date createTime;
@TableField(fill = FieldFill.INSERT_UPDATE)
private Date updateTime;
@TableField(fill = FieldFill.INSERT)
private Long createUser;
@TableField(fill = FieldFill.INSERT_UPDATE)
private Long updateUser;
2、按照框架要求编写元数据对象处理器，在此类中统一为公共字段赋值，此类需要实现MetaObjectHandler接口
@Slf4j
@Component
public class MyMateObjectHandler implements MetaObjectHandler {
@Override
public void insertFill(MetaObject metaObject) {
log.info("insertFill"+metaObject.toString());
Date date = new Date();
metaObject.setValue("createTime",date);
metaObject.setValue("updateTime",date);
}
@Override
public void updateFill(MetaObject metaObject) {
log.info("updateFill"+metaObject.toString());
metaObject.setValue("updateTime",new Date());
}
}
功能完善
我们可以在LoginCheckFilter的preHandle方法中获取当前登录用户id，并调用ThreadLocal的set方法来设置当前线程的线程局部变量的值（用户id)，然后在MyMetaObjectHandler的updateFill方法中调用ThreadLocal的get方法来获得当前线程所对应的线程局部变量的值
public class BaseContext {
private static ThreadLocal<Long> threadLocal = new ThreadLocal<>();
public static void setCurrentId(Long id){
threadLocal.set(id);
}
public static Long getCurrentId(){
return threadLocal.get();
}
public static void removeCurrentId(){
threadLocal.remove();
}
}
在登录拦截器的preHandle方法中保存当前id，为了缓解服务器的内存压力，我们可以在postHandle方法中移除当前线程id。
public class LoginCheckInterceptor implements HandlerInterceptor {
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
Long id = (Long) request.getSession().getAttribute("employee");
if(id==null){
response.getWriter().write(JSON.toJSONString(Result.error("NOTLOGIN")));
return false;
}
BaseContext.setCurrentId(id);
return true;
}
public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
BaseContext.removeCurrentId();
}
}
完整的MyMateObjectHandler类如下：
@Component
public class MyMateObjectHandler implements MetaObjectHandler {
@Override
public void insertFill(MetaObject metaObject) {
Date date = new Date();
metaObject.setValue("createTime",date);
metaObject.setValue("updateTime",date);
metaObject.setValue("createUser",BaseContext.getCurrentId());
metaObject.setValue("updateUser",BaseContext.getCurrentId());
}
@Override
public void updateFill(MetaObject metaObject) {
metaObject.setValue("updateTime",new Date());
metaObject.setValue("updateUser",BaseContext.getCurrentId());
}
}
2、新增分类
需求分析
后台系统中可以管理分类信息，分类包括两种类型，分别是菜品分类和套餐分类。当我们在后台系统中添加菜品时需要选择一个菜品分类，当我们在后台系统中添加一个套餐时需要选择一个套餐分类，在移动端也会按照菜品分类和套餐分类来展示对应的菜品和套餐。
实体类
@Data
public class Category implements Serializable {
private static final long serialVersionUID = 1L;
private Long id;
private Integer type; //类型 1 菜品分类 2 套餐分类
private String name; //分类名称
private Integer sort;//顺序
@TableField(fill = FieldFill.INSERT)
private LocalDateTime createTime;
@TableField(fill = FieldFill.INSERT_UPDATE)
private LocalDateTime updateTime;
@TableField(fill = FieldFill.INSERT)
private Long createUser;
@TableField(fill = FieldFill.INSERT_UPDATE)
private Long updateUser;
private Integer isDeleted;//是否删除
}
代码开发
这里的CategoryMapper，CategoryService，CategoryServiceImpl，CategoryController省略
@PostMapping
public Result<String> add(@RequestBody Category category){
categoryService.save(category);
return Result.success("分类添加成功");
}
3、菜品分页查询
需求分析
前端接口
const getCategoryPage = (params) => {
return $axios({
url: '/category/page',
method: 'get',
params
})
}
代码开发
@GetMapping("/page")
public Result<Page> page(
@RequestParam(value = "page", defaultValue = "1") Integer page,
@RequestParam(value = "pageSize", defaultValue = "10") Integer pageSize) {
log.info("page={},pageSize={}",page,pageSize);
Page<Category> pageInfo = new Page<>(page, pageSize);
LambdaQueryWrapper<Category> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.orderByAsc(Category::getSort);
categoryService.page(pageInfo,queryWrapper);
return Result.success(pageInfo);
}
4、删除分类
需求分析
在分类管理列表页面，可以对某个分类进行删除操作。需要注意的是当分类关联了菜品或者套餐时，此分类不允许删除。
前端接口：
const deleCategory = (ids) => {
return $axios({
url: '/category',
method: 'delete',
params: { id }
})
}
代码开发
由于我们需要查找菜品的分类和套餐的分类，要编写两张表的实体类
Dish实体类：
@Data
public class Dish implements Serializable {
private static final long serialVersionUID = 1L;
private Long id;
//菜品名称
private String name;
//菜品分类id
private Long categoryId;
//菜品价格
private BigDecimal price;
//商品码
private String code;
//图片
private String image;
//描述信息
private String description;
//0 停售 1 起售
private Integer status;
//顺序
private Integer sort;
@TableField(fill = FieldFill.INSERT)
private LocalDateTime createTime;
@TableField(fill = FieldFill.INSERT_UPDATE)
private LocalDateTime updateTime;
@TableField(fill = FieldFill.INSERT)
private Long createUser;
@TableField(fill = FieldFill.INSERT_UPDATE)
private Long updateUser;
//是否删除
private Integer isDeleted;
}
Setmeal实体类：
@Data
public class Setmeal implements Serializable {
private static final long serialVersionUID = 1L;
private Long id;
//分类id
private Long categoryId;
//套餐名称
private String name;
//套餐价格
private BigDecimal price;
//状态 0:停用 1:启用
private Integer status;
//编码
private String code;
//描述信息
private String description;
//图片
private String image;
@TableField(fill = FieldFill.INSERT)
private LocalDateTime createTime;
@TableField(fill = FieldFill.INSERT_UPDATE)
private LocalDateTime updateTime;
@TableField(fill = FieldFill.INSERT)
private Long createUser;
@TableField(fill = FieldFill.INSERT_UPDATE)
private Long updateUser;
//是否删除
private Integer isDeleted;
}
此处的Mapper,Service,ServiceImpl省略
@DeleteMapping
public Result<String> deleteCategory(Long id){
categoryService.remove(id);
return Result.success("删除成功");
}
@Override
public void remove(Long id) {
LambdaQueryWrapper<Dish> dishQueryWrapper = new LambdaQueryWrapper<>();
dishQueryWrapper.eq(Dish::getCategoryId,id);//根据分类查找菜品
if (dishService.count(dishQueryWrapper)>0){//判断关联菜品数量
throw new DishException("分类已关联菜品");
}
LambdaQueryWrapper<Setmeal> setmealQueryWrapper = new LambdaQueryWrapper<>();
setmealQueryWrapper.eq(Setmeal::getCategoryId,id);
if(setmealService.count(setmealQueryWrapper)>0){
throw new SetmealException("分类已关联套餐");
}
super.removeById(id);
}
两个自定义异常类：
public class DishException extends RuntimeException {
public DishException(String message){
super(message);
}
}
public class SetmealException extends RuntimeException{
public SetmealException(String message){
super(message);
}
}
并在全局异常处理器中捕获两个异常：
@ExceptionHandler(DishException.class)
public Result<String> exceptionHandler(DishException ex){
log.info("异常信息：{}",ex.getMessage());
return Result.error(ex.getMessage());
}
@ExceptionHandler(SetmealException.class)
public Result<String> exceptionHandler(SetmealException ex){
log.info("异常信息：{}",ex.getMessage());
return Result.error(ex.getMessage());
}
5、修改分类
代码开发
@PutMapping
public Result<String> updateCategory(@RequestBody Category category){
log.info("update category:{}",category);
categoryService.updateById(category);
return Result.success("分类修改成功");
}