wangpeng/yidaima
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add front user authentication

Browse Source
This commit is contained in:
王鹏
2026-05-22 09:43:30 +08:00
parent ebb529213a
commit 276da1cb20
6 changed files with 263 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
RuoYi-Vue/ruoyi-admin/src/main/java/com/ruoyi/web/controller/front/FrontAuthController.java Normal file
View File

@@ -0,0 +1,61 @@
package com.ruoyi.web.controller.front;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
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 com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.generator.domain.front.dto.FrontLoginBody;
import com.ruoyi.generator.domain.front.dto.FrontRegisterBody;
import com.ruoyi.generator.service.front.IFrontUserService;
import com.ruoyi.web.service.front.FrontAuthService;
@RestController
@RequestMapping("/front/auth")
public class FrontAuthController extends BaseController
{
@Autowired
private FrontAuthService frontAuthService;
@Autowired
private IFrontUserService frontUserService;
@Autowired
private TokenService tokenService;
@PostMapping("/register")
public AjaxResult register(@RequestBody FrontRegisterBody body)
{
return toAjax(frontUserService.register(body));
}
@PostMapping("/login")
public AjaxResult login(@RequestBody FrontLoginBody body)
{
AjaxResult ajax = AjaxResult.success();
String token = frontAuthService.login(body.getUsername(), body.getPassword());
ajax.put(Constants.TOKEN, token);
return ajax;
}
@GetMapping("/profile")
public AjaxResult profile()
{
return AjaxResult.success(frontUserService.selectById(SecurityUtils.getUserId()));
}
@PostMapping("/logout")
public AjaxResult logout()
{
LoginUser loginUser = SecurityUtils.getLoginUser();
tokenService.delLoginUser(loginUser.getToken());
return AjaxResult.success();
}
}

58
RuoYi-Vue/ruoyi-admin/src/main/java/com/ruoyi/web/service/front/FrontAuthService.java Normal file
View File

@@ -0,0 +1,58 @@
package com.ruoyi.web.service.front;
import java.util.Collections;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.enums.UserStatus;
import com.ruoyi.common.exception.ServiceException;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.generator.domain.front.FrontUser;
import com.ruoyi.generator.service.front.IFrontUserService;
@Component
public class FrontAuthService
{
@Autowired
private IFrontUserService frontUserService;
@Autowired
private TokenService tokenService;
public String login(String username, String password)
{
FrontUser frontUser = frontUserService.selectByUsername(username);
if (frontUser == null)
{
throw new ServiceException("前台账号或密码错误");
}
if (UserStatus.DISABLE.getCode().equals(frontUser.getStatus()))
{
throw new ServiceException("前台账号已停用");
}
if (UserStatus.DELETED.getCode().equals(frontUser.getDelFlag()))
{
throw new ServiceException("前台账号已删除");
}
if (!SecurityUtils.matchesPassword(password, frontUser.getPassword()))
{
throw new ServiceException("前台账号或密码错误");
}
SysUser tokenUser = new SysUser();
tokenUser.setUserId(frontUser.getUserId());
tokenUser.setUserName(frontUser.getUsername());
tokenUser.setNickName(StringUtils.nvl(frontUser.getNickname(), frontUser.getUsername()));
tokenUser.setPassword(frontUser.getPassword());
LoginUser loginUser = new LoginUser(frontUser.getUserId(), null, tokenUser, Collections.emptySet());
loginUser.setLoginType(Constants.LOGIN_TYPE_FRONT);
String token = tokenService.createToken(loginUser);
frontUserService.updateLastLoginTime(frontUser.getUserId());
return token;
}
}

72
RuoYi-Vue/ruoyi-admin/src/test/java/com/ruoyi/web/service/front/FrontAuthServiceTest.java Normal file
View File

@@ -0,0 +1,72 @@
package com.ruoyi.web.service.front;
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.generator.domain.front.FrontUser;
import com.ruoyi.generator.service.front.IFrontUserService;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import java.lang.reflect.Field;
import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
@RunWith(MockitoJUnitRunner.class)
public class FrontAuthServiceTest
{
private FrontAuthService frontAuthService;
@Mock
private IFrontUserService frontUserService;
@Mock
private TokenService tokenService;
@Before
public void setUp() throws Exception
{
frontAuthService = new FrontAuthService();
setField("frontUserService", frontUserService);
setField("tokenService", tokenService);
}
@Test
public void loginCreatesFrontToken()
{
FrontUser user = new FrontUser();
user.setUserId(100L);
user.setUsername("demo");
user.setNickname("Demo");
user.setPassword(SecurityUtils.encryptPassword("admin123"));
user.setStatus("0");
user.setDelFlag("0");
when(frontUserService.selectByUsername("demo")).thenReturn(user);
when(tokenService.createToken(any(LoginUser.class))).thenAnswer(invocation -> {
LoginUser loginUser = invocation.getArgument(0);
assertEquals(Constants.LOGIN_TYPE_FRONT, loginUser.getLoginType());
assertEquals(Long.valueOf(100L), loginUser.getUserId());
return "front-token";
});
String token = frontAuthService.login("demo", "admin123");
assertEquals("front-token", token);
verify(frontUserService).updateLastLoginTime(100L);
}
private void setField(String name, Object value) throws Exception
{
Field field = FrontAuthService.class.getDeclaredField(name);
field.setAccessible(true);
field.set(frontAuthService, value);
}
}