编程宝库 - 技术改变世界

编程宝库

Search

Java编程技术

Spring Boot项目抵御XSS攻击实战过程

如何将Java与C#时间进行互相转换

如何基于Springboot完成新增员工功能并设置全局异常处理器

1. 新增员工

1.1 需求分析

后台系统中可以管理员工信息，通过新增员工来添加后台系统用户。点击[添加员工]按钮跳转到新增页面，如下

当填写完表单信息, 点击"保存"按钮后, 会提交该表单的数据到服务端, 在服务端中需要接受数据, 然后将数据保存至数据库中。

1.2 数据模型

新增员工，其实就是将我们新增页面录入的员工数据插入到employee表。employee表中的status字段已经设置了默认值1，表示状态正常。

需要注意，employee表中对username字段加入了唯一约束，因为username是员工的登录账号，必须是唯一的。

1.3 程序执行流程

A. 点击"保存"按钮, 页面发送ajax请求，将新增员工页面中输入的数据以json的形式提交到服务端, 请求方式POST, 请求路径 /employee

B. 服务端Controller接收页面提交的数据并调用Service将数据进行保存

C. Service调用Mapper操作数据库，保存数据

1.4 代码实现

package com.itheima.reggie.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.itheima.reggie.common.R;
import com.itheima.reggie.entity.Employee;
import com.itheima.reggie.service.EmployeeService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.DigestUtils;
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.RestController;

import javax.servlet.http.HttpServletRequest;
import java.time.LocalDateTime;

@Slf4j
@RestController
@RequestMapping("/employee")
public class EmployeeController {
  @Autowired
  private EmployeeService employeeService;

  @PostMapping("/login")
  public R<Employee> login(HttpServletRequest request, @RequestBody Employee employee){
  /**@Description: 员工登陆
   * @param @RequestBody 传入的是json 故需要将其转化为实体类，json中的类名与实体类名对应才可以封装
   *          A. 由于需求分析时, 我们看到前端发起的请求为post请求, 所以服务端需要使用注解 @PostMapping
   *          B. 由于前端传递的请求参数为json格式的数据, 这里使用Employee对象接收, 但是将json格式数据封装到实体类中, 在形参前需要加注解@RequestBody
   * @return com.itheima.reggie.common.R<com.itheima.reggie.entity.Employee>
   * @version v1.0
   * @author LiBiGo
   * @date 2022/8/12 11:32
   */
      //  ①. 将页面提交的密码password进行md5加密处理, 得到加密后的字符串
      String password = employee.getPassword();
      password = DigestUtils.md5DigestAsHex(password.getBytes());
      //  ②. 根据页面提交的用户名username查询数据库中员工数据信息
      LambdaQueryWrapper<Employee> queryWrapper = new LambdaQueryWrapper<>();
      queryWrapper.eq(Employee::getUsername,employee.getUsername());
      Employee emp = employeeService.getOne(queryWrapper); //在数据库中用户名是唯一的，所以可以用getone()
      //  ③. 如果没有查询到, 则返回登录失败结果
      if (emp == null){
          return R.error("用户名不存在");
      }
      //  ④. 密码比对，如果不一致, 则返回登录失败结果
      if(!emp.getPassword().equals(password)){
          return R.error("密码失败");
      }
      //  ⑤. 查看员工状态，如果为已禁用状态，则返回员工已禁用结果
      if(emp.getStatus() == 0){
          return R.error("账户已禁用");
      }
      //  ⑥. 登录成功，将员工id存入Session, 并返回登录成功结果
      request.getSession().setAttribute("employee",emp.getId());
      return R.success(emp);
  }

  @PostMapping("/logout")
  public R<String> logout(HttpServletRequest request){
  /**@Description:  /employee/logout 员工退出
   * @author LiBiGo
   * @date 2022/8/12 12:09
   */
      //  清理Session中保存的当前登录员工的id
      request.getSession().removeAttribute("employee");
      //  返回结果
      return R.success("退出成功");
  }

  @PostMapping
  public R<String> save(HttpServletRequest request,@RequestBody Employee employee){
      /**@Description: 用于保存用户员工信息。
       *
       * B. 在组装员工信息时, 还需要封装创建时间、修改时间，创建人、修改人信息(从session中获取当前登录用户)。
       * @author LiBiGo
       * @date 2022/8/12 17:25
       */
      log.info("新增员工，员工信息：{}",employee.toString());
      // A. 在新增员工时，设置初始密码，需要进行MD5加密处理
      employee.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes()));
      // B. 在组装员工信息时, 需封装创建时间、修改时间，创建人、修改人信息(从session中获取当前登录用户)。
      employee.setCreateTime(LocalDateTime.now());
      employee.setUpdateTime(LocalDateTime.now());

      //获得当前登录员工用户的id
      Long empId = (Long) request.getSession().getAttribute("employee");

      employee.setCreateUser(empId); // 创建员工信息的人
      employee.setUpdateUser(empId); // 最后一次更新信息的人

      employeeService.save(employee); // 因为employeeService继承了，所以不用写，直接用
      log.info("新增员工成功");
      return R.success("新增员工成功");
  }
}

2 全局异常处理

2.1 新增员工存在的问题

当我们在测试中，添加用户时，输入了一个已存在的用户名时，前端界面出现错误提示信息：

而此时，服务端已经报错了，报错信息如下：

出现上述的错误，主要就是因为在 employee 表结构中，我们针对于username字段，建立了唯一索引，添加重复的username数据时，违背该约束，就会报错。但是此时前端提示的信息并不具体，用户并不知道是因为什么原因造成的该异常，我们需要给用户提示详细的错误信息。

2.2全局异常处理思路

我们需要对程序中可能出现的异常进行捕获，通常有两种处理方式：

A. 在Controller方法中加入 try...catch 进行异常捕获

形式如下：

如果采用这种方式，虽然可以解决，但是存在弊端，需要我们在保存其他业务数据时，也需要在Controller方法中加上try...catch进行处理，代码冗余，不通用。

B. 使用异常处理器进行全局异常捕获

采用这种方式来实现，只需要在项目中定义一个通用的全局异常处理器，就可以解决项目的所有异常。

2.3全局异常处理器

在项目中自定义一个全局异常处理器，在异常处理器上加上注解 @ControllerAdvice,可以通过属性annotations指定拦截哪一类的Controller方法。并在异常处理器的方法上加上注解 @ExceptionHandler 来指定拦截的是那一类型的异常。

异常处理方法逻辑:

指定捕获的异常类型为 SQLIntegrityConstraintViolationException
解析异常的提示信息, 获取出是那个值违背了唯一约束
组装错误信息并返回

2.4 全局异常处理器代码实现

package com.itheima.reggie.common;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import java.sql.SQLIntegrityConstraintViolationException;
/**
* Description: 全局异常处理器
*/

@ControllerAdvice(annotations = {RestController.class, Controller.class})// 处理@RestController、@Controller的函数异常 //指定拦截那些类型的控制器;
@ResponseBody   //将方法的返回值 R 对象转换为json格式的数据, 响应给页面;
@Slf4j
public class GlobalExceptionHandler {
  /**@Description: 异常处理方法
   * @author LiBiGo
   * @date 2022/8/12 17:46
   */
  @ExceptionHandler(SQLIntegrityConstraintViolationException.class) // 处理指定异常类
  public R<String> exceptionHandler(SQLIntegrityConstraintViolationException ex){
      log.error(ex.getMessage());
      //Duplicate entry '299067' for key 'idx_username'
      if(ex.getMessage().contains("Duplicate entry")){
          String[] split = ex.getMessage().split(" "); //提取重复字段，即哪个用户名重复 从0开始第2个即为用户名
          String msg = split[2] + "已存在";
          return R.error(msg);
      }
      return R.error("未知错误");
  }
}

2.5测试

全局异常处理器编写完毕后将项目重启, 完毕之后直接访问管理系统首页, 点击 "员工管理" 页面中的 "添加员工" 按钮。当添加用户时，输入了一个已存在的用户名时，前端界面出现如下错误提示信息：

总结

关于如何基于Springboot完成新增员工功能并设置全局异常处理器的文章就介绍至此，更多相关Springboot新增员工并设置全局异常内容请搜索编程宝库以前的文章，希望以后支持编程宝库！

下一节：如何将Java与C#时间进行互相转换

Java编程技术

JAVA中的日期计算方法JAVA统计从1970年1月1日起的毫秒的数量表示日期。也就是说，例如，1970年1月2日，是在1月1日后的86，400，000毫秒。同样的，1969年12月31日 ...

Spring Boot项目抵御XSS攻击实战过程

如何将Java与C#时间进行互相转换