mirror of https://github.com/apache/kafka.git
				
				
				
			
		
			
				
	
	
		
			75 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
			
		
		
	
	
			75 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
#!/usr/bin/env python
 | 
						|
 | 
						|
# Licensed to the Apache Software Foundation (ASF) under one or more
 | 
						|
# contributor license agreements.  See the NOTICE file distributed with
 | 
						|
# this work for additional information regarding copyright ownership.
 | 
						|
# The ASF licenses this file to You under the Apache License, Version 2.0
 | 
						|
# (the "License"); you may not use this file except in compliance with
 | 
						|
# the License.  You may obtain a copy of the License at
 | 
						|
#
 | 
						|
#    http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
#
 | 
						|
# Unless required by applicable law or agreed to in writing, software
 | 
						|
# distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
# See the License for the specific language governing permissions and
 | 
						|
# limitations under the License.
 | 
						|
 | 
						|
"""
 | 
						|
Python script to build and push a multiarch docker image
 | 
						|
This script is used to prepare and publish docker release candidate
 | 
						|
 | 
						|
Pre requisites:
 | 
						|
    Ensure that you are logged in the docker registry and you have access to push to that registry.
 | 
						|
    Ensure that docker buildx is enabled for you.
 | 
						|
 | 
						|
Usage:
 | 
						|
    docker_release.py --help
 | 
						|
        Get detailed description of argument
 | 
						|
 | 
						|
    Example command:-
 | 
						|
        docker_release <image> --kafka-url <kafka_url> --image-type <type>
 | 
						|
 | 
						|
        This command will build the multiarch image of type <type> (jvm by default),
 | 
						|
        named <image> using <kafka_url> to download kafka and push it to the docker image name <image> provided.
 | 
						|
        Make sure image is in the format of <registry>/<namespace>/<image_name>:<image_tag>.
 | 
						|
"""
 | 
						|
 | 
						|
from datetime import date
 | 
						|
import argparse
 | 
						|
 | 
						|
from common import execute, build_docker_image_runner
 | 
						|
 | 
						|
def build_push(image, kafka_url, image_type):
 | 
						|
    try:
 | 
						|
        create_builder()
 | 
						|
        build_docker_image_runner(f"docker buildx build -f $DOCKER_FILE --build-arg kafka_url={kafka_url} --build-arg build_date={date.today()} --push \
 | 
						|
              --platform linux/amd64,linux/arm64 --tag {image} $DOCKER_DIR", image_type)
 | 
						|
    except:
 | 
						|
        raise SystemError("Docker image push failed")
 | 
						|
    finally:
 | 
						|
        remove_builder()
 | 
						|
 | 
						|
def create_builder():
 | 
						|
    execute(["docker", "buildx", "create", "--name", "kafka-builder", "--use"])
 | 
						|
 | 
						|
def remove_builder():
 | 
						|
    execute(["docker", "buildx", "rm", "kafka-builder"])
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    print("\
 | 
						|
          This script will build and push docker images of apache kafka.\n \
 | 
						|
          Please ensure that image has been sanity tested before pushing the image. \n \
 | 
						|
          Please ensure you are logged in the docker registry that you are trying to push to.")
 | 
						|
    parser = argparse.ArgumentParser()
 | 
						|
    parser.add_argument("image", help="Dockerhub image that you want to push to (in the format <registry>/<namespace>/<image_name>:<image_tag>)")
 | 
						|
    parser.add_argument("--image-type", "-type", choices=["jvm", "native"], default="jvm", dest="image_type", help="Image type you want to build")
 | 
						|
    parser.add_argument("--kafka-url", "-u", dest="kafka_url", help="Kafka url to be used to download kafka binary tarball in the docker image")
 | 
						|
    args = parser.parse_args()
 | 
						|
 | 
						|
    print(f"Docker image of type {args.image_type} containing kafka downloaded from {args.kafka_url} will be pushed to {args.image}")
 | 
						|
 | 
						|
    print("Building and pushing the image")
 | 
						|
    build_push(args.image, args.kafka_url, args.image_type)
 | 
						|
    print(f"Image has been pushed to {args.image}")
 |