048953140ce99d823ad899ca221612c187776d4b
3 * Base class for memcached clients.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
25 * Base class for memcached clients.
29 class MemcachedBagOStuff
extends BagOStuff
{
30 /** @var MWMemcached|Memcached */
34 * Fill in the defaults for any parameters missing from $params, using the
35 * backwards-compatible global variables
36 * @param array $params
39 protected function applyDefaultParams( $params ) {
40 if ( !isset( $params['servers'] ) ) {
41 $params['servers'] = $GLOBALS['wgMemCachedServers'];
43 if ( !isset( $params['debug'] ) ) {
44 $params['debug'] = $GLOBALS['wgMemCachedDebug'];
46 if ( !isset( $params['persistent'] ) ) {
47 $params['persistent'] = $GLOBALS['wgMemCachedPersistent'];
49 if ( !isset( $params['compress_threshold'] ) ) {
50 $params['compress_threshold'] = 1500;
52 if ( !isset( $params['timeout'] ) ) {
53 $params['timeout'] = $GLOBALS['wgMemCachedTimeout'];
55 if ( !isset( $params['connect_timeout'] ) ) {
56 $params['connect_timeout'] = 0.5;
61 protected function doGet( $key, $flags = 0 ) {
64 return $this->getWithToken( $key, $casToken, $flags );
67 protected function getWithToken( $key, &$casToken, $flags = 0 ) {
68 return $this->client
->get( $this->validateKeyEncoding( $key ), $casToken );
71 public function set( $key, $value, $exptime = 0, $flags = 0 ) {
72 return $this->client
->set( $this->validateKeyEncoding( $key ), $value,
73 $this->fixExpiry( $exptime ) );
76 protected function cas( $casToken, $key, $value, $exptime = 0 ) {
77 return $this->client
->cas( $casToken, $this->validateKeyEncoding( $key ),
78 $value, $this->fixExpiry( $exptime ) );
81 public function delete( $key ) {
82 return $this->client
->delete( $this->validateKeyEncoding( $key ) );
85 public function add( $key, $value, $exptime = 0 ) {
86 return $this->client
->add( $this->validateKeyEncoding( $key ), $value,
87 $this->fixExpiry( $exptime ) );
90 public function merge( $key, $callback, $exptime = 0, $attempts = 10, $flags = 0 ) {
91 if ( !is_callable( $callback ) ) {
92 throw new Exception( "Got invalid callback." );
95 return $this->mergeViaCas( $key, $callback, $exptime, $attempts );
99 * Get the underlying client object. This is provided for debugging
103 public function getClient() {
104 return $this->client
;
108 * Construct a cache key.
111 * @param string $keyspace
115 public function makeKeyInternal( $keyspace, $args ) {
116 // Memcached keys have a maximum length of 255 characters. From that,
117 // subtract the number of characters we need for the keyspace and for
118 // the separator character needed for each argument.
119 $charsLeft = 255 - strlen( $keyspace ) - count( $args );
123 function ( $arg ) use ( $that, &$charsLeft ) {
124 // Make sure %, #, and non-ASCII chars are escaped
125 $arg = preg_replace_callback(
126 '/[^\x21-\x22\x24\x26-\x7e]+/',
128 return rawurlencode( $m[0] );
133 // 33 = 32 characters for the MD5 + 1 for the '#' prefix.
134 if ( $charsLeft > 33 && strlen( $arg ) > $charsLeft ) {
135 $arg = '#' . md5( $arg );
138 $charsLeft -= strlen( $arg );
144 if ( $charsLeft < 0 ) {
145 $args = array( '##' . md5( implode( ':', $args ) ) );
148 return parent
::makeKeyInternal( $keyspace, $args );
152 * Ensure that a key is safe to use (contains no control characters and no
153 * characters above the ASCII range.)
159 public function validateKeyEncoding( $key ) {
160 if ( preg_match( '/[^\x21-\x7e]+/', $key ) ) {
161 throw new Exception( "Key contains invalid characters: $key" );
167 * TTLs higher than 30 days will be detected as absolute TTLs
168 * (UNIX timestamps), and will result in the cache entry being
169 * discarded immediately because the expiry is in the past.
170 * Clamp expires >30d at 30d, unless they're >=1e9 in which
171 * case they are likely to really be absolute (1e9 = 2011-09-09)
175 function fixExpiry( $expiry ) {
176 if ( $expiry > 2592000 && $expiry < 1000000000 ) {
183 * Send a debug message to the log
184 * @param string $text
186 protected function debugLog( $text ) {
187 $this->logger
->debug( $text );
190 public function modifySimpleRelayEvent( array $event ) {
191 if ( array_key_exists( 'val', $event ) ) {
192 $event['flg'] = 0; // data is not serialized nor gzipped (for memcached driver)