public function execute()
 {
     $repo = RepoGroup::singleton()->getLocalRepo();
     $dbr = $repo->getSlaveDb();
     $this->output("Getting list of files to clean up...\n");
     $res = $dbr->select('uploadstash', 'us_key', 'us_timestamp < ' . $dbr->timestamp(time() - UploadStash::REPO_AGE * 3600), __METHOD__);
     if (!is_object($res) || $res->numRows() == 0) {
         // nothing to do.
         return false;
     }
     // finish the read before starting writes.
     $keys = array();
     foreach ($res as $row) {
         array_push($keys, $row->us_key);
     }
     $this->output('Removing ' . count($keys) . " file(s)...\n");
     // this could be done some other, more direct/efficient way, but using
     // UploadStash's own methods means it's less likely to fall accidentally
     // out-of-date someday
     $stash = new UploadStash($repo);
     foreach ($keys as $key) {
         $stash->getFile($key, true);
         $stash->removeFileNoAuth($key);
     }
 }
Example #2
0
 public function execute()
 {
     $repo = RepoGroup::singleton()->getLocalRepo();
     $dbr = $repo->getSlaveDb();
     // how far back should this look for files to delete?
     global $wgUploadStashMaxAge;
     $this->output("Getting list of files to clean up...\n");
     $res = $dbr->select('uploadstash', 'us_key', 'us_timestamp < ' . $dbr->addQuotes($dbr->timestamp(time() - $wgUploadStashMaxAge)), __METHOD__);
     if (!is_object($res) || $res->numRows() == 0) {
         $this->output("No files to cleanup!\n");
         // nothing to do.
         return;
     }
     // finish the read before starting writes.
     $keys = array();
     foreach ($res as $row) {
         array_push($keys, $row->us_key);
     }
     $this->output('Removing ' . count($keys) . " file(s)...\n");
     // this could be done some other, more direct/efficient way, but using
     // UploadStash's own methods means it's less likely to fall accidentally
     // out-of-date someday
     $stash = new UploadStash($repo);
     foreach ($keys as $key) {
         try {
             $stash->getFile($key, true);
             $stash->removeFileNoAuth($key);
         } catch (UploadStashBadPathException $ex) {
             $this->output("Failed removing stashed upload with key: {$key}\n");
         }
     }
 }
 public function execute()
 {
     global $wgUploadStashMaxAge;
     $repo = RepoGroup::singleton()->getLocalRepo();
     $dbr = $repo->getSlaveDb();
     // how far back should this look for files to delete?
     $cutoff = time() - $wgUploadStashMaxAge;
     $this->output("Getting list of files to clean up...\n");
     $res = $dbr->select('uploadstash', 'us_key', 'us_timestamp < ' . $dbr->addQuotes($dbr->timestamp($cutoff)), __METHOD__);
     if (!is_object($res) || $res->numRows() == 0) {
         $this->output("No files to cleanup!\n");
         // nothing to do.
         return;
     }
     // finish the read before starting writes.
     $keys = array();
     foreach ($res as $row) {
         array_push($keys, $row->us_key);
     }
     $this->output('Removing ' . count($keys) . " file(s)...\n");
     // this could be done some other, more direct/efficient way, but using
     // UploadStash's own methods means it's less likely to fall accidentally
     // out-of-date someday
     $stash = new UploadStash($repo);
     $i = 0;
     foreach ($keys as $key) {
         $i++;
         try {
             $stash->getFile($key, true);
             $stash->removeFileNoAuth($key);
         } catch (UploadStashBadPathException $ex) {
             $this->output("Failed removing stashed upload with key: {$key}\n");
         } catch (UploadStashZeroLengthFileException $ex) {
             $this->output("Failed removing stashed upload with key: {$key}\n");
         }
         if ($i % 100 == 0) {
             $this->output("{$i}\n");
         }
     }
     $this->output("{$i} done\n");
     $tempRepo = $repo->getTempRepo();
     $dir = $tempRepo->getZonePath('thumb');
     $iterator = $tempRepo->getBackend()->getFileList(array('dir' => $dir));
     $this->output("Deleting old thumbnails...\n");
     $i = 0;
     foreach ($iterator as $file) {
         $i++;
         if (wfTimestamp(TS_UNIX, $tempRepo->getFileTimestamp("{$dir}/{$file}")) < $cutoff) {
             $tempRepo->quickPurge("{$dir}/{$file}");
         }
         if ($i % 100 == 0) {
             $this->output("{$i}\n");
         }
     }
     $this->output("{$i} done\n");
 }
 public function testBug29408()
 {
     $this->setMwGlobals('wgUser', self::$users['uploader']->user);
     $repo = RepoGroup::singleton()->getLocalRepo();
     $stash = new UploadStash($repo);
     // Throws exception caught by PHPUnit on failure
     $file = $stash->stashFile($this->bug29408File);
     // We'll never reach this point if we hit bug 29408
     $this->assertTrue(true, 'Unrecognized file without extension');
     $stash->removeFile($file->getFileKey());
 }
 public function execute()
 {
     global $wgUploadStashMaxAge;
     $repo = RepoGroup::singleton()->getLocalRepo();
     $tempRepo = $repo->getTempRepo();
     $dbr = $repo->getSlaveDb();
     // how far back should this look for files to delete?
     $cutoff = time() - $wgUploadStashMaxAge;
     $this->output("Getting list of files to clean up...\n");
     $res = $dbr->select('uploadstash', 'us_key', 'us_timestamp < ' . $dbr->addQuotes($dbr->timestamp($cutoff)), __METHOD__);
     // Delete all registered stash files...
     if ($res->numRows() == 0) {
         $this->output("No stashed files to cleanup according to the DB.\n");
     } else {
         // finish the read before starting writes.
         $keys = array();
         foreach ($res as $row) {
             array_push($keys, $row->us_key);
         }
         $this->output('Removing ' . count($keys) . " file(s)...\n");
         // this could be done some other, more direct/efficient way, but using
         // UploadStash's own methods means it's less likely to fall accidentally
         // out-of-date someday
         $stash = new UploadStash($repo);
         $i = 0;
         foreach ($keys as $key) {
             $i++;
             try {
                 $stash->getFile($key, true);
                 $stash->removeFileNoAuth($key);
             } catch (UploadStashException $ex) {
                 $type = get_class($ex);
                 $this->output("Failed removing stashed upload with key: {$key} ({$type})\n");
             }
             if ($i % 100 == 0) {
                 wfWaitForSlaves();
                 $this->output("{$i}\n");
             }
         }
         $this->output("{$i} done\n");
     }
     // Delete all the corresponding thumbnails...
     $dir = $tempRepo->getZonePath('thumb');
     $iterator = $tempRepo->getBackend()->getFileList(array('dir' => $dir, 'adviseStat' => 1));
     $this->output("Deleting old thumbnails...\n");
     $i = 0;
     $batch = array();
     // operation batch
     foreach ($iterator as $file) {
         if (wfTimestamp(TS_UNIX, $tempRepo->getFileTimestamp("{$dir}/{$file}")) < $cutoff) {
             $batch[] = array('op' => 'delete', 'src' => "{$dir}/{$file}");
             if (count($batch) >= $this->mBatchSize) {
                 $this->doOperations($tempRepo, $batch);
                 $i += count($batch);
                 $batch = array();
                 $this->output("{$i}\n");
             }
         }
     }
     if (count($batch)) {
         $this->doOperations($tempRepo, $batch);
         $i += count($batch);
     }
     $this->output("{$i} done\n");
     // Apparently lots of stash files are not registered in the DB...
     $dir = $tempRepo->getZonePath('public');
     $iterator = $tempRepo->getBackend()->getFileList(array('dir' => $dir, 'adviseStat' => 1));
     $this->output("Deleting orphaned temp files...\n");
     if (strpos($dir, '/local-temp') === false) {
         // sanity check
         $this->error("Temp repo is not using the temp container.", 1);
         // die
     }
     $i = 0;
     $batch = array();
     // operation batch
     foreach ($iterator as $file) {
         if (wfTimestamp(TS_UNIX, $tempRepo->getFileTimestamp("{$dir}/{$file}")) < $cutoff) {
             $batch[] = array('op' => 'delete', 'src' => "{$dir}/{$file}");
             if (count($batch) >= $this->mBatchSize) {
                 $this->doOperations($tempRepo, $batch);
                 $i += count($batch);
                 $batch = array();
                 $this->output("{$i}\n");
             }
         }
     }
     if (count($batch)) {
         $this->doOperations($tempRepo, $batch);
         $i += count($batch);
     }
     $this->output("{$i} done\n");
 }
Example #6
0
 /**
  * Static callback for the HTMLForm in showUploads, to process 
  * Note the stash has to be recreated since this is being called in a static context.
  * This works, because there really is only one stash per logged-in user, despite appearances.
  *
  * @return Status
  */
 public static function tryClearStashedUploads($formData)
 {
     wfDebug(__METHOD__ . " form data : " . print_r($formData, 1));
     if (isset($formData['clear']) and $formData['clear']) {
         $stash = new UploadStash();
         wfDebug("stash has: " . print_r($stash->listFiles(), 1));
         if (!$stash->clear()) {
             return Status::newFatal('uploadstash-errclear');
         }
     }
     return Status::newGood();
 }
	public function execute() {
		global $wgUploadStashMaxAge;

		$repo = RepoGroup::singleton()->getLocalRepo();
		$tempRepo = $repo->getTempRepo();

		$dbr = $repo->getSlaveDb();

		// how far back should this look for files to delete?
		$cutoff = time() - $wgUploadStashMaxAge;

		$this->output( "Getting list of files to clean up...\n" );
		$res = $dbr->select(
			'uploadstash',
			'us_key',
			'us_timestamp < ' . $dbr->addQuotes( $dbr->timestamp( $cutoff ) ),
			__METHOD__
		);

		// Delete all registered stash files...
		if ( $res->numRows() == 0 ) {
			$this->output( "No stashed files to cleanup according to the DB.\n" );
		} else {
			// finish the read before starting writes.
			$keys = array();
			foreach ( $res as $row ) {
				array_push( $keys, $row->us_key );
			}

			$this->output( 'Removing ' . count( $keys ) . " file(s)...\n" );
			// this could be done some other, more direct/efficient way, but using
			// UploadStash's own methods means it's less likely to fall accidentally
			// out-of-date someday
			$stash = new UploadStash( $repo );

			$i = 0;
			foreach ( $keys as $key ) {
				$i++;
				try {
					$stash->getFile( $key, true );
					$stash->removeFileNoAuth( $key );
				} catch ( UploadStashBadPathException $ex ) {
					$this->output( "Failed removing stashed upload with key: $key\n" );
				} catch ( UploadStashZeroLengthFileException $ex ) {
					$this->output( "Failed removing stashed upload with key: $key\n" );
				}
				if ( $i % 100 == 0 ) {
					$this->output( "$i\n" );
				}
			}
			$this->output( "$i done\n" );
		}

		// Delete all the corresponding thumbnails...
		$dir = $tempRepo->getZonePath( 'thumb' );
		$iterator = $tempRepo->getBackend()->getFileList( array( 'dir' => $dir ) );
		$this->output( "Deleting old thumbnails...\n" );
		$i = 0;
		foreach ( $iterator as $file ) {
			if ( wfTimestamp( TS_UNIX, $tempRepo->getFileTimestamp( "$dir/$file" ) ) < $cutoff ) {
				$status = $tempRepo->quickPurge( "$dir/$file" );
				if ( !$status->isOK() ) {
					$this->error( print_r( $status->getErrorsArray(), true ) );
				}
				if ( ( ++$i % 100 ) == 0 ) {
					$this->output( "$i\n" );
				}
			}
		}
		$this->output( "$i done\n" );

		// Apparently lots of stash files are not registered in the DB...
		$dir = $tempRepo->getZonePath( 'public' );
		$iterator = $tempRepo->getBackend()->getFileList( array( 'dir' => $dir, 'adviseStat' => 1 ) );
		$this->output( "Deleting orphaned temp files...\n" );
		if ( strpos( $dir, '/local-temp' ) === false ) { // sanity check
			$this->error( "Temp repo is not using the temp container.", 1 ); // die
		}
		$i = 0;
		foreach ( $iterator as $file ) {
			// Absolute sanity check for stashed files and file segments
			if ( !preg_match( '#(^\d{14}!|\.\d+\.\w+\.\d+$)#', basename( $file ) ) ) {
				$this->output( "Skipped non-stash $file\n" );
				continue;
			}
			if ( wfTimestamp( TS_UNIX, $tempRepo->getFileTimestamp( "$dir/$file" ) ) < $cutoff ) {
				$status = $tempRepo->quickPurge( "$dir/$file" );
				if ( !$status->isOK() ) {
					$this->error( print_r( $status->getErrorsArray(), true ) );
				}
				if ( ( ++$i % 100 ) == 0 ) {
					$this->output( "$i\n" );
				}
			}
		}
		$this->output( "$i done\n" );
	}