Skip to content
This repository was archived by the owner on Nov 21, 2020. It is now read-only.

Commit 39f9b19

Browse files
committed
feat: add method get token in service model
1 parent e0d1d3a commit 39f9b19

File tree

2 files changed

+32
-3
lines changed

2 files changed

+32
-3
lines changed

src/main/java/org/code4everything/boot/service/BootBaseService.java

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,26 @@
66
**/
77
public interface BootBaseService {
88

9+
/**
10+
* 获取令牌,抛出未登录异常
11+
*
12+
* @return 令牌
13+
*
14+
* @since 1.1.3
15+
*/
16+
String getToken();
17+
18+
/**
19+
* 获取令牌
20+
*
21+
* @param require 是否抛出未登录异常
22+
*
23+
* @return 令牌
24+
*
25+
* @since 1.1.3
26+
*/
27+
String getToken(boolean require);
28+
929
/**
1030
* 获取用户,当用户未登陆时抛出异常
1131
*

src/main/java/org/code4everything/boot/service/impl/BaseServiceImpl.java

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ public class BaseServiceImpl<U> implements BootBaseService {
2020
public HttpServletRequest request;
2121

2222
@Resource
23-
private BootUserService<U> userService;
23+
protected BootUserService<U> userService;
2424

2525
/**
2626
* 只能继承
@@ -42,6 +42,16 @@ public BaseServiceImpl(HttpServletRequest request, BootUserService<U> userServic
4242
this.userService = userService;
4343
}
4444

45+
@Override
46+
public String getToken() {
47+
return HttpUtils.requireToken(request);
48+
}
49+
50+
@Override
51+
public String getToken(boolean require) {
52+
return require ? HttpUtils.requireToken(request) : HttpUtils.getToken(request);
53+
}
54+
4555
@Override
4656
public U getUser() {
4757
return getUser(true);
@@ -50,8 +60,7 @@ public U getUser() {
5060
@Override
5161
public U getUser(boolean require) {
5262
Objects.requireNonNull(userService, "please set interface 'BootUserService<T>'");
53-
String token = require ? HttpUtils.requireToken(request) : HttpUtils.getToken(request);
54-
U user = userService.getUserByToken(StrUtil.nullToEmpty(token));
63+
U user = userService.getUserByToken(StrUtil.nullToEmpty(getToken(require)));
5564
return require ? AssertUtils.assertUserLoggedIn(user) : user;
5665
}
5766
}

0 commit comments

Comments
 (0)