summaryrefslogtreecommitdiff
path: root/subprojects/boost-sqlite/include/boost/sqlite/result.hpp
blob: b609eb4a55609c7a756f83b4a9ac4cdc69bba4ed (plain)
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
//
// Copyright (c) 2022 Klemens Morgenstern (klemens.morgenstern@gmx.net)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//

#ifndef BOOST_SQLITE_RESULT_HPP
#define BOOST_SQLITE_RESULT_HPP

#include <boost/sqlite/blob.hpp>
#include <boost/sqlite/detail/config.hpp>
#include <boost/sqlite/value.hpp>

#include <boost/variant2/variant.hpp>

#include <type_traits>


BOOST_SQLITE_BEGIN_NAMESPACE


struct set_result_tag {};


inline void tag_invoke(set_result_tag, sqlite3_context * ctx, blob b)
{
  auto sz = b.size();
  sqlite3_result_blob(ctx, std::move(b).release(), sz, &operator delete);
}


inline void tag_invoke(set_result_tag, sqlite3_context * ctx, zero_blob zb)
{
  sqlite3_result_zeroblob64(ctx, static_cast<sqlite3_uint64>(zb));
}

inline void tag_invoke(set_result_tag, sqlite3_context * ctx, double dbl) { sqlite3_result_double(ctx, dbl); }

inline void tag_invoke(set_result_tag, sqlite3_context * ctx, sqlite3_int64 value)
{
  sqlite3_result_int64(ctx, static_cast<sqlite3_int64>(value));
}

template<typename = std::enable_if_t<!std::is_same<std::int64_t, sqlite3_int64>::value>>
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, std::int64_t value)

{
  sqlite3_result_int64(ctx, static_cast<sqlite3_int64>(value));
}

inline void tag_invoke(set_result_tag, sqlite3_context * ctx, std::nullptr_t) { sqlite3_result_null(ctx); }
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, string_view str)
{
  sqlite3_result_text(ctx, str.data(), str.size(), SQLITE_TRANSIENT);
}
template<typename String>
inline auto tag_invoke(set_result_tag, sqlite3_context * ctx, String && str)
  -> typename std::enable_if<std::is_convertible<String, string_view>::value>::type
{
  return tag_invoke(set_result_tag{}, ctx, string_view(str));
}


inline void tag_invoke(set_result_tag, sqlite3_context * , variant2::monostate) { }
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, const value & val)
{
  sqlite3_result_value(ctx, val.handle());
}

struct set_variant_result
{
  sqlite3_context * ctx;
  template<typename T>
  void operator()(T && val)
  {
    tag_invoke(set_result_tag{}, ctx, std::forward<T>(val));
  }
};

template<typename ... Args>
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, const variant2::variant<Args...> & var)
{
  visit(set_variant_result{ctx}, var);
}

template<typename T>
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, std::unique_ptr<T> ptr)
{
  sqlite3_result_pointer(ctx, ptr.release(), typeid(T).name(), +[](void * ptr){delete static_cast<T*>(ptr);});
}

template<typename T>
inline void tag_invoke(set_result_tag, sqlite3_context * ctx, std::unique_ptr<T, void(*)(T*)> ptr)
{
  sqlite3_result_pointer(ctx, ptr.release(), typeid(T).name(), static_cast<void(*)(void*)>(ptr.get_deleter()));
}

template<typename T, typename Deleter>
inline auto tag_invoke(set_result_tag, sqlite3_context * ctx, std::unique_ptr<T, Deleter> ptr)
    -> typename std::enable_if<std::is_empty<Deleter>::value &&
                               std::is_default_constructible<Deleter>::value>::type
{
  sqlite3_result_pointer(ctx, ptr.release(), typeid(T).name(), +[](void * ptr){Deleter()(static_cast<T*>(ptr));});
}

inline void tag_invoke(set_result_tag, sqlite3_context * ctx, error err)
{
  if (err.info)
    sqlite3_result_error(ctx, err.info.message().c_str(), -1);
  sqlite3_result_error_code(ctx, err.code);
}


template<typename T>
inline void tag_invoke(set_result_tag tag, sqlite3_context * ctx, result<T> res)
{
  if (res.has_value())
    tag_invoke(tag, ctx, std::move(res).value());
  else
    tag_invoke(tag, ctx, std::move(res).error());
}

inline void tag_invoke(set_result_tag tag, sqlite3_context * ctx, result<void> res)
{
  if (res.has_error())
    tag_invoke(tag, ctx, std::move(res).error());
}


namespace detail
{

template<typename Value>
inline auto set_result(sqlite3_context * ctx, Value && value)
    -> decltype(tag_invoke(set_result_tag{}, ctx, std::forward<Value>(value)))
{
  tag_invoke(set_result_tag{}, ctx, std::forward<Value>(value));
}


}


BOOST_SQLITE_END_NAMESPACE

#endif //BOOST_SQLITE_RESULT_HPP