60 lines
1.7 KiB
PHP
60 lines
1.7 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\ORM\Tests\DataObjectTest;
|
|
|
|
use SilverStripe\Dev\TestOnly;
|
|
use SilverStripe\ORM\Connect\Query;
|
|
use SilverStripe\ORM\DataObject;
|
|
use SilverStripe\ORM\DB;
|
|
use SilverStripe\ORM\HasManyList;
|
|
use SilverStripe\ORM\Hierarchy\Hierarchy;
|
|
use SilverStripe\ORM\ManyManyList;
|
|
use SilverStripe\ORM\Queries\SQLUpdate;
|
|
|
|
/**
|
|
* The purpose of this test class is to test recursive writes and make sure we don't get stuck in an infinite loop.
|
|
* @property int $WriteCount Number of times this object was written sine the last call of `resetCount`
|
|
*/
|
|
class TreeNode extends DataObject implements TestOnly
|
|
{
|
|
private static $table_name = 'DataObjectTest_TreeNode';
|
|
|
|
private static $db = [
|
|
'Title' => 'Varchar',
|
|
'WriteCount' => 'Int'
|
|
];
|
|
|
|
private static $has_one = [
|
|
'Parent' => self::class,
|
|
'Cycle' => self::class,
|
|
];
|
|
|
|
private static $has_many = [
|
|
'Children' => self::class,
|
|
];
|
|
|
|
public function write($showDebug = false, $forceInsert = false, $forceWrite = false, $writeComponents = false, bool $skipValidation = false)
|
|
{
|
|
// Force the component to fetch its Parent and Cycle relation so we have components to recursively write
|
|
$this->Parent;
|
|
$this->Cycle;
|
|
|
|
// Count a write attempts
|
|
$this->WriteCount++;
|
|
|
|
return parent::write($showDebug, $forceInsert, $forceWrite, $writeComponents, $skipValidation);
|
|
}
|
|
|
|
/**
|
|
* Reset the WriteCount on all TreeNodes
|
|
*/
|
|
public function resetCounts()
|
|
{
|
|
$update = new SQLUpdate(
|
|
sprintf('"%s"', self::baseTable()),
|
|
['"WriteCount"' => 0]
|
|
);
|
|
$results = $update->execute();
|
|
}
|
|
}
|