diff --git a/core/lib/Drupal/Core/FileUsage/DatabaseFileUsageBackend.php b/core/lib/Drupal/Core/FileUsage/DatabaseFileUsageBackend.php index 2a2b7c2..24662ac 100644 --- a/core/lib/Drupal/Core/FileUsage/DatabaseFileUsageBackend.php +++ b/core/lib/Drupal/Core/FileUsage/DatabaseFileUsageBackend.php @@ -17,23 +17,39 @@ class DatabaseFileUsageBackend extends FileUsageBase { /** + * The database connection used to store file usage information. + * + * @var Drupal\Core\Database\Connection + */ + protected $connection; + + /** + * The name of the SQL table used to store file usage information. + * + * @var string + */ + protected $tableName; + + /** * Construct the DatabaseFileUsageBackend. * * @param Drupal\Core\Database\Connection $connection - * The database connection which will be used to store the route + * The database connection which will be used to store the file usage * information. * @param string $table - * (optional) The table to store the route info in. Defaults to 'router'. + * (optional) The table to store file usage info. Defaults to 'file_usage'. */ - public function __construct(Connection $connection) { + public function __construct(Connection $connection, $table = 'file_usage') { $this->connection = $connection; + + $this->tableName = $table; } /** * Implements Drupal\Core\FileUsage\FileUsageInterface::add(). */ public function add(File $file, $module, $type, $id, $count = 1) { - $this->connection->merge('file_usage') + $this->connection->merge($this->tableName) ->key(array( 'fid' => $file->fid, 'module' => $module, @@ -52,7 +68,7 @@ public function add(File $file, $module, $type, $id, $count = 1) { */ public function delete(File $file, $module, $type = NULL, $id = NULL, $count = 1) { // Delete rows that have a exact or less value to prevent empty rows. - $query = $this->connection->delete('file_usage') + $query = $this->connection->delete($this->tableName) ->condition('module', $module) ->condition('fid', $file->fid); if ($type && $id) { @@ -67,7 +83,7 @@ public function delete(File $file, $module, $type = NULL, $id = NULL, $count = 1 // If the row has more than the specified count decrement it by that number. if (!$result && $count > 0) { - $query = $this->connection->update('file_usage') + $query = $this->connection->update($this->tableName) ->condition('module', $module) ->condition('fid', $file->fid); if ($type && $id) { @@ -86,7 +102,7 @@ public function delete(File $file, $module, $type = NULL, $id = NULL, $count = 1 * Implements Drupal\Core\FileUsage\FileUsageInterface::listUsage(). */ public function listUsage(File $file) { - $result = $this->connection->select('file_usage', 'f') + $result = $this->connection->select($this->tableName, 'f') ->fields('f', array('module', 'type', 'id', 'count')) ->condition('fid', $file->fid) ->condition('count', 0, '>')