Mercurial > hg > orthanc
annotate Core/SQLite/Connection.h @ 1220:9b9026560a5f
SQLite wrapper is now fully independent of Orthanc
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Mon, 10 Nov 2014 16:33:51 +0100 |
| parents | 2d0a347e8cfc |
| children | 2255e66da726 |
| rev | line source |
|---|---|
| 0 | 1 /** |
| 59 | 2 * Orthanc - A Lightweight, RESTful DICOM Store |
|
1220
9b9026560a5f
SQLite wrapper is now fully independent of Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
3 * |
|
9b9026560a5f
SQLite wrapper is now fully independent of Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
4 * Copyright (C) 2012-2014 Sebastien Jodogne <s.jodogne@gmail.com>, |
|
9b9026560a5f
SQLite wrapper is now fully independent of Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
689
diff
changeset
|
5 * Medical Physics Department, CHU of Liege, Belgium |
| 0 | 6 * |
| 17 | 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: | |
| 0 | 12 * |
| 17 | 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. | |
| 0 | 35 **/ |
| 36 | |
| 37 | |
| 38 #pragma once | |
| 39 | |
| 40 #include "Statement.h" | |
| 41 #include "IScalarFunction.h" | |
| 42 | |
| 43 #include <string> | |
| 44 #include <boost/noncopyable.hpp> | |
| 45 #include <map> | |
| 46 | |
| 47 struct sqlite3; | |
| 48 struct sqlite3_stmt; | |
| 49 | |
| 50 #define SQLITE_FROM_HERE SQLite::StatementId(__FILE__, __LINE__) | |
| 51 | |
| 59 | 52 namespace Orthanc |
| 0 | 53 { |
| 54 namespace SQLite | |
| 55 { | |
| 56 class Connection : boost::noncopyable | |
| 57 { | |
| 58 friend class Statement; | |
| 59 friend class Transaction; | |
| 60 | |
| 61 private: | |
| 62 // All cached statements. Keeping a reference to these statements means that | |
| 63 // they'll remain active. | |
| 64 typedef std::map<StatementId, StatementReference*> CachedStatements; | |
| 65 CachedStatements cachedStatements_; | |
| 66 | |
| 67 // The actual sqlite database. Will be NULL before Init has been called or if | |
| 68 // Init resulted in an error. | |
| 69 sqlite3* db_; | |
| 70 | |
| 71 // Number of currently-nested transactions. | |
| 72 int transactionNesting_; | |
| 73 | |
| 74 // True if any of the currently nested transactions have been rolled back. | |
| 75 // When we get to the outermost transaction, this will determine if we do | |
| 76 // a rollback instead of a commit. | |
| 77 bool needsRollback_; | |
| 78 | |
| 79 void ClearCache(); | |
| 80 | |
| 81 void CheckIsOpen() const; | |
| 82 | |
| 83 sqlite3* GetWrappedObject() | |
| 84 { | |
| 85 return db_; | |
| 86 } | |
| 87 | |
| 88 StatementReference& GetCachedStatement(const StatementId& id, | |
| 89 const char* sql); | |
| 90 | |
| 91 bool DoesTableOrIndexExist(const char* name, | |
| 92 const char* type) const; | |
| 93 | |
| 94 void DoRollback(); | |
| 95 | |
| 96 public: | |
| 97 // The database is opened by calling Open[InMemory](). Any uncommitted | |
| 98 // transactions will be rolled back when this object is deleted. | |
| 99 Connection(); | |
| 100 ~Connection(); | |
| 101 | |
| 102 void Open(const std::string& path); | |
| 103 | |
| 104 void OpenInMemory(); | |
| 105 | |
| 106 void Close(); | |
| 107 | |
| 108 bool Execute(const char* sql); | |
| 109 | |
| 110 bool Execute(const std::string& sql) | |
| 111 { | |
| 112 return Execute(sql.c_str()); | |
| 113 } | |
| 114 | |
|
206
4453a010d0db
flush to disk thread
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
59
diff
changeset
|
115 void FlushToDisk(); |
|
4453a010d0db
flush to disk thread
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
59
diff
changeset
|
116 |
| 0 | 117 IScalarFunction* Register(IScalarFunction* func); // Takes the ownership of the function |
| 118 | |
| 119 // Info querying ------------------------------------------------------------- | |
| 120 | |
| 121 // Used to check a |sql| statement for syntactic validity. If the | |
| 122 // statement is valid SQL, returns true. | |
| 123 bool IsSQLValid(const char* sql); | |
| 124 | |
| 125 // Returns true if the given table exists. | |
| 126 bool DoesTableExist(const char* table_name) const; | |
| 127 | |
| 128 // Returns true if the given index exists. | |
| 129 bool DoesIndexExist(const char* index_name) const; | |
| 130 | |
| 131 // Returns true if a column with the given name exists in the given table. | |
| 132 bool DoesColumnExist(const char* table_name, const char* column_name) const; | |
| 133 | |
| 134 // Returns sqlite's internal ID for the last inserted row. Valid only | |
| 135 // immediately after an insert. | |
| 136 int64_t GetLastInsertRowId() const; | |
| 137 | |
| 138 // Returns sqlite's count of the number of rows modified by the last | |
| 139 // statement executed. Will be 0 if no statement has executed or the database | |
| 140 // is closed. | |
| 141 int GetLastChangeCount() const; | |
| 142 | |
| 143 // Errors -------------------------------------------------------------------- | |
| 144 | |
| 145 // Returns the error code associated with the last sqlite operation. | |
| 146 int GetErrorCode() const; | |
| 147 | |
| 148 // Returns the errno associated with GetErrorCode(). See | |
| 149 // SQLITE_LAST_ERRNO in SQLite documentation. | |
| 150 int GetLastErrno() const; | |
| 151 | |
| 152 // Returns a pointer to a statically allocated string associated with the | |
| 153 // last sqlite operation. | |
| 154 const char* GetErrorMessage() const; | |
| 155 | |
| 156 | |
| 157 // Diagnostics (for unit tests) ---------------------------------------------- | |
| 158 | |
| 159 int ExecuteAndReturnErrorCode(const char* sql); | |
| 160 | |
| 161 bool HasCachedStatement(const StatementId& id) const | |
| 162 { | |
| 163 return cachedStatements_.find(id) != cachedStatements_.end(); | |
| 164 } | |
| 165 | |
| 166 int GetTransactionNesting() const | |
| 167 { | |
| 168 return transactionNesting_; | |
| 169 } | |
| 170 | |
| 171 // Transactions -------------------------------------------------------------- | |
| 172 | |
| 173 bool BeginTransaction(); | |
| 174 void RollbackTransaction(); | |
|
206
4453a010d0db
flush to disk thread
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
59
diff
changeset
|
175 bool CommitTransaction(); |
| 0 | 176 }; |
| 177 } | |
| 178 } |