diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-01-21 15:08:31 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-01-21 15:08:31 +0300 |
commit | 28ce39a3e0e7b47e53939a15fb823af9c433327a (patch) | |
tree | f703cd7b431827880ccb74b98d16fab0f60dd245 /db/fixtures | |
parent | afe057a8ff8546f0032e439a9a200307fb6de86a (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'db/fixtures')
-rw-r--r-- | db/fixtures/development/26_container_images.rb | 65 |
1 files changed, 65 insertions, 0 deletions
diff --git a/db/fixtures/development/26_container_images.rb b/db/fixtures/development/26_container_images.rb new file mode 100644 index 00000000000..5b5879673b9 --- /dev/null +++ b/db/fixtures/development/26_container_images.rb @@ -0,0 +1,65 @@ +# frozen_string_literal: true + +class Gitlab::Seeder::ContainerImages + attr_reader :tmp_dir, :project, :images_count + + DOCKER_FILE_CONTENTS = <<~EOS + FROM scratch + ARG tag + ENV tag=$tag + EOS + + def initialize(project, images_count) + @project = project + @images_count = images_count + initialize_tmp_dir + end + + def seed! + images_count.times do |i| + image_path = "#{project.container_registry_url}:tag_#{i}" + build_image(image_path) + push_image(image_path) + puts '.' + end + ensure + FileUtils.remove_entry tmp_dir + end + + private + + def build_image(image_path) + system(*%W[docker build -t #{image_path} --build-arg tag=gitlab_container_image_seed .], chdir: @tmp_dir) + end + + def push_image(image_path) + system(*%W[docker push #{image_path}], chdir: @tmp_dir) + end + + def initialize_tmp_dir + @tmp_dir = Dir.mktmpdir('gitlab_seeder_container_images') + + File.write(File.join(@tmp_dir, 'Dockerfile'), DOCKER_FILE_CONTENTS) + end +end + +Gitlab::Seeder.quiet do + flag = 'SEED_CONTAINER_IMAGES' + + if ENV[flag] + admin_user = User.admins.first + images_count = Integer(ENV[flag]) + + Project.not_mass_generated.visible_to_user(admin_user).sample(1).each do |project| + puts "\nSeeding #{images_count} container images to the '#{project.full_path}' project." + + seeder = Gitlab::Seeder::ContainerImages.new(project, images_count) + seeder.seed! + rescue => e + puts "\nSeeding container images failed with #{e.message}." + puts "Make sure that the registry is running (https://gitlab.com/gitlab-org/gitlab-development-kit/blob/master/doc/howto/registry.md) and that Docker CLI (https://www.docker.com/products/docker-desktop) is installed." + end + else + puts "Skipped. Use the `#{flag}` environment variable to seed container images to the registry." + end +end |