OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef SQL_STATEMENT_H_ | 5 #ifndef SQL_STATEMENT_H_ |
6 #define SQL_STATEMENT_H_ | 6 #define SQL_STATEMENT_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <string> | 9 #include <string> |
10 #include <vector> | 10 #include <vector> |
(...skipping 11 matching lines...) Expand all Loading... |
22 enum ColType { | 22 enum ColType { |
23 COLUMN_TYPE_INTEGER = 1, | 23 COLUMN_TYPE_INTEGER = 1, |
24 COLUMN_TYPE_FLOAT = 2, | 24 COLUMN_TYPE_FLOAT = 2, |
25 COLUMN_TYPE_TEXT = 3, | 25 COLUMN_TYPE_TEXT = 3, |
26 COLUMN_TYPE_BLOB = 4, | 26 COLUMN_TYPE_BLOB = 4, |
27 COLUMN_TYPE_NULL = 5, | 27 COLUMN_TYPE_NULL = 5, |
28 }; | 28 }; |
29 | 29 |
30 // Normal usage: | 30 // Normal usage: |
31 // sql::Statement s(connection_.GetUniqueStatement(...)); | 31 // sql::Statement s(connection_.GetUniqueStatement(...)); |
32 // if (!s) // You should check for errors before using the statement. | |
33 // return false; | |
34 // | |
35 // s.BindInt(0, a); | 32 // s.BindInt(0, a); |
36 // if (s.Step()) | 33 // if (s.Step()) |
37 // return s.ColumnString(0); | 34 // return s.ColumnString(0); |
38 // | 35 // |
| 36 // If there are errors getting the statement, the statement will be inert; no |
| 37 // mutating or database-access methods will work. If you need to check for |
| 38 // validity, use: |
| 39 // if (!s.is_valid()) |
| 40 // return false; |
| 41 // |
39 // Step() and Run() just return true to signal success. If you want to handle | 42 // Step() and Run() just return true to signal success. If you want to handle |
40 // specific errors such as database corruption, install an error handler in | 43 // specific errors such as database corruption, install an error handler in |
41 // in the connection object using set_error_delegate(). | 44 // in the connection object using set_error_delegate(). |
42 class SQL_EXPORT Statement { | 45 class SQL_EXPORT Statement { |
43 public: | 46 public: |
44 // Creates an uninitialized statement. The statement will be invalid until | 47 // Creates an uninitialized statement. The statement will be invalid until |
45 // you initialize it via Assign. | 48 // you initialize it via Assign. |
46 Statement(); | 49 Statement(); |
47 | 50 |
48 explicit Statement(scoped_refptr<Connection::StatementRef> ref); | 51 explicit Statement(scoped_refptr<Connection::StatementRef> ref); |
49 ~Statement(); | 52 ~Statement(); |
50 | 53 |
51 // Initializes this object with the given statement, which may or may not | 54 // Initializes this object with the given statement, which may or may not |
52 // be valid. Use is_valid() to check if it's OK. | 55 // be valid. Use is_valid() to check if it's OK. |
53 void Assign(scoped_refptr<Connection::StatementRef> ref); | 56 void Assign(scoped_refptr<Connection::StatementRef> ref); |
54 | 57 |
55 // Returns true if the statement can be executed. All functions can still | 58 // Returns true if the statement can be executed. All functions can still |
56 // be used if the statement is invalid, but they will return failure or some | 59 // be used if the statement is invalid, but they will return failure or some |
57 // default value. This is because the statement can become invalid in the | 60 // default value. This is because the statement can become invalid in the |
58 // middle of executing a command if there is a serioud error and the database | 61 // middle of executing a command if there is a serioud error and the database |
59 // has to be reset. | 62 // has to be reset. |
60 bool is_valid() const { return ref_->is_valid(); } | 63 bool is_valid() const { return ref_->is_valid(); } |
61 | 64 |
62 // These operators allow conveniently checking if the statement is valid | 65 // These operators allow conveniently checking if the statement is valid |
63 // or not. See the pattern above for an example. | 66 // or not. See the pattern above for an example. |
| 67 // TODO(shess,gbillock): Remove these once clients are converted. |
64 operator bool() const { return is_valid(); } | 68 operator bool() const { return is_valid(); } |
65 bool operator!() const { return !is_valid(); } | 69 bool operator!() const { return !is_valid(); } |
66 | 70 |
67 // Running ------------------------------------------------------------------- | 71 // Running ------------------------------------------------------------------- |
68 | 72 |
69 // Executes the statement, returning true on success. This is like Step but | 73 // Executes the statement, returning true on success. This is like Step but |
70 // for when there is no output, like an INSERT statement. | 74 // for when there is no output, like an INSERT statement. |
71 bool Run(); | 75 bool Run(); |
72 | 76 |
73 // Executes the statement, returning true if there is a row of data returned. | 77 // Executes the statement, returning true if there is a row of data returned. |
(...skipping 15 matching lines...) Expand all Loading... |
89 // the bound variables and any current result row. | 93 // the bound variables and any current result row. |
90 void Reset(); | 94 void Reset(); |
91 | 95 |
92 // Returns true if the last executed thing in this statement succeeded. If | 96 // Returns true if the last executed thing in this statement succeeded. If |
93 // there was no last executed thing or the statement is invalid, this will | 97 // there was no last executed thing or the statement is invalid, this will |
94 // return false. | 98 // return false. |
95 bool Succeeded() const; | 99 bool Succeeded() const; |
96 | 100 |
97 // Binding ------------------------------------------------------------------- | 101 // Binding ------------------------------------------------------------------- |
98 | 102 |
99 // These all take a 0-based argument index and return true on failure. You | 103 // These all take a 0-based argument index and return true on success. You |
100 // may not always care about the return value (they'll DCHECK if they fail). | 104 // may not always care about the return value (they'll DCHECK if they fail). |
101 // The main thing you may want to check is when binding large blobs or | 105 // The main thing you may want to check is when binding large blobs or |
102 // strings there may be out of memory. | 106 // strings there may be out of memory. |
103 bool BindNull(int col); | 107 bool BindNull(int col); |
104 bool BindBool(int col, bool val); | 108 bool BindBool(int col, bool val); |
105 bool BindInt(int col, int val); | 109 bool BindInt(int col, int val); |
106 bool BindInt64(int col, int64 val); | 110 bool BindInt64(int col, int64 val); |
107 bool BindDouble(int col, double val); | 111 bool BindDouble(int col, double val); |
108 bool BindCString(int col, const char* val); | 112 bool BindCString(int col, const char* val); |
109 bool BindString(int col, const std::string& val); | 113 bool BindString(int col, const std::string& val); |
(...skipping 20 matching lines...) Expand all Loading... |
130 double ColumnDouble(int col) const; | 134 double ColumnDouble(int col) const; |
131 std::string ColumnString(int col) const; | 135 std::string ColumnString(int col) const; |
132 string16 ColumnString16(int col) const; | 136 string16 ColumnString16(int col) const; |
133 | 137 |
134 // When reading a blob, you can get a raw pointer to the underlying data, | 138 // When reading a blob, you can get a raw pointer to the underlying data, |
135 // along with the length, or you can just ask us to copy the blob into a | 139 // along with the length, or you can just ask us to copy the blob into a |
136 // vector. Danger! ColumnBlob may return NULL if there is no data! | 140 // vector. Danger! ColumnBlob may return NULL if there is no data! |
137 int ColumnByteLength(int col) const; | 141 int ColumnByteLength(int col) const; |
138 const void* ColumnBlob(int col) const; | 142 const void* ColumnBlob(int col) const; |
139 bool ColumnBlobAsString(int col, std::string* blob); | 143 bool ColumnBlobAsString(int col, std::string* blob); |
140 void ColumnBlobAsVector(int col, std::vector<char>* val) const; | 144 bool ColumnBlobAsVector(int col, std::vector<char>* val) const; |
141 void ColumnBlobAsVector(int col, std::vector<unsigned char>* val) const; | 145 bool ColumnBlobAsVector(int col, std::vector<unsigned char>* val) const; |
142 | 146 |
143 // Diagnostics -------------------------------------------------------------- | 147 // Diagnostics -------------------------------------------------------------- |
144 | 148 |
145 // Returns the original text of sql statement. Do not keep a pointer to it. | 149 // Returns the original text of sql statement. Do not keep a pointer to it. |
146 const char* GetSQLStatement(); | 150 const char* GetSQLStatement(); |
147 | 151 |
148 private: | 152 private: |
149 // This is intended to check for serious errors and report them to the | 153 // This is intended to check for serious errors and report them to the |
150 // connection object. It takes a sqlite error code, and returns the same | 154 // connection object. It takes a sqlite error code, and returns the same |
151 // code. Currently this function just updates the succeeded flag, but will be | 155 // code. Currently this function just updates the succeeded flag, but will be |
152 // enhanced in the future to do the notification. | 156 // enhanced in the future to do the notification. |
153 int CheckError(int err); | 157 int CheckError(int err); |
154 | 158 |
| 159 // Contraction for checking an error code against SQLITE_OK. Does not set the |
| 160 // succeeded flag. |
| 161 bool CheckOk(int err) const; |
| 162 |
| 163 // Should be called by all mutating methods to check that the statement is |
| 164 // valid. Returns true if the statement is valid. DCHECKS and returns false |
| 165 // if it is not. |
| 166 // The reason for this is to handle two specific cases in which a Statement |
| 167 // may be invalid. The first case is that the programmer made an SQL error. |
| 168 // Those cases need to be DCHECKed so that we are guaranteed to find them |
| 169 // before release. The second case is that the computer has an error (probably |
| 170 // out of disk space) which is prohibiting the correct operation of the |
| 171 // database. Our testing apparatus should not exhibit this defect, but release |
| 172 // situations may. Therefore, the code is handling disjoint situations in |
| 173 // release and test. In test, we're ensuring correct SQL. In release, we're |
| 174 // ensuring that contracts are honored in error edge cases. |
| 175 bool CheckValid() const; |
| 176 |
155 // The actual sqlite statement. This may be unique to us, or it may be cached | 177 // The actual sqlite statement. This may be unique to us, or it may be cached |
156 // by the connection, which is why it's refcounted. This pointer is | 178 // by the connection, which is why it's refcounted. This pointer is |
157 // guaranteed non-NULL. | 179 // guaranteed non-NULL. |
158 scoped_refptr<Connection::StatementRef> ref_; | 180 scoped_refptr<Connection::StatementRef> ref_; |
159 | 181 |
160 // See Succeeded() for what this holds. | 182 // See Succeeded() for what this holds. |
161 bool succeeded_; | 183 bool succeeded_; |
162 | 184 |
163 DISALLOW_COPY_AND_ASSIGN(Statement); | 185 DISALLOW_COPY_AND_ASSIGN(Statement); |
164 }; | 186 }; |
165 | 187 |
166 } // namespace sql | 188 } // namespace sql |
167 | 189 |
168 #endif // SQL_STATEMENT_H_ | 190 #endif // SQL_STATEMENT_H_ |
OLD | NEW |