Mercurial > hg > orthanc-databases

comparison Framework/Odbc/OdbcStatement.cpp @ 329:b5fb8b77ce4d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
initial commit of ODBC framework
author Sebastien Jodogne <s.jodogne@gmail.com>
date Tue, 10 Aug 2021 20:08:53 +0200
parents
children 79e21c33962d
comparison
equal deleted inserted replaced
328:6a49c495c940 329:b5fb8b77ce4d
1 /**
2 * Orthanc - A Lightweight, RESTful DICOM Store
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
4 * Department, University Hospital of Liege, Belgium
5 * Copyright (C) 2017-2021 Osimis S.A., Belgium
6 *
7 * This program is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU Affero General Public License
9 * as published by the Free Software Foundation, either version 3 of
10 * the License, or (at your option) any later version.
11 *
12 * This program is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Affero General Public License for more details.
16 *
17 * You should have received a copy of the GNU Affero General Public License
18 * along with this program. If not, see <http://www.gnu.org/licenses/>.
19 **/
20
21
22 #include "OdbcStatement.h"
23
24 #include "OdbcEnvironment.h"
25
26 #include <Logging.h>
27 #include <OrthancException.h>
28
29 #include <sqlext.h>
30
31
32 namespace OrthancDatabases
33 {
34 OdbcStatement::OdbcStatement(SQLHSTMT databaseHandle)
35 {
36 if (!SQL_SUCCEEDED(SQLAllocHandle(SQL_HANDLE_STMT, databaseHandle, &handle_)))
37 {
38 throw Orthanc::OrthancException(Orthanc::ErrorCode_Database, "Cannot allocate statement");
39 }
40 }
41
42
43 OdbcStatement::~OdbcStatement()
44 {
45 if (!SQL_SUCCEEDED(SQLFreeHandle(SQL_HANDLE_STMT, handle_)))
46 {
47 LOG(ERROR) << "Cannot destruct statement";
48 }
49 }
50
51
52 std::string OdbcStatement::FormatError()
53 {
54 return OdbcEnvironment::FormatError(handle_, SQL_HANDLE_STMT);
55 }
56
57
58 void OdbcStatement::CheckCollision(Dialect dialect)
59 {
60 SQLINTEGER native = -1;
61 SQLCHAR stateBuf[SQL_SQLSTATE_SIZE + 1];
62 SQLSMALLINT stateLength = 0;
63
64 for (SQLSMALLINT recNum = 1; ; recNum++)
65 {
66 if (SQL_SUCCEEDED(SQLGetDiagField(SQL_HANDLE_STMT, handle_,
67 recNum, SQL_DIAG_NATIVE, &native, SQL_IS_INTEGER, 0)) &&
68 SQL_SUCCEEDED(SQLGetDiagField(SQL_HANDLE_STMT, handle_,
69 recNum, SQL_DIAG_SQLSTATE, &stateBuf, sizeof(stateBuf), &stateLength)))
70 {
71 const std::string state(reinterpret_cast<const char*>(stateBuf));
72
73 if (state == "40001" ||
74 (dialect == Dialect_MySQL && native == 1213) ||
75 (dialect == Dialect_MSSQL && native == 1205))
76 {
77 throw Orthanc::OrthancException(Orthanc::ErrorCode_DatabaseCannotSerialize);
78 }
79 }
80 else
81 {
82 return;
83 }
84 }
85 }
86 }