ReactiveMongoデータベースフレームをPlay Framework 2.5でダンプする

Question

mongoデータベースをjsonオブジェクトにダンプしようとしていますが、データベースに対するクエリが非同期であるため、問題が発生しています。

私のデータベースの各コレクションにはユーザーデータが含まれており、各コレクション名はユーザー名です。

私はすべてのコレクションの名前を回復してから、各コレクションを1つずつ回復するためにそれらの上にループします。

def databaseDump(prom : Promise[JsObject]) = { 
    for{ 
     dbUsers <- getUsers 
    } yield dbUsers 

var rebuiltJson = Json.obj() 
var array = JsArray() 
res.map{ users => 
    users.map{ userNames => 
    if(userNames.size == 0){ 
     prom failure new Throwable("Empty database") 
    } 
    var counter = 0 
    userNames.foreach { username => 
     getUserTables(username).map { tables => 
     /* Add data to array*/ 
      ... 
     counter += 1 
     if(counter == userNames.size){ 
      /*Add data to new json*/ 
      ... 
      prom success rebuiltJson 
     } 

     } 
    } 
    } 
} 

このように動作しますが、すべてのデータがまだ復元されていなくても、約束が正常に開始されることがあります。これは、私のカウンタ変数が信頼できる解決策ではないという事実によるものです。

すべてのユーザーをループし、データベースにクエリを送信し、すべてのデータが回復するのを待ってから、約束をうまく起動させる方法はありますか？私は理解のために使用しようとしましたが、それを行う方法が見つかりませんでした。 1つのJsonに全体のDBをダンプする方法はありますか：{ username : data, username : data ..}？

Answer 1

ユーザー/テーブルの用語が混乱していたので、データベースを1つのJsObjectにダンプする新しい関数を書きました。

// helper function to find all documents inside a collection c 
// and return them as a single JsArray 
def getDocs(c: JSONCollection)(implicit ec: ExecutionContext) = c.find(Json.obj()).cursor[JsObject]().jsArray() 

def dumpToJsObject(db: DefaultDB)(implicit ec: ExecutionContext): Future[JsObject] = { 
    // get a list of all collections in the db 
    val collectionNames = db.collectionNames 
    val collections = collectionNames.map(_.map(db.collection[JSONCollection](_))) 

    // each entry is a tuple collectionName -> content (as JsArray) 
    val namesToDocs = collections.flatMap { 
    colls => Future.sequence(colls.map(c => getDocs(c).map(c.name -> _))) 
    } 

    // convert to a single JsObject 
    namesToDocs.map(JsObject(_)) 
} 

私は（私が後でそうだろう）まだそれをテストしていませんが、この機能は、少なくともあなたの一般的なアイデアを与える必要があります。データベース内のすべてのコレクションのリストを取得します。コレクションごとに、そのコレクション内のすべての文書を取得するためのクエリを実行します。ドキュメントのリストはJsArrayに変換され、最後にすべてのコレクションはキーとしてコレクション名を持つ単一のJsObjectで構成されます。

Answer 2

目的がデータを出力ストリーム（ローカル/ファイルまたはネットワーク）に、副作用で書き込むことである場合。

import scala.concurrent.{ ExecutionContext, Future } 
import reactivemongo.bson.BSONDocument 
import reactivemongo.api.{ Cursor, MongoDriver, MongoConnection } 

val mongoUri = "mongodb://localhost:27017/my_db" 

val driver = new MongoDriver 
val maxDocs = Int.MaxValue // max per collection 

// Requires to have an ExecutionContext in the scope 
// (e.g. `import scala.concurrent.ExecutionContext.Implicits.global`) 
def dump()(implicit ec: ExecutionContext): Future[Unit] = for { 
    uri <- Future.fromTry(MongoConnection.parseURI(mongoUri)) 
    con = driver.connection(uri) 
    dn <- Future(uri.db.get) 
    db <- con.database(dn) 
    cn <- db.collectionNames 
    _ <- Future.sequence(cn.map { collName => 
    println(s"Collection: $collName") 

    db.collection(collName).find(BSONDocument.empty). // findAll 
     cursor[BSONDocument]().foldWhile({}, maxDocs) { (_, doc) => 
     // Replace println by appropriate side-effect 
     Cursor.Cont(println(s"- ${BSONDocument pretty doc}")) 
     } 
    }) 
} yield() 

だけJsObject（例えばBSONDocument.empty〜>Json.obj()）でBSONDocumentを置き換え、JSONのシリアライズパックを使用している場合。

Scala REPLからテストする場合、前のコードを貼り付けた後、次のように実行できます。

dump().onComplete { 
    case result => 
    println(s"Dump result: $result") 
    //driver.close() 
}