1 files changed, 44 insertions, 0 deletions
diff --git a/gems/gitlab-schema-validation/lib/gitlab/schema/validation/schema_objects/table.rb b/gems/gitlab-schema-validation/lib/gitlab/schema/validation/schema_objects/table.rb
new file mode 100644
index 00000000000..591131cb220
--- /dev/null
+++ b/gems/gitlab-schema-validation/lib/gitlab/schema/validation/schema_objects/table.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Schema
+    module Validation
+      module SchemaObjects
+        class Table
+          def initialize(name, columns)
+            @name = name
+            @columns = columns
+          end
+
+          attr_reader :name, :columns
+
+          def table_name
+            name
+          end
+
+          def statement
+            format('CREATE TABLE %s (%s)', name, columns_statement)
+          end
+
+          def fetch_column_by_name(column_name)
+            columns.find { |column| column.name == column_name }
+          end
+
+          def column_exists?(column_name)
+            column = fetch_column_by_name(column_name)
+
+            return false if column.nil?
+
+            true
+          end
+
+          private
+
+          def columns_statement
+            columns.reject(&:partition_key?).map(&:statement).join(', ')
+          end
+        end
+      end
+    end
+  end
+end