cloud/blockstore/libs/common/block_range.cpp - main - githubmirror/nbs

cloud/blockstore/libs/common/block_range.cpp

69 строк2 KB

Kirill Pleshivtsev

Print TBlockRange the same way (#2638)

10 дек 2024, 12:02

10 дек 2024, 12:0289c75e4

#include "block_range.h"

#include <util/generic/strbuf.h>
#include <util/string/builder.h>
#include <util/string/cast.h>

namespace NCloud::NBlockStore {

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

template <std::unsigned_integral TBlockIndex>
bool TBlockRange<TBlockIndex>::TryParse(
    TStringBuf s,
    TBlockRange<TBlockIndex>& range)
{
    TStringBuf l, r;
    if (s.TrySplit(':', l, r)) {
        TBlockIndex start = 0, end = 0;
        if (TryFromString(l, start) && TryFromString(r, end) && start <= end) {
            range = TBlockRange<TBlockIndex>::MakeClosedInterval(start, end);
            return true;
        }
    } else {
        TBlockIndex blockIndex = 0;
        if (TryFromString(s, blockIndex)) {
            range = TBlockRange<TBlockIndex>::MakeOneBlock(blockIndex);
            return true;
        }
    }
    return false;
}

template bool TBlockRange32::TryParse(TStringBuf s, TBlockRange32& range);
template bool TBlockRange64::TryParse(TStringBuf s, TBlockRange64& range);

template <std::unsigned_integral TBlockIndex>
TString TBlockRange<TBlockIndex>::Print() const
{
    return TStringBuilder() << "[" << Start << ".." << End << "]";
}

template TString TBlockRange32::Print() const;
template TString TBlockRange64::Print() const;

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

template <std::unsigned_integral TBlockIndex>
TString DescribeRange(const TBlockRange<TBlockIndex>& blockRange)
{
    return blockRange.Print();
}

template TString DescribeRange(const TBlockRange32& blockRange);
template TString DescribeRange(const TBlockRange64& blockRange);

template <std::unsigned_integral TBlockIndex>
TString DescribeRange(const TVector<TBlockIndex>& blocks)
{
    if (blocks) {
        return TStringBuilder()
            << "[" << blocks.front() << ".." << blocks.back() << "]";
    }
    return "<none>";
}

template TString DescribeRange(const TVector<ui32>& blocks);
template TString DescribeRange(const TVector<ui64>& blocks);

}   // namespace NCloud::NBlockStore