feat: update partition spec

Author: zhjwpku

src/iceberg/CMakeLists.txt

@@ -72,6 +72,7 @@ set(ICEBERG_SOURCES
     transform.cc
     transform_function.cc
     type.cc
+    update/update_partition_spec.cc
     update/update_properties.cc
     util/bucket_util.cc
     util/conversions.cc

src/iceberg/meson.build

@@ -94,6 +94,7 @@ iceberg_sources = files(
     'transform.cc',
     'transform_function.cc',
     'type.cc',
+    'update/update_partition_spec.cc',
     'update/update_properties.cc',
     'util/bucket_util.cc',
     'util/conversions.cc',

src/iceberg/table.cc

@@ -26,6 +26,7 @@
 #include "iceberg/table_metadata.h"
 #include "iceberg/table_properties.h"
 #include "iceberg/table_scan.h"
+#include "iceberg/update/update_partition_spec.h"
 #include "iceberg/update/update_properties.h"
 #include "iceberg/util/macros.h"
 
@@ -93,6 +94,8 @@ const TableProperties& Table::properties() const { return *properties_; }
 
 const std::string& Table::location() const { return metadata_->location; }
 
+const TableMetadata& Table::metadata() const { return *metadata_; }
+
 Result<std::shared_ptr<Snapshot>> Table::current_snapshot() const {
   return metadata_->Snapshot();
 }
@@ -113,6 +116,10 @@ std::unique_ptr<UpdateProperties> Table::UpdateProperties() const {
   return std::make_unique<iceberg::UpdateProperties>(identifier_, catalog_, metadata_);
 }
 
+std::unique_ptr<UpdatePartitionSpec> Table::UpdateSpec() {
+  return std::make_unique<UpdatePartitionSpec>(identifier_, catalog_, metadata_);
+}
+
 std::unique_ptr<Transaction> Table::NewTransaction() const {
   throw NotImplemented("Table::NewTransaction is not implemented");
 }

src/iceberg/table.h

@@ -87,6 +87,9 @@ class ICEBERG_EXPORT Table {
   /// \brief Return the table's base location
   const std::string& location() const;
 
+  /// \brief Return the table's metadata
+  const TableMetadata& metadata() const;
+
   /// \brief Return the table's current snapshot, return NotFoundError if not found
   Result<std::shared_ptr<Snapshot>> current_snapshot() const;
 
@@ -110,6 +113,11 @@ class ICEBERG_EXPORT Table {
   /// \return a new UpdateProperties instance
   virtual std::unique_ptr<iceberg::UpdateProperties> UpdateProperties() const;
 
+  /// \brief Create a new UpdatePartitionSpec to alter the partition spec of this table
+  /// and commit the changes.
+  /// \return a pointer to the new UpdatePartitionSpec
+  virtual std::unique_ptr<UpdatePartitionSpec> UpdateSpec();
+
   /// \brief Create a new table scan builder for this table
   ///
   /// Once a table scan builder is created, it can be refined to project columns and

src/iceberg/test/CMakeLists.txt

@@ -77,6 +77,7 @@ add_iceberg_test(table_test
                  table_requirement_test.cc
                  table_requirements_test.cc
                  table_update_test.cc
+                 update_partition_spec_test.cc
                  update_properties_test.cc)
 
 add_iceberg_test(expression_test

src/iceberg/test/meson.build

@@ -53,6 +53,7 @@ iceberg_tests = {
             'table_requirement_test.cc',
             'table_test.cc',
             'table_update_test.cc',
+            'update_partition_spec_test.cc',
             'update_properties_test.cc',
         ),
     },

src/iceberg/test/mock_catalog.h

@@ -23,6 +23,8 @@
 #include <gtest/gtest.h>
 
 #include "iceberg/catalog.h"
+#include "iceberg/table.h"
+#include "iceberg/transaction.h"
 
 namespace iceberg {