src/util/db/sqlstringformatter.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#include "util/db/sqlstringformatter.h"

#include <QSqlDriver>
#include <QSqlField>

#include "util/assert.h"


QString SqlStringFormatter::format(
        const QSqlDatabase& database,
        const QString& value) {
    QSqlDriver* pDriver = database.driver();
    VERIFY_OR_DEBUG_ASSERT(pDriver != nullptr) {
        return value; // unformatted
    }
#if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
    QSqlField stringField(QString(), QMetaType(QMetaType::QString));
#else
    QSqlField stringField(QString(), QVariant::String);
#endif
    stringField.setValue(value);
    return pDriver->formatValue(stringField);
}

namespace {
    template<typename T> // forward-iterable container type with values of type QString
    QString formatListValues(
            const QSqlDatabase& database,
            const T& values) {
        QString result;
        QString previousString;
        for (const QString& value: values) {
            if (!previousString.isEmpty()) {
                result.append(',');
            }
            previousString = SqlStringFormatter::format(database, value);
            result += previousString;
        }
        return result;
    }
    } // namespace

QString SqlStringFormatter::formatList(
        const QSqlDatabase& database,
        const QStringList& values) {
    return formatListValues(database, values);
}

QString SqlStringFormatter::formatList(
        const QSqlDatabase& database,
        const QSet<QString>& values) {
    return formatListValues(database, values);
}