/
SingleFileArrayStorage.php
119 lines (98 loc) · 3.88 KB
/
SingleFileArrayStorage.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
<?php
/**
* @copyright 2016-2023 Roman Parpalak
* @license MIT
*/
namespace S2\Rose\Storage\File;
use S2\Rose\Entity\ExternalId;
use S2\Rose\Entity\Metadata\Img;
use S2\Rose\Entity\Metadata\ImgCollection;
use S2\Rose\Entity\Metadata\SnippetSource;
use S2\Rose\Entity\TocEntry;
use S2\Rose\Helper\ProfileHelper;
use S2\Rose\Storage\ArrayFulltextStorage;
use S2\Rose\Storage\ArrayStorage;
use S2\Rose\Storage\FulltextProxyInterface;
class SingleFileArrayStorage extends ArrayStorage
{
protected FulltextProxyInterface $fulltextProxy;
protected string $filename;
public function __construct($filename)
{
$this->filename = $filename;
$this->fulltextProxy = new ArrayFulltextStorage();
}
public function load(bool $isDebug = false): array
{
$return = [];
if (\count($this->toc)) {
return $return;
}
if (!is_file($this->filename)) {
return $return;
}
if ($isDebug) {
$start_time = microtime(true);
}
$data = file_get_contents($this->filename);
if ($isDebug) {
$return[] = ProfileHelper::getProfilePoint('Reading index file', -$start_time + ($start_time = microtime(true)));
}
$end = strpos($data, "\n");
$myData = substr($data, 8, $end - 8);
$data = substr($data, $end + 1);
$unserializeOptions = ['allowed_classes' => [
\DateTime::class,
TocEntry::class,
Img::class,
ImgCollection::class,
SnippetSource::class,
]];
$this->fulltextProxy->setFulltextIndex(unserialize($myData, $unserializeOptions) ?: []);
$end = strpos($data, "\n");
$myData = substr($data, 8, $end - 8);
$data = substr($data, $end + 1);
$this->excludedWords = unserialize($myData, $unserializeOptions) ?: [];
$end = strpos($data, "\n");
$myData = substr($data, 8, $end - 8);
$data = substr($data, $end + 1);
$this->metadata = unserialize($myData, $unserializeOptions) ?: [];
$end = strpos($data, "\n");
$myData = substr($data, 8, $end - 8);
// $data = substr($data, $end + 1);
$this->toc = unserialize($myData, $unserializeOptions) ?: [];
if ($isDebug) {
$return[] = ProfileHelper::getProfilePoint('Unserializing index', -$start_time + ($start_time = microtime(true)));
}
$this->externalIdMap = [];
foreach ($this->toc as $serializedExtId => $entry) {
$this->externalIdMap[$entry->getInternalId()] = ExternalId::fromString($serializedExtId);
}
return $return;
}
public function save(): void
{
@unlink($this->filename);
file_put_contents($this->filename, '<?php //' . 'a:' . \count($this->fulltextProxy->getFulltextIndex()) . ':{');
$buffer = '';
$length = 0;
foreach ($this->fulltextProxy->getFulltextIndex() as $word => $data) {
$chunk = serialize($word) . serialize($data);
$length += \strlen($chunk);
$buffer .= $chunk;
if ($length > 100000) {
file_put_contents($this->filename, $buffer, FILE_APPEND);
$buffer = '';
$length = 0;
}
}
file_put_contents($this->filename, $buffer . '}' . "\n", FILE_APPEND);
$this->fulltextProxy->setFulltextIndex([]);
file_put_contents($this->filename, ' //' . serialize($this->excludedWords) . "\n", FILE_APPEND);
$this->excludedWords = [];
file_put_contents($this->filename, ' //' . serialize($this->metadata) . "\n", FILE_APPEND);
$this->metadata = [];
file_put_contents($this->filename, ' //' . serialize($this->toc) . "\n", FILE_APPEND);
$this->toc = [];
}
}