lidavidm commented on code in PR #31: URL: https://github.com/apache/iceberg-cpp/pull/31#discussion_r1919669923
########## src/iceberg/type.h: ########## @@ -0,0 +1,120 @@ +/* + * 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 + +/// \file iceberg/type.h +/// Data types for Iceberg. + +#include <memory> +#include <span> +#include <string> +#include <vector> + +#include "iceberg/iceberg_export.h" +#include "iceberg/type_fwd.h" + +namespace iceberg { + +/// \brief Interface for a data type for a field. +class ICEBERG_EXPORT Type { + public: + virtual ~Type() = default; + + /// \brief Get the type ID. + [[nodiscard]] virtual TypeId type_id() const = 0; + + /// \brief Get a user-readable string representation of the type. + [[nodiscard]] virtual std::string ToString() const = 0; + + /// \brief Compare two types for equality. + [[nodiscard]] virtual bool Equals(const Type& other) const = 0; + + friend bool operator==(const Type& lhs, const Type& rhs) { return lhs.Equals(rhs); } + + friend bool operator!=(const Type& lhs, const Type& rhs) { return !(lhs == rhs); } +}; + +/// \brief A type combined with a name. +class ICEBERG_EXPORT Field { + public: + Field(int32_t field_id, std::string name, std::shared_ptr<Type> type, bool optional); + + static Field MakeOptional(int32_t field_id, std::string name, + std::shared_ptr<Type> type); + static Field MakeRequired(int32_t field_id, std::string name, + std::shared_ptr<Type> type); + + /// \brief Get the field ID. + [[nodiscard]] int32_t field_id() const; + + /// \brief Get the field name. + [[nodiscard]] std::string_view name() const; + + /// \brief Get the field type. + [[nodiscard]] const std::shared_ptr<Type>& type() const; + + /// \brief Get whether the field is optional. + [[nodiscard]] bool optional() const; + + /// \brief Get a user-readable string representation of the field. + [[nodiscard]] std::string ToString() const; + + /// \brief Compare two fields for equality. + [[nodiscard]] bool Equals(const Field& other) const; + + friend bool operator==(const Field& lhs, const Field& rhs) { return lhs.Equals(rhs); } + + friend bool operator!=(const Field& lhs, const Field& rhs) { return !(lhs == rhs); } +}; + +/// \brief A data type representing a boolean. +class ICEBERG_EXPORT BooleanType : public Type { + public: + BooleanType() = default; + ~BooleanType() = default; + + TypeId type_id() const override; + + std::string ToString() const override; + + bool Equals(const Type& other) const override; +}; + +/// \brief A data type representing a struct with nested fields. +class ICEBERG_EXPORT StructType : public Type { + public: + explicit StructType(std::vector<Field> fields); + + StructType() = default; + ~StructType() = default; + + [[nodiscard]] std::span<Field> fields() const; + + const Field& field(int i) const; + const Field& field(std::string_view name) const; Review Comment: Changed to GetFieldBy{Index,Name} -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@iceberg.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@iceberg.apache.org For additional commands, e-mail: issues-h...@iceberg.apache.org
