silverstripe-environmentcheck/src/Checks/DatabaseCheck.php

48 lines
1.1 KiB
PHP

<?php
namespace SilverStripe\EnvironmentCheck\Checks;
use SilverStripe\EnvironmentCheck\EnvironmentCheck;
use SilverStripe\ORM\DB;
/**
* Check that the connection to the database is working, by ensuring that the table exists and that
* the table contains some records.
*
* @package environmentcheck
*/
class DatabaseCheck implements EnvironmentCheck
{
protected $checkTable;
/**
* By default, Member will be checked.
*
* @param string $checkTable
*/
public function __construct($checkTable = 'Member')
{
$this->checkTable = $checkTable;
}
/**
* {@inheritDoc}
*
* @return array
*/
public function check()
{
if (!DB::get_schema()->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, '');
}
return array(EnvironmentCheck::WARNING, "$this->checkTable queried ok but has no records");
}
}