Add check to prevent online guests query before upgrade script is run

NamelessMC · Jan 5, 2020 · 5129bca · 5129bca
1 parent 8cf093e
commit 5129bca
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/modules/Core/widgets/StatsWidget.php b/modules/Core/widgets/StatsWidget.php
@@ -73,9 +73,14 @@ public function initialise(){
 	    }
 
 	    if(!$this->_cache->isCached('online_guests')){
-		    $online_guests = DB::getInstance()->query('SELECT count(*) FROM nl2_online_guests WHERE last_seen > ?', array(strtotime('-5 minutes')))->first();
-		    $online_guests = $online_guests->{'count(*)'};
-		    $this->_cache->store('online_guests', $online_guests, 60);
+	    	try {
+			    $online_guests = DB::getInstance()->query('SELECT count(*) FROM nl2_online_guests WHERE last_seen > ?', array(strtotime('-5 minutes')))->first();
+			    $online_guests = $online_guests->{'count(*)'};
+			    $this->_cache->store('online_guests', $online_guests, 60);
+		    } catch(Exception $e){
+	    		// Upgrade script hasn't been run
+			    $online_guests = 0;
+		    }
 	    } else {
 		    $online_guests = $this->_cache->retrieve('online_guests');
 	    }