Wed, 26 Nov 2014 08:12:01 -0800
Merge
1 #!/bin/sh
2 #
3 # Copyright (c) 2009, 2014, Oracle and/or its affiliates. All rights reserved.
4 # DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5 #
6 # This code is free software; you can redistribute it and/or modify it
7 # under the terms of the GNU General Public License version 2 only, as
8 # published by the Free Software Foundation.
9 #
10 # This code is distributed in the hope that it will be useful, but WITHOUT
11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12 # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13 # version 2 for more details (a copy is included in the LICENSE file that
14 # accompanied this code).
15 #
16 # You should have received a copy of the GNU General Public License version
17 # 2 along with this work; if not, write to the Free Software Foundation,
18 # Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
19 #
20 # Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
21 # or visit www.oracle.com if you need additional information or have any
22 # questions.
23 #
25 # Shell script for a fast parallel forest/trees command
27 usage() {
28 echo "usage: $0 [-h|--help] [-q|--quiet] [-v|--verbose] [-s|--sequential] [--] <command> [commands...]" > ${status_output}
29 echo "Environment variables which modify behaviour:"
30 echo " HGFOREST_QUIET : (boolean) If 'true' then standard output is redirected to /dev/null"
31 echo " HGFOREST_VERBOSE : (boolean) If 'true' then Mercurial asked to produce verbose output"
32 echo " HGFOREST_SEQUENTIAL : (boolean) If 'true' then repos are processed sequentially. Disables concurrency"
33 echo " HGFOREST_GLOBALOPTS : (string, must begin with space) Additional Mercurial global options"
34 echo " HGFOREST_REDIRECT : (file path) Redirect standard output to specified file"
35 echo " HGFOREST_FIFOS : (boolean) Default behaviour for FIFO detection. Does not override FIFOs disabled"
36 echo " HGFOREST_CONCURRENCY: (positive integer) Number of repos to process concurrently"
37 echo " HGFOREST_DEBUG : (boolean) If 'true' then temp files are retained"
38 exit 1
39 }
41 global_opts="${HGFOREST_GLOBALOPTS:-}"
42 status_output="${HGFOREST_REDIRECT:-/dev/stdout}"
43 qflag="${HGFOREST_QUIET:-false}"
44 vflag="${HGFOREST_VERBOSE:-false}"
45 sflag="${HGFOREST_SEQUENTIAL:-false}"
46 while [ $# -gt 0 ]
47 do
48 case $1 in
49 -h | --help )
50 usage
51 ;;
53 -q | --quiet )
54 qflag="true"
55 ;;
57 -v | --verbose )
58 vflag="true"
59 ;;
61 -s | --sequential )
62 sflag="true"
63 ;;
65 '--' ) # no more options
66 shift; break
67 ;;
69 -*) # bad option
70 usage
71 ;;
73 * ) # non option
74 break
75 ;;
76 esac
77 shift
78 done
80 # silence standard output?
81 if [ ${qflag} = "true" ] ; then
82 global_opts="${global_opts} -q"
83 status_output="/dev/null"
84 fi
86 # verbose output?
87 if [ ${vflag} = "true" ] ; then
88 global_opts="${global_opts} -v"
89 fi
91 # Make sure we have a command.
92 if [ $# -lt 1 -o -z "${1:-}" ] ; then
93 echo "ERROR: No command to hg supplied!"
94 usage
95 fi
97 command="$1"; shift
98 command_args="${@:-}"
100 # Clean out the temporary directory that stores the pid files.
101 tmp=/tmp/forest.$$
102 rm -f -r ${tmp}
103 mkdir -p ${tmp}
106 if [ "${HGFOREST_DEBUG:-false}" = "true" ] ; then
107 echo "DEBUG: temp files are in: ${tmp}"
108 fi
110 # Check if we can use fifos for monitoring sub-process completion.
111 echo "1" > ${tmp}/read
112 while_subshell=1
113 while read line; do
114 while_subshell=0
115 break;
116 done < ${tmp}/read
117 rm ${tmp}/read
119 on_windows=`uname -s | egrep -ic -e 'cygwin|msys'`
121 if [ ${while_subshell} = "1" -o ${on_windows} = "1" ]; then
122 # cygwin has (2014-04-18) broken (single writer only) FIFOs
123 # msys has (2014-04-18) no FIFOs.
124 # older shells create a sub-shell for redirect to while
125 have_fifos="false"
126 else
127 have_fifos="${HGFOREST_FIFOS:-true}"
128 fi
130 safe_interrupt () {
131 if [ -d ${tmp} ]; then
132 if [ "`ls ${tmp}/*.pid`" != "" ]; then
133 echo "Waiting for processes ( `cat ${tmp}/.*.pid ${tmp}/*.pid 2> /dev/null | tr '\n' ' '`) to terminate nicely!" > ${status_output}
134 sleep 1
135 # Pipe stderr to dev/null to silence kill, that complains when trying to kill
136 # a subprocess that has already exited.
137 kill -TERM `cat ${tmp}/*.pid | tr '\n' ' '` 2> /dev/null
138 wait
139 echo "Interrupt complete!" > ${status_output}
140 fi
141 rm -f -r ${tmp}
142 fi
143 exit 130
144 }
146 nice_exit () {
147 if [ -d ${tmp} ]; then
148 if [ "`ls -A ${tmp} 2> /dev/null`" != "" ]; then
149 wait
150 fi
151 if [ "${HGFOREST_DEBUG:-false}" != "true" ] ; then
152 rm -f -r ${tmp}
153 fi
154 fi
155 }
157 trap 'safe_interrupt' INT QUIT
158 trap 'nice_exit' EXIT
160 subrepos="corba jaxp jaxws langtools jdk hotspot nashorn"
161 subrepos_extra="jdk/src/closed jdk/make/closed jdk/test/closed hotspot/make/closed hotspot/src/closed hotspot/test/closed deploy install sponsors pubs"
163 # Only look in specific locations for possible forests (avoids long searches)
164 pull_default=""
165 repos=""
166 repos_extra=""
167 if [ "${command}" = "clone" -o "${command}" = "fclone" -o "${command}" = "tclone" ] ; then
168 # we must be a clone
169 if [ ! -f .hg/hgrc ] ; then
170 echo "ERROR: Need initial repository to use this script" > ${status_output}
171 exit 1
172 fi
174 # the clone must know where it came from (have a default pull path).
175 pull_default=`hg paths default`
176 if [ "${pull_default}" = "" ] ; then
177 echo "ERROR: Need initial clone with 'hg paths default' defined" > ${status_output}
178 exit 1
179 fi
181 # determine which sub repos need to be cloned.
182 for i in ${subrepos} ; do
183 if [ ! -f ${i}/.hg/hgrc ] ; then
184 repos="${repos} ${i}"
185 fi
186 done
188 pull_default_tail=`echo ${pull_default} | sed -e 's@^.*://[^/]*/\(.*\)@\1@'`
190 if [ -n "${command_args}" ] ; then
191 # if there is an "extra sources" path then reparent "extra" repos to that path
192 if [ "x${pull_default}" = "x${pull_default_tail}" ] ; then
193 echo "ERROR: Need initial clone from non-local source" > ${status_output}
194 exit 1
195 fi
196 pull_extra="${command_args}/${pull_default_tail}"
198 # determine which extra subrepos need to be cloned.
199 for i in ${subrepos_extra} ; do
200 if [ ! -f ${i}/.hg/hgrc ] ; then
201 repos_extra="${repos_extra} ${i}"
202 fi
203 done
204 else
205 if [ "x${pull_default}" = "x${pull_default_tail}" ] ; then
206 # local source repo. Clone the "extra" subrepos that exist there.
207 for i in ${subrepos_extra} ; do
208 if [ -f ${pull_default}/${i}/.hg/hgrc -a ! -f ${i}/.hg/hgrc ] ; then
209 # sub-repo there in source but not here
210 repos_extra="${repos_extra} ${i}"
211 fi
212 done
213 fi
214 fi
216 # Any repos to deal with?
217 if [ "${repos}" = "" -a "${repos_extra}" = "" ] ; then
218 echo "No repositories to process." > ${status_output}
219 exit
220 fi
222 # Repos to process concurrently. Clone does better with low concurrency.
223 at_a_time="${HGFOREST_CONCURRENCY:-2}"
224 else
225 # Process command for all of the present repos
226 for i in . ${subrepos} ${subrepos_extra} ; do
227 if [ -d ${i}/.hg ] ; then
228 repos="${repos} ${i}"
229 fi
230 done
232 # Any repos to deal with?
233 if [ "${repos}" = "" ] ; then
234 echo "No repositories to process." > ${status_output}
235 exit
236 fi
238 # any of the repos locked?
239 locked=""
240 for i in ${repos} ; do
241 if [ -h ${i}/.hg/store/lock -o -f ${i}/.hg/store/lock ] ; then
242 locked="${i} ${locked}"
243 fi
244 done
245 if [ "${locked}" != "" ] ; then
246 echo "ERROR: These repositories are locked: ${locked}" > ${status_output}
247 exit 1
248 fi
250 # Repos to process concurrently.
251 at_a_time="${HGFOREST_CONCURRENCY:-8}"
252 fi
254 # Echo out what repositories we do a command on.
255 echo "# Repositories: ${repos} ${repos_extra}" > ${status_output}
257 if [ "${command}" = "serve" ] ; then
258 # "serve" is run for all the repos as one command.
259 (
260 (
261 cwd=`pwd`
262 serving=`basename ${cwd}`
263 (
264 echo "[web]"
265 echo "description = ${serving}"
266 echo "allow_push = *"
267 echo "push_ssl = False"
269 echo "[paths]"
270 for i in ${repos} ; do
271 if [ "${i}" != "." ] ; then
272 echo "/${serving}/${i} = ${i}"
273 else
274 echo "/${serving} = ${cwd}"
275 fi
276 done
277 ) > ${tmp}/serve.web-conf
279 echo "serving root repo ${serving}" > ${status_output}
281 echo "hg${global_opts} serve" > ${status_output}
282 (PYTHONUNBUFFERED=true hg${global_opts} serve -A ${status_output} -E ${status_output} --pid-file ${tmp}/serve.pid --web-conf ${tmp}/serve.web-conf; echo "$?" > ${tmp}/serve.pid.rc ) 2>&1 &
283 ) 2>&1 | sed -e "s@^@serve: @" > ${status_output}
284 ) &
285 else
286 # Run the supplied command on all repos in parallel.
288 # n is the number of subprocess started or which might still be running.
289 n=0
290 if [ ${have_fifos} = "true" ]; then
291 # if we have fifos use them to detect command completion.
292 mkfifo ${tmp}/fifo
293 exec 3<>${tmp}/fifo
294 fi
296 # iterate over all of the subrepos.
297 for i in ${repos} ${repos_extra} ; do
298 n=`expr ${n} '+' 1`
299 repopidfile=`echo ${i} | sed -e 's@./@@' -e 's@/@_@g'`
300 reponame=`echo ${i} | sed -e :a -e 's/^.\{1,20\}$/ &/;ta'`
301 pull_base="${pull_default}"
303 # regular repo or "extra" repo?
304 for j in ${repos_extra} ; do
305 if [ "${i}" = "${j}" ] ; then
306 # it's an "extra"
307 pull_base="${pull_extra}"
308 fi
309 done
311 # remove trailing slash
312 pull_base="`echo ${pull_base} | sed -e 's@[/]*$@@'`"
314 # execute the command on the subrepo
315 (
316 (
317 if [ "${command}" = "clone" -o "${command}" = "fclone" -o "${command}" = "tclone" ] ; then
318 # some form of clone
319 clone_newrepo="${pull_base}/${i}"
320 parent_path="`dirname ${i}`"
321 if [ "${parent_path}" != "." ] ; then
322 times=0
323 while [ ! -d "${parent_path}" ] ; do ## nested repo, ensure containing dir exists
324 if [ "${sflag}" = "true" ] ; then
325 # Missing parent is fatal during sequential operation.
326 echo "ERROR: Missing parent path: ${parent_path}" > ${status_output}
327 exit 1
328 fi
329 times=`expr ${times} '+' 1`
330 if [ `expr ${times} '%' 10` -eq 0 ] ; then
331 echo "${parent_path} still not created, waiting..." > ${status_output}
332 fi
333 sleep 5
334 done
335 fi
336 # run the clone command.
337 echo "hg${global_opts} clone ${clone_newrepo} ${i}" > ${status_output}
338 (PYTHONUNBUFFERED=true hg${global_opts} clone ${clone_newrepo} ${i}; echo "$?" > ${tmp}/${repopidfile}.pid.rc ) 2>&1 &
339 else
340 # run the command.
341 echo "cd ${i} && hg${global_opts} ${command} ${command_args}" > ${status_output}
342 cd ${i} && (PYTHONUNBUFFERED=true hg${global_opts} ${command} ${command_args}; echo "$?" > ${tmp}/${repopidfile}.pid.rc ) 2>&1 &
343 fi
345 echo $! > ${tmp}/${repopidfile}.pid
346 ) 2>&1 | sed -e "s@^@${reponame}: @" > ${status_output}
347 # tell the fifo waiter that this subprocess is done.
348 if [ ${have_fifos} = "true" ]; then
349 echo "${i}" >&3
350 fi
351 ) &
353 if [ "${sflag}" = "true" ] ; then
354 # complete this task before starting another.
355 wait
356 else
357 if [ "${have_fifos}" = "true" ]; then
358 # check on count of running subprocesses and possibly wait for completion
359 if [ ${n} -ge ${at_a_time} ] ; then
360 # read will block until there are completed subprocesses
361 while read repo_done; do
362 n=`expr ${n} '-' 1`
363 if [ ${n} -lt ${at_a_time} ] ; then
364 # we should start more subprocesses
365 break;
366 fi
367 done <&3
368 fi
369 else
370 # Compare completions to starts
371 completed="`(ls -a1 ${tmp}/*.pid.rc 2> /dev/null | wc -l) || echo 0`"
372 while [ `expr ${n} '-' ${completed}` -ge ${at_a_time} ] ; do
373 # sleep a short time to give time for something to complete
374 sleep 1
375 completed="`(ls -a1 ${tmp}/*.pid.rc 2> /dev/null | wc -l) || echo 0`"
376 done
377 fi
378 fi
379 done
380 fi
382 # Wait for all subprocesses to complete
383 wait
385 # Terminate with exit 0 only if all subprocesses were successful
386 ec=0
387 if [ -d ${tmp} ]; then
388 rcfiles="`(ls -a ${tmp}/*.pid.rc 2> /dev/null) || echo ''`"
389 for rc in ${rcfiles} ; do
390 exit_code=`cat ${rc} | tr -d ' \n\r'`
391 if [ "${exit_code}" != "0" ] ; then
392 repo="`echo ${rc} | sed -e 's@^'${tmp}'@@' -e 's@/*\([^/]*\)\.pid\.rc$@\1@' -e 's@_@/@g'`"
393 echo "WARNING: ${repo} exited abnormally (${exit_code})" > ${status_output}
394 ec=1
395 fi
396 done
397 fi
398 exit ${ec}