R/tpch_raw_01.R

In duckdblabs/duckplyr: A 'DuckDB'-Backed Version of 'dplyr'

# Generated by meta_replay_to_fun_file(), do not edit by hand
tpch_raw_01 <- function(con, experimental) {
  df1 <- lineitem
  "select"
  rel1 <- duckdb$rel_from_df(con, df1, experimental = experimental)
  "select"
  rel2 <- duckdb$rel_project(
    rel1,
    list(
      {
        tmp_expr <- duckdb$expr_reference("l_shipdate")
        duckdb$expr_set_alias(tmp_expr, "l_shipdate")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_returnflag")
        duckdb$expr_set_alias(tmp_expr, "l_returnflag")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_linestatus")
        duckdb$expr_set_alias(tmp_expr, "l_linestatus")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_quantity")
        duckdb$expr_set_alias(tmp_expr, "l_quantity")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_extendedprice")
        duckdb$expr_set_alias(tmp_expr, "l_extendedprice")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_discount")
        duckdb$expr_set_alias(tmp_expr, "l_discount")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_tax")
        duckdb$expr_set_alias(tmp_expr, "l_tax")
        tmp_expr
      }
    )
  )
  "filter"
  rel3 <- duckdb$rel_filter(
    rel2,
    list(
      duckdb$expr_function(
        "r_base::<=",
        list(
          duckdb$expr_reference("l_shipdate"),
          if ("experimental" %in% names(formals(duckdb$expr_constant))) {
            duckdb$expr_constant(as.Date("1998-09-02"), experimental = experimental)
          } else {
            duckdb$expr_constant(as.Date("1998-09-02"))
          }
        )
      )
    )
  )
  "select"
  rel4 <- duckdb$rel_project(
    rel3,
    list(
      {
        tmp_expr <- duckdb$expr_reference("l_returnflag")
        duckdb$expr_set_alias(tmp_expr, "l_returnflag")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_linestatus")
        duckdb$expr_set_alias(tmp_expr, "l_linestatus")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_quantity")
        duckdb$expr_set_alias(tmp_expr, "l_quantity")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_extendedprice")
        duckdb$expr_set_alias(tmp_expr, "l_extendedprice")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_discount")
        duckdb$expr_set_alias(tmp_expr, "l_discount")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("l_tax")
        duckdb$expr_set_alias(tmp_expr, "l_tax")
        tmp_expr
      }
    )
  )
  "summarise"
  rel5 <- duckdb$rel_aggregate(
    rel4,
    groups = list(duckdb$expr_reference("l_returnflag"), duckdb$expr_reference("l_linestatus")),
    aggregates = list(
      {
        tmp_expr <- duckdb$expr_function("sum", list(duckdb$expr_reference("l_quantity")))
        duckdb$expr_set_alias(tmp_expr, "sum_qty")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function("sum", list(duckdb$expr_reference("l_extendedprice")))
        duckdb$expr_set_alias(tmp_expr, "sum_base_price")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function(
          "sum",
          list(
            duckdb$expr_function(
              "*",
              list(
                duckdb$expr_reference("l_extendedprice"),
                duckdb$expr_function(
                  "-",
                  list(
                    if ("experimental" %in% names(formals(duckdb$expr_constant))) {
                      duckdb$expr_constant(1, experimental = experimental)
                    } else {
                      duckdb$expr_constant(1)
                    },
                    duckdb$expr_reference("l_discount")
                  )
                )
              )
            )
          )
        )
        duckdb$expr_set_alias(tmp_expr, "sum_disc_price")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function(
          "sum",
          list(
            duckdb$expr_function(
              "*",
              list(
                duckdb$expr_function(
                  "*",
                  list(
                    duckdb$expr_reference("l_extendedprice"),
                    duckdb$expr_function(
                      "-",
                      list(
                        if ("experimental" %in% names(formals(duckdb$expr_constant))) {
                          duckdb$expr_constant(1, experimental = experimental)
                        } else {
                          duckdb$expr_constant(1)
                        },
                        duckdb$expr_reference("l_discount")
                      )
                    )
                  )
                ),
                duckdb$expr_function(
                  "+",
                  list(
                    if ("experimental" %in% names(formals(duckdb$expr_constant))) {
                      duckdb$expr_constant(1, experimental = experimental)
                    } else {
                      duckdb$expr_constant(1)
                    },
                    duckdb$expr_reference("l_tax")
                  )
                )
              )
            )
          )
        )
        duckdb$expr_set_alias(tmp_expr, "sum_charge")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function("mean", list(duckdb$expr_reference("l_quantity")))
        duckdb$expr_set_alias(tmp_expr, "avg_qty")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function("mean", list(duckdb$expr_reference("l_extendedprice")))
        duckdb$expr_set_alias(tmp_expr, "avg_price")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function("mean", list(duckdb$expr_reference("l_discount")))
        duckdb$expr_set_alias(tmp_expr, "avg_disc")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function("n", list())
        duckdb$expr_set_alias(tmp_expr, "count_order")
        tmp_expr
      }
    )
  )
  "arrange"
  rel6 <- duckdb$rel_order(
    rel5,
    list(duckdb$expr_reference("l_returnflag"), duckdb$expr_reference("l_linestatus"))
  )
  rel6
  duckdb$rel_to_altrep(rel6)
}