123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251 |
- <?php
- if (!defined('STATUSNET')) {
-
-
- exit(1);
- }
- class MemcachePlugin extends Plugin
- {
- static $cacheInitialized = false;
- private $_conn = null;
- public $servers = array('127.0.0.1;11211');
- public $compressThreshold = 20480;
- public $compressMinSaving = 0.2;
- public $persistent = null;
- public $defaultExpiry = 86400;
-
- function onInitializePlugin()
- {
- if (self::$cacheInitialized) {
- $this->persistent = true;
- } else {
-
-
-
-
-
-
-
-
- $this->persistent = (php_sapi_name() == 'cli') ? false : true;
- }
- $this->_ensureConn();
- self::$cacheInitialized = true;
- return true;
- }
-
- function onStartCacheGet(&$key, &$value)
- {
- $this->_ensureConn();
- $value = $this->_conn->get($key);
- Event::handle('EndCacheGet', array($key, &$value));
- return false;
- }
-
- function onStartCacheSet(&$key, &$value, &$flag, &$expiry, &$success)
- {
- $this->_ensureConn();
- if ($expiry === null) {
- $expiry = $this->defaultExpiry;
- }
- $success = $this->_conn->set($key, $value, $this->flag(intval($flag)), $expiry);
- Event::handle('EndCacheSet', array($key, $value, $flag,
- $expiry));
- return false;
- }
-
- function onStartCacheIncrement(&$key, &$step, &$value)
- {
- $this->_ensureConn();
- $value = $this->_conn->increment($key, $step);
- Event::handle('EndCacheIncrement', array($key, $step, $value));
- return false;
- }
-
- function onStartCacheDelete(&$key, &$success)
- {
- $this->_ensureConn();
- $success = $this->_conn->delete($key);
- Event::handle('EndCacheDelete', array($key));
- return false;
- }
- function onStartCacheReconnect(&$success)
- {
- if (empty($this->_conn)) {
-
- return true;
- }
- if ($this->persistent) {
- common_log(LOG_ERR, "Cannot close persistent memcached connection");
- $success = false;
- } else {
- common_log(LOG_INFO, "Closing memcached connection");
- $success = $this->_conn->close();
- $this->_conn = null;
- }
- return false;
- }
-
- private function _ensureConn()
- {
- if (empty($this->_conn)) {
- $this->_conn = new Memcache();
- if (is_array($this->servers)) {
- $servers = $this->servers;
- } else {
- $servers = array($this->servers);
- }
- foreach ($servers as $server) {
- if (strpos($server, ';') !== false) {
- list($host, $port) = explode(';', $server);
- } else {
- $host = $server;
- $port = 11211;
- }
- $this->_conn->addServer($host, $port, $this->persistent);
- }
-
-
-
-
-
- $this->_conn->setCompressThreshold($this->compressThreshold,
- $this->compressMinSaving);
- }
- }
-
- protected function flag($flag)
- {
- $out = 0;
- if ($flag & Cache::COMPRESSED == Cache::COMPRESSED) {
- $out |= MEMCACHE_COMPRESSED;
- }
- return $out;
- }
- function onPluginVersion(&$versions)
- {
- $versions[] = array('name' => 'Memcache',
- 'version' => GNUSOCIAL_VERSION,
- 'author' => 'Evan Prodromou, Craig Andrews',
- 'homepage' => 'http://status.net/wiki/Plugin:Memcache',
- 'rawdescription' =>
-
- _m('Use <a href="http://memcached.org/">Memcached</a> to cache query results.'));
- return true;
- }
- }
|