init templates

Author: mapleFU

src/iceberg/CMakeLists.txt

@@ -21,6 +21,7 @@ set(ICEBERG_SOURCES
     arrow_c_data_internal.cc
     catalog/in_memory_catalog.cc
     demo.cc
+    datum.cc
     expression/expression.cc
     file_reader.cc
     json_internal.cc

src/iceberg/datum.cc

@@ -0,0 +1,101 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#include "iceberg/datum.h"
+#include "iceberg/exception.h"
+
+#include <sstream>
+
+namespace iceberg {
+
+// Constructor
+PrimitiveLiteral::PrimitiveLiteral(PrimitiveLiteralValue value, std::shared_ptr<PrimitiveType> type)
+    : value_(std::move(value)), type_(std::move(type)) {}
+
+// Factory methods
+PrimitiveLiteral PrimitiveLiteral::Boolean(bool value) {
+  return PrimitiveLiteral(value, std::make_shared<BooleanType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::Integer(int32_t value) {
+  return PrimitiveLiteral(value, std::make_shared<IntType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::Long(int64_t value) {
+  return PrimitiveLiteral(value, std::make_shared<LongType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::Float(float value) {
+  return PrimitiveLiteral(value, std::make_shared<FloatType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::Double(double value) {
+  return PrimitiveLiteral(value, std::make_shared<DoubleType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::String(std::string value) {
+  return PrimitiveLiteral(std::move(value), std::make_shared<StringType>());
+}
+
+PrimitiveLiteral PrimitiveLiteral::Binary(std::vector<uint8_t> value) {
+  return PrimitiveLiteral(std::move(value), std::make_shared<BinaryType>());
+}
+
+Result<PrimitiveLiteral> PrimitiveLiteral::Deserialize(std::span<const uint8_t> data) {
+  return NotImplemented("Deserialization of PrimitiveLiteral is not implemented yet");
+}
+
+Result<std::vector<uint8_t>> PrimitiveLiteral::Serialize() const {
+  return NotImplemented("Serialization of PrimitiveLiteral is not implemented yet");
+}
+
+// Getters
+const PrimitiveLiteralValue& PrimitiveLiteral::value() const {
+  return value_;
+}
+
+const std::shared_ptr<PrimitiveType>& PrimitiveLiteral::type() const {
+  return type_;
+}
+
+// Cast method
+Result<PrimitiveLiteral> PrimitiveLiteral::CastTo(const std::shared_ptr<PrimitiveType>& target_type) const {
+  if (*type_ == *target_type) {
+    // If types are the same, return a copy of the current literal
+    return PrimitiveLiteral(value_, target_type);
+  }
+
+  return NotImplemented("Cast from {} to {} is not implemented",
+                          type_->ToString(), target_type->ToString());
+
+}
+
+// Three-way comparison operator
+std::partial_ordering PrimitiveLiteral::operator<=>(const PrimitiveLiteral& other) const {
+  // If types are different, comparison is unordered
+  if (type_->type_id() != other.type_->type_id()) {
+    return std::partial_ordering::unordered;
+  }
+  if (value_ == other.value_) {
+    return std::partial_ordering::equivalent;
+  }
+  throw IcebergError("Not implemented: comparison between different primitive types");
+}
+
+}  // namespace iceberg

src/iceberg/datum.h

@@ -0,0 +1,95 @@
+/*
+* Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#pragma once
+
+#include <compare>
+#include <memory>
+#include <string>
+#include <variant>
+#include <vector>
+
+#include "iceberg/type.h"
+#include "iceberg/result.h"
+
+namespace iceberg {
+
+/// \brief Exception type for values that are below the minimum allowed value for a primitive type.
+///
+/// When casting a value to a narrow primitive type, if the value exceeds the maximum of dest type,
+/// it might be above the maximum allowed value for that type.
+struct BelowMin {
+  bool operator==(const BelowMin&) const = default;
+  std::strong_ordering operator<=>(const BelowMin&) const = default;
+};
+
+/// \brief Exception type for values that are above the maximum allowed value for a primitive type.
+///
+/// When casting a value to a narrow primitive type, if the value exceeds the maximum of dest type,
+/// it might be above the maximum allowed value for that type.
+struct AboveMax {
+  bool operator==(const AboveMax&) const = default;
+  std::strong_ordering operator<=>(const AboveMax&) const = default;
+};
+
+// TODO(mwish): Supports More types
+using PrimitiveLiteralValue =
+    std::variant<bool, int32_t, int64_t, float, double, std::string, std::vector<uint8_t>, BelowMin, AboveMax>;
+
+/// \brief PrimitiveLiteral is owned literal of a primitive type.
+class PrimitiveLiteral {
+ public:
+  explicit PrimitiveLiteral(PrimitiveLiteralValue value, std::shared_ptr<PrimitiveType> type);
+
+  // Factory methods for primitive types
+  static PrimitiveLiteral Boolean(bool value);
+  static PrimitiveLiteral Integer(int32_t value);
+  static PrimitiveLiteral Long(int64_t value);
+  static PrimitiveLiteral Float(float value);
+  static PrimitiveLiteral Double(double value);
+  static PrimitiveLiteral String(std::string value);
+  static PrimitiveLiteral Binary(std::vector<uint8_t> value);
+
+  /// Create iceberg value from bytes.
+  ///
+  /// See [this spec](https://iceberg.apache.org/spec/#binary-single-value-serialization) for reference.
+  static Result<PrimitiveLiteral> Deserialize(std::span<const uint8_t> data);
+  /// Serialize iceberg value to bytes.
+  ///
+  /// See [this spec](https://iceberg.apache.org/spec/#binary-single-value-serialization) for reference.
+  Result<std::vector<uint8_t>> Serialize() const;
+
+  // Get the value as a variant
+  const PrimitiveLiteralValue& value() const;
+
+  // Get the Iceberg Type of the literal
+  const std::shared_ptr<PrimitiveType>& type() const;
+
+  // Cast the literal to a specific type
+  Result<PrimitiveLiteral> CastTo(const std::shared_ptr<PrimitiveType>& target_type) const;
+
+  std::partial_ordering operator<=>(const PrimitiveLiteral& other) const;
+
+ private:
+  PrimitiveLiteralValue value_;
+  std::shared_ptr<PrimitiveType> type_;
+};
+
+}  // namespace iceberg
+