add find many

README.md

@@ -2,7 +2,7 @@
 
 > **deno_mongo** is a **MongoDB** database driver developed for deno, based on rust's official [`mongodb`](https://crates.io/crates/mongodb) library package.
 
-[![tag](https://img.shields.io/github/tag/manyuanrong/deno_mongo.svg)](https://github.com/manyuanrong/deno_mongo)
+[![tag](https://img.shields.io/github/tag/manyuanrong/deno_mongo.svg)](https://github.com/manyuanrong/deno_mongo/releases)
 [![Build Status](https://github.com/manyuanrong/deno_mongo/workflows/ci/badge.svg?branch=master)](https://github.com/manyuanrong/deno_mongo/actions)
 [![license](https://img.shields.io/github/license/manyuanrong/deno_mongo.svg)](https://github.com/manyuanrong/deno_mongo)
 [![tag](https://img.shields.io/badge/deno-v0.35.0-green.svg)](https://github.com/denoland/deno)
@@ -16,10 +16,10 @@
 ## Examples
 
 ```ts
-import { init, MongoClient } from "https://deno.land/x/mongo/mod.ts";
+import { init, MongoClient } from "https://deno.land/x/mongo@v0.3.0/mod.ts";
 
-// Initialize the plugin and specify the binary release version (because the binary currently has no idea how to associate the version in ts and the binary)
-await init("0.1.0");
+// Initialize the plugin
+await init();
 
 const client = new MongoClient();
 client.connectWithUri("mongodb://localhost:27017");
@@ -33,9 +33,27 @@ const insertId = await users.insertOne({
   password: "pass1"
 });
 
+// insertMany
+const insertIds = await users.insertMany([
+  {
+    username: "user1",
+    password: "pass1"
+  },
+  {
+    username: "user2",
+    password: "pass2"
+  }
+]);
+
 // findOne
 const user1 = await users.findOne({ _id: insertId });
 
+// find
+const users = await users.find({ username: { $ne: null } });
+
 // deleteOne
 const deleteCount = await users.deleteOne({ _id: insertId });
+
+// deleteMany
+const deleteCount2 = await users.deleteMany({ usename: "test" });
 ```

mod.ts

@@ -1,3 +1,4 @@
 export { ClientOptions, MongoClient } from "./ts/client.ts";
 export { Database } from "./ts/database.ts";
 export { init } from "./ts/util.ts";
+export const VERSION = "v0.3.0";

src/command/find.rs

@@ -0,0 +1,42 @@
+use crate::*;
+use bson::Document;
+use mongodb::error::Result;
+use serde_json::Value;
+use util::maybe_json_to_document;
+
+#[derive(Deserialize, Debug)]
+#[serde(rename_all = "camelCase")]
+struct FindArgs {
+    db_name: String,
+    collection_name: String,
+    filter: Option<Value>,
+    find_one: bool,
+}
+
+pub fn find(command: Command) -> CoreOp {
+    let fut = async move {
+        let client = command.get_client();
+        let data = command.data;
+        let args: FindArgs = serde_json::from_slice(data.unwrap().as_ref()).unwrap();
+        let db_name = args.db_name;
+        let collection_name = args.collection_name;
+        let filter = maybe_json_to_document(args.filter);
+        let database = client.database(&db_name);
+        let collection = database.collection(&collection_name);
+
+        if args.find_one {
+            let doc = collection.find_one(filter, None).unwrap();
+            Ok(util::async_result(&command.args, doc))
+        } else {
+            let cursor = collection.find(filter, None).unwrap();
+            let docs: Vec<Document> = cursor
+                .filter_map(|doc: Result<Document>| match doc {
+                    Ok(doc) => Some(doc),
+                    _ => None,
+                })
+                .collect();
+            Ok(util::async_result(&command.args, docs))
+        }
+    };
+    CoreOp::Async(fut.boxed())
+}

src/command/mod.rs

@@ -1,13 +1,13 @@
 mod connect;
 mod delete;
-mod find_one;
+mod find;
 mod insert;
 mod list_collection_names;
 mod list_database_names;
 
 pub use connect::{connect_with_options, connect_with_uri};
 pub use delete::delete;
-pub use find_one::find_one;
+pub use find::find;
 pub use insert::{insert_many, insert_one};
 pub use list_collection_names::list_collection_names;
 pub use list_database_names::list_database_names;

src/lib.rs

@@ -33,7 +33,7 @@ pub enum CommandType {
     ConnectWithOptions,
     ConnectWithUri,
     ListDatabases,
-    FindOne,
+    Find,
     ListCollectionNames,
     InsertMany,
     InsertOne,
@@ -93,7 +93,7 @@ fn op_command(data: &[u8], zero_copy: Option<ZeroCopyBuf>) -> CoreOp {
         CommandType::ConnectWithUri => command::connect_with_uri,
         CommandType::ListDatabases => command::list_database_names,
         CommandType::ListCollectionNames => command::list_collection_names,
-        CommandType::FindOne => command::find_one,
+        CommandType::Find => command::find,
         CommandType::InsertOne => command::insert_one,
         CommandType::InsertMany => command::insert_many,
         CommandType::Delete => command::delete,

test.ts

@@ -55,6 +55,24 @@ test(async function testInsertOne() {
   });
 });
 
+test(async function testFindOne() {
+  const db = getClient().database("test");
+  const users = db.collection("mongo_test_users");
+  const user1 = await users.findOne();
+  assert(user1 instanceof Object);
+  assertEquals(Object.keys(user1), ["_id", "username", "password"]);
+
+  const findNull = await users.findOne({ test: 1 });
+  assertEquals(findNull, null);
+});
+
+test(async function testDeleteOne() {
+  const db = getClient().database("test");
+  const users = db.collection("mongo_test_users");
+  const deleteCount = await users.deleteOne({});
+  assertEquals(deleteCount, 1);
+});
+
 test(async function testInsertMany() {
   const db = getClient().database("test");
   const users = db.collection("mongo_test_users");
@@ -72,22 +90,15 @@ test(async function testInsertMany() {
   assertEquals(insertIds.length, 2);
 });
 
-test(async function testFindOne() {
+test(async function testFind() {
   const db = getClient().database("test");
   const users = db.collection("mongo_test_users");
-  const user1 = await users.findOne();
-  assert(user1 instanceof Object);
-  assertEquals(Object.keys(user1), ["_id", "username", "password"]);
+  const findUsers = await users.find({ username: "many" });
+  assert(findUsers instanceof Array);
+  assertEquals(findUsers.length, 2);
 
-  const findNull = await users.findOne({ test: 1 });
-  assertEquals(findNull, null);
-});
-
-test(async function testDeleteOne() {
-  const db = getClient().database("test");
-  const users = db.collection("mongo_test_users");
-  const deleteCount = await users.deleteOne({});
-  assertEquals(deleteCount, 1);
+  const notFound = await users.find({ test: 1 });
+  assertEquals(notFound, []);
 });
 
 test(async function testDeleteMany() {
@@ -100,5 +111,5 @@ test(async function testDeleteMany() {
 });
 
 await cargoBuild();
-await init("master");
+await init();
 await runTests();

ts/collection.ts

@@ -9,23 +9,32 @@ export class Collection {
     private readonly collectionName: string
   ) {}
 
-  public async findOne(filter?: Object): Promise<any> {
+  private async _find(filter?: Object, findOne: boolean = false): Promise<any> {
     const doc = await dispatchAsync(
       {
-        command_type: CommandType.FindOne,
+        command_type: CommandType.Find,
         client_id: this.client.clientId
       },
       encode(
         JSON.stringify({
           dbName: this.dbName,
           collectionName: this.collectionName,
-          filter
+          filter,
+          findOne
         })
       )
     );
     return doc;
   }
 
+  public async findOne(filter?: Object): Promise<any> {
+    return this._find(filter, true);
+  }
+
+  public async find(filter?: Object): Promise<any> {
+    return this._find(filter, false);
+  }
+
   public async insertOne(doc: Object): Promise<any> {
     const _id = await dispatchAsync(
       {

ts/types.ts

@@ -3,7 +3,7 @@ export enum CommandType {
   ConnectWithOptions = "ConnectWithOptions",
   ListDatabases = "ListDatabases",
   ListCollectionNames = "ListCollectionNames",
-  FindOne = "FindOne",
+  Find = "Find",
   InsertOne = "InsertOne",
   InsertMany = "InsertMany",
   Delete = "Delete"

ts/util.ts

@@ -1,4 +1,5 @@
 import { prepare } from "https://raw.githubusercontent.com/manyuanrong/deno-plugin-prepare/master/mod.ts";
+import { VERSION } from "../mod.ts";
 import { CommandType } from "./types.ts";
 
 const PLUGIN_NAME = "deno_mongo";
@@ -17,7 +18,7 @@ interface Command {
   command_id?: number;
 }
 
-export async function init(binVer: string) {
+export async function init(binVer: string = VERSION) {
   let releaseUrl = `https://github.com/manyuanrong/deno_mongo/releases/download/${binVer}`;
 
   const options = {