Add count function

jakubknejzlik · Mar 14, 2024 · 1c954b2 · 1c954b2
1 parent 5371b2a
commit 1c954b2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/Function.test.ts b/src/Function.test.ts
@@ -6,6 +6,8 @@ const flavor = Q.flavors.mysql;
 
 describe("Expression", () => {
   it("should produce valid SQL", () => {
+    expect(Fn.count("*").toSQL(flavor)).toEqual("COUNT(*)");
+    expect(Fn.count("foo").toSQL(flavor)).toEqual("COUNT(`foo`)");
     expect(Fn.sum("foo").toSQL(flavor)).toEqual("SUM(`foo`)");
     expect(Fn.max("foo").toSQL(flavor)).toEqual("MAX(`foo`)");
     expect(Fn.add("foo", "blah").toSQL(flavor)).toEqual("(`foo` + `blah`)");

diff --git a/src/Function.ts b/src/Function.ts
@@ -19,6 +19,9 @@ const ifFn = (
 };
 
 export const Function = {
+  count: (column: ExpressionValue) => {
+    return new FunctionExpression("COUNT", column);
+  },
   add: (...columns: ExpressionValue[]) => {
     return new OperationExpression("+", ...columns);
   },