Sqlite: Add check constraint

Change-Id: Ib4b909da40c7fe07dcb6a07c4650a720313391c2
Reviewed-by: Tim Jenssen <tim.jenssen@qt.io>

src/libs/sqlite/constraints.h

@@ -0,0 +1,185 @@
+/****************************************************************************
+**
+** Copyright (C) 2020 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of Qt Creator.
+**
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see https://www.qt.io/terms-conditions. For further
+** information use the contact form at https://www.qt.io/contact-us.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3 as published by the Free Software
+** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
+** included in the packaging of this file. Please review the following
+** information to ensure the GNU General Public License requirements will
+** be met: https://www.gnu.org/licenses/gpl-3.0.html.
+**
+****************************************************************************/
+
+#pragma once
+
+#include "sqliteglobal.h"
+
+#include <sqlitevalue.h>
+#include <utils/smallstring.h>
+#include <utils/variant.h>
+
+namespace Sqlite {
+
+class Unique
+{
+    friend bool operator==(Unique, Unique) { return true; }
+};
+
+class PrimaryKey
+{
+    friend bool operator==(PrimaryKey, PrimaryKey) { return true; }
+};
+
+class NotNull
+{
+    friend bool operator==(NotNull, NotNull) { return true; }
+};
+
+class Check
+{
+public:
+    Check(Utils::SmallStringView expression)
+        : expression(expression)
+    {}
+
+    friend bool operator==(const Check &first, const Check &second)
+    {
+        return first.expression == second.expression;
+    }
+
+public:
+    Utils::SmallString expression;
+};
+
+class DefaultValue
+{
+public:
+    DefaultValue(long long value)
+        : value(value)
+    {}
+
+    DefaultValue(double value)
+        : value(value)
+    {}
+
+    DefaultValue(Utils::SmallStringView value)
+        : value(value)
+    {}
+
+    friend bool operator==(const DefaultValue &first, const DefaultValue &second)
+    {
+        return first.value == second.value;
+    }
+
+public:
+    Sqlite::Value value;
+};
+
+class DefaultExpression
+{
+public:
+    DefaultExpression(Utils::SmallStringView expression)
+        : expression(expression)
+    {}
+
+    friend bool operator==(const DefaultExpression &first, const DefaultExpression &second)
+    {
+        return first.expression == second.expression;
+    }
+
+public:
+    Utils::SmallString expression;
+};
+
+class Collate
+{
+public:
+    Collate(Utils::SmallStringView collation)
+        : collation(collation)
+    {}
+
+    friend bool operator==(const Collate &first, const Collate &second)
+    {
+        return first.collation == second.collation;
+    }
+
+public:
+    Utils::SmallString collation;
+};
+
+class ForeignKey
+{
+public:
+    ForeignKey() = default;
+    ForeignKey(Utils::SmallStringView table,
+               Utils::SmallStringView column,
+               ForeignKeyAction updateAction = {},
+               ForeignKeyAction deleteAction = {},
+               Enforment enforcement = {})
+        : table(table)
+        , column(column)
+        , updateAction(updateAction)
+        , deleteAction(deleteAction)
+        , enforcement(enforcement)
+    {}
+
+    friend bool operator==(const ForeignKey &first, const ForeignKey &second)
+    {
+        return first.table == second.table && first.column == second.column
+               && first.updateAction == second.updateAction
+               && first.deleteAction == second.deleteAction;
+    }
+
+public:
+    Utils::SmallString table;
+    Utils::SmallString column;
+    ForeignKeyAction updateAction = {};
+    ForeignKeyAction deleteAction = {};
+    Enforment enforcement = {};
+};
+
+enum class GeneratedAlwaysStorage { Stored, Virtual };
+
+class GeneratedAlways
+{
+public:
+    GeneratedAlways(Utils::SmallStringView expression, GeneratedAlwaysStorage storage)
+        : expression(expression)
+        , storage(storage)
+    {}
+
+    friend bool operator==(const GeneratedAlways &first, const GeneratedAlways &second)
+    {
+        return first.expression == second.expression;
+    }
+
+public:
+    Utils::SmallString expression;
+    GeneratedAlwaysStorage storage = {};
+};
+
+using Constraint = Utils::variant<Unique,
+                                  PrimaryKey,
+                                  ForeignKey,
+                                  NotNull,
+                                  Check,
+                                  DefaultValue,
+                                  DefaultExpression,
+                                  Collate,
+                                  GeneratedAlways>;
+using Constraints = std::vector<Constraint>;
+
+} // namespace Sqlite

src/libs/sqlite/createtablesqlstatementbuilder.cpp

@@ -154,6 +154,13 @@ public:
 
     void operator()(const NotNull &) { columnDefinitionString.append(" NOT NULL"); }
 
+    void operator()(const Check &check)
+    {
+        columnDefinitionString.append(" CHECK (");
+        columnDefinitionString.append(check.expression);
+        columnDefinitionString.append(")");
+    }
+
     void operator()(const DefaultValue &defaultValue)
     {
         columnDefinitionString.append(" DEFAULT ");

src/libs/sqlite/sqlite-lib.pri

@@ -24,12 +24,12 @@ SOURCES += \
     $$PWD/sqlitedatabase.cpp \
     $$PWD/sqlitebasestatement.cpp
 HEADERS += \
+    $$PWD/constraints.h \
     $$PWD/createtablesqlstatementbuilder.h \
     $$PWD/lastchangedrowid.h \
     $$PWD/sqlitedatabasebackend.h \
     $$PWD/sqlitedatabaseinterface.h \
     $$PWD/sqliteexception.h \
-    $$PWD/sqliteforeignkey.h \
     $$PWD/sqliteglobal.h \
     $$PWD/sqlitereadstatement.h \
     $$PWD/sqlitereadwritestatement.h \

src/libs/sqlite/sqlitecolumn.h

@@ -25,116 +25,12 @@
 
 #pragma once
 
-#include "sqliteforeignkey.h"
-
-#include <sqlitevalue.h>
-#include <utils/smallstring.h>
-#include <utils/variant.h>
+#include "constraints.h"
 
 #include <functional>
 
 namespace Sqlite {
 
-class Unique
-{
-    friend bool operator==(Unique, Unique) { return true; }
-};
-
-class PrimaryKey
-{
-    friend bool operator==(PrimaryKey, PrimaryKey) { return true; }
-};
-
-class NotNull
-{
-    friend bool operator==(NotNull, NotNull) { return true; }
-};
-
-class DefaultValue
-{
-public:
-    DefaultValue(long long value)
-        : value(value)
-    {}
-
-    DefaultValue(double value)
-        : value(value)
-    {}
-
-    DefaultValue(Utils::SmallStringView value)
-        : value(value)
-    {}
-
-    friend bool operator==(const DefaultValue &first, const DefaultValue &second)
-    {
-        return first.value == second.value;
-    }
-
-public:
-    Sqlite::Value value;
-};
-
-class DefaultExpression
-{
-public:
-    DefaultExpression(Utils::SmallStringView expression)
-        : expression(expression)
-    {}
-
-    friend bool operator==(const DefaultExpression &first, const DefaultExpression &second)
-    {
-        return first.expression == second.expression;
-    }
-
-public:
-    Utils::SmallString expression;
-};
-
-class Collate
-{
-public:
-    Collate(Utils::SmallStringView collation)
-        : collation(collation)
-    {}
-
-    friend bool operator==(const Collate &first, const Collate &second)
-    {
-        return first.collation == second.collation;
-    }
-
-public:
-    Utils::SmallString collation;
-};
-
-enum class GeneratedAlwaysStorage { Stored, Virtual };
-
-class GeneratedAlways
-{
-public:
-    GeneratedAlways(Utils::SmallStringView expression, GeneratedAlwaysStorage storage)
-        : expression(expression)
-        , storage(storage)
-    {}
-
-    friend bool operator==(const GeneratedAlways &first, const GeneratedAlways &second)
-    {
-        return first.expression == second.expression;
-    }
-
-public:
-    Utils::SmallString expression;
-    GeneratedAlwaysStorage storage = {};
-};
-
-using Constraint = Utils::variant<Unique,
-                                  PrimaryKey,
-                                  ForeignKey,
-                                  NotNull,
-                                  DefaultValue,
-                                  DefaultExpression,
-                                  Collate,
-                                  GeneratedAlways>;
-using Constraints = std::vector<Constraint>;
 
 class Column
 {

src/libs/sqlite/sqliteforeignkey.h

@@ -1,65 +0,0 @@
-/****************************************************************************
-**
-** Copyright (C) 2020 The Qt Company Ltd.
-** Contact: https://www.qt.io/licensing/
-**
-** This file is part of Qt Creator.
-**
-** Commercial License Usage
-** Licensees holding valid commercial Qt licenses may use this file in
-** accordance with the commercial license agreement provided with the
-** Software or, alternatively, in accordance with the terms contained in
-** a written agreement between you and The Qt Company. For licensing terms
-** and conditions see https://www.qt.io/terms-conditions. For further
-** information use the contact form at https://www.qt.io/contact-us.
-**
-** GNU General Public License Usage
-** Alternatively, this file may be used under the terms of the GNU
-** General Public License version 3 as published by the Free Software
-** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
-** included in the packaging of this file. Please review the following
-** information to ensure the GNU General Public License requirements will
-** be met: https://www.gnu.org/licenses/gpl-3.0.html.
-**
-****************************************************************************/
-
-#pragma once
-
-#include "sqliteglobal.h"
-
-#include <utils/smallstring.h>
-
-namespace Sqlite {
-
-class ForeignKey
-{
-public:
-    ForeignKey() = default;
-    ForeignKey(Utils::SmallStringView table,
-               Utils::SmallStringView column,
-               ForeignKeyAction updateAction = {},
-               ForeignKeyAction deleteAction = {},
-               Enforment enforcement = {})
-        : table(table)
-        , column(column)
-        , updateAction(updateAction)
-        , deleteAction(deleteAction)
-        , enforcement(enforcement)
-    {}
-
-    friend bool operator==(const ForeignKey &first, const ForeignKey &second)
-    {
-        return first.table == second.table && first.column == second.column
-               && first.updateAction == second.updateAction
-               && first.deleteAction == second.deleteAction;
-    }
-
-public:
-    Utils::SmallString table;
-    Utils::SmallString column;
-    ForeignKeyAction updateAction = {};
-    ForeignKeyAction deleteAction = {};
-    Enforment enforcement = {};
-};
-
-} // namespace Sqlite