feat: Add API to set location in the transacation

crates/iceberg/src/transaction/mod.rs

@@ -183,6 +183,12 @@ impl<'a> Transaction<'a> {
         Ok(self)
     }
 
+    /// Set the location of table
+    pub fn set_location(mut self, location: String) -> Result<Self> {
+        self.apply(vec![TableUpdate::SetLocation { location }], vec![])?;
+        Ok(self)
+    }
+
     /// Commit transaction.
     pub async fn commit(self, catalog: &dyn Catalog) -> Result<Table> {
         let table_commit = TableCommit::builder()
@@ -335,6 +341,22 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_set_location() {
+        let table = make_v2_table();
+        let tx = Transaction::new(&table);
+        let tx = tx
+            .set_location(String::from("s3://bucket/prefix/new_table"))
+            .unwrap();
+
+        assert_eq!(
+            vec![TableUpdate::SetLocation {
+                location: String::from("s3://bucket/prefix/new_table")
+            }],
+            tx.updates
+        )
+    }
+
     #[tokio::test]
     async fn test_transaction_apply_upgrade() {
         let table = make_v1_table();