source: trunk/cluster_status.php@ 35

Last change on this file since 35 was 35, checked in by gegorbet, 7 years ago

mods mostly for use of mysqli
File size: 11.8 KB
Line 
1<?php
2
3$us3bin = exec( "ls -d ~us3/lims/bin" );
4include "$us3bin/listen-config.php";
5
6if ( ! preg_match( "/_local/", $class_dir ) )
7{
8 $xml = get_data();
9
10 if ( $xml != "" )
11 parse( $xml );
12}
13
14$data = array();
15
16local_status();
17
18foreach ( $data as $item )
19{
20 if ( ! preg_match( "/error/", $item[ 'running' ] ) )
21 update( $item[ 'cluster' ], $item[ 'queued' ], $item[ 'status' ], $item[ 'running' ] );
22}
23
24//exit();
25exit(0);
26
27// Get the cluster status
28
29function get_data()
30{
31 global $self;
32 $url = "http://community.ucs.indiana.edu:19444/orps-service/XML/gateway/ultrascan";
33
34 try
35 {
36 $post = new HttpRequest( $url, HttpRequest::METH_GET );
37 $http = $post->send();
38 $xml = $post->getResponseBody();
39 }
40 catch ( HttpException $e )
41 {
42// write_log( "$self: Cluster Status not available" );
43 return "";
44 }
45
46 return $xml;
47}
48
49// Parse the xml
50
51function parse( $xml )
52{
53 global $data;
54
55 $data = array();
56
57 $x = new XML_Array( $xml );
58 $d = $x->ReturnArray();
59
60 if ( ! isset( $d[ 'summaries' ] ) ) exit(); // Bad input
61
62 foreach ( $d[ 'summaries' ] as $item )
63 {
64 $a = Array();
65
66
67 $a[ 'queued' ] = $item[ 'waitingJobs' ];
68 $a[ 'running' ] = $item[ 'runningJobs' ];
69
70 if ( $a[ 'queued' ] == "" || $a[ 'queued' ] < 0 ) $a[ 'queued' ] = 0;
71 if ( $a[ 'running' ] == "" || $a[ 'running' ] < 0 ) $a[ 'running' ] = 0;
72
73 $clusterParts = explode( ".", $item[ 'resourceId' ] );
74 $cluster = preg_replace( '/\d+$/', "", $clusterParts[ 0 ] );
75
76 if ( $cluster == 'uthscsa-bcf' ) $cluster = 'bcf';
77 if ( $cluster == 'uthscsa-alamo' ) $cluster = 'alamo';
78
79 $a[ 'cluster' ] = $cluster;
80
81 switch ( $item[ 'resourceStatus' ] )
82 {
83 case 'UP' :
84 $status = "up";
85 break;
86
87 case 'DOWN' :
88 $status = "down";
89 break;
90
91 case 'WARN' :
92 $status = "warn";
93 break;
94
95 case 'FAILED' :
96 default :
97 $status = "unknown";
98 break;
99 }
100
101 $a[ 'status' ] = $status;
102
103 $data[] = $a;
104 }
105}
106
107// Put it in the DB
108
109function update( $cluster, $queued, $status, $running )
110{
111 global $dbhost;
112 global $guser;
113 global $gpasswd;
114 global $gDB;
115 global $self;
116//echo " $cluster $queued, $status, $running\n";
117
118 $gfac_link = mysqli_connect( $dbhost, $guser, $gpasswd, $gDB );
119
120 if ( ! $gfac_link )
121 {
122 write_log( "$self: Could not connect to DB $gDB" );
123 echo "Could not connect to DB $gDB.\n";
124 exit();
125 }
126
127 $query = "SELECT * FROM cluster_status WHERE cluster='$cluster'";
128 $result = mysqli_query( $gfac_link, $query );
129
130 if ( ! $result )
131 {
132 write_log( "$self: Query failed $query - " . mysqli_error( $gfac_link ) );
133 echo "$self: Query failed $query - " . mysqli_error( $gfac_link ) . "\n";
134 exit();
135 }
136
137 $rows = mysqli_num_rows( $result );
138
139 if ( $rows == 0 ) // INSERT
140 {
141 $query = "INSERT INTO cluster_status SET " .
142 "cluster='$cluster', " .
143 "queued=$queued, " .
144 "running=$running, " .
145 "status='$status'";
146 }
147 else // UPDATE
148 {
149 $query = "UPDATE cluster_status SET " .
150 "queued=$queued, " .
151 "running=$running, " .
152 "status='$status' " .
153 "WHERE cluster='$cluster'";
154 }
155
156 $result = mysqli_query( $gfac_link, $query );
157
158 if ( ! $result )
159 {
160 write_log( "$self: Query failed $query - " . mysqli_error( $gfac_link ) );
161 echo "$self: Query failed $query - " . mysqli_error( $gfac_link ) . "\n";
162 }
163}
164
165// Get local cluster status
166
167function local_status()
168{
169 global $self;
170 global $data;
171 global $dbhost;
172 global $org_domain;
173 global $class_dir;
174
175 if ( preg_match( "/_local/", $class_dir ) )
176 {
177 if ( preg_match( "/attlocal/", $org_domain ) )
178 $clusters = array( "us3iab-devel", "alamo-local" );
179 else
180 $clusters = array( "us3iab-node0" );
181 }
182 else
183 {
184// $clusters = array( "alamo", "lonestar5", "comet",
185// "stampede2", "jetstream", "jureca", "jacinto" );
186 $clusters = array( "alamo", "lonestar5", "comet",
187 "stampede2", "jetstream" );
188 }
189
190 foreach ( $clusters as $clname )
191 {
192 $a = Array();
193//echo "$self: clname=$clname\n";
194
195 switch( $clname )
196 {
197 case 'us3iab-node0':
198 case 'us3iab-node1':
199 case 'us3iab-devel':
200 {
201 $qstat = `/usr/bin/qstat -B 2>&1|tail -1`;
202
203 $sparts = preg_split( '/\s+/', $qstat );
204 $que = $sparts[ 3 ];
205 $run = $sparts[ 4 ];
206 $sta = $sparts[ 10 ];
207 if ( $sta == "Active" )
208 $sta = "up";
209 else
210 $sta = "down";
211 break;
212 }
213 case 'alamo-local':
214 case 'alamo':
215 {
216 $host = "us3@alamo.uthscsa.edu";
217 $qstat = `ssh $host '/usr/bin/qstat -B 2>&1|tail -1'`;
218 $sparts = preg_split( '/\s+/', $qstat );
219 $que = $sparts[ 3 ];
220 $run = $sparts[ 4 ];
221 $sta = $sparts[ 10 ];
222 if ( $sta == "Active" )
223 $sta = "up";
224 else
225 $sta = "down";
226 break;
227 }
228 case 'jacinto':
229 {
230 $host = "us3@jacinto.uthscsa.edu";
231 $qstat = `ssh $host '/opt/torque/bin/qstat -B 2>&1|tail -1'`;
232 $sparts = preg_split( '/\s+/', $qstat );
233 $que = $sparts[ 3 ];
234 $run = $sparts[ 4 ];
235 $sta = $sparts[ 9 ];
236 if ( $sta == "Active" )
237 $sta = "up";
238 else
239 $sta = "down";
240 break;
241 }
242 case 'stampede2':
243 {
244 $host = "us3@stampede2.tacc.utexas.edu";
245 $qstat = `ssh $host '/usr/local/bin/showq 2>/dev/null|grep "Total Jobs"'`;
246 $sparts = preg_split( '/\s+/', $qstat );
247 $tot = $sparts[ 2 ];
248 $run = $sparts[ 5 ];
249 $que = $sparts[ 8 ];
250 $sta = "up";
251 if ( $tot == '' || $tot == '0' )
252 $sta = "down";
253 break;
254 }
255 case 'lonestar5':
256 {
257 $host = "us3@ls5.tacc.utexas.edu";
258 $qstat = `ssh $host '/usr/local/bin/showq 2>/dev/null|grep "Total Jobs"'`;
259 $sparts = preg_split( '/\s+/', $qstat );
260 $tot = $sparts[ 2 ];
261 $run = '0';
262 $que = '0';
263 $sta = "up";
264 if ( $tot == '' || $tot == '0' )
265 {
266 $sta = "down";
267 }
268 else
269 {
270 $run = $sparts[ 5 ];
271 $que = $sparts[ 8 ];
272// $que = $sparts[ 11 ];
273 }
274 break;
275 }
276 case 'comet':
277 {
278 $host = "us3@comet.sdsc.edu";
279 //$qstat = `ssh $host '/usr/bin/sinfo -s -p compute -o "%a %F" |tail -1'`;
280 $qstat = `ssh $host '/home/us3/scripts/cstat 2>&1'`;
281 $sparts = preg_split( '/\s+/', $qstat );
282 $tot = $sparts[ 1 ];
283 $run = '0';
284 $que = '0';
285 $sta = "up";
286 if ( $tot == '' || $tot == '0' )
287 {
288 $sta = "down";
289 }
290 else
291 {
292 $run = $sparts[ 3 ];
293 $que = $sparts[ 5 ];
294 }
295 break;
296 }
297 case 'jureca':
298 {
299 $host = "swus1@jureca.fz-juelich.de";
300 $qstat = `ssh $host '~swus1/scripts/qstat-jureca 2>&1'`;
301 $sparts = preg_split( '/\s+/', $qstat );
302 $sta = $sparts[ 0 ];
303 $run = $sparts[ 1 ];
304 $que = $sparts[ 2 ];
305 break;
306 }
307 case 'jetstream-local':
308 case 'jetstream':
309 {
310 $host = "us3@js-169-137.jetstream-cloud.org";
311// $qstat = `ssh $host '/usr/bin/sinfo -s -p batch -o "%a %F" |tail -1'`;
312 $qstat = `ssh $host '/home/us3/bin/clusstat |tail -n 1'`;
313 $sparts = preg_split( '/\s+/', $qstat );
314 $sta = $sparts[ 0 ];
315 $knts = $sparts[ 1 ];
316 $sparts = preg_split( '/\//', $knts );
317 $run = $sparts[ 0 ];
318 $que = $sparts[ 2 ];
319 $tot = $sparts[ 3 ];
320 if ( $sta == "" )
321 $sta = "down";
322 break;
323 }
324 }
325
326 if ( $sta == "" )
327 $sta = "down";
328
329 if ( $sta == "down" )
330 {
331 $que = "0";
332 $run = "0";
333 }
334
335 $a[ 'cluster' ] = $clname;
336 $a[ 'queued' ] = $que;
337 $a[ 'running' ] = $run;
338 $a[ 'status' ] = $sta;
339echo "$self: $clname $que $run $sta\n";
340
341 $data[] = $a;
342
343 if ( $clname == 'alamo' ||
344 $clname == 'jacinto' ||
345 $clname == 'jetstream' )
346 {
347 $a[ 'cluster' ] = $clname . "-local";
348 $data[] = $a;
349 }
350 }
351}
352
353class XML_Array
354{
355 var $_data = Array();
356 var $_name = Array();
357 var $_rep = Array();
358 var $_parser = 0;
359 var $_level = 0;
360 var $_index = 0;
361
362 function XML_Array( &$data )
363 {
364 $this->_parser = xml_parser_create();
365
366 xml_set_object ( $this->_parser, $this );
367 xml_parser_set_option ( $this->_parser, XML_OPTION_CASE_FOLDING, false );
368 xml_set_element_handler ( $this->_parser, "_startElement", "_endElement" );
369 xml_set_character_data_handler( $this->_parser, "_cdata" );
370
371 $this->_data = array();
372 $this->_level = 0;
373
374 if ( ! xml_parse( $this->_parser, $data, true ) )
375 return false;
376
377 xml_parser_free( $this->_parser );
378 }
379
380 function & ReturnArray()
381 {
382 return $this->_data[ 0 ];
383 }
384
385 function _startElement( $parser, $name, $attrs )
386 {
387 if ( $name == "resourceHealth" )
388 {
389## $name .= $this->_index;
390 $this->_index++;
391 }
392
393 if ( ! isset( $this->_rep[ $name ] ) ) $this->_rep[ $name ] = 0;
394
395 $this->_addElement( $name, $this->_data[ $this->_level ], $attrs );
396 $this->_name[ $this->_level ] = $name;
397 $this->_level++;
398 }
399
400 function _endElement( $parser, $name )
401 {
402 if ( isset( $this->_data[ $this->_level ] ) )
403 {
404 $this->_addElement( $this->_name[ $this->_level - 1 ],
405 $this->_data[ $this->_level - 1 ],
406 $this->_data[ $this->_level ]
407 );
408 }
409
410 unset( $this->_data[ $this->_level ] );
411 $this->_level--;
412 $this->_rep[ $name ]++;
413 }
414
415 function _cdata( $parser, $data )
416 {
417 if ( $this->_name[ $this->_level - 1 ] )
418 {
419 $this->_addElement( $this->_name[ $this->_level - 1 ],
420 $this->_data[ $this->_level - 1 ],
421 str_replace( array( "&gt;", "&lt;","&quot;", "&amp;" ),
422 array( ">" , "<" , '"' , "&" ),
423 $data
424 )
425 );
426 }
427 }
428
429 function _addElement( &$name, &$start, $add = array() )
430 {
431 if ( ( sizeof( $add ) == 0 && is_array( $add ) ) || ! $add )
432 {
433 if ( ! isset( $start[ $name ] ) ) $start[ $name ] = '';
434 $add = '';
435 }
436
437 $update = &$start[ $name ];
438
439 if ( is_array( $add) &&
440 is_array( $update ) ) $update += $add;
441 elseif ( is_array( $update ) ) return;
442 elseif ( is_array( $add ) ) $update = $add;
443 elseif ( $add ) $update .= $add;
444 }
445}
446?>
Note: See TracBrowser for help on using the repository browser.