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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
#pragma once
#include <exception>
#include <sstream>
#include <string>
#include <utility>
#include "arrow/type_fwd.h"
#include "arrow/util/string_builder.h"
#include "parquet/platform.h"
// PARQUET-1085
#if !defined(ARROW_UNUSED)
#define ARROW_UNUSED(x) UNUSED(x)
#endif
// Parquet exception to Arrow Status
#define BEGIN_PARQUET_CATCH_EXCEPTIONS try {
#define END_PARQUET_CATCH_EXCEPTIONS \
} \
catch (const ::parquet::ParquetStatusException& e) { \
return e.status(); \
} \
catch (const ::parquet::ParquetException& e) { \
return ::arrow::Status::IOError(e.what()); \
}
// clang-format off
#define PARQUET_CATCH_NOT_OK(s) \
BEGIN_PARQUET_CATCH_EXCEPTIONS \
(s); \
END_PARQUET_CATCH_EXCEPTIONS
// clang-format on
#define PARQUET_CATCH_AND_RETURN(s) \
BEGIN_PARQUET_CATCH_EXCEPTIONS \
return (s); \
END_PARQUET_CATCH_EXCEPTIONS
// Arrow Status to Parquet exception
#define PARQUET_IGNORE_NOT_OK(s) \
do { \
::arrow::Status _s = ::arrow::internal::GenericToStatus(s); \
ARROW_UNUSED(_s); \
} while (0)
#define PARQUET_THROW_NOT_OK(s) \
do { \
::arrow::Status _s = ::arrow::internal::GenericToStatus(s); \
if (!_s.ok()) { \
throw ::parquet::ParquetStatusException(std::move(_s)); \
} \
} while (0)
#define PARQUET_ASSIGN_OR_THROW_IMPL(status_name, lhs, rexpr) \
auto status_name = (rexpr); \
PARQUET_THROW_NOT_OK(status_name.status()); \
lhs = std::move(status_name).ValueOrDie();
#define PARQUET_ASSIGN_OR_THROW(lhs, rexpr) \
PARQUET_ASSIGN_OR_THROW_IMPL(ARROW_ASSIGN_OR_RAISE_NAME(_error_or_value, __COUNTER__), \
lhs, rexpr);
namespace parquet {
class ParquetException : public std::exception {
public:
PARQUET_NORETURN static void EofException(const std::string& msg = "") {
static std::string prefix = "Unexpected end of stream";
if (msg.empty()) {
throw ParquetException(prefix);
}
throw ParquetException(prefix, ": ", msg);
}
PARQUET_NORETURN static void NYI(const std::string& msg = "") {
throw ParquetException("Not yet implemented: ", msg, ".");
}
template <typename... Args>
explicit ParquetException(Args&&... args)
: msg_(::arrow::util::StringBuilder(std::forward<Args>(args)...)) {}
explicit ParquetException(std::string msg) : msg_(std::move(msg)) {}
explicit ParquetException(const char* msg, const std::exception&) : msg_(msg) {}
ParquetException(const ParquetException&) = default;
ParquetException& operator=(const ParquetException&) = default;
ParquetException(ParquetException&&) = default;
ParquetException& operator=(ParquetException&&) = default;
const char* what() const noexcept override { return msg_.c_str(); }
private:
std::string msg_;
};
// Support printing a ParquetException.
// This is needed for clang-on-MSVC as there operator<< is not defined for
// std::exception.
PARQUET_EXPORT
std::ostream& operator<<(std::ostream& os, const ParquetException& exception);
class ParquetStatusException : public ParquetException {
public:
explicit ParquetStatusException(::arrow::Status status)
: ParquetException(status.ToString()), status_(std::move(status)) {}
const ::arrow::Status& status() const { return status_; }
private:
::arrow::Status status_;
};
// This class exists for the purpose of detecting an invalid or corrupted file.
class ParquetInvalidOrCorruptedFileException : public ParquetStatusException {
public:
ParquetInvalidOrCorruptedFileException(const ParquetInvalidOrCorruptedFileException&) =
default;
template <typename Arg,
typename std::enable_if<
!std::is_base_of<ParquetInvalidOrCorruptedFileException, Arg>::value,
int>::type = 0,
typename... Args>
explicit ParquetInvalidOrCorruptedFileException(Arg arg, Args&&... args)
: ParquetStatusException(::arrow::Status::Invalid(std::forward<Arg>(arg),
std::forward<Args>(args)...)) {}
};
template <typename StatusReturnBlock>
void ThrowNotOk(StatusReturnBlock&& b) {
PARQUET_THROW_NOT_OK(b());
}
} // namespace parquet
|
