job_statistics.h

Go to the documentation of this file.

#pragma once

 
#include "fwd.h"
 
#include <library/cpp/yson/node/node.h>
 
#include <util/system/defaults.h>
#include <util/generic/maybe.h>
#include <util/generic/ptr.h>
 
namespace NYT {


template <typename T>
T ConvertJobStatisticsEntry(i64 value);


class TJobStatistics
{
public:
    TJobStatistics();

    TJobStatistics(const NYT::TNode& statistics);
 
    TJobStatistics(const TJobStatistics& jobStatistics);
    TJobStatistics(TJobStatistics&& jobStatistics) noexcept;
 
    TJobStatistics& operator=(const TJobStatistics& jobStatistics);
    TJobStatistics& operator=(TJobStatistics&& jobStatistics) noexcept;
 
    ~TJobStatistics();

    TJobStatistics TaskName(TVector<TTaskName> taskNames) const;

    TJobStatistics JobState(TVector<EJobState> filter) const;

    TJobStatistics JobType(TVector<EJobType> filter) const;

    bool HasStatistics(TStringBuf name) const;

    TJobStatisticsEntry<i64> GetStatistics(TStringBuf name) const;

    template <typename T>
    TJobStatisticsEntry<T> GetStatisticsAs(TStringBuf name) const;

    TVector<TString> GetStatisticsNames() const;

    bool HasCustomStatistics(TStringBuf name) const;

    TJobStatisticsEntry<i64> GetCustomStatistics(TStringBuf name) const;

    template <typename T>
    TJobStatisticsEntry<T> GetCustomStatisticsAs(TStringBuf name) const;

    TVector<TString> GetCustomStatisticsNames() const;
 
private:
    class TData;
    struct TFilter;
 
    struct TDataEntry {
        i64 Max;
        i64 Min;
        i64 Sum;
        i64 Count;
    };
 
    static const TString CustomStatisticsNamePrefix_;
 
private:
    TJobStatistics(::TIntrusivePtr<TData> data, ::TIntrusivePtr<TFilter> filter);
 
    TMaybe<TDataEntry> GetStatisticsImpl(TStringBuf name) const;
 
private:
    ::TIntrusivePtr<TData> Data_;
    ::TIntrusivePtr<TFilter> Filter_;
 
private:
    template<typename T>
    friend class TJobStatisticsEntry;
};


template <typename T>
class TJobStatisticsEntry
{
public:
    TJobStatisticsEntry(TMaybe<TJobStatistics::TDataEntry> data)
        : Data_(std::move(data))
    { }

    TMaybe<T> Sum() const
    {
        if (Data_) {
            return ConvertJobStatisticsEntry<T>(Data_->Sum);
        }
        return Nothing();
    }

    TMaybe<T> Avg() const
    {
        if (Data_ && Data_->Count) {
            return ConvertJobStatisticsEntry<T>(Data_->Sum / Data_->Count);
        }
        return Nothing();
    }

    TMaybe<T> Count() const
    {
        if (Data_) {
            return ConvertJobStatisticsEntry<T>(Data_->Count);
        }
        return Nothing();
    }

    TMaybe<T> Max() const
    {
        if (Data_) {
            return ConvertJobStatisticsEntry<T>(Data_->Max);
        }
        return Nothing();
    }

    TMaybe<T> Min() const
    {
        if (Data_) {
            return ConvertJobStatisticsEntry<T>(Data_->Min);
        }
        return Nothing();
    }
 
private:
    TMaybe<TJobStatistics::TDataEntry> Data_;
 
private:
    friend class TJobStatistics;
};

 
template <typename T>
TJobStatisticsEntry<T> TJobStatistics::GetStatisticsAs(TStringBuf name) const
{
    return TJobStatisticsEntry<T>(GetStatisticsImpl(name));
}
 
template <typename T>
TJobStatisticsEntry<T> TJobStatistics::GetCustomStatisticsAs(TStringBuf name) const
{
    return TJobStatisticsEntry<T>(GetStatisticsImpl(CustomStatisticsNamePrefix_ + name));
}


void WriteCustomStatistics(TStringBuf path, i64 value);

void WriteCustomStatistics(const TNode& statistics);

void FlushCustomStatisticsStream();
 
} // namespace NYT