danog/AsyncOrm
1
0
Fork 0
mirror of https://github.com/danog/AsyncOrm.git synced 2024-11-26 12:24:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

Refactor ObjectArray

Browse Source
This commit is contained in:
Daniil Gentili 2024-03-31 22:02:36 +02:00
parent e2b2c5674b
commit 9a5641d414
2 changed files with 59 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/Internal/Containers/ObjectContainer.php
View File

@ -25,6 +25,8 @@ use danog\AsyncOrm\DbObject;
use Revolt\EventLoop;
use Traversable;
use function Amp\async;
/**
* @template TKey as array-key
* @template TValue as DbObject
@ -87,25 +89,35 @@ final class ObjectContainer
{
if (isset($this->cache[$index])) {
$obj = $this->cache[$index];
$ref = $obj->reference->get();
$ref = $obj->get();
if ($ref !== null) {
$obj->ttl = \time() + $this->cacheTtl;
return $ref;
}
// @codeCoverageIgnoreStart
unset($this->cache[$index]);
// @codeCoverageIgnoreEnd
}
$result = $this->inner->get($index);
if (isset($this->cache[$index])) {
return $this->cache[$index]->reference->get();
return $this->cache[$index]->get();
}
if ($result === null) {
return null;
}
$result->initDb($this, $index, $this->config);
$this->cache[$index] = new ObjectReference($result, \time() + $this->cacheTtl);
$this->cache[$index] = new ObjectReference(
$result,
\time() + $this->cacheTtl,
$f = async(
$result->initDb(...),
$this,
$index,
$this->config
)
);
$f->await();
return $result;
}
@ -116,11 +128,19 @@ final class ObjectContainer
*/
public function set(string|int $key, DbObject $value): void
{
if (isset($this->cache[$key]) && $this->cache[$key]->reference->get() === $value) {
if (isset($this->cache[$key]) && $this->cache[$key]->get() === $value) {
return;
}
$value->initDb($this, $key, $this->config);
$this->cache[$key] = new ObjectReference($value, \time() + $this->cacheTtl);
$value->initDb(
$this,
$key,
$this->config
);
$this->cache[$key] = new ObjectReference(
$value,
\time() + $this->cacheTtl,
null
);
$value->save();
}
@ -137,15 +157,24 @@ final class ObjectContainer
foreach ($this->inner->getIterator() as $key => $value) {
if (isset($this->cache[$key])) {
$obj = $this->cache[$key];
$ref = $obj->reference->get();
$ref = $obj->get();
if ($ref !== null) {
$obj->ttl = \time() + $this->cacheTtl;
yield $key => $ref;
continue;
}
}
$value->initDb($this, $key, $this->config);
$this->cache[$key] = new ObjectReference($value, \time() + $this->cacheTtl);
$this->cache[$key] = new ObjectReference(
$value,
\time() + $this->cacheTtl,
$f = async(
$value->initDb(...),
$this,
$key,
$this->config
)
);
$f->await();
yield $key => $value;
}
}
@ -177,7 +206,7 @@ final class ObjectContainer
if ($value->ttl <= $now) {
$value->obj = null;
}
if ($value->reference->get() !== null) {
if ($value->get() !== null) {
$new[$key] = $value;
}
}

20
src/Internal/Containers/ObjectReference.php
View File

@ -18,6 +18,7 @@
namespace danog\AsyncOrm\Internal\Containers;
use Amp\Future;
use danog\AsyncOrm\DbObject;
use WeakReference;
@ -29,14 +30,29 @@ use WeakReference;
final class ObjectReference
{
/** @var WeakReference<TObject> */
public readonly WeakReference $reference;
private readonly WeakReference $reference;
public ?DbObject $obj;
/** @param TObject $object */
public function __construct(
DbObject $object,
public int $ttl
public int $ttl,
private ?Future $initFuture
) {
$this->obj = $object;
$this->reference = WeakReference::create($object);
}
/** @return ?TObject */
public function get(): ?DbObject
{
$ref = $this->reference->get();
if ($ref === null) {
return $ref;
}
if ($this->initFuture !== null) {
$this->initFuture->await();
$this->initFuture = null;
}
return $ref;
}
}