R/tpch_raw_20.R

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

# Generated by meta_replay_to_fun_file(), do not edit by hand
tpch_raw_20 <- function(con, experimental) {
  df1 <- nation
  "filter"
  rel1 <- duckdb$rel_from_df(con, df1, experimental = experimental)
  "filter"
  rel2 <- duckdb$rel_filter(
    rel1,
    list(
      duckdb$expr_function(
        "r_base::==",
        list(
          duckdb$expr_reference("n_name"),
          if ("experimental" %in% names(formals(duckdb$expr_constant))) {
            duckdb$expr_constant("CANADA", experimental = experimental)
          } else {
            duckdb$expr_constant("CANADA")
          }
        )
      )
    )
  )
  df2 <- supplier
  "inner_join"
  rel3 <- duckdb$rel_from_df(con, df2, experimental = experimental)
  "inner_join"
  rel4 <- duckdb$rel_set_alias(rel3, "lhs")
  "inner_join"
  rel5 <- duckdb$rel_set_alias(rel2, "rhs")
  "inner_join"
  rel6 <- duckdb$rel_join(
    rel4,
    rel5,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("s_nationkey", rel4), duckdb$expr_reference("n_nationkey", rel5))
      )
    ),
    "inner"
  )
  "inner_join"
  rel7 <- duckdb$rel_project(
    rel6,
    list(
      {
        tmp_expr <- duckdb$expr_reference("s_suppkey")
        duckdb$expr_set_alias(tmp_expr, "s_suppkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_name")
        duckdb$expr_set_alias(tmp_expr, "s_name")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_address")
        duckdb$expr_set_alias(tmp_expr, "s_address")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function(
          "___coalesce",
          list(duckdb$expr_reference("s_nationkey", rel4), duckdb$expr_reference("n_nationkey", rel5))
        )
        duckdb$expr_set_alias(tmp_expr, "s_nationkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_phone")
        duckdb$expr_set_alias(tmp_expr, "s_phone")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_acctbal")
        duckdb$expr_set_alias(tmp_expr, "s_acctbal")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_comment")
        duckdb$expr_set_alias(tmp_expr, "s_comment")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("n_name")
        duckdb$expr_set_alias(tmp_expr, "n_name")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("n_regionkey")
        duckdb$expr_set_alias(tmp_expr, "n_regionkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("n_comment")
        duckdb$expr_set_alias(tmp_expr, "n_comment")
        tmp_expr
      }
    )
  )
  "select"
  rel8 <- duckdb$rel_project(
    rel7,
    list(
      {
        tmp_expr <- duckdb$expr_reference("s_suppkey")
        duckdb$expr_set_alias(tmp_expr, "s_suppkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_name")
        duckdb$expr_set_alias(tmp_expr, "s_name")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_address")
        duckdb$expr_set_alias(tmp_expr, "s_address")
        tmp_expr
      }
    )
  )
  df3 <- part
  "filter"
  rel9 <- duckdb$rel_from_df(con, df3, experimental = experimental)
  "filter"
  rel10 <- duckdb$rel_filter(
    rel9,
    list(
      duckdb$expr_function(
        "grepl",
        list(
          if ("experimental" %in% names(formals(duckdb$expr_constant))) {
            duckdb$expr_constant("^forest", experimental = experimental)
          } else {
            duckdb$expr_constant("^forest")
          },
          duckdb$expr_reference("p_name")
        )
      )
    )
  )
  df4 <- partsupp
  "semi_join"
  rel11 <- duckdb$rel_from_df(con, df4, experimental = experimental)
  "semi_join"
  rel12 <- duckdb$rel_set_alias(rel11, "lhs")
  "semi_join"
  rel13 <- duckdb$rel_set_alias(rel8, "rhs")
  "semi_join"
  rel14 <- duckdb$rel_join(
    rel12,
    rel13,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("ps_suppkey", rel12), duckdb$expr_reference("s_suppkey", rel13))
      )
    ),
    "semi"
  )
  "semi_join"
  rel15 <- duckdb$rel_set_alias(rel14, "lhs")
  "semi_join"
  rel16 <- duckdb$rel_set_alias(rel10, "rhs")
  "semi_join"
  rel17 <- duckdb$rel_join(
    rel15,
    rel16,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("ps_partkey", rel15), duckdb$expr_reference("p_partkey", rel16))
      )
    ),
    "semi"
  )
  df5 <- lineitem
  "filter"
  rel18 <- duckdb$rel_from_df(con, df5, experimental = experimental)
  "filter"
  rel19 <- duckdb$rel_filter(
    rel18,
    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("1994-01-01"), experimental = experimental)
          } else {
            duckdb$expr_constant(as.Date("1994-01-01"))
          }
        )
      ),
      duckdb$expr_function(
        "r_base::<",
        list(
          duckdb$expr_reference("l_shipdate"),
          if ("experimental" %in% names(formals(duckdb$expr_constant))) {
            duckdb$expr_constant(as.Date("1995-01-01"), experimental = experimental)
          } else {
            duckdb$expr_constant(as.Date("1995-01-01"))
          }
        )
      )
    )
  )
  "semi_join"
  rel20 <- duckdb$rel_set_alias(rel19, "lhs")
  "semi_join"
  rel21 <- duckdb$rel_set_alias(rel17, "rhs")
  "semi_join"
  rel22 <- duckdb$rel_join(
    rel20,
    rel21,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("l_partkey", rel20), duckdb$expr_reference("ps_partkey", rel21))
      ),
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("l_suppkey", rel20), duckdb$expr_reference("ps_suppkey", rel21))
      )
    ),
    "semi"
  )
  "summarise"
  rel23 <- duckdb$rel_aggregate(
    rel22,
    groups = list(duckdb$expr_reference("l_suppkey")),
    aggregates = list(
      {
        tmp_expr <- duckdb$expr_function(
          "*",
          list(
            if ("experimental" %in% names(formals(duckdb$expr_constant))) {
              duckdb$expr_constant(0.5, experimental = experimental)
            } else {
              duckdb$expr_constant(0.5)
            },
            duckdb$expr_function("sum", list(duckdb$expr_reference("l_quantity")))
          )
        )
        duckdb$expr_set_alias(tmp_expr, "qty_threshold")
        tmp_expr
      }
    )
  )
  "inner_join"
  rel24 <- duckdb$rel_set_alias(rel17, "lhs")
  "inner_join"
  rel25 <- duckdb$rel_set_alias(rel23, "rhs")
  "inner_join"
  rel26 <- duckdb$rel_join(
    rel24,
    rel25,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("ps_suppkey", rel24), duckdb$expr_reference("l_suppkey", rel25))
      )
    ),
    "inner"
  )
  "inner_join"
  rel27 <- duckdb$rel_project(
    rel26,
    list(
      {
        tmp_expr <- duckdb$expr_reference("ps_partkey")
        duckdb$expr_set_alias(tmp_expr, "ps_partkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_function(
          "___coalesce",
          list(duckdb$expr_reference("ps_suppkey", rel24), duckdb$expr_reference("l_suppkey", rel25))
        )
        duckdb$expr_set_alias(tmp_expr, "ps_suppkey")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("ps_availqty")
        duckdb$expr_set_alias(tmp_expr, "ps_availqty")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("ps_supplycost")
        duckdb$expr_set_alias(tmp_expr, "ps_supplycost")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("ps_comment")
        duckdb$expr_set_alias(tmp_expr, "ps_comment")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("qty_threshold")
        duckdb$expr_set_alias(tmp_expr, "qty_threshold")
        tmp_expr
      }
    )
  )
  "filter"
  rel28 <- duckdb$rel_filter(
    rel27,
    list(
      duckdb$expr_function(
        "r_base::>",
        list(duckdb$expr_reference("ps_availqty"), duckdb$expr_reference("qty_threshold"))
      )
    )
  )
  "semi_join"
  rel29 <- duckdb$rel_set_alias(rel8, "lhs")
  "semi_join"
  rel30 <- duckdb$rel_set_alias(rel28, "rhs")
  "semi_join"
  rel31 <- duckdb$rel_join(
    rel29,
    rel30,
    list(
      duckdb$expr_function(
        "==",
        list(duckdb$expr_reference("s_suppkey", rel29), duckdb$expr_reference("ps_suppkey", rel30))
      )
    ),
    "semi"
  )
  "select"
  rel32 <- duckdb$rel_project(
    rel31,
    list(
      {
        tmp_expr <- duckdb$expr_reference("s_name")
        duckdb$expr_set_alias(tmp_expr, "s_name")
        tmp_expr
      },
      {
        tmp_expr <- duckdb$expr_reference("s_address")
        duckdb$expr_set_alias(tmp_expr, "s_address")
        tmp_expr
      }
    )
  )
  "arrange"
  rel33 <- duckdb$rel_order(rel32, list(duckdb$expr_reference("s_name")))
  rel33
  duckdb$rel_to_altrep(rel33)
}