comparison OrthancFramework/Sources/SQLite/Statement.h @ 4044:d25f4c0fa160 framework

splitting code into OrthancFramework and OrthancServer
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Jun 2020 20:30:34 +0200
parents Core/SQLite/Statement.h@f9863630ec7f
children 50b0c69b653a
comparison
equal deleted inserted replaced
4043:6c6239aec462 4044:d25f4c0fa160
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 *
4 * Copyright (C) 2012-2016 Sebastien Jodogne <s.jodogne@orthanc-labs.com>,
5 * Medical Physics Department, CHU of Liege, Belgium
6 *
7 * Copyright (c) 2012 The Chromium Authors. All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are
11 * met:
12 *
13 * * Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * * Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following disclaimer
17 * in the documentation and/or other materials provided with the
18 * distribution.
19 * * Neither the name of Google Inc., the name of the CHU of Liege,
20 * nor the names of its contributors may be used to endorse or promote
21 * products derived from this software without specific prior written
22 * permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35 **/
36
37
38 #pragma once
39
40 #include "NonCopyable.h"
41 #include "OrthancSQLiteException.h"
42 #include "StatementId.h"
43 #include "StatementReference.h"
44
45 #include <vector>
46 #include <stdint.h>
47
48 #if ORTHANC_BUILD_UNIT_TESTS == 1
49 # include <gtest/gtest_prod.h>
50 #endif
51
52
53 namespace Orthanc
54 {
55 namespace SQLite
56 {
57 class Connection;
58
59 // Possible return values from ColumnType in a statement. These
60 // should match the values in sqlite3.h.
61 enum ColumnType
62 {
63 COLUMN_TYPE_INTEGER = 1,
64 COLUMN_TYPE_FLOAT = 2,
65 COLUMN_TYPE_TEXT = 3,
66 COLUMN_TYPE_BLOB = 4,
67 COLUMN_TYPE_NULL = 5
68 };
69
70 class ORTHANC_PUBLIC Statement : public NonCopyable
71 {
72 friend class Connection;
73
74 #if ORTHANC_BUILD_UNIT_TESTS == 1
75 FRIEND_TEST(SQLStatementTest, Run);
76 FRIEND_TEST(SQLStatementTest, Reset);
77 #endif
78
79 private:
80 StatementReference reference_;
81
82 int CheckError(int err,
83 ErrorCode code) const;
84
85 void CheckOk(int err,
86 ErrorCode code) const;
87
88 struct sqlite3_stmt* GetStatement() const
89 {
90 return reference_.GetWrappedObject();
91 }
92
93 public:
94 Statement(Connection& database,
95 const std::string& sql);
96
97 Statement(Connection& database,
98 const StatementId& id,
99 const std::string& sql);
100
101 Statement(Connection& database,
102 const char* sql);
103
104 Statement(Connection& database,
105 const StatementId& id,
106 const char* sql);
107
108 ~Statement()
109 {
110 Reset();
111 }
112
113 bool Run();
114
115 bool Step();
116
117 // Diagnostics --------------------------------------------------------------
118
119 std::string GetOriginalSQLStatement();
120
121
122 // Binding -------------------------------------------------------------------
123
124 // These all take a 0-based argument index
125 void BindNull(int col);
126 void BindBool(int col, bool val);
127 void BindInt(int col, int val);
128 void BindInt64(int col, int64_t val);
129 void BindDouble(int col, double val);
130 void BindCString(int col, const char* val);
131 void BindString(int col, const std::string& val);
132 //void BindString16(int col, const string16& value);
133 void BindBlob(int col, const void* value, int value_len);
134
135
136 // Retrieving ----------------------------------------------------------------
137
138 // Returns the number of output columns in the result.
139 int ColumnCount() const;
140
141 // Returns the type associated with the given column.
142 //
143 // Watch out: the type may be undefined if you've done something to cause a
144 // "type conversion." This means requesting the value of a column of a type
145 // where that type is not the native type. For safety, call ColumnType only
146 // on a column before getting the value out in any way.
147 ColumnType GetColumnType(int col) const;
148 ColumnType GetDeclaredColumnType(int col) const;
149
150 // These all take a 0-based argument index.
151 bool ColumnIsNull(int col) const ;
152 bool ColumnBool(int col) const;
153 int ColumnInt(int col) const;
154 int64_t ColumnInt64(int col) const;
155 double ColumnDouble(int col) const;
156 std::string ColumnString(int col) const;
157 //string16 ColumnString16(int col) const;
158
159 // When reading a blob, you can get a raw pointer to the underlying data,
160 // along with the length, or you can just ask us to copy the blob into a
161 // vector. Danger! ColumnBlob may return NULL if there is no data!
162 int ColumnByteLength(int col) const;
163 const void* ColumnBlob(int col) const;
164 bool ColumnBlobAsString(int col, std::string* blob);
165 //bool ColumnBlobAsString16(int col, string16* val) const;
166 //bool ColumnBlobAsVector(int col, std::vector<char>* val) const;
167 //bool ColumnBlobAsVector(int col, std::vector<unsigned char>* val) const;
168
169 // Resets the statement to its initial condition. This includes any current
170 // result row, and also the bound variables if the |clear_bound_vars| is true.
171 void Reset(bool clear_bound_vars = true);
172 };
173 }
174 }