feat: add alias() method for DataFrame

datafusion/core/src/dataframe/mod.rs

@@ -1903,6 +1903,17 @@ impl DataFrame {
         let mem_table = MemTable::try_new(schema, partitions)?;
         context.read_table(Arc::new(mem_table))
     }
+
+    /// Apply an alias to the DataFrame.
+    ///
+    /// This method replaces the qualifiers of output columns with the given alias.
+    pub fn alias(self, alias: &str) -> Result<DataFrame> {
+        let plan = LogicalPlanBuilder::from(self.plan).alias(alias)?.build()?;
+        Ok(DataFrame {
+            session_state: self.session_state,
+            plan,
+        })
+    }
 }
 
 #[derive(Debug)]

datafusion/core/tests/dataframe/mod.rs

@@ -2646,3 +2646,58 @@ async fn boolean_dictionary_as_filter() {
 
     assert_batches_eq!(expected, &result_df.collect().await.unwrap());
 }
+
+#[tokio::test]
+async fn test_alias() -> Result<()> {
+    let df = create_test_table("test")
+        .await?
+        .select(vec![col("a"), col("test.b"), lit(1).alias("one")])?
+        .alias("table_alias")?;
+    // All ouput column qualifiers are changed to "table_alias"
+    df.schema().columns().iter().for_each(|c| {
+        assert_eq!(c.relation, Some("table_alias".into()));
+    });
+    let expected = "SubqueryAlias: table_alias [a:Utf8, b:Int32, one:Int32]\
+     \n  Projection: test.a, test.b, Int32(1) AS one [a:Utf8, b:Int32, one:Int32]\
+     \n    TableScan: test [a:Utf8, b:Int32]";
+    let plan = df
+        .clone()
+        .into_unoptimized_plan()
+        .display_indent_schema()
+        .to_string();
+    assert_eq!(plan, expected);
+
+    // Select over the aliased DataFrame
+    let df = df.select(vec![
+        col("table_alias.a"),
+        col("b") + col("table_alias.one"),
+    ])?;
+    let expected = [
+        "+-----------+---------------------------------+",
+        "| a         | table_alias.b + table_alias.one |",
+        "+-----------+---------------------------------+",
+        "| abcDEF    | 2                               |",
+        "| abc123    | 11                              |",
+        "| CBAdef    | 11                              |",
+        "| 123AbcDef | 101                             |",
+        "+-----------+---------------------------------+",
+    ];
+    assert_batches_sorted_eq!(expected, &df.collect().await?);
+
+    // Use alias to perform a self-join
+    let left = create_test_table("t1").await?;
+    let right = left.clone().alias("t2")?;
+    let joined = left.join(right, JoinType::Full, &["a"], &["a"], None)?;
+    let expected = [
+        "+-----------+-----+-----------+-----+",
+        "| a         | b   | a         | b   |",
+        "+-----------+-----+-----------+-----+",
+        "| abcDEF    | 1   | abcDEF    | 1   |",
+        "| abc123    | 10  | abc123    | 10  |",
+        "| CBAdef    | 10  | CBAdef    | 10  |",
+        "| 123AbcDef | 100 | 123AbcDef | 100 |",
+        "+-----------+-----+-----------+-----+",
+    ];
+    assert_batches_sorted_eq!(expected, &joined.collect().await?);
+    Ok(())
+}