AWS Lambda筆記-函數(shù)依賴注入-9【含源代碼】

在J2EE或Sping項(xiàng)目一般分為三次，表現(xiàn)層，業(yè)務(wù)層，數(shù)據(jù)層。表現(xiàn)層一般是項(xiàng)目的起點(diǎn)，業(yè)務(wù)層被設(shè)計(jì)為對(duì)業(yè)務(wù)對(duì)象進(jìn)行操作處理，數(shù)據(jù)層則負(fù)責(zé)存儲(chǔ)對(duì)象本身。處于后端的層次實(shí)際不關(guān)心水在使用他們的數(shù)據(jù)，他們經(jīng)常被依賴注入框架創(chuàng)建后，自動(dòng)注入他們需要的對(duì)象中。

  1. 依賴注入框架Guice
  2. 創(chuàng)建用戶管理服務(wù)
  3. Lambda添加依賴注入
  4. 測(cè)試服務(wù)正確性

源代碼

代碼下載地址：https://pan.baidu.com/s/1ePh8zMVj5V5NRbukeYbmCw
提取碼：v49p

工程說(shuō)明

工程創(chuàng)建一個(gè)用戶管理服務(wù)(services-user)，通過(guò)依賴注入，讓認(rèn)證Lambda函數(shù)(lambda-authorizer)可以使用該服務(wù),這里用的依賴注入框架是Google Guice（以下是工程的關(guān)聯(lián)圖，可以結(jié)合目錄工程了解具體的目錄結(jié)構(gòu)。


工程服務(wù)關(guān)聯(lián)

工程目錄結(jié)構(gòu)


函數(shù)依賴注入工程目錄結(jié)構(gòu)

1. 依賴注入框架Guice

這里使用Guice將用戶管理服務(wù)工程(services-user)，注入到認(rèn)證Lambda函數(shù)工程(lambda-authorizer)中，即在lambda-authorizer的Handler中創(chuàng)建Guice容器，通過(guò)setter的方法將UserService注入，這樣就可以在Handler中使用UserService的getUserByToken方法獲得用戶的信息。了解Guice戳：《三分鐘快速了解 Google Guice依賴注入框架》

public class Handler extends LambdaHandler<I, O> {
    //注入器對(duì)象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í)行以來(lái)注入操作
    public Handler() {
        //要求框架執(zhí)行以來(lái)注入操作
        INJECTOR.injectMembers(this);
        //判斷一個(gè)對(duì)象是否為空,確?？蚣茏⑷胗行У囊蕾噷?duì)象
        Objects.requireNonNull(userService);
    }

    @Override
    public AuthorizationOutput handleRequest(AuthorizationInput input, Context context) {
       ......
    }
}

3. 創(chuàng)建用戶管理服務(wù)

在工程根目錄下創(chuàng)建一個(gè)services-user目錄，按照“工程目錄結(jié)構(gòu)”創(chuàng)建相關(guān)的文件。主要實(shí)現(xiàn)根據(jù)用戶輸入的令牌："Authorization:Bearer xxxxx"字符串返回一個(gè)User對(duì)象，如果令牌不在數(shù)據(jù)庫(kù)(AWS DynamoDB)中，則拋出UserNotFoundException異常。下面對(duì)這幾個(gè)類進(jìn)行簡(jiǎn)單的說(shuō)明，詳細(xì)內(nèi)容下載源代碼查看。

  1. User : 用戶model
  2. UserNotFoundException: 用戶不存在異常類
  3. UserRepository: 數(shù)據(jù)層接口，定義通過(guò)toke獲取用戶對(duì)象
  4. UserRepositoryDynamoDB: UserRepository 實(shí)現(xiàn)類，通過(guò)toke到DynamoDB數(shù)據(jù)庫(kù)查詢用戶數(shù)據(jù)
  5. UserService: 業(yè)務(wù)層接口，根據(jù)toke獲取用戶信息
  6. 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ù)庫(kù)取
    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ù)，傳入用戶存儲(chǔ)服務(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)建對(duì)象，因此，我們可以利用構(gòu)造對(duì)象的時(shí)候初始化依賴注入。同時(shí)我們需要準(zhǔn)備好一個(gè)Guice的Module類用來(lái)配置用戶管理服務(wù)接口UserService以及實(shí)現(xiàn)類UserServiceImpl之間的關(guān)聯(lián)。

  1. DependencyInjectionModule: Guice配置依賴注入關(guān)系。
  2. 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接口訪問(wèn)用戶信息時(shí)
        //創(chuàng)建并返回UserServiceImpl對(duì)象實(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);

    //注入器對(duì)象，從這個(gè)對(duì)象中獲取實(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í)行以來(lái)注入操作
    public Handler() {
        //要求框架執(zhí)行以來(lái)注入操作
        INJECTOR.injectMembers(this);
        //判斷一個(gè)對(duì)象是否為空,確保框架注入有效的依賴對(duì)象
        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. 測(cè)試服務(wù)正確性

Lambda注入代碼寫(xiě)完，由于本次Java代碼較多需要在本地測(cè)試完基礎(chǔ)代碼確保沒(méi)問(wèn)題，在deploy到AWS上。我們使用的junit，easymock來(lái)模擬測(cè)試。通過(guò)easymock來(lái)模擬產(chǎn)生一個(gè)權(quán)限令牌AuthorizationInput，省去部署和Http請(qǐng)求的方式獲取令牌。

public class HandlerTest {
    @Test
    public void testFailingToken() throws Exception {
        Handler testHandler = new Handler();
        //創(chuàng)建一個(gè)AuthorizationInput模擬對(duì)象
        AuthorizationInput mockAuthorizationInput = createNiceMock(AuthorizationInput.class);
        //模擬getAuthorizationToken對(duì)象的行為，并返回"INVALID_TOKEN"
        //即當(dāng)調(diào)用getAuthorizationToken()方法時(shí)，返回"INVALID_TOKEN"
       expect(mockAuthorizationInput.getAuthorizationToken()).andReturn("INVALID_TOKEN").anyTimes();
        //使模擬對(duì)象備用
        replay(mockAuthorizationInput);

        AuthorizationOutput authorizationOutput = testHandler.handleRequest(mockAuthorizationInput, null);
        assertEquals(PolicyStatement.Effect.DENY, authorizationOutput.getPolicyDocument().getPolicyStatements().get(0).getEffect());
    }
}

在根目錄下運(yùn)行./gradlew test 檢查代碼是否測(cè)試通過(guò)。確認(rèn)沒(méi)問(wèn)題發(fā)布./gradlew deploy到AWS云上，

測(cè)試通過(guò)

目前數(shù)據(jù)庫(kù)操作層都是返回empty對(duì)象，所以我們的測(cè)試都是沒(méi)有權(quán)限。這個(gè)在后續(xù)通過(guò)對(duì)接DynamoDB數(shù)據(jù)庫(kù)解決。
無(wú)權(quán)限訪問(wèn)

源代碼

代碼下載地址：https://pan.baidu.com/s/1ePh8zMVj5V5NRbukeYbmCw
提取碼：v49p

最后編輯于 ：
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書(shū)系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容