xref: /freebsd/contrib/kyua/utils/sqlite/statement.hpp (revision b0d29bc47dba79f6f38e67eabadfb4b32ffd9390)
1*b0d29bc4SBrooks Davis // Copyright 2011 The Kyua Authors.
2*b0d29bc4SBrooks Davis // All rights reserved.
3*b0d29bc4SBrooks Davis //
4*b0d29bc4SBrooks Davis // Redistribution and use in source and binary forms, with or without
5*b0d29bc4SBrooks Davis // modification, are permitted provided that the following conditions are
6*b0d29bc4SBrooks Davis // met:
7*b0d29bc4SBrooks Davis //
8*b0d29bc4SBrooks Davis // * Redistributions of source code must retain the above copyright
9*b0d29bc4SBrooks Davis //   notice, this list of conditions and the following disclaimer.
10*b0d29bc4SBrooks Davis // * Redistributions in binary form must reproduce the above copyright
11*b0d29bc4SBrooks Davis //   notice, this list of conditions and the following disclaimer in the
12*b0d29bc4SBrooks Davis //   documentation and/or other materials provided with the distribution.
13*b0d29bc4SBrooks Davis // * Neither the name of Google Inc. nor the names of its contributors
14*b0d29bc4SBrooks Davis //   may be used to endorse or promote products derived from this software
15*b0d29bc4SBrooks Davis //   without specific prior written permission.
16*b0d29bc4SBrooks Davis //
17*b0d29bc4SBrooks Davis // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18*b0d29bc4SBrooks Davis // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19*b0d29bc4SBrooks Davis // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20*b0d29bc4SBrooks Davis // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21*b0d29bc4SBrooks Davis // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22*b0d29bc4SBrooks Davis // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23*b0d29bc4SBrooks Davis // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24*b0d29bc4SBrooks Davis // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25*b0d29bc4SBrooks Davis // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26*b0d29bc4SBrooks Davis // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27*b0d29bc4SBrooks Davis // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28*b0d29bc4SBrooks Davis 
29*b0d29bc4SBrooks Davis /// \file utils/sqlite/statement.hpp
30*b0d29bc4SBrooks Davis /// Wrapper classes and utilities for SQLite statement processing.
31*b0d29bc4SBrooks Davis ///
32*b0d29bc4SBrooks Davis /// This module contains thin RAII wrappers around the SQLite 3 structures
33*b0d29bc4SBrooks Davis /// representing statements.
34*b0d29bc4SBrooks Davis 
35*b0d29bc4SBrooks Davis #if !defined(UTILS_SQLITE_STATEMENT_HPP)
36*b0d29bc4SBrooks Davis #define UTILS_SQLITE_STATEMENT_HPP
37*b0d29bc4SBrooks Davis 
38*b0d29bc4SBrooks Davis #include "utils/sqlite/statement_fwd.hpp"
39*b0d29bc4SBrooks Davis 
40*b0d29bc4SBrooks Davis extern "C" {
41*b0d29bc4SBrooks Davis #include <stdint.h>
42*b0d29bc4SBrooks Davis }
43*b0d29bc4SBrooks Davis 
44*b0d29bc4SBrooks Davis #include <memory>
45*b0d29bc4SBrooks Davis #include <string>
46*b0d29bc4SBrooks Davis 
47*b0d29bc4SBrooks Davis #include "utils/sqlite/database_fwd.hpp"
48*b0d29bc4SBrooks Davis 
49*b0d29bc4SBrooks Davis namespace utils {
50*b0d29bc4SBrooks Davis namespace sqlite {
51*b0d29bc4SBrooks Davis 
52*b0d29bc4SBrooks Davis 
53*b0d29bc4SBrooks Davis /// Representation of a BLOB.
54*b0d29bc4SBrooks Davis class blob {
55*b0d29bc4SBrooks Davis public:
56*b0d29bc4SBrooks Davis     /// Memory representing the contents of the blob, or NULL if empty.
57*b0d29bc4SBrooks Davis     ///
58*b0d29bc4SBrooks Davis     /// This memory must remain valid throughout the life of this object, as we
59*b0d29bc4SBrooks Davis     /// do not grab ownership of the memory.
60*b0d29bc4SBrooks Davis     const void* memory;
61*b0d29bc4SBrooks Davis 
62*b0d29bc4SBrooks Davis     /// Number of bytes in memory.
63*b0d29bc4SBrooks Davis     int size;
64*b0d29bc4SBrooks Davis 
65*b0d29bc4SBrooks Davis     /// Constructs a new blob.
66*b0d29bc4SBrooks Davis     ///
67*b0d29bc4SBrooks Davis     /// \param memory_ Pointer to the contents of the blob.
68*b0d29bc4SBrooks Davis     /// \param size_ The size of memory_.
blob(const void * memory_,const int size_)69*b0d29bc4SBrooks Davis     blob(const void* memory_, const int size_) :
70*b0d29bc4SBrooks Davis         memory(memory_), size(size_)
71*b0d29bc4SBrooks Davis     {
72*b0d29bc4SBrooks Davis     }
73*b0d29bc4SBrooks Davis };
74*b0d29bc4SBrooks Davis 
75*b0d29bc4SBrooks Davis 
76*b0d29bc4SBrooks Davis /// Representation of a SQL NULL value.
77*b0d29bc4SBrooks Davis class null {
78*b0d29bc4SBrooks Davis };
79*b0d29bc4SBrooks Davis 
80*b0d29bc4SBrooks Davis 
81*b0d29bc4SBrooks Davis /// A RAII model for an SQLite 3 statement.
82*b0d29bc4SBrooks Davis class statement {
83*b0d29bc4SBrooks Davis     struct impl;
84*b0d29bc4SBrooks Davis 
85*b0d29bc4SBrooks Davis     /// Pointer to the shared internal implementation.
86*b0d29bc4SBrooks Davis     std::shared_ptr< impl > _pimpl;
87*b0d29bc4SBrooks Davis 
88*b0d29bc4SBrooks Davis     statement(database&, void*);
89*b0d29bc4SBrooks Davis     friend class database;
90*b0d29bc4SBrooks Davis 
91*b0d29bc4SBrooks Davis public:
92*b0d29bc4SBrooks Davis     ~statement(void);
93*b0d29bc4SBrooks Davis 
94*b0d29bc4SBrooks Davis     bool step(void);
95*b0d29bc4SBrooks Davis     void step_without_results(void);
96*b0d29bc4SBrooks Davis 
97*b0d29bc4SBrooks Davis     int column_count(void);
98*b0d29bc4SBrooks Davis     std::string column_name(const int);
99*b0d29bc4SBrooks Davis     type column_type(const int);
100*b0d29bc4SBrooks Davis     int column_id(const char*);
101*b0d29bc4SBrooks Davis 
102*b0d29bc4SBrooks Davis     blob column_blob(const int);
103*b0d29bc4SBrooks Davis     double column_double(const int);
104*b0d29bc4SBrooks Davis     int column_int(const int);
105*b0d29bc4SBrooks Davis     int64_t column_int64(const int);
106*b0d29bc4SBrooks Davis     std::string column_text(const int);
107*b0d29bc4SBrooks Davis     int column_bytes(const int);
108*b0d29bc4SBrooks Davis 
109*b0d29bc4SBrooks Davis     blob safe_column_blob(const char*);
110*b0d29bc4SBrooks Davis     double safe_column_double(const char*);
111*b0d29bc4SBrooks Davis     int safe_column_int(const char*);
112*b0d29bc4SBrooks Davis     int64_t safe_column_int64(const char*);
113*b0d29bc4SBrooks Davis     std::string safe_column_text(const char*);
114*b0d29bc4SBrooks Davis     int safe_column_bytes(const char*);
115*b0d29bc4SBrooks Davis 
116*b0d29bc4SBrooks Davis     void reset(void);
117*b0d29bc4SBrooks Davis 
118*b0d29bc4SBrooks Davis     void bind(const int, const blob&);
119*b0d29bc4SBrooks Davis     void bind(const int, const double);
120*b0d29bc4SBrooks Davis     void bind(const int, const int);
121*b0d29bc4SBrooks Davis     void bind(const int, const int64_t);
122*b0d29bc4SBrooks Davis     void bind(const int, const null&);
123*b0d29bc4SBrooks Davis     void bind(const int, const std::string&);
124*b0d29bc4SBrooks Davis     template< class T > void bind(const char*, const T&);
125*b0d29bc4SBrooks Davis 
126*b0d29bc4SBrooks Davis     int bind_parameter_count(void);
127*b0d29bc4SBrooks Davis     int bind_parameter_index(const std::string&);
128*b0d29bc4SBrooks Davis     std::string bind_parameter_name(const int);
129*b0d29bc4SBrooks Davis 
130*b0d29bc4SBrooks Davis     void clear_bindings(void);
131*b0d29bc4SBrooks Davis };
132*b0d29bc4SBrooks Davis 
133*b0d29bc4SBrooks Davis 
134*b0d29bc4SBrooks Davis }  // namespace sqlite
135*b0d29bc4SBrooks Davis }  // namespace utils
136*b0d29bc4SBrooks Davis 
137*b0d29bc4SBrooks Davis #endif  // !defined(UTILS_SQLITE_STATEMENT_HPP)
138