add r-dbi user-agent

duckdb · Dec 26, 2024 · ea8af42 · ea8af42
1 parent 677d1c9
commit ea8af42
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/database.cpp b/src/database.cpp
@@ -29,6 +29,7 @@ static bool CastRstringToVarchar(Vector &source, Vector &result, idx_t count, Ca
 	if (readonly) {
 		config.options.access_mode = AccessMode::READ_ONLY;
 	}
+	config.options.duckdb_api = "r-dbi";
 
 	auto confignames = configsexp.names();
 

diff --git a/tests/testthat/test-connect.R b/tests/testthat/test-connect.R
@@ -184,3 +184,12 @@ test_that("config only applies to the first driver object for a path", {
   gc()
 })
 
+test_that("user agent is set to r", {
+  skip_if_not(TEST_RE2)
+
+  drv <- duckdb()
+  con <- dbConnect(drv)
+  expect_match(dbGetQuery(con, "PRAGMA user_agent")[1, "user_agent"], "duckdb/.*(.*) r-dbi")
+
+  gc()
+})