S3 implementation of db_compute for Athena

Source: R/dplyr_integration.R

This is a backend function for dplyr's compute function. Users won't be required to access and run this function.

db_compute.AthenaConnection(con, table, sql, ...)

Arguments

con

A dbConnect object, as returned by dbConnect()

table

Table name, if left default RAthena will use the default from dplyr's compute function.

sql

SQL code to be sent to the data

...

passes RAthena table creation parameters: [file_type,s3_location,partition]

  • file_type: What file type to store data.frame on s3, RAthena currently supports ["NULL","csv", "parquet", "json"]. "NULL" will let Athena set the file_type for you.

  • s3_location: s3 bucket to store Athena table, must be set as a s3 uri for example ("s3://mybucket/data/")

  • partition: Partition Athena table, requires to be a partitioned variable from previous table.

Value

db_compute returns table name

Examples

if (FALSE) {
# Note: 
# - Require AWS Account to run below example.
# - Different connection methods can be used please see `RAthena::dbConnect` documentation

library(DBI)
library(dplyr)

# Demo connection to Athena using profile name 
con <- dbConnect(RAthena::athena())

# Write data.frame to Athena table
copy_to(con, mtcars,
        s3_location = "s3://mybucket/data/")
             
# Write Athena table from tbl_sql
athena_mtcars <- tbl(con, "mtcars")
mtcars_filter <- athena_mtcars %>% filter(gear >=4)

# create athena with unique table name
mtcars_filer %>% 
  compute()

# create athena with specified name and s3 location
mtcars_filer %>% 
    compute("mtcars_filer",
            s3_location = "s3://mybucket/mtcars_filer/")

# Disconnect from Athena
dbDisconnect(con)
}