Merge pull request #24112 from owncloud/backport-24111-chunking-users-in-background-jobs

[9.0] Chunk the users correctly in the trashbin and versions background job
This commit is contained in:
Thomas Müller 2016-04-21 16:21:12 +02:00
commit ed746a7ec2
3 changed files with 26 additions and 58 deletions

View File

@ -23,6 +23,7 @@
namespace OCA\Files_Trashbin\BackgroundJob; namespace OCA\Files_Trashbin\BackgroundJob;
use OCP\IConfig; use OCP\IConfig;
use OCP\IUser;
use OCP\IUserManager; use OCP\IUserManager;
use OCA\Files_Trashbin\AppInfo\Application; use OCA\Files_Trashbin\AppInfo\Application;
use OCA\Files_Trashbin\Expiration; use OCA\Files_Trashbin\Expiration;
@ -31,40 +32,28 @@ use OCA\Files_Trashbin\Trashbin;
class ExpireTrash extends \OC\BackgroundJob\TimedJob { class ExpireTrash extends \OC\BackgroundJob\TimedJob {
const ITEMS_PER_SESSION = 1000;
/** /**
* @var Expiration * @var Expiration
*/ */
private $expiration; private $expiration;
/**
* @var IConfig
*/
private $config;
/** /**
* @var IUserManager * @var IUserManager
*/ */
private $userManager; private $userManager;
const USERS_PER_SESSION = 1000;
/** /**
* @param IConfig|null $config
* @param IUserManager|null $userManager * @param IUserManager|null $userManager
* @param Expiration|null $expiration * @param Expiration|null $expiration
*/ */
public function __construct(IConfig $config = null, public function __construct(IUserManager $userManager = null,
IUserManager $userManager = null,
Expiration $expiration = null) { Expiration $expiration = null) {
// Run once per 30 minutes // Run once per 30 minutes
$this->setInterval(60 * 30); $this->setInterval(60 * 30);
if (is_null($expiration) || is_null($userManager) || is_null($config)) { if (is_null($expiration) || is_null($userManager)) {
$this->fixDIForJobs(); $this->fixDIForJobs();
} else { } else {
$this->config = $config;
$this->userManager = $userManager; $this->userManager = $userManager;
$this->expiration = $expiration; $this->expiration = $expiration;
} }
@ -72,7 +61,6 @@ class ExpireTrash extends \OC\BackgroundJob\TimedJob {
protected function fixDIForJobs() { protected function fixDIForJobs() {
$application = new Application(); $application = new Application();
$this->config = \OC::$server->getConfig();
$this->userManager = \OC::$server->getUserManager(); $this->userManager = \OC::$server->getUserManager();
$this->expiration = $application->getContainer()->query('Expiration'); $this->expiration = $application->getContainer()->query('Expiration');
} }
@ -87,25 +75,14 @@ class ExpireTrash extends \OC\BackgroundJob\TimedJob {
return; return;
} }
$offset = $this->config->getAppValue('files_trashbin', 'cronjob_user_offset', 0); $this->userManager->callForAllUsers(function(IUser $user) {
$users = $this->userManager->search('', self::USERS_PER_SESSION, $offset);
if (!count($users)) {
// No users found, reset offset and retry
$offset = 0;
$users = $this->userManager->search('', self::USERS_PER_SESSION);
}
$offset += self::USERS_PER_SESSION;
$this->config->setAppValue('files_trashbin', 'cronjob_user_offset', $offset);
foreach ($users as $user) {
$uid = $user->getUID(); $uid = $user->getUID();
if (!$this->setupFS($uid)) { if (!$this->setupFS($uid)) {
continue; return;
} }
$dirContent = Helper::getTrashFiles('/', $uid, 'mtime'); $dirContent = Helper::getTrashFiles('/', $uid, 'mtime');
Trashbin::deleteExpiredFiles($dirContent, $uid); Trashbin::deleteExpiredFiles($dirContent, $uid);
} });
\OC_Util::tearDownFS(); \OC_Util::tearDownFS();
} }
@ -115,20 +92,16 @@ class ExpireTrash extends \OC\BackgroundJob\TimedJob {
* @param string $user * @param string $user
* @return boolean * @return boolean
*/ */
private function setupFS($user){ protected function setupFS($user) {
if (!$this->userManager->userExists($user)) {
return false;
}
//Check if this user has a trashbin directory
$view = new \OC\Files\View('/' . $user);
if (!$view->is_dir('/files_trashbin/files')){
return false;
}
\OC_Util::tearDownFS(); \OC_Util::tearDownFS();
\OC_Util::setupFS($user); \OC_Util::setupFS($user);
// Check if this user has a trashbin directory
$view = new \OC\Files\View('/' . $user);
if (!$view->is_dir('/files_trashbin/files')) {
return false;
}
return true; return true;
} }
} }

View File

@ -26,7 +26,6 @@ use \OCA\Files_Trashbin\BackgroundJob\ExpireTrash;
class ExpireTrash_Test extends \Test\TestCase { class ExpireTrash_Test extends \Test\TestCase {
public function testConstructAndRun() { public function testConstructAndRun() {
$backgroundJob = new ExpireTrash( $backgroundJob = new ExpireTrash(
$this->getMock('OCP\IConfig'),
$this->getMock('OCP\IUserManager'), $this->getMock('OCP\IUserManager'),
$this->getMockBuilder('OCA\Files_Trashbin\Expiration')->disableOriginalConstructor()->getMock() $this->getMockBuilder('OCA\Files_Trashbin\Expiration')->disableOriginalConstructor()->getMock()
); );

View File

@ -21,6 +21,7 @@
namespace OCA\Files_Versions\BackgroundJob; namespace OCA\Files_Versions\BackgroundJob;
use OCP\IUser;
use OCP\IUserManager; use OCP\IUserManager;
use OCA\Files_Versions\AppInfo\Application; use OCA\Files_Versions\AppInfo\Application;
use OCA\Files_Versions\Storage; use OCA\Files_Versions\Storage;
@ -64,20 +65,13 @@ class ExpireVersions extends \OC\BackgroundJob\TimedJob {
return; return;
} }
$users = $this->userManager->search(''); $this->userManager->callForAllUsers(function(IUser $user) {
$isFSready = false;
foreach ($users as $user) {
$uid = $user->getUID(); $uid = $user->getUID();
if (!$isFSready) {
if (!$this->setupFS($uid)) { if (!$this->setupFS($uid)) {
continue; return;
}
$isFSready = true;
} }
Storage::expireOlderThanMaxForUser($uid); Storage::expireOlderThanMaxForUser($uid);
} });
\OC_Util::tearDownFS();
} }
/** /**
@ -85,14 +79,16 @@ class ExpireVersions extends \OC\BackgroundJob\TimedJob {
* @param string $user * @param string $user
* @return boolean * @return boolean
*/ */
private function setupFS($user){ protected function setupFS($user) {
if (!$this->userManager->userExists($user)) {
return false;
}
\OC_Util::tearDownFS(); \OC_Util::tearDownFS();
\OC_Util::setupFS($user); \OC_Util::setupFS($user);
// Check if this user has a versions directory
$view = new \OC\Files\View('/' . $user);
if (!$view->is_dir('/files_versions')) {
return false;
}
return true; return true;
} }
} }