MongoDB Wire Protocol を利用した count の実装方法

Question

MongoDB の count についての質問です。

現在、以下のページの MongoDB のプロトコル仕様を見ながら独自ドライバを作ってみています。
MongoDB Wire Protocol

コレクション中の、条件に合致するドキュメント数を返す count メソッドを実装したいのですが、opCode 一覧を見てもそれらしいものが見当たりません。

今のところ OP_QUERY で find を実行して、取得できたドキュメント数をドライバ側でカウントするくらいしか思いつかないのですが、それだと不要なドキュメントを取得することになり、無駄が多そうな気がします。

何かよい方法があるのでしょうか？

Answer 1

MongoDB プロトコルをプロキシするプログラムを作成し、標準の mongo クライアントで

use test-db

db.TestCollection.find({'a': 10}).count()

を実行してどのようなリクエストが流れているかを確認しました。
その結果、count() を実行したタイミングで、test-db.$cmd コレクションに対して、

query={'count': 'TestCollection', 'query': {'a': 10.0}, 'fields': {}}

という OP_QUERY リクエストが流れていることが分かりました。

Answer 2

間違ってたらすみません。
OP_REPLYのnumberReturnedは使えないでしょうか。

Comments

[bobunderson]

回答ありがとうございます。
numberReturned は試してみたのですが、OP_QUERY の numberToReturn より大きい数は返ってこないように見えます。
例えば、コレクションに 3 件のドキュメントを格納した状態で、query={}, numberToReturn=2 の OP_QUERY を送信してみたところ、numberReturned=2 で返って来ました。
これだとやはり OP_GET_MORE で全件取得しないとカウントできないように思えます。。。

[sekitaka_1214]

なるほど、そうでしたか。
ご確認頂きありがとうございました。