增加sae相关驱动

2026-07-06 07:02:47 +08:00 · 2015-12-10 15:17:32 +08:00
parent 9f0ee6acbb
commit 7b0fe7286d
5 changed files with 291 additions and 34 deletions
--- a/library/think/cache/driver/sae.php
+++ b/library/think/cache/driver/sae.php
@@ -0,0 +1,155 @@
+<?php
+// +----------------------------------------------------------------------
+// | ThinkPHP [ WE CAN DO IT JUST THINK ]
+// +----------------------------------------------------------------------
+// | Copyright (c) 2006~2015 http://thinkphp.cn All rights reserved.
+// +----------------------------------------------------------------------
+// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
+// +----------------------------------------------------------------------
+// | Author: liu21st <liu21st@gmail.com>
+// +----------------------------------------------------------------------
+
+namespace think\cache\driver;
+
+use think\Exception;
+
+/**
+ * SAE Memcache缓存驱动
+ * @author    liu21st <liu21st@gmail.com>
+ */
+class Sae
+{
+    protected $handler = null;
+    protected $options = [
+        'host'       => '127.0.0.1',
+        'port'       => 11211,
+        'expire'     => 0,
+        'timeout'    => false,
+        'persistent' => false,
+        'length'     => 0,
+    ];
+
+    /**
+     * 架构函数
+     * @param array $options 缓存参数
+     * @access public
+     */
+    public function __construct($options = [])
+    {
+        if (!function_exists('memcache_init')) {
+            throw new Exception('请在SAE平台上运行代码。');
+        }
+        $this->handler = memcache_init();
+        if (!$this->handler) {
+            throw new Exception('您未开通Memcache服务，请在SAE管理平台初始化Memcache服务');
+        }
+        if (!empty($options)) {
+            $this->options = array_merge($this->options, $options);
+        }
+    }
+
+    /**
+     * 读取缓存
+     * @access public
+     * @param string $name 缓存变量名
+     * @return mixed
+     */
+    public function get($name)
+    {
+        return $this->handler->get($_SERVER['HTTP_APPVERSION'] . '/' . $this->options['prefix'] . $name);
+    }
+
+    /**
+     * 写入缓存
+     * @access public
+     * @param string $name 缓存变量名
+     * @param mixed $value  存储数据
+     * @param integer $expire  有效时间（秒）
+     * @return bool
+     */
+    public function set($name, $value, $expire = null)
+    {
+        if (is_null($expire)) {
+            $expire = $this->options['expire'];
+        }
+        $name = $this->options['prefix'] . $name;
+        if ($this->handler->set($_SERVER['HTTP_APPVERSION'] . '/' . $name, $value, 0, $expire)) {
+            if ($this->options['length'] > 0) {
+                // 记录缓存队列
+                $this->queue($name);
+            }
+            return true;
+        }
+        return false;
+    }
+
+    /**
+     * 删除缓存
+     *
+     * @param    string  $name 缓存变量名
+     * @param bool|false $ttl
+     *
+     * @return bool
+     */
+    public function rm($name, $ttl = false)
+    {
+        $name = $_SERVER['HTTP_APPVERSION'] . '/' . $this->options['prefix'] . $name;
+        return false === $ttl ?
+        $this->handler->delete($name) :
+        $this->handler->delete($name, $ttl);
+    }
+
+    /**
+     * 清除缓存
+     * @access public
+     * @return bool
+     */
+    public function clear()
+    {
+        return $this->handler->flush();
+    }
+
+    /**
+     * 获得SaeKv对象
+     */
+    private function getKv()
+    {
+        static $kv;
+        if (!$kv) {
+            $kv = new \SaeKV();
+            if (!$kv->init()) {
+                throw new Exception('您没有初始化KVDB，请在SAE管理平台初始化KVDB服务');
+            }
+        }
+        return $kv;
+    }
+
+    /**
+     * 队列缓存
+     * @access protected
+     * @param string $key 队列名
+     * @return mixed
+     */
+    //[sae] 下重写queque队列缓存方法
+    protected function queue($key)
+    {
+        $queue_name = isset($this->options['queue_name']) ? $this->options['queue_name'] : 'think_queue';
+        $kv         = $this->getKv();
+        $value      = $kv->get($queue_name);
+        if (!$value) {
+            $value = [];
+        }
+        // 进列
+        if (false === array_search($key, $value)) {
+            array_push($value, $key);
+        }
+
+        if (count($value) > $this->options['length']) {
+            // 出列
+            $key = array_shift($value);
+            // 删除缓存
+            $this->rm($key);
+        }
+        return $kv->set($queue_name, $value);
+    }
+}