fix:初始化

wanglizhong

2025-05-05 9b8a7157bb9c401de973a4107f74ff3e723ec156

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
<?php
class LtDbConnectionManager
{
    /**
     * Connection management
     * array(
     *     "connection"  => connection resource id,
     *     "expire_time" => expire time,
     *     "schema"      => default schema name,
     *     "charset"     => char set / encoding
     * )
     */
    static public $connectionPool;
    public $configHandle;
    protected $connectionAdapter;
    protected $sqlAdapter;
    private $servers;
 
    public function getConnection($group, $node, $role = "master")
    {
        if(empty($this->servers))
        {
            $this->servers = $this->configHandle->get("db.servers");
        }
        if (($connection = $this->getNewConnection($group, $node, $role)) ||($connection = $this->getCachedConnection($group, $node, $role)))
        {
            return array(
                "connectionAdapter" => $this->connectionAdapter,
                "connectionResource" => $connection
            );
        }
        else
        {
            trigger_error("db server can not be connected: group=$group, node=$node, role=$role", E_USER_ERROR);
            return false;
        }
    }
 
    protected function getConnectionKey($connConf)
    {
        return $connConf['adapter'] . $connConf['host'] . $connConf['port'] . $connConf['username'] . $connConf['dbname'];
    }
 
    protected function saveConnection($connConf, $connection, $ttl)
    {
        $connectionInfo = array(
            "connection"  => $connection,
            "expire_time" => time() + $ttl,
            "schema"      => $connConf["schema"],
            "charset"     => $connConf["charset"],
        );
        self::$connectionPool[$this->getConnectionKey($connConf)] = $connectionInfo;
    }
 
    protected function getCachedConnection($group, $node, $role)
    {
        foreach($this->servers[$group][$node][$role] as $hostConfig)
        {
            $key = $this->getConnectionKey($hostConfig);
            if(isset(self::$connectionPool[$key]) && time() < self::$connectionPool[$key]['expire_time'])
            {//cached connection resource FOUND
                $connectionInfo = self::$connectionPool[$key];
                if ($connectionInfo["schema"] != $hostConfig["schema"] || $connectionInfo["charset"] != $hostConfig["charset"])
                {//检查当前schema和charset与用户要操作的目标不一致
                    $hostConfig = $this->servers[$group][$node][$role][$hostIndexArray[$hashNumber]];
                    $dbFactory = new LtDbAdapterFactory;
                    $this->connectionAdapter = $dbFactory->getConnectionAdapter($hostConfig["connection_adapter"]);
                    $this->sqlAdapter = $dbFactory->getSqlAdapter($hostConfig["sql_adapter"]);
                    if ($connectionInfo["schema"] != $hostConfig["schema"])
                    {
                        $this->connectionAdapter->exec($this->sqlAdapter->setSchema($hostConfig["schema"]), $connectionInfo["connection"]);
                    }
                    if ($connectionInfo["charset"] != $hostConfig["charset"])
                    {
                        $this->connectionAdapter->exec($this->sqlAdapter->setCharset($hostConfig["charset"]), $connectionInfo["connection"]);
                    }
                    $this->saveConnection($hostConfig, $connectionInfo["connection"], $hostConfig["connection_ttl"]);
                }
                return $connectionInfo["connection"];
            }
        }
        return false;
    }
 
    protected function getNewConnection($group, $node, $role)
    {
        $hostTotal = count($this->servers[$group][$node][$role]);
        $hostIndexArray = array_keys($this->servers[$group][$node][$role]);
        while ($hostTotal)
        {
            $hashNumber = substr(microtime(),7,1) % $hostTotal;
            $hostConfig = $this->servers[$group][$node][$role][$hostIndexArray[$hashNumber]];
            $dbFactory = new LtDbAdapterFactory;
            $this->connectionAdapter = $dbFactory->getConnectionAdapter($hostConfig["connection_adapter"]);
            $this->sqlAdapter = $dbFactory->getSqlAdapter($hostConfig["sql_adapter"]);
            if ($connection = $this->connectionAdapter->connect($hostConfig))
            {
                $this->connectionAdapter->exec($this->sqlAdapter->setSchema($hostConfig["schema"]), $connection);
                $this->connectionAdapter->exec($this->sqlAdapter->setCharset($hostConfig["charset"]), $connection);
                $this->saveConnection($hostConfig, $connection, $hostConfig["connection_ttl"]);
                return $connection;
            }
            else
            {
                //trigger_error('connection fail', E_USER_WARNING);
                //delete the unavailable server
                for ($i = $hashNumber; $i < $hostTotal - 1; $i ++)
                {
                    $hostIndexArray[$i] = $hostIndexArray[$i+1];
                }
                unset($hostIndexArray[$hostTotal-1]);
                $hostTotal --;
            }//end else
        }//end while
        return false;
    }
}