1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Copyright Epic Games, Inc. All Rights Reserved.
#include <zenremotestore/operationlogoutput.h>
#include <zencore/logging.h>
ZEN_THIRD_PARTY_INCLUDES_START
#include <gsl/gsl-lite.hpp>
ZEN_THIRD_PARTY_INCLUDES_END
namespace zen {
class StandardLogOutput;
class StandardLogOutputProgressBar : public OperationLogOutput::ProgressBar
{
public:
StandardLogOutputProgressBar(StandardLogOutput& Output, std::string_view InSubTask) : m_Output(Output), m_SubTask(InSubTask) {}
virtual void UpdateState(const State& NewState, bool DoLinebreak) override;
virtual void Finish() override;
private:
StandardLogOutput& m_Output;
std::string m_SubTask;
State m_State;
};
class StandardLogOutput : public OperationLogOutput
{
public:
StandardLogOutput(LoggerRef& Log) : m_Log(Log) {}
virtual void EmitLogMessage(int LogLevel, std::string_view Format, fmt::format_args Args) override
{
if (m_Log.ShouldLog(LogLevel))
{
fmt::basic_memory_buffer<char, 250> MessageBuffer;
fmt::vformat_to(fmt::appender(MessageBuffer), Format, Args);
ZEN_LOG(m_Log, LogLevel, "{}", std::string_view(MessageBuffer.data(), MessageBuffer.size()));
}
}
virtual void SetLogOperationName(std::string_view Name) override
{
m_LogOperationName = Name;
ZEN_OPERATION_LOG_INFO(*this, "{}", m_LogOperationName);
}
virtual void SetLogOperationProgress(uint32_t StepIndex, uint32_t StepCount) override
{
const size_t PercentDone = StepCount > 0u ? gsl::narrow<uint8_t>((100 * StepIndex) / StepCount) : 0u;
ZEN_OPERATION_LOG_INFO(*this, "{}: {}%", m_LogOperationName, PercentDone);
}
virtual uint32_t GetProgressUpdateDelayMS() override { return 2000; }
virtual ProgressBar* CreateProgressBar(std::string_view InSubTask) override
{
return new StandardLogOutputProgressBar(*this, InSubTask);
}
private:
LoggerRef m_Log;
std::string m_LogOperationName;
};
void
StandardLogOutputProgressBar::UpdateState(const State& NewState, bool DoLinebreak)
{
ZEN_UNUSED(DoLinebreak);
const size_t PercentDone =
NewState.TotalCount > 0u ? gsl::narrow<uint8_t>((100 * (NewState.TotalCount - NewState.RemainingCount)) / NewState.TotalCount) : 0u;
std::string Task = NewState.Task;
switch (NewState.Status)
{
case State::EStatus::Aborted:
Task = "Aborting";
break;
case State::EStatus::Paused:
Task = "Paused";
break;
default:
break;
}
ZEN_OPERATION_LOG_INFO(m_Output, "{}: {}%{}", Task, PercentDone, NewState.Details.empty() ? "" : fmt::format(" {}", NewState.Details));
m_State = NewState;
}
void
StandardLogOutputProgressBar::Finish()
{
if (m_State.RemainingCount > 0)
{
State NewState = m_State;
NewState.RemainingCount = 0;
NewState.Details = "";
UpdateState(NewState, /*DoLinebreak*/ true);
}
}
OperationLogOutput*
CreateStandardLogOutput(LoggerRef Log)
{
return new StandardLogOutput(Log);
}
} // namespace zen
|