job_counters.h

#pragma once
 
#include "fwd.h"
 
#include <yt/cpp/mapreduce/interface/node.h>
 
namespace NYT {
 
class TJobCounter
{
private:
    TNode Data_;
    i64 Total_ = 0;
 
public:
    TJobCounter() = default;
 
    TJobCounter(TNode data);
    TJobCounter(i64 total);
 
    i64 GetTotal() const;
 
    i64 GetValue(const TStringBuf key) const;
};

class TJobCounters
{
public:
    TJobCounters() = default;

    TJobCounters(const TNode& counters);
 
    const TJobCounter& GetAborted() const;
    const TJobCounter& GetAbortedScheduled() const;
    const TJobCounter& GetAbortedNonScheduled() const;
    const TJobCounter& GetCompleted() const;
    const TJobCounter& GetCompletedNonInterrupted() const;
    const TJobCounter& GetCompletedInterrupted() const;
    const TJobCounter& GetLost() const;
    const TJobCounter& GetInvalidated() const;
    const TJobCounter& GetFailed() const;
    const TJobCounter& GetRunning() const;
    const TJobCounter& GetSuspended() const;
    const TJobCounter& GetPending() const;
    const TJobCounter& GetBlocked() const;
 
    i64 GetTotal() const;
 
private:
    i64 Total_ = 0;
 
    TJobCounter Aborted_;
    TJobCounter AbortedScheduled_;
    TJobCounter AbortedNonScheduled_;
    TJobCounter Completed_;
    TJobCounter CompletedNonInterrupted_;
    TJobCounter CompletedInterrupted_;
    TJobCounter Lost_;
    TJobCounter Invalidated_;
    TJobCounter Failed_;
    TJobCounter Running_;
    TJobCounter Suspended_;
    TJobCounter Pending_;
    TJobCounter Blocked_;
};

 
} // namespace NYT