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 故需要將其轉(zhuǎn)化為實體類，json中的類名與實體類名對應(yīng)才可以封裝
     *          A. 由于需求分析時, 我們看到前端發(fā)起的請求為post請求, 所以服務(wù)端需要使用注解 @PostMapping
     *          B. 由于前端傳遞的請求參數(shù)為json格式的數(shù)據(jù), 這里使用Employee對象接收, 但是將json格式數(shù)據(jù)封裝到實體類中, 在形參前需要加注解@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());
        //  ②. 根據(jù)頁面提交的用戶名username查詢數(shù)據(jù)庫中員工數(shù)據(jù)信息
        LambdaQueryWrapper<Employee> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(Employee::getUsername,employee.getUsername());
        Employee emp = employeeService.getOne(queryWrapper); //在數(shù)據(jù)庫中用戶名是唯一的，所以可以用getone()
        //  ③. 如果沒有查詢到, 則返回登錄失敗結(jié)果
        if (emp == null){
            return R.error("用戶名不存在");
        }
        //  ④. 密碼比對，如果不一致, 則返回登錄失敗結(jié)果
        if(!emp.getPassword().equals(password)){
            return R.error("密碼失敗");
        }
        //  ⑤. 查看員工狀態(tài)，如果為已禁用狀態(tài)，則返回員工已禁用結(jié)果
        if(emp.getStatus() == 0){
            return R.error("賬戶已禁用");
        }
        //  ⑥. 登錄成功，將員工id存入Session, 并返回登錄成功結(jié)果
        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");
        //  返回結(jié)果
        return R.success("退出成功");
    }

    @PostMapping
    public R<String> save(HttpServletRequest request,@RequestBody Employee employee){
        /**@Description: 用于保存用戶員工信息。
         *
         * B. 在組裝員工信息時, 還需要封裝創(chuàng)建時間、修改時間，創(chuàng)建人、修改人信息(從session中獲取當前登錄用戶)。
         * @author LiBiGo
         * @date 2022/8/12 17:25
         */
        log.info("新增員工，員工信息：{}",employee.toString());
        // A. 在新增員工時，設(shè)置初始密碼，需要進行MD5加密處理
        employee.setPassword(DigestUtils.md5DigestAsHex("123456".getBytes()));
        // B. 在組裝員工信息時, 需封裝創(chuàng)建時間、修改時間，創(chuàng)建人、修改人信息(從session中獲取當前登錄用戶)。
        employee.setCreateTime(LocalDateTime.now());
        employee.setUpdateTime(LocalDateTime.now());

        //獲得當前登錄員工用戶的id
        Long empId = (Long) request.getSession().getAttribute("employee");

        employee.setCreateUser(empId); // 創(chuàng)建員工信息的人
        employee.setUpdateUser(empId); // 最后一次更新信息的人

        employeeService.save(employee); // 因為employeeService繼承了，所以不用寫，直接用
        log.info("新增員工成功");
        return R.success("新增員工成功");
    }
}

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的函數(shù)異常 //指定攔截那些類型的控制器;
@ResponseBody   //將方法的返回值 R 對象轉(zhuǎn)換為json格式的數(shù)據(jù), 響應(yīng)給頁面;
@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("未知錯誤");
    }
}