common/bin/hgforest.sh

changeset 0
75a576e87639
child 1
2bcf2a10b799
equal deleted inserted replaced
-1:000000000000 0:75a576e87639
1 #!/bin/sh
2
3 #
4 # Copyright (c) 2009, 2013, Oracle and/or its affiliates. All rights reserved.
5 # DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 #
7 # This code is free software; you can redistribute it and/or modify it
8 # under the terms of the GNU General Public License version 2 only, as
9 # published by the Free Software Foundation.
10 #
11 # This code is distributed in the hope that it will be useful, but WITHOUT
12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 # version 2 for more details (a copy is included in the LICENSE file that
15 # accompanied this code).
16 #
17 # You should have received a copy of the GNU General Public License version
18 # 2 along with this work; if not, write to the Free Software Foundation,
19 # Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 #
21 # Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 # or visit www.oracle.com if you need additional information or have any
23 # questions.
24 #
25
26 # Shell script for a fast parallel forest command
27 command="$1"
28 pull_extra_base="$2"
29
30 if [ "" = "$command" ] ; then
31 echo No command to hg supplied!
32 exit 1
33 fi
34
35 # Clean out the temporary directory that stores the pid files.
36 tmp=/tmp/forest.$$
37 rm -f -r ${tmp}
38 mkdir -p ${tmp}
39
40 safe_interrupt () {
41 if [ -d ${tmp} ]; then
42 if [ "`ls ${tmp}/*.pid`" != "" ]; then
43 echo "Waiting for processes ( `cat ${tmp}/*.pid | tr '\n' ' '`) to terminate nicely!"
44 sleep 1
45 # Pipe stderr to dev/null to silence kill, that complains when trying to kill
46 # a subprocess that has already exited.
47 kill -TERM `cat ${tmp}/*.pid | tr '\n' ' '` 2> /dev/null
48 wait
49 echo Interrupt complete!
50 fi
51 fi
52 rm -f -r ${tmp}
53 exit 1
54 }
55
56 nice_exit () {
57 if [ -d ${tmp} ]; then
58 if [ "`ls ${tmp}`" != "" ]; then
59 wait
60 fi
61 fi
62 rm -f -r ${tmp}
63 }
64
65 trap 'safe_interrupt' INT QUIT
66 trap 'nice_exit' EXIT
67
68 # Only look in specific locations for possible forests (avoids long searches)
69 pull_default=""
70 repos=""
71 repos_extra=""
72 if [ "${command}" = "clone" -o "${command}" = "fclone" ] ; then
73 subrepos="corba jaxp jaxws langtools jdk hotspot nashorn"
74 if [ -f .hg/hgrc ] ; then
75 pull_default=`hg paths default`
76 if [ "${pull_default}" = "" ] ; then
77 echo "ERROR: Need initial clone with 'hg paths default' defined"
78 exit 1
79 fi
80 fi
81 if [ "${pull_default}" = "" ] ; then
82 echo "ERROR: Need initial repository to use this script"
83 exit 1
84 fi
85 for i in ${subrepos} ; do
86 if [ ! -f ${i}/.hg/hgrc ] ; then
87 repos="${repos} ${i}"
88 fi
89 done
90 if [ "${pull_extra_base}" != "" ] ; then
91 subrepos_extra="jdk/src/closed jdk/make/closed jdk/test/closed hotspot/make/closed hotspot/src/closed hotspot/test/closed deploy install sponsors pubs"
92 pull_default_tail=`echo ${pull_default} | sed -e 's@^.*://[^/]*/\(.*\)@\1@'`
93 pull_extra="${pull_extra_base}/${pull_default_tail}"
94 for i in ${subrepos_extra} ; do
95 if [ ! -f ${i}/.hg/hgrc ] ; then
96 repos_extra="${repos_extra} ${i}"
97 fi
98 done
99 fi
100 at_a_time=2
101 # Any repos to deal with?
102 if [ "${repos}" = "" -a "${repos_extra}" = "" ] ; then
103 exit
104 fi
105 else
106 hgdirs=`ls -d ./.hg ./*/.hg ./*/*/.hg ./*/*/*/.hg ./*/*/*/*/.hg 2>/dev/null`
107 # Derive repository names from the .hg directory locations
108 for i in ${hgdirs} ; do
109 repos="${repos} `echo ${i} | sed -e 's@/.hg$@@'`"
110 done
111 for i in ${repos} ; do
112 if [ -h ${i}/.hg/store/lock -o -f ${i}/.hg/store/lock ] ; then
113 locked="${i} ${locked}"
114 fi
115 done
116 at_a_time=8
117 # Any repos to deal with?
118 if [ "${repos}" = "" ] ; then
119 echo "No repositories to process."
120 exit
121 fi
122 if [ "${locked}" != "" ] ; then
123 echo "These repositories are locked: ${locked}"
124 exit
125 fi
126 fi
127
128 # Echo out what repositories we do a command on.
129 echo "# Repositories: ${repos} ${repos_extra}"
130 echo
131
132 # Run the supplied command on all repos in parallel.
133 n=0
134 for i in ${repos} ${repos_extra} ; do
135 n=`expr ${n} '+' 1`
136 repopidfile=`echo ${i} | sed -e 's@./@@' -e 's@/@_@g'`
137 reponame=`echo ${i} | sed -e :a -e 's/^.\{1,20\}$/ &/;ta'`
138 pull_base="${pull_default}"
139 for j in $repos_extra ; do
140 if [ "$i" = "$j" ] ; then
141 pull_base="${pull_extra}"
142 fi
143 done
144 (
145 (
146 if [ "${command}" = "clone" -o "${command}" = "fclone" ] ; then
147 pull_newrepo="`echo ${pull_base}/${i} | sed -e 's@\([^:]/\)//*@\1@g'`"
148 echo hg clone ${pull_newrepo} ${i}
149 path="`dirname ${i}`"
150 if [ "${path}" != "." ] ; then
151 times=0
152 while [ ! -d "${path}" ] ## nested repo, ensure containing dir exists
153 do
154 times=`expr ${times} '+' 1`
155 if [ `expr ${times} '%' 10` -eq 0 ] ; then
156 echo ${path} still not created, waiting...
157 fi
158 sleep 5
159 done
160 fi
161 (PYTHONUNBUFFERED=true hg clone ${pull_newrepo} ${i}; echo "$?" > ${tmp}/${repopidfile}.pid.rc )&
162 else
163 echo "cd ${i} && hg $*"
164 cd ${i} && (PYTHONUNBUFFERED=true hg "$@"; echo "$?" > ${tmp}/${repopidfile}.pid.rc )&
165 fi
166 echo $! > ${tmp}/${repopidfile}.pid
167 ) 2>&1 | sed -e "s@^@${reponame}: @") &
168
169 if [ `expr ${n} '%' ${at_a_time}` -eq 0 ] ; then
170 sleep 2
171 echo Waiting 5 secs before spawning next background command.
172 sleep 3
173 fi
174 done
175 # Wait for all hg commands to complete
176 wait
177
178 # Terminate with exit 0 only if all subprocesses were successful
179 ec=0
180 if [ -d ${tmp} ]; then
181 for rc in ${tmp}/*.pid.rc ; do
182 exit_code=`cat ${rc} | tr -d ' \n\r'`
183 if [ "${exit_code}" != "0" ] ; then
184 echo "WARNING: ${rc} exited abnormally."
185 ec=1
186 fi
187 done
188 fi
189 exit ${ec}

mercurial