-
Notifications
You must be signed in to change notification settings - Fork 22
/
manageHooks.sh
executable file
·209 lines (209 loc) · 8.35 KB
/
manageHooks.sh
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
#!/bin/bash
#----------------------------------------------------------------
# writeStatus.sh: Writes configuration, ASGS instance status,
# and resource status info to metadata file.
#----------------------------------------------------------------
# Copyright(C) 2021 Jason Fleming
#
# This file is part of the ADCIRC Surge Guidance System (ASGS).
#
# The ASGS is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# ASGS is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with the ASGS. If not, see <http://www.gnu.org/licenses/>.
#----------------------------------------------------------------
#
# Executed upon startup initialization
nullifyHooks()
{
local THIS="asgs_main->manageHooks->nullifyHooks()"
logMessage "$THIS: Nullifying the time values associated with each hook."
for k in ${allHooks[@]} ; do
hooksTimes[$k]="null"
logMessage "$THIS: Setting hooksTimes['$k'] to '${hooksTimes[$k]}'."
done
}
#
# nullify just the nowcast and forecast hook times;
# executed when a new nowcast/forecast cycle starts
nullifyNowcastForecastHooks()
{
local THIS="asgs_main->manageHooks->nullifyNowcastForecastHooks()"
logMessage "$THIS: Nullifying the time values associated with each nowcast and forecast hook."
for k in "${nowcastHooks[@]}" "${forecastHooks[@]}" ; do
hooksTimes["$k"]="null"
logMessage "$THIS: Setting hooksTimes['$k'] to '${hooksTimes[$k]}'."
done
}
#
# set (or append) timestamp to hooksTimes entry
timestampHook()
{
hook=$1
local THIS="asgs_main->manageHooks->timestampHook()"
logMessage "$THIS: Updating timestamp for the '$hook' hook."
dateTime=`date +'%Y-%h-%d-T%H:%M:%S%z'`
local status="null"
local statusURL="null"
case $hook in
"START_INIT")
mypath="null"
;;
"FINISH_INIT")
mypath="status"
;;
"START_SPINUP_STAGE"|"HOT_SPINUP"|"FINISH_SPINUP_STAGE")
mypath="initialize"
;;
"BUILD_SPINUP"|"SUBMIT_SPINUP"|"FINISH_SPINUP_SCENARIO")
mypath="initialize/hindcast"
;;
"START_NOWCAST_STAGE"|"NOWCAST_POLLING")
mypath="status"
;;
"NOWCAST_TRIGGERED"|"BUILD_NOWCAST_SCENARIO"|"SUBMIT_NOWCAST_SCENARIO"|"FINISH_NOWCAST_SCENARIO")
mypath="$ADVISORY/nowcast"
;;
"FINISH_NOWCAST_STAGE"|"START_FORECAST_STAGE")
mypath="$ADVISORY"
;;
"INITIALIZE_FORECAST_SCENARIO"|"CAPACITY_WAIT"|"BUILD_FORECAST_SCENARIO"|"SUBMIT_FORECAST_SCENARIO")
mypath="$ADVISORY/$SCENARIO"
;;
"FINISH_FORECAST_STAGE")
mypath=$ADVISORY
;;
"EXIT_STAGE")
mypath="status"
;;
*)
warn "$THIS: Unrecognized hook '${hooksTimes[$hook]}'."
;;
esac
# TODO: go to that directory and get the statusURL from the status.json file
# and put the statusURL into this status.json file
for file in cycle.status.json scenario.status.json ; do
if [[ $mypath != "null" && -e $RUNDIR/$mypath/$file ]]; then
status=\"${mypath}/${file}\"
# the URL should be in the $SCENARIODIR/status/run.properties file
# as the downloadurl property
if [[ $file == "scenario.status.json" || -e $RUNDIR/$mypath/status/run.properties ]]; then
local downloadurl=$(grep downloadurl $RUNDIR/$mypath/status/run.properties)
if [[ ! -z $downloadurl && $downloadurl != "" ]]; then
statusURL=\"${downloadurl##downloadurl : }/$file\"
fi
fi
fi
done
if [[ $mypath != "null" ]]; then
mypath=\"$mypath\"
fi
json="{ \"time\" : \"$dateTime\", \"path\" : $mypath, \"statusfile\" : $status, \"statusURL\" : $statusURL }"
# determine number of spaces required to get the status objects to line up
longestHookKey=0
for h in ${allHooks[@]} ; do
if [[ ${#h} -gt $longestHookKey ]]; then longestHookKey=${#h} ; fi
done
len=$[ 11 + $longestHookKey ]
if [[ ${hooksTimes[$hook]} == "null" ]]; then
firstlen=$[ $longestHookKey - ${#hook} ]
printf -v spaces "%*s%s" $firstlen " "
if [[ ${#hook} -eq $longestHookKey ]]; then spaces="" ; fi
hooksTimes[$hook]="$spaces$json" # nuke out the null entry
else
printf -v spaces "%*s%s" $len " "
hooksTimes[$hook]+=",\n$spaces$json" # add it to the list
fi
latestHook=$hook # to be written into the status file
}
#
writeHookStatus()
{
local THIS="asgs_main->manageHooks.sh->writeHookStatus()"
local jsonfile="$statusDir/hook.status.json"
logMessage "$THIS: Writing status associated with ASGS hooks to '$jsonfile'."
#
# write the time value(s) associated with each hook; will be null
# if that hook has not been reached for this cycle
echo "{" > $jsonfile # <-<< OVERWRITE
echo \""hpc.hpcenv\" : \"$HPCENV\"," >> $jsonfile
echo \""instancename\" : \"$INSTANCENAME\"," >> $jsonfile
echo \""path.rundir\" : \"$RUNDIR\"," >> $jsonfile
echo \""path.scriptdir\" : \"$SCRIPTDIR\"," >> $jsonfile
if [[ $LASTSUBDIR != "null" ]]; then
echo \""path.lastsubdir\" : \"$LASTSUBDIR\"," >> $jsonfile
else
echo \""path.lastsubdir\" : null," >> $jsonfile
fi
echo \""monitoring.logging.file.syslog\" : \"$SYSLOG\"," >> $jsonfile
echo \""config.file\" : \"$CONFIG\"," >> $jsonfile
echo "\"monitoring.hook\" : {" >> $jsonfile
for k in ${allHooks[@]} ; do
comma="," ; if [[ $k == "EXIT_STAGE" ]]; then comma="" ; fi
if [[ ${hooksTimes[$k]} != "null" ]]; then
echo -n " \"$k\" : [ " >> $jsonfile
echo -e "${hooksTimes[$k]} ]$comma" >> $jsonfile
else
json=" \"$k\" : [ { \"time\" : null, \"path\" : null, \"statusfile\" : null, \"statusURL\" : null } ]$comma"
echo "$json" >> $jsonfile
fi
done
echo "}," >> $jsonfile
# update time stamp
dateTime=`date +'%Y-%h-%d-T%H:%M:%S%z'`
echo "\"asgs.instance.status.file\" : \"asgs.instance.status.json\"," >> $jsonfile
echo "\"asgs.instance.status.url\" : \"$asgsInstanceStatusURL\"," >> $jsonfile
echo "\"time.hook.status.lastupdated\" : \"$dateTime\"," >> $jsonfile
echo "\"hook.status.url\" : \"$hookStatusURL\"," >> $jsonfile
echo "\"hook.status.file.previous\" : \"$previousHookStatusFile\"," >> $jsonfile
echo "\"hook.status.url.previous\" : \"$previousHookStatusURL\"," >> $jsonfile
echo "\"hook.status.latest\" : \"$latestHook\"" >> $jsonfile
echo "}" >> $jsonfile
# redact the username for security
sed --in-place "s/$USER/\$USER/g" $jsonfile
}
#
# execute hook scripts
executeHookScripts()
{
hook=$1
local THIS="asgs_main->manageHooks->executeHookScripts()"
timestampHook $hook
if [[ $hook != "START_INIT" ]]; then
logMessage "$THIS: Executing scripts for the $hook hook."
# write status immediately, in case one of the hook scripts
# wants to post the status file somewhere
# (can't write status file immediately for the START_INIT script
# b/c RUNDIR not established yet)
if [[ $hook == "FINISH_INIT" ]]; then
notifyNow=$enableStatusNotify
else
notifyNow="no"
fi
writeASGSInstanceStatus
writeHookStatus
if [[ $enablePostStatus == "yes" ]]; then
postStatus
fi
fi
for hs in ${hooksScripts[$hook]} ; do
if [[ $hook != "START_INIT" ]]; then
logMessage "$THIS: Executing $hook hook script $SCRIPTDIR/$hs."
fi
$SCRIPTDIR/$hs >> ${SYSLOG} 2>&1
done
if [[ $hook == "FINISH_FORECAST_STAGE" ]]; then
previousHookStatusFile=${ADVISORY}.hook.status.json
previousHookStatusURL=$hookStatusURL
mv $RUNDIR/status/hook.status.json $RUNDIR/status/$previousHookStatusFile 2>> $SYSLOG
nullifyNowcastForecastHooks # clears out the timestamps and statuses of these hooks
fi
}