Ask the schema whether the table and column exist

Signed-off-by: Joas Schilling <coding@schilljs.com>
This commit is contained in:
Joas Schilling 2017-09-08 11:37:22 +02:00
parent 194f880073
commit 29e1aa57e1
No known key found for this signature in database
GPG Key ID: E166FD8976B3BAC8
1 changed files with 6 additions and 13 deletions

View File

@ -23,8 +23,6 @@
namespace OC\Repair\Owncloud; namespace OC\Repair\Owncloud;
use Doctrine\DBAL\Exception\InvalidFieldNameException;
use Doctrine\DBAL\Exception\TableNotFoundException;
use OCP\DB\QueryBuilder\IQueryBuilder; use OCP\DB\QueryBuilder\IQueryBuilder;
use OCP\IConfig; use OCP\IConfig;
use OCP\IDBConnection; use OCP\IDBConnection;
@ -86,20 +84,15 @@ class SaveAccountsTableData implements IRepairStep {
* @return bool * @return bool
*/ */
protected function shouldRun() { protected function shouldRun() {
$query = $this->db->getQueryBuilder(); $schema = $this->db->createSchema();
$query->select('*')
->from('accounts')
->where($query->expr()->isNotNull('user_id'))
->setMaxResults(1);
try { $tableName = $this->config->getSystemValue('dbtableprefix', 'oc_') . 'accounts';
$query->execute(); if (!$schema->hasTable($tableName)) {
return true;
} catch (InvalidFieldNameException $e) {
return false;
} catch (TableNotFoundException $e) {
return false; return false;
} }
$table = $schema->getTable($tableName);
return $table->hasColumn('user_id');
} }
/** /**