1 files changed, 75 insertions, 37 deletions
diff --git a/lib/gitlab/ci/config/entry/cache.rb b/lib/gitlab/ci/config/entry/cache.rb
index 6b036182706..cf599ce5294 100644
--- a/lib/gitlab/ci/config/entry/cache.rb
+++ b/lib/gitlab/ci/config/entry/cache.rb
@@ -7,52 +7,90 @@ module Gitlab
         ##
         # Entry that represents a cache configuration
         #
-        class Cache < ::Gitlab::Config::Entry::Node
-          include ::Gitlab::Config::Entry::Configurable
-          include ::Gitlab::Config::Entry::Validatable
-          include ::Gitlab::Config::Entry::Attributable
-
-          ALLOWED_KEYS = %i[key untracked paths when policy].freeze
-          ALLOWED_POLICY = %w[pull-push push pull].freeze
-          DEFAULT_POLICY = 'pull-push'
-          ALLOWED_WHEN = %w[on_success on_failure always].freeze
-          DEFAULT_WHEN = 'on_success'
-
-          validations do
-            validates :config, type: Hash, allowed_keys: ALLOWED_KEYS
-            validates :policy,
-              inclusion: { in: ALLOWED_POLICY, message: 'should be pull-push, push, or pull' },
-              allow_blank: true
-
-            with_options allow_nil: true do
-              validates :when,
-                  inclusion: {
-                    in: ALLOWED_WHEN,
-                    message: 'should be on_success, on_failure or always'
-                  }
+        class Cache < ::Gitlab::Config::Entry::Simplifiable
+          strategy :Caches, if: -> (config) { Feature.enabled?(:multiple_cache_per_job) }
+          strategy :Cache, if: -> (config) { Feature.disabled?(:multiple_cache_per_job) }
+
+          class Caches < ::Gitlab::Config::Entry::ComposableArray
+            include ::Gitlab::Config::Entry::Validatable
+
+            MULTIPLE_CACHE_LIMIT = 4
+
+            validations do
+              validates :config, presence: true
+
+              validate do
+                unless config.is_a?(Hash) || config.is_a?(Array)
+                  errors.add(:config, 'can only be a Hash or an Array')
+                end
+
+                if config.is_a?(Array) && config.count > MULTIPLE_CACHE_LIMIT
+                  errors.add(:config, "no more than #{MULTIPLE_CACHE_LIMIT} caches can be created")
+                end
+              end
+            end
+
+            def initialize(*args)
+              super
+
+              @key = nil
+            end
+
+            def composable_class
+              Entry::Cache::Cache
             end
           end
 
-          entry :key, Entry::Key,
-            description: 'Cache key used to define a cache affinity.'
+          class Cache < ::Gitlab::Config::Entry::Node
+            include ::Gitlab::Config::Entry::Configurable
+            include ::Gitlab::Config::Entry::Validatable
+            include ::Gitlab::Config::Entry::Attributable
+
+            ALLOWED_KEYS = %i[key untracked paths when policy].freeze
+            ALLOWED_POLICY = %w[pull-push push pull].freeze
+            DEFAULT_POLICY = 'pull-push'
+            ALLOWED_WHEN = %w[on_success on_failure always].freeze
+            DEFAULT_WHEN = 'on_success'
+
+            validations do
+              validates :config, type: Hash, allowed_keys: ALLOWED_KEYS
+              validates :policy,
+                inclusion: { in: ALLOWED_POLICY, message: 'should be pull-push, push, or pull' },
+                allow_blank: true
+
+              with_options allow_nil: true do
+                validates :when,
+                    inclusion: {
+                      in: ALLOWED_WHEN,
+                      message: 'should be on_success, on_failure or always'
+                    }
+              end
+            end
 
-          entry :untracked, ::Gitlab::Config::Entry::Boolean,
-            description: 'Cache all untracked files.'
+            entry :key, Entry::Key,
+              description: 'Cache key used to define a cache affinity.'
 
-          entry :paths, Entry::Paths,
-            description: 'Specify which paths should be cached across builds.'
+            entry :untracked, ::Gitlab::Config::Entry::Boolean,
+              description: 'Cache all untracked files.'
 
-          attributes :policy, :when
+            entry :paths, Entry::Paths,
+              description: 'Specify which paths should be cached across builds.'
 
-          def value
-            result = super
+            attributes :policy, :when
 
-            result[:key] = key_value
-            result[:policy] = policy || DEFAULT_POLICY
-            # Use self.when to avoid conflict with reserved word
-            result[:when] = self.when || DEFAULT_WHEN
+            def value
+              result = super
+
+              result[:key] = key_value
+              result[:policy] = policy || DEFAULT_POLICY
+              # Use self.when to avoid conflict with reserved word
+              result[:when] = self.when || DEFAULT_WHEN
+
+              result
+            end
+          end
 
-            result
+          class UnknownStrategy < ::Gitlab::Config::Entry::Node
           end
         end
       end