Пример #1
0
	public static function set_verbose($value) {
		self::$verbose = (bool)$value;
	}
	/**
	 * Handle orphaned rows in intermediate many_many tables.
	 */
	public function handleBrokenManyManyRelations() {
		JanitorDebug::message("handleBrokenManyManyRelations()");
		$dbTables = DataObjectOnDeleteDecorator::get_db_tables();
		foreach ((array)$this->dataObject->many_many() as $relationName => $relationClass) {
			list($parentClass, $componentClass, $parentField, $componentField, $table) = $this->dataObject->many_many($relationName);
			if (!in_array($table, DataObjectOnDeleteDecorator_ManyManyCleaner::get_ignored_tables()) &&
				in_array(strtolower($table), $dbTables)) {
				$oneWayTables = DataObjectOnDeleteDecorator_ManyManyCleaner::get_one_way_tables();
				$oneWayTable = array_key_exists($table, $oneWayTables);
				if ($oneWayTable && $oneWayTables[$table] != $relationName) {
					JanitorDebug::message("Handling one-way table \"{$table}\" from \"{$relationName}\", aborting");
					continue;
				}
				$query = "SELECT \"{$parentField}\" FROM \"{$table}\" WHERE 1=1";
				$relations = DB::query($query)->column($parentField);
				foreach ((array)$relations as $ID) {
					$baseDataObject = DataObject::get_by_id($this->baseClassName, $ID);
					if ($baseDataObject)
						continue;
					$query = "DELETE FROM \"{$table}\" WHERE \"{$parentField}\" = {$ID}";
					JanitorDebug::message("Running query: {$query}", 'p', 'color:#ff0000');
					DB::query($query);
				}
			}
		}
	}
	/**
	 * Perform retroactive DataObject cleaning.
	 * 
	 * @param SS_HTTPRequest $request
	 */
	public function run($request) {
		if (!Object::get_static('SapphireTest', 'is_running_test')) {
			JanitorDebug::set_verbose(true);
		}
		if (JanitorDBP::available()) {
			$this->backupPath = JanitorDBP::backup_database();
		}
		$dataObjectSubClasses = (array)ClassInfo::subclassesFor('DataObject');
		// Remove DataObject
		array_shift($dataObjectSubClasses);
		foreach ($dataObjectSubClasses as $class) {
			$retroactiveCleaner = new DataObjectRetroactiveCleaner($class);
			$retroactiveCleaner->clean();
		}
	}
	public function testWithOrphanedManyManyRelations() {
		DataObjectOnDeleteDecorator::set_disabled(true);
		//JanitorDebug::set_verbose(true);
		
		$page1 = new Page();
		$page1->write();
		$page1->doPublish();
		$page1ID = $page1->ID;
		
		$page2 = new Page();
		$page2->Content .= "<p><a href=\"[sitetree_link id={$page1->ID}]\">page1 link</a></p>";
		$page2->write();
		$page2->doPublish();
		$page2ID = $page2->ID;
		
		$query = "SELECT \"ID\" FROM \"SiteTree_LinkTracking\" WHERE \"SiteTreeID\" = $page2ID AND \"ChildID\" = $page1ID";
		
		$page2->deleteFromStage('Live');
		$page2 = DataObject::get_by_id('Page', $page2ID);
		$page2->delete();
		
		$this->assertTrue((bool)DB::query($query)->value(),
			"many_many table SiteTree_LinkTracking cleaned prematurely (possibly due to SilverStripe core changes)");
		
		DataObjectOnDeleteDecorator::set_disabled(false);
		$task = new DataObjectRetroactiveCleanerTask();
		$task->run(null);
		$task->deleteBackup();
		
		$this->assertFalse((bool)DB::query($query)->value(),
			"many_many table SiteTree_LinkTracking not cleaned properly (retroactively)");
		JanitorDebug::set_verbose(false);
	}