feat: Add temporary option to CreateTableBuilder (#398)

Author: nuno-faria

core/src/sql/arrow_sql_gen/statement.rs

@@ -37,6 +37,7 @@ pub struct CreateTableBuilder {
     schema: SchemaRef,
     table_name: String,
     primary_keys: Vec<String>,
+    temporary: bool,
 }
 
 impl CreateTableBuilder {
@@ -46,6 +47,7 @@ impl CreateTableBuilder {
             schema,
             table_name: table_name.to_string(),
             primary_keys: Vec::new(),
+            temporary: false,
         }
     }
 
@@ -58,6 +60,13 @@ impl CreateTableBuilder {
         self
     }
 
+    #[must_use]
+    /// Set whether the table is temporary or not.
+    pub fn temporary(mut self, temporary: bool) -> Self {
+        self.temporary = temporary;
+        self
+    }
+
     #[must_use]
     #[cfg(feature = "postgres")]
     pub fn build_postgres(self) -> Vec<String> {
@@ -145,6 +154,10 @@ impl CreateTableBuilder {
             create_stmt.primary_key(&mut index);
         }
 
+        if self.temporary {
+            create_stmt.temporary();
+        }
+
         create_stmt.to_string(query_builder)
     }
 }
@@ -1550,6 +1563,20 @@ mod tests {
         assert_eq!(sql, "CREATE TABLE IF NOT EXISTS \"users\" ( \"id\" integer NOT NULL, \"id2\" integer NOT NULL, \"name\" text NOT NULL, \"age\" integer, PRIMARY KEY (\"id\", \"id2\") )");
     }
 
+    #[test]
+    fn test_temporary_table_creation() {
+        let schema = Schema::new(vec![
+            Field::new("id", DataType::Int32, false),
+            Field::new("name", DataType::Utf8, false),
+        ]);
+        let sql = CreateTableBuilder::new(SchemaRef::new(schema), "users")
+            .primary_keys(vec!["id"])
+            .temporary(true)
+            .build_sqlite();
+
+        assert_eq!(sql, "CREATE TEMPORARY TABLE IF NOT EXISTS \"users\" ( \"id\" integer NOT NULL, \"name\" text NOT NULL, PRIMARY KEY (\"id\") )");
+    }
+
     #[test]
     fn test_table_insertion_with_list() {
         let schema1 = Schema::new(vec![Field::new(