From 38dc510b2dba02dba5a60fbc00947aac4fd24aab Mon Sep 17 00:00:00 2001 From: alecpl <alec@alec.pl> Date: Wed, 29 Feb 2012 06:53:52 -0500 Subject: [PATCH] - Fix warning when properties array is empty --- program/include/rcube_cache.php | 174 ++++++++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 134 insertions(+), 40 deletions(-) diff --git a/program/include/rcube_cache.php b/program/include/rcube_cache.php index 8c5a750..e501185 100644 --- a/program/include/rcube_cache.php +++ b/program/include/rcube_cache.php @@ -7,7 +7,10 @@ | This file is part of the Roundcube Webmail client | | Copyright (C) 2011, The Roundcube Dev Team | | Copyright (C) 2011, Kolab Systems AG | - | Licensed under the GNU GPL | + | | + | Licensed under the GNU General Public License version 3 or | + | any later version with exceptions for skins & plugins. | + | See the README file for a full license statement. | | | | PURPOSE: | | Caching engine | @@ -28,7 +31,7 @@ * @package Cache * @author Thomas Bruederli <roundcube@gmail.com> * @author Aleksander Machniak <alec@alec.pl> - * @version 1.0 + * @version 1.1 */ class rcube_cache { @@ -41,6 +44,8 @@ private $type; private $userid; private $prefix; + private $ttl; + private $packed; private $index; private $cache = array(); private $cache_keys = array(); @@ -54,12 +59,16 @@ * @param string $type Engine type ('db' or 'memcache' or 'apc') * @param int $userid User identifier * @param string $prefix Key name prefix + * @param int $ttl Expiration time of memcache/apc items in seconds (max.2592000) + * @param bool $packed Enables/disabled data serialization. + * It's possible to disable data serialization if you're sure + * stored data will be always a safe string */ - function __construct($type, $userid, $prefix='') + function __construct($type, $userid, $prefix='', $ttl=0, $packed=true) { $rcmail = rcmail::get_instance(); $type = strtolower($type); - + if ($type == 'memcache') { $this->type = 'memcache'; $this->db = $rcmail->get_memcache(); @@ -73,8 +82,10 @@ $this->db = $rcmail->get_dbh(); } - $this->userid = (int) $userid; - $this->prefix = $prefix; + $this->userid = (int) $userid; + $this->ttl = (int) $ttl; + $this->packed = $packed; + $this->prefix = $prefix; } @@ -106,6 +117,38 @@ $this->cache[$key] = $data; $this->cache_changed = true; $this->cache_changes[$key] = true; + } + + + /** + * Returns cached value without storing it in internal memory. + * + * @param string $key Cache key name + * + * @return mixed Cached value + */ + function read($key) + { + if (array_key_exists($key, $this->cache)) { + return $this->cache[$key]; + } + + return $this->read_record($key, true); + } + + + /** + * Sets (add/update) value in cache and immediately saves + * it in the backend, no internal memory will be used. + * + * @param string $key Cache key name + * @param mixed $data Cache data + * + * @param boolean True on success, False on failure + */ + function write($key, $data) + { + return $this->write_record($key, $this->packed ? serialize($data) : $data); } @@ -148,6 +191,24 @@ /** + * Remove cache records older than ttl + */ + function expunge() + { + if ($this->type == 'db' && $this->db) { + $this->db->query( + "DELETE FROM ".get_table_name('cache'). + " WHERE user_id = ?". + " AND cache_key LIKE ?". + " AND " . $this->db->unixtimestamp('created')." < ?", + $this->userid, + $this->prefix.'.%', + time() - $this->ttl); + } + } + + + /** * Writes the cache back to the DB. */ function close() @@ -161,7 +222,7 @@ if ($this->cache_changes[$key]) { // Make sure we're not going to write unchanged data // by comparing current md5 sum with the sum calculated on DB read - $data = serialize($data); + $data = $this->packed ? serialize($data) : $data; if (!$this->cache_sums[$key] || $this->cache_sums[$key] != md5($data)) { $this->write_record($key, $data); @@ -176,29 +237,41 @@ /** * Reads cache entry. * - * @param string $key Cache key name + * @param string $key Cache key name + * @param boolean $nostore Enable to skip in-memory store * * @return mixed Cached value - * @access private */ - private function read_record($key) + private function read_record($key, $nostore=false) { if (!$this->db) { return null; } - if ($this->type == 'memcache') { - $data = $this->db->get($this->ckey($key)); - } - else if ($this->type == 'apc') { - $data = apc_fetch($this->ckey($key)); - } + if ($this->type != 'db') { + if ($this->type == 'memcache') { + $data = $this->db->get($this->ckey($key)); + } + else if ($this->type == 'apc') { + $data = apc_fetch($this->ckey($key)); + } - if ($data) { - $this->cache_sums[$key] = md5($data); - $this->cache[$key] = unserialize($data); + if ($data) { + $md5sum = md5($data); + $data = $this->packed ? unserialize($data) : $data; + + if ($nostore) { + return $data; + } + + $this->cache_sums[$key] = $md5sum; + $this->cache[$key] = $data; + } + else { + $this->cache[$key] = null; + } } - else if ($this->type == 'db') { + else { $sql_result = $this->db->limitquery( "SELECT cache_id, data, cache_key". " FROM ".get_table_name('cache'). @@ -212,10 +285,20 @@ if ($sql_arr = $this->db->fetch_assoc($sql_result)) { $key = substr($sql_arr['cache_key'], strlen($this->prefix)+1); $md5sum = $sql_arr['data'] ? md5($sql_arr['data']) : null; - $data = $sql_arr['data'] ? unserialize($sql_arr['data']) : null; + if ($sql_arr['data']) { + $data = $this->packed ? unserialize($sql_arr['data']) : $sql_arr['data']; + } + + if ($nostore) { + return $data; + } + $this->cache[$key] = $data; $this->cache_sums[$key] = $md5sum; $this->cache_keys[$key] = $sql_arr['cache_id']; + } + else { + $this->cache[$key] = null; } } @@ -228,7 +311,8 @@ * * @param string $key Cache key name * @param mxied $data Serialized cache data - * @access private + * + * @param boolean True on success, False on failure */ private function write_record($key, $data) { @@ -256,7 +340,7 @@ // update existing cache record if ($key_exists) { - $this->db->query( + $result = $this->db->query( "UPDATE ".get_table_name('cache'). " SET created = ". $this->db->now().", data = ?". " WHERE user_id = ?". @@ -267,14 +351,14 @@ else { // for better performance we allow more records for one key // so, no need to check if record exist (see rcube_cache::read_record()) - $this->db->query( + $result = $this->db->query( "INSERT INTO ".get_table_name('cache'). " (created, user_id, cache_key, data)". " VALUES (".$this->db->now().", ?, ?, ?)", $this->userid, $key, $data); } - return true; + return $this->db->affected_rows($result); } @@ -285,7 +369,6 @@ * @param boolean $prefix_mode Enable it to clear all keys starting * with prefix specified in $key * - * @access private; */ private function remove_record($key=null, $prefix_mode=false) { @@ -341,30 +424,45 @@ /** * Adds entry into memcache/apc DB. - * @access private + * + * @param string $key Cache key name + * @param mxied $data Serialized cache data + * @param bollean $index Enables immediate index update + * + * @param boolean True on success, False on failure */ - private function add_record($key, $data) + private function add_record($key, $data, $index=false) { if ($this->type == 'memcache') { - $result = $this->db->replace($key, $data, MEMCACHE_COMPRESSED); + $result = $this->db->replace($key, $data, MEMCACHE_COMPRESSED, $this->ttl); if (!$result) - $result = $this->db->set($key, $data, MEMCACHE_COMPRESSED); - return $result; + $result = $this->db->set($key, $data, MEMCACHE_COMPRESSED, $this->ttl); } - - if ($this->type == 'apc') { + else if ($this->type == 'apc') { if (apc_exists($key)) apc_delete($key); - return apc_store($key, $data); + $result = apc_store($key, $data, $this->ttl); } + + // Update index + if ($index && $result) { + $this->load_index(); + + if (array_search($key, $this->index) === false) { + $this->index[] = $key; + $data = serialize($this->index); + $this->add_record($this->ikey(), $data); + } + } + + return $result; } /** * Deletes entry from memcache/apc DB. - * @access private */ - private function delete_record($index=true) + private function delete_record($key, $index=true) { if ($this->type == 'memcache') $this->db->delete($this->ckey($key)); @@ -381,7 +479,6 @@ /** * Writes the index entry into memcache/apc DB. - * @access private */ private function write_index() { @@ -411,7 +508,6 @@ /** * Gets the index entry from memcache/apc DB. - * @access private */ private function load_index() { @@ -441,7 +537,6 @@ * @param string $key Cache key name * * @return string Cache key - * @access private */ private function ckey($key) { @@ -453,7 +548,6 @@ * Creates per-user index cache key name (for memcache and apc) * * @return string Cache key - * @access private */ private function ikey() { -- Gitblit v1.9.1