initial commit

src/ZM/Utils/SQLPool.php

@@ -0,0 +1,101 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: jerry
+ * Date: 2019/1/5
+ * Time: 4:48 PM
+ */
+
+namespace ZM\Utils;
+
+use framework\Console;
+use framework\ZMBuf;
+use SplQueue;
+use Swoole\Coroutine;
+use Swoole\Coroutine\Mysql;
+
+class SQLPool
+{
+    protected $available = true;
+    protected $pool;
+    private $info;
+
+    public $co_list = [];
+    public $connect_cnt = 0;
+
+    public function __construct() {
+        $this->pool = new SplQueue;
+        $this->info = [
+            "host" => ZMBuf::globals("sql_config")["sql_host"],
+            "port" => ZMBuf::globals("sql_config")["sql_port"],
+            "user" => ZMBuf::globals("sql_config")["sql_username"],
+            "password" => ZMBuf::globals("sql_config")["sql_password"],
+            "database" => ZMBuf::globals("sql_config")["sql_database"]
+        ];
+    }
+
+    /**
+     * 将利用过的连接入队
+     * @param $mysql
+     */
+    public function put($mysql) {
+        $this->pool->push($mysql);
+        if (($a = array_shift($this->co_list)) !== null) {
+            Coroutine::resume($a);
+        }
+    }
+
+    /**
+     * 获取队中的连接，如果不存在则创建新的
+     * @param bool $no_new_conn
+     * @return bool|mixed|Mysql
+     */
+    public function get($no_new_conn = false) {
+        if (count($this->pool) == 0 && $this->connect_cnt <= 70) {
+            if($no_new_conn) return false;
+            $this->connect_cnt += 1;
+            $r = $this->newConnect();
+            if ($r !== false) {
+                return $r;
+            } else {
+                $this->connect_cnt -= 1;
+                return false;
+            }
+        } elseif (count($this->pool) > 0) {
+            $con = $this->pool->pop();
+            if ($con->connected !== false) return $con;
+        } elseif ($this->connect_cnt > 70) {
+            $this->co_list[]=Coroutine::getuid();
+            Console::warning("数据库连接过多，协程等待重复利用中...当前协程数 ".Coroutine::stats()["coroutine_num"]);
+            Coroutine::suspend();
+            return $this->get($no_new_conn);
+        }
+        return false;
+    }
+
+    public function getCount() {
+        return $this->pool->count();
+    }
+
+    public function destruct() {
+        // 连接池销毁, 置不可用状态, 防止新的客户端进入常驻连接池, 导致服务器无法平滑退出
+        $this->available = false;
+        while (!$this->pool->isEmpty()) {
+            $this->pool->pop();
+        }
+    }
+
+    private function newConnect() {
+        //无空闲连接，创建新连接
+        $mysql = new Mysql();
+
+        Console::info("创建SQL连接中，当前有" . $this->connect_cnt . "个连接");
+        $res = $mysql->connect($this->info);
+        if ($res == false) {
+            echo $mysql->error . PHP_EOL;
+            return false;
+        } else {
+            return $mysql;
+        }
+    }
+}