AWSラムダの（Firebase）コールバックを待つ方法は？

Question

私はいくつかの同様の質問を探していましたが、私が知る限り、どれもJava関連です。 Firebaseデータベースに接続しているAWSラムダ関数を呼び出したいとします。問題は、Firebaseから必要なデータを取得する前にハンドラが実行されることです。

@Override 
public String handleRequest(Request input, Context context) { 
    try { 
     FileInputStream serviceAccountInputStream = new FileInputStream(FIREBASE_SERVICE_ACCOUNT_CREDENTIALS_PATH); 

     FirebaseOptions options = new FirebaseOptions.Builder() 
       .setCredential(FirebaseCredentials.fromCertificate(serviceAccountInputStream)) 
       .setDatabaseUrl(FIREBASE_DATABASE_URL) 
       .build(); 
     FirebaseApp.initializeApp(options); 

     DatabaseReference ref = FirebaseDatabase 
       .getInstance() 
       .getReference("users/" + input.getUid()); 
     ref.addListenerForSingleValueEvent(new ValueEventListener() { 
      @Override 
      public void onDataChange(DataSnapshot dataSnapshot) { 
       System.out.print(dataSnapshot); 
       // TODO: Do computations on data and return results 
      } 

      @Override 
      public void onCancelled(DatabaseError databaseError) { 
       System.out.print("Canceled"); 
       // TODO: Return error 
      } 
     }); 
    } catch (Exception e) { 
     e.printStackTrace(); 
    } 
    // TODO: Return computed results 
    return "This could be the start of something new."; 
} 

Firebaseデータベースからデータを読み込み、それに計算を行い、計算結果をユーザに返す必要があります。どうすればこれを達成できますか？ :)

Answer 1

CountDownLatchを使って非同期コードが終了するのを待つことができます。あなたがたCountDownLatchを作成する非同期コードを呼び出す前に

：

final CountDownLatch countDownLatch = new CountDownLatch(1); 

そして、コールバックメソッドの最後に、あなたは、ラッチをカウントダウン：

countDownLatch.countDown(); 

を次に非同期メソッドの後、あなたは待ちますCountDownLatchをカウントダウンする：

waitForCountdownLatch(countDownLatch); 

private static void waitForCountdownLatch(CountDownLatch countDownLatch) { 
    try { 
     countDownLatch.await(); 
    } catch (InterruptedException e) { 
     log.error(e); 
     e.printStackTrace(); 
    } 
} 

コードでは、次のようになります。

@Override 
public String handleRequest(Request input, Context context) { 

    final CountDownLatch countDownLatch = new CountDownLatch(1); 
    final Object[] singleValue = new Object[1]; 
    final DatabaseError[] firebaseError = new DatabaseError[1]; 

    try { 
     FileInputStream serviceAccountInputStream = new FileInputStream(FIREBASE_SERVICE_ACCOUNT_CREDENTIALS_PATH); 

     FirebaseOptions options = new FirebaseOptions.Builder() 
       .setCredential(FirebaseCredentials.fromCertificate(serviceAccountInputStream)) 
       .setDatabaseUrl(FIREBASE_DATABASE_URL) 
       .build(); 
     FirebaseApp.initializeApp(options); 

     DatabaseReference ref = FirebaseDatabase 
       .getInstance() 
       .getReference("users/" + input.getUid()); 

     ref.addListenerForSingleValueEvent(new ValueEventListener() { 
      @Override 
      public void onDataChange(DataSnapshot dataSnapshot) { 
       System.out.print(dataSnapshot); 
       Object snapshotValue = dataSnapshot.getValue(); 
       if(snapshotValue != null) { 
        singleValue[0] = snapshotValue; 
       } 
       countDownLatch.countDown(); 
      } 

      @Override 
      public void onCancelled(DatabaseError databaseError) { 
       System.out.print("Canceled"); 
       firebaseError0] = databaseError; 
       countDownLatch.countDown(); 
      } 
     }); 
    } catch (Exception e) { 
     e.printStackTrace(); 
     countDownLatch.countDown(); 
    } 

    waitForCountdownLatch(countDownLatch); 
    if(firebaseError[0] != null) { 
     System.out.print(firebaseError[0].toException().getMessage()); 
    } 
    if(singleValue[0] != null) { 
     // do something with result 
    } 
    return "This could be the start of something new."; 
} 

private void waitForCountdownLatch(CountDownLatch countDownLatch) { 
    try { 
     countDownLatch.await(); 
    } catch (InterruptedException e) { 
     log.error(e); 
     e.printStackTrace(); 
    } 
}