AdminAuthServiceImpl.java
3.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package com.diligrp.rider.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.diligrp.rider.common.exception.BizException;
import com.diligrp.rider.config.JwtUtil;
import com.diligrp.rider.dto.AdminLoginDTO;
import com.diligrp.rider.entity.AdminUser;
import com.diligrp.rider.entity.Substation;
import com.diligrp.rider.mapper.AdminUserMapper;
import com.diligrp.rider.mapper.SubstationMapper;
import com.diligrp.rider.vo.AdminLoginVO;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;
import java.nio.charset.StandardCharsets;
@Service
@RequiredArgsConstructor
public class AdminAuthServiceImpl {
private final AdminUserMapper adminUserMapper;
private final SubstationMapper substationMapper;
private final JwtUtil jwtUtil;
/**
* 统一登录入口
* role=admin：超级管理员登录（admin_user表）
* role=substation：分站管理员登录（substation表）
*/
public AdminLoginVO login(AdminLoginDTO dto) {
if ("admin".equals(dto.getRole())) {
return loginAdmin(dto.getAccount(), dto.getPass());
}
return loginSubstation(dto.getAccount(), dto.getPass());
}
private AdminLoginVO loginAdmin(String account, String pass) {
AdminUser user = adminUserMapper.selectOne(new LambdaQueryWrapper<AdminUser>()
.eq(AdminUser::getUserLogin, account).last("LIMIT 1"));
if (user == null) throw new BizException("账号不存在");
if (!encryptPass(pass).equals(user.getUserPass())) throw new BizException("密码错误");
if (user.getUserStatus() == null || user.getUserStatus() == 0) throw new BizException("账号已被禁用");
AdminLoginVO vo = new AdminLoginVO();
vo.setId(user.getId());
vo.setUserLogin(user.getUserLogin());
vo.setUserNickname(user.getUserNickname());
vo.setRole("admin");
vo.setToken(jwtUtil.generateAdminToken(user.getId(), "admin"));
return vo;
}
private AdminLoginVO loginSubstation(String account, String pass) {
Substation sub = substationMapper.selectOne(new LambdaQueryWrapper<Substation>()
.eq(Substation::getUserLogin, account).last("LIMIT 1"));
if (sub == null) throw new BizException("账号不存在");
if (!encryptPass(pass).equals(sub.getUserPass())) throw new BizException("密码错误");
if (sub.getUserStatus() == null || sub.getUserStatus() == 0) throw new BizException("账号已被禁用");
AdminLoginVO vo = new AdminLoginVO();
vo.setId(sub.getId());
vo.setUserLogin(sub.getUserLogin());
vo.setUserNickname(sub.getUserNickname());
vo.setRole("substation");
vo.setCityId(sub.getCityId());
vo.setToken(jwtUtil.generateAdminToken(sub.getId(), "substation"));
return vo;
}
private String encryptPass(String pass) {
return DigestUtils.md5DigestAsHex(pass.getBytes(StandardCharsets.UTF_8));
}
}