Sql server count big

NEWS.md

@@ -17,6 +17,8 @@
 
 * `across(everything())` doesn't select grouping columns created via `.by` in
   `summarise()` (@mgirlich, #1493).
+*  Use `COUNT_BIG` instead of `COUNT` for SQL server so that `tally()` and
+  `count()` work regardless of size of the data (@edward-burn, #1498).
 
 *  New translations of clock function `date_count_between()` for SQL server, Redshift, Snowflake, Postgres, and Spark (@edward-burn, #1495).
 

R/backend-mssql.R

@@ -425,6 +425,8 @@ simulate_mssql <- function(version = "15.0") {
   sql_variant(
     mssql_scalar,
     sql_translator(.parent = base_odbc_agg,
+     # https://learn.microsoft.com/en-us/sql/t-sql/functions/count-big-transact-sql?view=sql-server-ver17
+      n           = function() sql("COUNT_BIG(*)"),
       sd            = sql_aggregate("STDEV", "sd"),
       var           = sql_aggregate("VAR", "var"),
       str_flatten = function(x, collapse = "") sql_expr(string_agg(!!x, !!collapse)),

tests/testthat/_snaps/backend-mssql.md

@@ -497,6 +497,24 @@
       FROM `df`
       ORDER BY `y`
 
+# count_big
+
+    Code
+      count(mf)
+    Output
+      <SQL>
+      SELECT COUNT_BIG(*) AS `n`
+      FROM `df`
+
+---
+
+    Code
+      tally(mf)
+    Output
+      <SQL>
+      SELECT COUNT_BIG(*) AS `n`
+      FROM `df`
+
 # add prefix to temporary table
 
     Code

tests/testthat/test-backend-mssql.R

@@ -400,6 +400,12 @@ test_that("row_number() with and without group_by() and arrange(): unordered def
   expect_snapshot(mf %>% arrange(y) %>% mutate(rown = row_number()))
 })
 
+test_that("count_big", {
+  mf <- lazy_frame(x = c(1:5), y = c(rep("A", 5)), con = simulate_mssql())
+  expect_snapshot(count(mf))
+  expect_snapshot(tally(mf))
+})
+
 # Live database -----------------------------------------------------------
 
 test_that("can copy_to() and compute() with temporary tables (#438)", {