ydb/library/cpp/yt/misc/source_location.cpp

#include "source_location.h"

#include <library/cpp/yt/string/format.h>

#include <string.h>

namespace NYT {

////////////////////////////////////////////////////////////////////////////////

#ifdef __cpp_lib_source_location

void FormatValue(TStringBuilderBase* builder, const std::source_location& location, TStringBuf /*spec*/)
{
    if (location.file_name() != nullptr) {
        builder->AppendFormat(
            "%v:%v:%v",
            location.file_name(),
            location.line(),
            location.column());
    } else {
        builder->AppendString("<unknown>");
    }
}

#endif // __cpp_lib_source_location

////////////////////////////////////////////////////////////////////////////////

const char* TSourceLocation::GetFileName() const
{
    return FileName_;
}

int TSourceLocation::GetLine() const
{
    return Line_;
}

bool TSourceLocation::IsValid() const
{
    return FileName_ != nullptr;
}

bool TSourceLocation::operator<(const TSourceLocation& other) const
{
    const char* fileName = FileName_ ? FileName_ : "";
    const char* otherFileName = other.FileName_ ? other.FileName_ : "";
    int fileNameResult = strcmp(fileName, otherFileName);
    if (fileNameResult != 0) {
        return fileNameResult < 0;
    }

    if (Line_ < other.Line_) {
        return true;
    }
    if (Line_ > other.Line_) {
        return false;
    }

    return false;
}

bool TSourceLocation::operator==(const TSourceLocation& other) const
{
    const char* fileName = FileName_ ? FileName_ : "";
    const char* otherFileName = other.FileName_ ? other.FileName_ : "";
    return
        strcmp(fileName, otherFileName) == 0 &&
        Line_ == other.Line_;
}

#ifdef __cpp_lib_source_location
TSourceLocation TSourceLocation::FromStd(const std::source_location& location)
{
    return TSourceLocation(location.file_name(), location.line());
}
#endif // __cpp_lib_source_location

void FormatValue(TStringBuilderBase* builder, const TSourceLocation& location, TStringBuf /*spec*/)
{
    if (location.GetFileName() != nullptr) {
        builder->AppendFormat(
            "%v:%v",
            location.GetFileName(),
            location.GetLine());
    } else {
        builder->AppendString("<unknown>");
    }
}

////////////////////////////////////////////////////////////////////////////////

} // namespace NYT