aboutsummaryrefslogtreecommitdiffhomepage
path: root/tools/build_defs/docker/incremental_load.sh.tpl
blob: d74c231fd5e5e40b21cedc2b9b101d0c75828c2c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#!/bin/bash
#
# Copyright 2015 The Bazel Authors. All rights reserved.
#
# Licensed 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.

set -eu

# This is a generated file that loads all docker layers built by "docker_build".

RUNFILES="${PYTHON_RUNFILES:-${BASH_SOURCE[0]}.runfiles}"

DOCKER="${DOCKER:-docker}"

FULL_DOCKER_VERSION=$(docker version -f {{.Server.Version}} 2> /dev/null \
    || echo "1.5.0")
DOCKER_MAJOR_VERSION=$(echo "$FULL_DOCKER_VERSION" | awk -F\. '{ print $1 }')
DOCKER_MINOR_VERSION=$(echo "$FULL_DOCKER_VERSION" | awk -F\. '{ print $2 }')
if [ "$DOCKER_MAJOR_VERSION" -eq "1" ] && [ "$DOCKER_MINOR_VERSION" -lt "10" ]; then
  LEGACY_DOCKER=true
else
  LEGACY_DOCKER=false
fi

# List all images identifier (only the identifier) from the local
# docker registry.
IMAGES="$("${DOCKER}" images -aq)"
IMAGE_LEN=$(for i in $IMAGES; do echo -n $i | wc -c; done | sort -g | head -1 | xargs)

[ -n "$IMAGE_LEN" ] || IMAGE_LEN=64

function incr_load() {
  # Load a layer if and only if the layer is not in "$IMAGES", that is
  # in the local docker registry.
  if [ "$LEGACY_DOCKER" = true ]; then
    name=$(cat ${RUNFILES}/$1)
  else
    name=$(cat ${RUNFILES}/$2)
  fi

  if (echo "$IMAGES" | grep -q ^${name:0:$IMAGE_LEN}$); then
    echo "Skipping $name, already loaded."
  else
    echo "Loading $name..."
    "${DOCKER}" load -i ${RUNFILES}/$3
  fi
}

function tag_layer() {
  if [ "$LEGACY_DOCKER" = true ]; then
    name=$(cat ${RUNFILES}/$2)
  else
    name=$(cat ${RUNFILES}/$3)
  fi

  TAG="$1"
  echo "Tagging ${name} as ${TAG}"
  if [ "$LEGACY_DOCKER" = true ]; then
    "${DOCKER}" tag -f ${name} ${TAG}
  else
    "${DOCKER}" tag sha256:${name} ${TAG}
  fi
}

# List of 'incr_load' statements for all layers.
# This generated and injected by docker_build.
%{load_statements}

# List of 'tag_layer' statements for all tags.
# This generated and injected by docker_build.
%{tag_statements}