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
|
#pragma once
#include "yql_issue_message.h"
#include <util/generic/deque.h>
#include <util/string/join.h>
#include <util/stream/output.h>
namespace NYql {
using namespace NIssue::NProto;
template<typename TIssueMessage>
TIssue IssueFromMessage(const TIssueMessage& issueMessage) {
TIssue topIssue;
TDeque<std::pair<TIssue*, const TIssueMessage*>> queue;
queue.push_front(std::make_pair(&topIssue, &issueMessage));
while (!queue.empty()) {
TIssue& issue = *queue.back().first;
const auto& message = *queue.back().second;
queue.pop_back();
TPosition position(message.position().column(), message.position().row(), message.position().file());
TPosition endPosition(message.end_position().column(), message.end_position().row());
if (position.HasValue()) {
if (endPosition.HasValue()) {
issue = TIssue(position, endPosition, message.message());
} else {
issue = TIssue(position, message.message());
}
} else {
issue = TIssue(message.message());
}
for (const auto& subMessage : message.issues()) {
auto subIssue = new TIssue();
issue.AddSubIssue(subIssue);
queue.push_front(std::make_pair(subIssue, &subMessage));
}
issue.SetCode(message.issue_code(), static_cast<ESeverity>(message.severity()));
}
return topIssue;
}
template<typename TIssueMessage>
void IssuesFromMessage(const ::google::protobuf::RepeatedPtrField<TIssueMessage> &message, TIssues &issues) {
issues.Clear();
if (message.size()) {
issues.Reserve(message.size());
for (auto &x : message)
issues.AddIssue(IssueFromMessage(x));
}
}
template<typename TIssueMessage>
void IssueToMessage(const TIssue& topIssue, TIssueMessage* issueMessage) {
TDeque<std::pair<const TIssue*, TIssueMessage*>> queue;
queue.push_front(std::make_pair(&topIssue, issueMessage));
while (!queue.empty()) {
const TIssue& issue = *queue.back().first;
auto& message = *queue.back().second;
queue.pop_back();
if (issue.Position) {
auto& position = *message.mutable_position();
position.set_row(issue.Position.Row);
position.set_column(issue.Position.Column);
position.set_file(issue.Position.File);
}
if (issue.EndPosition) {
auto& endPosition = *message.mutable_end_position();
endPosition.set_row(issue.EndPosition.Row);
endPosition.set_column(issue.EndPosition.Column);
}
message.set_message(issue.GetMessage());
message.set_issue_code(issue.GetCode());
message.set_severity(issue.GetSeverity());
for (auto subIssue : issue.GetSubIssues()) {
TIssueMessage* subMessage = message.add_issues();
queue.push_front(std::make_pair(subIssue.Get(), subMessage));
}
}
}
template<typename TIssueMessage>
void IssuesToMessage(const TIssues& issues, ::google::protobuf::RepeatedPtrField<TIssueMessage> *message) {
message->Clear();
if (!issues)
return;
message->Reserve(issues.Size());
for (const auto &issue : issues) {
IssueToMessage(issue, message->Add());
}
}
}
|
