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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#!/bin/bash
# Copyright The containerd Authors.
# 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 -euo pipefail
CONTAINERD_ROOT=/var/lib/containerd/
CONTAINERD_CONFIG_DIR=/etc/containerd/
REMOTE_SNAPSHOTTER_SOCKET=/run/containerd-stargz-grpc/containerd-stargz-grpc.sock
REMOTE_SNAPSHOTTER_ROOT=/var/lib/containerd-stargz-grpc/
REMOTE_SNAPSHOTTER_CONFIG_DIR=/etc/containerd-stargz-grpc/
RETRYNUM=30
RETRYINTERVAL=1
TIMEOUTSEC=180
function retry {
local SUCCESS=false
for i in $(seq ${RETRYNUM}) ; do
if eval "timeout ${TIMEOUTSEC} ${@}" ; then
SUCCESS=true
break
fi
echo "Fail(${i}). Retrying..."
sleep ${RETRYINTERVAL}
done
if [ "${SUCCESS}" == "true" ] ; then
return 0
else
return 1
fi
}
function kill_all {
if [ "${1}" != "" ] ; then
ps aux | grep "${1}" \
| grep -v "benchmark" \
| grep -v grep \
| grep -v "hello.py" \
| grep -v $(basename ${0}) \
| sed -E 's/ +/ /g' | cut -f 2 -d ' ' | xargs -I{} kill -9 {} || true
fi
}
function cleanup {
rm -rf "${CONTAINERD_ROOT}"*
if [ -f "${REMOTE_SNAPSHOTTER_SOCKET}" ] ; then
rm "${REMOTE_SNAPSHOTTER_SOCKET}"
fi
if [ -d "${REMOTE_SNAPSHOTTER_ROOT}snapshotter/snapshots/" ] ; then
find "${REMOTE_SNAPSHOTTER_ROOT}snapshotter/snapshots/" \
-maxdepth 1 -mindepth 1 -type d -exec umount "{}/fs" \;
fi
rm -rf "${REMOTE_SNAPSHOTTER_ROOT}"*
}
echo "cleaning up the environment..."
kill_all "containerd"
kill_all "containerd-stargz-grpc"
cleanup
if [ "${DISABLE_PREFETCH:-}" == "true" ] ; then
sed -i 's/noprefetch = .*/noprefetch = true/g' "${REMOTE_SNAPSHOTTER_CONFIG_DIR}config.toml"
else
sed -i 's/noprefetch = .*/noprefetch = false/g' "${REMOTE_SNAPSHOTTER_CONFIG_DIR}config.toml"
fi
if [ "${DISABLE_ESTARGZ:-}" == "true" ] ; then
echo "DO NOT RUN remote snapshotter"
else
echo "running remote snaphsotter..."
if [ "${LOG_FILE:-}" == "" ] ; then
LOG_FILE=/dev/null
fi
containerd-stargz-grpc --log-level=debug \
--address="${REMOTE_SNAPSHOTTER_SOCKET}" \
--config="${REMOTE_SNAPSHOTTER_CONFIG_DIR}config.toml" \
2>&1 | tee -a "${LOG_FILE}" & # Dump all log
retry ls "${REMOTE_SNAPSHOTTER_SOCKET}"
fi
echo "running containerd..."
containerd --config="${CONTAINERD_CONFIG_DIR}config.toml" &
CTRCMD=ctr-remote
if ! which "${CTRCMD}" ; then
if ! which ctr ; then
echo "ctr nor ctr-remote not found"
exit 1
fi
CTRCMD=ctr
fi
retry "${CTRCMD}" version
|
