heyuntao
2023-10-07 579e58522c83dfc2fe54f3d0fc225ef12091b269
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
<?php
 
/*
 * This file is part of the overtrue/pinyin.
 *
 * (c) overtrue <i@overtrue.me>
 *
 * This source file is subject to the MIT license that is bundled
 * with this source code in the file LICENSE.
 */
 
namespace Overtrue\Pinyin;
 
use Closure;
 
/**
 * Memory Dict File loader.
 */
class MemoryFileDictLoader implements DictLoaderInterface
{
    /**
     * Data directory.
     *
     * @var string
     */
    protected $path;
 
    /**
     * Words segment name.
     *
     * @var string
     */
    protected $segmentName = 'words_%s';
 
    /**
     * Segment files.
     *
     * @var array
     */
    protected $segments = array();
 
    /**
     * Surname cache.
     *
     * @var array
     */
    protected $surnames = array();
 
    /**
     * Constructor.
     *
     * @param string $path
     */
    public function __construct($path)
    {
        $this->path = $path;
 
        for ($i = 0; $i < 100; ++$i) {
            $segment = $path.'/'.sprintf($this->segmentName, $i);
 
            if (file_exists($segment)) {
                $this->segments[] = (array) include $segment;
            }
        }
    }
 
    /**
     * Load dict.
     *
     * @param Closure $callback
     */
    public function map(Closure $callback)
    {
        foreach ($this->segments as $dictionary) {
            $callback($dictionary);
        }
    }
 
    /**
     * Load surname dict.
     *
     * @param Closure $callback
     */
    public function mapSurname(Closure $callback)
    {
        if (empty($this->surnames)) {
            $surnames = $this->path.'/surnames';
 
            if (file_exists($surnames)) {
                $this->surnames = (array) include $surnames;
            }
        }
 
        $callback($this->surnames);
    }
}