silverstripe / silverstripe-environmentcheck

code/checks/DatabaseCheck.php

<?php

/**
 * Check that the connection to the database is working, by ensuring that the table exists and that
 * the table contains some records.
 */
class DatabaseCheck implements EnvironmentCheck {

As per PSR2, the opening brace for this class should be on a new line.

	protected $checkTable;

	/**
	 * By default, Member will be checked.
	 *
	 * @param string $checkTable
	 */
	function __construct($checkTable = "Member") {
		$this->checkTable = $checkTable;
	}

	/**
	 * @inheritdoc
	 *
	 * @return array
	 */
	function check() {
		if(!DB::getConn()->hasTable($this->checkTable)) {
			return array(EnvironmentCheck::ERROR, "$this->checkTable not present in the database");
		}

		$count = DB::query("SELECT COUNT(*) FROM \"$this->checkTable\"")->value();
		
		if($count > 0) {
			return array(EnvironmentCheck::OK, "");
		} else {
			return array(EnvironmentCheck::WARNING, "$this->checkTable queried ok but has no records");
		}
	}
}