generated from CDCgov/template
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add nullable filter helpers and use in scoring filter and soql query …
…generation (#33)
- Loading branch information
1 parent
c5bc613
commit e42ff17
Showing
13 changed files
with
370 additions
and
129 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,139 @@ | ||
| #' Compare two quantities `a` and `b` with a given | ||
| #' `comparison operator`, | ||
| #' returning a single `TRUE` if `b` is | ||
| #' `NULL`. | ||
| #' | ||
| #' Useful for letting `NULL` reflect "all values" | ||
| #' if [dplyr::filter()] calls. Internal function. | ||
| #' | ||
| #' @param a First set of values for the comparison | ||
| #' @param comparison_operator Comparison operator for the comparison, | ||
| #' as a string, e.g. `"=="`, `">="` `"%in%"`. | ||
| #' @param b Second set of values for the comparison, or `NULL`. | ||
| #' @return A logical vector. Equivalent to `b {comparison_operator} a` | ||
| #' if `b` is not `NULL` and to `TRUE` if `b` is `NULL`. | ||
| #' #' | ||
| #' @examples | ||
| #' | ||
| #' x <- 6 | ||
| #' nullable_comparison(5, ">", x) | ||
| #' | ||
| #' x <- NULL | ||
| #' nullable_comparison(5, ">", x) | ||
| #' | ||
| #' df <- tibble::tibble(y = 1:6) | ||
| #' x <- 3 | ||
| #' df |> dplyr::filter( | ||
| #' nullable_comparison(y, ">", !!x), | ||
| #' y < 5 | ||
| #' ) | ||
| #' x <- NULL | ||
| #' df |> dplyr::filter( | ||
| #' nullable_comparison(y, ">", !!x), | ||
| #' y < 5 | ||
| #' ) | ||
| #' | ||
| #' @export | ||
| nullable_comparison <- function(a, | ||
| comparison_operator, | ||
| b) { | ||
| comparison_func <- getFunction(comparison_operator) | ||
| return( | ||
| if (!is.null(b)) { | ||
| comparison_func(a, b) | ||
| } else { | ||
| TRUE | ||
| } | ||
| ) | ||
| } | ||
|
|
||
|
|
||
| #' Add a [soql::soql_where()] statement to a SOQL query, | ||
| #' or return the original query if `where_value` is `NULL`. | ||
| #' | ||
| #' @param soql_list A `soql` query object, which | ||
| #' can be piped in. If one hasn't been | ||
| #' created yet, use or pipe in [soql::soql()]. | ||
| #' @param column column name for the [soql::soql_where()] component | ||
| #' of the query, as a string. | ||
| #' @param comparison_operator comparison operator for the | ||
| #' [soql::soql_where()] component of the query, as a string. | ||
| #' @param where_value A value for the comparison, or `NULL` | ||
| #' for no filtering. | ||
| #' @return A new [soql::soql()] object with the filter added, or | ||
| #' simply the input object if `where_value` is `NULL`. | ||
| #' @export | ||
| soql_nullable_where <- function(soql_list, | ||
| column, | ||
| comparison_operator, | ||
| where_value) { | ||
| return( | ||
| if (!is.null(where_value)) { | ||
| soql::soql_where( | ||
| soql_list, | ||
| glue::glue(paste0( | ||
| "{column} ", | ||
| "{comparison_operator} ", | ||
| "'{where_value}'" | ||
| )) | ||
| ) | ||
| } else { | ||
| soql_list | ||
| } | ||
| ) | ||
| } | ||
|
|
||
|
|
||
| #' Add an "is in" statement to a SOQL query, or return the original | ||
| #' query if `match_values` is `NULL`. | ||
| #' | ||
| #' An is in statement is a [soql::soql_where()] statement | ||
| #' that requires the values of a given column to match one of the | ||
| #' entries of a vector of `match_values`. | ||
| #' | ||
| #' @param soql_list A `soql` query object, which | ||
| #' can be piped in. If one hasn't been | ||
| #' created yet, use or pipe in [soql::soql()]. | ||
| #' @param column The column to filter on | ||
| #' @param match_values A vector of values that column | ||
| #' must match, or `NULL` for no filtering. | ||
| #' @return A new [soql::soql()] object with the filter added, | ||
| #' or simply the input object if `match_value` is `NULL`. | ||
| #' @export | ||
| soql_nullable_is_in <- function(soql_list, column, match_values) { | ||
| if (is.null(match_values)) { | ||
| return(soql_list) | ||
| } else { | ||
| query <- glue::glue( | ||
| "{column}='{unique(match_values)}'" | ||
| ) |> | ||
| paste(collapse = " OR ") | ||
| return(soql::soql_where(soql_list, query)) | ||
| } | ||
| } | ||
|
|
||
| #' Add a [soql::soql_select()] statement to a query | ||
| #' to select a set of columns `columns`, or return | ||
| #' the original query if `columns` is `NULL`. | ||
| #' | ||
| #' @param soql_list A `soql` query object, which | ||
| #' can be piped in. If one hasn't been | ||
| #' created yet, use or pipe in [soql::soql()]. | ||
| #' @param columns The columns to select, or `NULL`. | ||
| #' @return A new [soql::soql()] object with the selection statement | ||
| #' added, or the input object if `columns` is `NULL`. | ||
| #' @export | ||
| soql_nullable_select <- function(soql_list, columns) { | ||
| return( | ||
| if (!is.null(columns)) { | ||
| soql::soql_select( | ||
| soql_list, | ||
| paste(unique(columns), | ||
| collapse = "," | ||
| ) | ||
| ) | ||
| } else { | ||
| soql_list | ||
| } | ||
| ) | ||
| } |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Oops, something went wrong.