From 773849a43d9ab25df77c38e00de5d6b914c0b0f2 Mon Sep 17 00:00:00 2001
From: Gleb Novikov <gnovikov@iponweb.net>
Date: Sun, 19 May 2019 09:08:25 +0300
Subject: [PATCH] Added constraints getter and setter to IStorage

---
 dbms/src/Storages/IStorage.cpp | 10 ++++++++++
 dbms/src/Storages/IStorage.h   |  5 +++++
 2 files changed, 15 insertions(+)

diff --git a/dbms/src/Storages/IStorage.cpp b/dbms/src/Storages/IStorage.cpp
index 06320cc1f30..7c19fd94aea 100644
--- a/dbms/src/Storages/IStorage.cpp
+++ b/dbms/src/Storages/IStorage.cpp
@@ -47,6 +47,16 @@ void IStorage::setIndices(IndicesDescription indices_)
     indices = std::move(indices_);
 }
 
+const ConstraintsDescription & IStorage::getConstraints() const
+{
+    return constraints;
+}
+
+void IStorage::setConstraints(ConstraintsDescription constraints_)
+{
+    constraints = std::move(constraints_);
+}
+
 NameAndTypePair IStorage::getColumn(const String & column_name) const
 {
     /// By default, we assume that there are no virtual columns in the storage.
diff --git a/dbms/src/Storages/IStorage.h b/dbms/src/Storages/IStorage.h
index f18592ebce5..b01244ba111 100644
--- a/dbms/src/Storages/IStorage.h
+++ b/dbms/src/Storages/IStorage.h
@@ -11,6 +11,7 @@
 #include <Common/ActionLock.h>
 #include <Common/Exception.h>
 #include <Common/RWLock.h>
+#include <Storages/ConstraintsDescription.h>
 
 #include <optional>
 #include <shared_mutex>
@@ -88,6 +89,9 @@ public: /// thread-unsafe part. lockStructure must be acquired
     const IndicesDescription & getIndices() const;
     void setIndices(IndicesDescription indices_);
 
+    const ConstraintsDescription & getConstraints() const;
+    void setConstraints(ConstraintsDescription constraints_);
+
     /// NOTE: these methods should include virtual columns,
     ///       but should NOT include ALIAS columns (they are treated separately).
     virtual NameAndTypePair getColumn(const String & column_name) const;
@@ -115,6 +119,7 @@ public: /// thread-unsafe part. lockStructure must be acquired
 private:
     ColumnsDescription columns;
     IndicesDescription indices;
+    ConstraintsDescription constraints;
 
 public:
     /// Acquire this lock if you need the table structure to remain constant during the execution of