在J2EE或Sping項(xiàng)目一般分為三次纸泄，表現(xiàn)層基公，業(yè)務(wù)層幅慌，數(shù)據(jù)層。表現(xiàn)層一般是項(xiàng)目的起點(diǎn)轰豆，業(yè)務(wù)層被設(shè)計(jì)為對業(yè)務(wù)對象進(jìn)行操作處理胰伍，數(shù)據(jù)層則負(fù)責(zé)存儲對象本身。處于后端的層次實(shí)際不關(guān)心水在使用他們的數(shù)據(jù)酸休，他們經(jīng)常被依賴注入框架創(chuàng)建后骂租，自動注入他們需要的對象中。
- 依賴注入框架Guice
- 創(chuàng)建用戶管理服務(wù)
- Lambda添加依賴注入
- 測試服務(wù)正確性
源代碼
代碼下載地址：https://pan.baidu.com/s/1ePh8zMVj5V5NRbukeYbmCw
提取碼：v49p
工程說明
工程創(chuàng)建一個用戶管理服務(wù)(services-user)斑司，通過依賴注入渗饮，讓認(rèn)證Lambda函數(shù)(lambda-authorizer)可以使用該服務(wù),這里用的依賴注入框架是Google Guice（以下是工程的關(guān)聯(lián)圖，可以結(jié)合目錄工程了解具體的目錄結(jié)構(gòu)宿刮。
工程目錄結(jié)構(gòu)
1. 依賴注入框架Guice
這里使用Guice將用戶管理服務(wù)工程(services-user)互站，注入到認(rèn)證Lambda函數(shù)工程(lambda-authorizer)中，即在lambda-authorizer的Handler中創(chuàng)建Guice容器僵缺，通過setter的方法將UserService注入胡桃，這樣就可以在Handler中使用UserService的getUserByToken方法獲得用戶的信息。了解Guice戳：《三分鐘快速了解 Google Guice依賴注入框架》
public class Handler extends LambdaHandler<I, O> {
//注入器對象DependencyInjectionModule繼承AbstractModule
private static final Injector INJECTOR = Guice.createInjector(new DependencyInjectionModule());
//用戶服務(wù)接口磕潮，需要注入實(shí)例
private UserService userService;
@Inject
public void setUserService(UserService userService) {
this.userService = userService;
}
//構(gòu)造函數(shù)翠胰，執(zhí)行以來注入操作
public Handler() {
//要求框架執(zhí)行以來注入操作
INJECTOR.injectMembers(this);
//判斷一個對象是否為空,確比荼矗框架注入有效的依賴對象
Objects.requireNonNull(userService);
}
@Override
public AuthorizationOutput handleRequest(AuthorizationInput input, Context context) {
......
}
}
3. 創(chuàng)建用戶管理服務(wù)
在工程根目錄下創(chuàng)建一個services-user目錄，按照“工程目錄結(jié)構(gòu)”創(chuàng)建相關(guān)的文件之景。主要實(shí)現(xiàn)根據(jù)用戶輸入的令牌："Authorization:Bearer xxxxx"字符串返回一個User對象斤富，如果令牌不在數(shù)據(jù)庫(AWS DynamoDB)中，則拋出UserNotFoundException異常锻狗。下面對這幾個類進(jìn)行簡單的說明茂缚，詳細(xì)內(nèi)容下載源代碼查看。
-
User: 用戶model -
UserNotFoundException: 用戶不存在異常類 -
UserRepository: 數(shù)據(jù)層接口屋谭，定義通過toke獲取用戶對象 -
UserRepositoryDynamoDB: UserRepository 實(shí)現(xiàn)類脚囊，通過toke到DynamoDB數(shù)據(jù)庫查詢用戶數(shù)據(jù) -
UserService: 業(yè)務(wù)層接口，根據(jù)toke獲取用戶信息 -
UserServiceImpl: UserService的實(shí)現(xiàn)類桐磁，注入U(xiǎn)serRepository獲取用戶信息悔耘。
User
public class User {
private String id;
private String username;
private String email;
//getter/setter
}
UserNotFoundException
public class UserNotFoundException extends Exception {
private static final long serialVersionUID = -3235669501483817417L;
}
UserRepository
public interface UserRepository {
Optional<User> getUserByToken(String token);
}
UserRepositoryDynamoDB
public class UserRepositoryDynamoDB implements UserRepository {
//暫不去數(shù)據(jù)庫取
public Optional<User> getUserByToken(String token) {
return Optional.empty();
}
}
UserService
public interface UserService {
//根據(jù)toke獲取用戶信息
User getUserByToken(String token) throws UserNotFoundException;
}
UserServiceImpl
public class UserServiceImpl implements UserService {
private final UserRepository userRepository;
//構(gòu)造函數(shù)，傳入用戶存儲服務(wù)
@Inject
public UserServiceImpl(UserRepository userRepository) {
this.userRepository = userRepository;
Objects.requireNonNull(userRepository);
}
@Override
public User getUserByToken(String token) throws UserNotFoundException {
return userRepository.getUserByToken(token).orElseThrow(UserNotFoundException::new);
}
}
4. Lambda添加依賴注入
用戶授權(quán)Lambda(lambda-authorizer)我擂，在調(diào)用入口Handler類使用默認(rèn)構(gòu)造函數(shù)創(chuàng)建對象衬以，因此，我們可以利用構(gòu)造對象的時候初始化依賴注入校摩。同時我們需要準(zhǔn)備好一個Guice的Module類用來配置用戶管理服務(wù)接口UserService以及實(shí)現(xiàn)類UserServiceImpl之間的關(guān)聯(lián)看峻。
-
DependencyInjectionModule: Guice配置依賴注入關(guān)系。 -
Handler: 具體的認(rèn)證衙吩，這邊主要關(guān)注userService的注入
DependencyInjectionModule
public class DependencyInjectionModule extends AbstractModule {
@Override
protected void configure() {
//將接口與實(shí)現(xiàn)類關(guān)聯(lián)互妓，當(dāng)需要UserServie接口訪問用戶信息時
//創(chuàng)建并返回UserServiceImpl對象實(shí)例。（同Spring的Bean注入）
bind(UserService.class).to(UserServiceImpl.class);
bind(UserRepository.class).to(UserRepositoryDynamoDB.class);
}
}
Handler
public class Handler extends LambdaHandler<AuthorizationInput, AuthorizationOutput> {
private static final Logger LOGGER = Logger.getLogger(Handler.class);
//注入器對象坤塞，從這個對象中獲取實(shí)例
private static final Injector INJECTOR = Guice.createInjector(new DependencyInjectionModule());
//用戶服務(wù)接口冯勉，需要注入實(shí)例
private UserService userService;
@Inject
public void setUserService(UserService userService) {
this.userService = userService;
}
//構(gòu)造函數(shù)，執(zhí)行以來注入操作
public Handler() {
//要求框架執(zhí)行以來注入操作
INJECTOR.injectMembers(this);
//判斷一個對象是否為空,確蹦≤剑框架注入有效的依賴對象
Objects.requireNonNull(userService);
}
@Override
public AuthorizationOutput handleRequest(AuthorizationInput input, Context context) {
try {
//根據(jù)token獲取用戶信息
User authenticatedUser = userService.getUserByToken(authenticationToken);
} catch (UserNotFoundException userNotFoundException) {
LOGGER.info("User authentication failed for token " + authenticationToken);
}
}
5. 測試服務(wù)正確性
Lambda注入代碼寫完灼狰，由于本次Java代碼較多需要在本地測試完基礎(chǔ)代碼確保沒問題，在deploy到AWS上浮禾。我們使用的junit交胚，easymock來模擬測試。通過easymock來模擬產(chǎn)生一個權(quán)限令牌AuthorizationInput盈电，省去部署和Http請求的方式獲取令牌蝴簇。
public class HandlerTest {
@Test
public void testFailingToken() throws Exception {
Handler testHandler = new Handler();
//創(chuàng)建一個AuthorizationInput模擬對象
AuthorizationInput mockAuthorizationInput = createNiceMock(AuthorizationInput.class);
//模擬getAuthorizationToken對象的行為，并返回"INVALID_TOKEN"
//即當(dāng)調(diào)用getAuthorizationToken()方法時挣轨，返回"INVALID_TOKEN"
expect(mockAuthorizationInput.getAuthorizationToken()).andReturn("INVALID_TOKEN").anyTimes();
//使模擬對象備用
replay(mockAuthorizationInput);
AuthorizationOutput authorizationOutput = testHandler.handleRequest(mockAuthorizationInput, null);
assertEquals(PolicyStatement.Effect.DENY, authorizationOutput.getPolicyDocument().getPolicyStatements().get(0).getEffect());
}
}
在根目錄下運(yùn)行./gradlew test 檢查代碼是否測試通過军熏。確認(rèn)沒問題發(fā)布./gradlew deploy到AWS云上，
目前數(shù)據(jù)庫操作層都是返回empty對象卷扮，所以我們的測試都是沒有權(quán)限荡澎。這個在后續(xù)通過對接DynamoDB數(shù)據(jù)庫解決均践。
源代碼
代碼下載地址：https://pan.baidu.com/s/1ePh8zMVj5V5NRbukeYbmCw
提取碼：v49p
