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
|
/*
* Copyright (C) 2017 The Android Open Source Project
*
* Licensed 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.
*/
#ifndef NETUTILS_STATUSOR_H
#define NETUTILS_STATUSOR_H
#include <cassert>
#include "netdutils/Status.h"
namespace android {
namespace netdutils {
// Wrapper around a combination of Status and application value type.
// T may be any copyable or movable type.
template <typename T>
class [[nodiscard]] StatusOr {
public:
// Constructs a new StatusOr with status::undefined status.
// This is marked 'explicit' to try to catch cases like 'return {};',
// where people think StatusOr<std::vector<int>> will be initialized
// with an empty vector, instead of a status::undefined.
explicit StatusOr() = default;
// Implicit copy constructor and construction from T.
// NOLINTNEXTLINE(google-explicit-constructor)
StatusOr(Status status) : mStatus(std::move(status)) { assert(!isOk(mStatus)); }
// Implicit construction from T. It is convenient and sensible to be able
// to do 'return T()' when the return type is StatusOr<T>.
// NOLINTNEXTLINE(google-explicit-constructor)
StatusOr(const T& value) : mStatus(status::ok), mValue(value) {}
// NOLINTNEXTLINE(google-explicit-constructor)
StatusOr(T&& value) : mStatus(status::ok), mValue(std::move(value)) {}
// Move constructor ok (if T supports move)
StatusOr(StatusOr&&) noexcept = default;
// Move assignment ok (if T supports move)
StatusOr& operator=(StatusOr&&) noexcept = default;
// Copy constructor ok (if T supports copy)
StatusOr(const StatusOr&) = default;
// Copy assignment ok (if T supports copy)
StatusOr& operator=(const StatusOr&) = default;
// Returns a const reference to wrapped type.
// It is an error to call value() when !isOk(status())
const T& value() const & { return mValue; }
const T&& value() const && { return mValue; }
// Returns an rvalue reference to wrapped type
// It is an error to call value() when !isOk(status())
//
// If T is expensive to copy but supports efficient move, it can be moved
// out of a StatusOr as follows:
// T value = std::move(statusor).value();
T& value() & { return mValue; }
T&& value() && { return mValue; }
// Returns the Status object assigned at construction time.
const Status status() const { return mStatus; }
// Explicitly ignores the Status without triggering [[nodiscard]] errors.
void ignoreError() const {}
// Implicit cast to Status.
// NOLINTNEXTLINE(google-explicit-constructor)
operator Status() const { return status(); }
private:
Status mStatus = status::undefined;
T mValue;
};
template <typename T>
inline std::ostream& operator<<(std::ostream& os, const StatusOr<T>& s) {
return os << "StatusOr[status: " << s.status() << "]";
}
#define ASSIGN_OR_RETURN_IMPL(tmp, lhs, stmt) \
auto tmp = (stmt); \
RETURN_IF_NOT_OK(tmp); \
lhs = std::move(tmp.value());
#define ASSIGN_OR_RETURN_CONCAT(line, lhs, stmt) \
ASSIGN_OR_RETURN_IMPL(__CONCAT(_status_or_, line), lhs, stmt)
// Macro to allow exception-like handling of error return values.
//
// If the evaluation of stmt results in an error, return that error
// from the current function. Otherwise, assign the result to lhs.
//
// This macro supports both move and copy assignment operators. lhs
// may be either a new local variable or an existing non-const
// variable accessible in the current scope.
//
// Example usage:
// StatusOr<MyType> foo() { ... }
//
// ASSIGN_OR_RETURN(auto myVar, foo());
// ASSIGN_OR_RETURN(myExistingVar, foo());
// ASSIGN_OR_RETURN(myMemberVar, foo());
#define ASSIGN_OR_RETURN(lhs, stmt) ASSIGN_OR_RETURN_CONCAT(__LINE__, lhs, stmt)
} // namespace netdutils
} // namespace android
#endif /* NETUTILS_STATUSOR_H */
|
