Add count() endpoint to database and datastore

Author: Chris

src/api/rest/v1/db/controller.ts

@@ -133,6 +133,45 @@ export class DbController {
         }
     }
 
+    public async count(req: Request, res: Response) {
+        try {
+            const dbName = req.params.database
+            const { context } = req.veridaNetworkConnection
+            const permissions = Utils.buildPermissions(req)
+
+            const db = await context.openDatabase(dbName, {
+                // @ts-ignore
+                permissions
+            })
+
+            const selector = req.body.query
+            const limit = 1000
+            const fields = ['_id']
+
+            const loops = 0
+            let count = 0
+            while (true) {
+                const result = await db.getMany(selector, {
+                    fields,
+                    limit,
+                    skip: loops * limit
+                })
+
+                if (result.length < limit) {
+                    count = loops*limit + result.length
+                    break
+                }
+            }
+
+            res.json({
+                count
+            })
+        } catch (error: any) {
+            console.log(error)
+            res.status(500).send(error.message);
+        }
+    }
+
     public async query(req: Request, res: Response) {
         try {
             const dbName = req.params.database

src/api/rest/v1/db/routes.ts

@@ -11,6 +11,12 @@ router.get("/get/:database/:recordId", auth({
     credits: CONFIG.verida.billing.defaultCredits
 }), controller.getById)
 
+router.post("/count/:database", auth({
+    scopes: ["api:db-query"],
+    dbScope: "r",
+    credits: 0
+}), controller.count)
+
 router.post("/query/:database", auth({
     scopes: ["api:db-query"],
     dbScope: "r",

src/api/rest/v1/ds/controller.ts

@@ -132,6 +132,51 @@ export class DsController {
         }
     }
 
+    public async count(req: Request, res: Response) {
+        try {
+            const schemaName = Utils.getSchemaFromParams(req.params.schema)
+            const { context } = req.veridaNetworkConnection
+            const permissions = Utils.buildPermissions(req)
+
+            const ds = await context.openDatastore(schemaName, {
+                // @ts-ignore
+                permissions
+            })
+
+            const selector = req.body.query
+            const limit = 1000
+            const fields = ['_id']
+
+            const loops = 0
+            let count = 0
+            while (true) {
+                const result = await ds.getMany(selector, {
+                    fields,
+                    limit,
+                    skip: loops * limit
+                })
+
+                if (result.length < limit) {
+                    count = loops*limit + result.length
+                    break
+                }
+            }
+
+            res.json({
+                count
+            })
+        } catch (error: any) {
+            let message = error.message
+            if (error.message.match('invalid encoding')) {
+                message = 'Invalid encoding (check permissions header)'
+            }
+
+            res.status(500).send({
+                error: message
+            });
+        }
+    }
+
     public async query(req: Request, res: Response) {
         try {
             const schemaName = Utils.getSchemaFromParams(req.params.schema)

src/api/rest/v1/ds/routes.ts

@@ -11,6 +11,12 @@ router.get("/get/:schema/:recordId", auth({
     credits: CONFIG.verida.billing.defaultCredits
 }), controller.getById)
 
+router.post("/count/:schema", auth({
+    scopes: ["api:ds-query"],
+    dsScope: "r",
+    credits: 0,
+}),  controller.count)
+
 router.post("/query/:schema", auth({
     scopes: ["api:ds-query"],
     dsScope: "r",