feat: update partition spec

Author: zhjwpku

src/iceberg/CMakeLists.txt

@@ -66,6 +66,7 @@ set(ICEBERG_SOURCES
     transform.cc
     transform_function.cc
     type.cc
+    update_partition_spec.cc
     util/bucket_util.cc
     util/conversions.cc
     util/decimal.cc

src/iceberg/meson.build

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

src/iceberg/table.cc

@@ -28,6 +28,7 @@
 #include "iceberg/table_metadata.h"
 #include "iceberg/table_properties.h"
 #include "iceberg/table_scan.h"
+#include "iceberg/update_partition_spec.h"
 #include "iceberg/util/macros.h"
 
 namespace iceberg {
@@ -94,6 +95,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();
 }
@@ -110,6 +113,10 @@ const std::vector<SnapshotLogEntry>& Table::history() const {
   return metadata_->snapshot_log;
 }
 
+std::unique_ptr<UpdatePartitionSpec> Table::UpdateSpec() {
+  return std::make_unique<UpdatePartitionSpec>(this);
+}
+
 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;
 
@@ -104,6 +107,11 @@ class ICEBERG_EXPORT Table {
   /// \return a vector of history entries
   const std::vector<SnapshotLogEntry>& history() 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
+  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/type_fwd.h

@@ -161,6 +161,16 @@ class PendingUpdate;
 template <typename T>
 class PendingUpdateTyped;
 
+class BoundPredicate;
+class UnboundPredicate;
+
+class BoundReference;
+class BoundTransform;
+template <typename B>
+class UnboundTerm;
+
+class UpdatePartitionSpec;
+
 /// ----------------------------------------------------------------------------
 /// TODO: Forward declarations below are not added yet.
 /// ----------------------------------------------------------------------------