@aceitadev/adatabase 0.3.5 → 0.4.0

package/dist/QueryBuilder.d.ts

@@ -16,6 +16,7 @@ export declare class QueryBuilder<T extends ActiveRecord> {
     limit(count: number): this;
     offset(count: number): this;
     first(): Promise<T | null>;
+    count(): Promise<number>;
     get(): Promise<T[]>;
     private getColumnName;
     private mapRowsToEntities;

package/dist/QueryBuilder.js

@@ -63,6 +63,47 @@ class QueryBuilder {
             return rows.length > 0 ? rows[0] : null;
         });
     }
+    count() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const table = (0, Table_1.getTableName)(this.model);
+            if (!table) {
+                throw new PersistenceException_1.PersistenceException("Model has no @Table", null);
+            }
+            const columnsMeta = (0, Column_1.getColumnMeta)(this.model);
+            if (!columnsMeta) {
+                throw new PersistenceException_1.PersistenceException("Model has no @Column decorators", null);
+            }
+            const allowedOperators = ['=', '!=', '<>', '>', '<', '>=', '<=', 'LIKE', 'IN', 'IS NULL', 'IS NOT NULL'];
+            const params = [];
+            const baseAlias = 't1';
+            let sql = `SELECT COUNT(*) as count FROM \`${table}\` AS ${baseAlias}`;
+            if (this.whereClauses.length > 0) {
+                sql += " WHERE ";
+                this.whereClauses.forEach((c, i) => {
+                    if (!allowedOperators.includes(c.operator.toUpperCase())) {
+                        throw new Error(`Invalid operator used: ${c.operator}`);
+                    }
+                    if (i > 0) {
+                        sql += ` ${c.booleanOp} `;
+                    }
+                    const colName = this.getColumnName(c.field, columnsMeta);
+                    sql += `\`${baseAlias}\`.\`${colName}\` ${c.operator} ?`;
+                    params.push(c.value);
+                });
+            }
+            try {
+                const rows = yield (0, Database_1.query)(sql, params);
+                if (rows && rows.length > 0 && rows[0].count !== undefined) {
+                    return Number(rows[0].count);
+                }
+                return 0;
+            }
+            catch (error) {
+                console.error("Failed to execute count query:", error);
+                throw new PersistenceException_1.PersistenceException("Failed to count records");
+            }
+        });
+    }
     get() {
         return __awaiter(this, void 0, void 0, function* () {
             const table = (0, Table_1.getTableName)(this.model);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@aceitadev/adatabase",
-    "version": "0.3.5",
+    "version": "0.4.0",
     "description": "Uma biblioteca para facilitar a interação com bancos de dados MySQL e PostgreSQL em projetos TypeScript/Node.js.",
     "main": "dist/index.js",
     "types": "dist/index.d.ts",