includes/libs/objectcache/RESTBagOStuff.php

   1 <?php
   2
   3 /**
   4  * Interface to key-value storage behind an HTTP server.
   5  *
   6  * Uses URL of the form "baseURL/{KEY}" to store, fetch, and delete values.
   7  *
   8  * E.g., when base URL is `/v1/sessions/`, then the store would do:
   9  *
  10  * `PUT /v1/sessions/12345758`
  11  *
  12  * and fetch would do:
  13  *
  14  * `GET /v1/sessions/12345758`
  15  *
  16  * delete would do:
  17  *
  18  * `DELETE /v1/sessions/12345758`
  19  *
  20  * Configure with:
  21  *
  22  * @code
  23  * $wgObjectCaches['sessions'] = array(
  24  *      'class' => 'RESTBagOStuff',
  25  *      'url' => 'http://localhost:7231/wikimedia.org/v1/sessions/'
  26  * );
  27  * @endcode
  28  */
  29 class RESTBagOStuff extends BagOStuff {
  30
  31         /**
  32          * @var MultiHttpClient
  33          */
  34         private $client;
  35
  36         /**
  37          * REST URL to use for storage.
  38          * @var string
  39          */
  40         private $url;
  41
  42         public function __construct( $params ) {
  43                 if ( empty( $params['url'] ) ) {
  44                         throw new InvalidArgumentException( 'URL parameter is required' );
  45                 }
  46                 parent::__construct( $params );
  47                 if ( empty( $params['client'] ) ) {
  48                         $this->client = new MultiHttpClient( [] );
  49                 } else {
  50                         $this->client = $params['client'];
  51                 }
  52                 // Make sure URL ends with /
  53                 $this->url = rtrim( $params['url'], '/' ) . '/';
  54         }
  55
  56         /**
  57          * @param string  $key
  58          * @param integer $flags Bitfield of BagOStuff::READ_* constants [optional]
  59          * @return mixed Returns false on failure and if the item does not exist
  60          */
  61         protected function doGet( $key, $flags = 0 ) {
  62                 $req = [
  63                         'method' => 'GET',
  64                     'url' => $this->url . rawurlencode( $key ),
  65
  66                 ];
  67                 list( $rcode, $rdesc, $rhdrs, $rbody, $rerr ) = $this->client->run( $req );
  68                 if ( $rcode === 200 ) {
  69                         if ( is_string( $rbody ) ) {
  70                                 return unserialize( $rbody );
  71                         }
  72                         return false;
  73                 }
  74                 if ( $rcode === 0 || ( $rcode >= 400 && $rcode != 404 ) ) {
  75                         return $this->handleError( "Failed to fetch $key", $rcode, $rerr );
  76                 }
  77                 return false;
  78         }
  79
  80         /**
  81          * Handle storage error
  82          * @param string $msg Error message
  83          * @param int    $rcode Error code from client
  84          * @param string $rerr Error message from client
  85          * @return false
  86          */
  87         protected function handleError( $msg, $rcode, $rerr ) {
  88                 $this->logger->error( "$msg : ({code}) {error}", [
  89                         'code' => $rcode,
  90                         'error' => $rerr
  91                 ] );
  92                 $this->setLastError( $rcode === 0 ? self::ERR_UNREACHABLE : self::ERR_UNEXPECTED );
  93                 return false;
  94         }
  95
  96         /**
  97          * Set an item
  98          *
  99          * @param string $key
 100          * @param mixed  $value
 101          * @param int    $exptime Either an interval in seconds or a unix timestamp for expiry
 102          * @param int    $flags Bitfield of BagOStuff::WRITE_* constants
 103          * @return bool Success
 104          */
 105         public function set( $key, $value, $exptime = 0, $flags = 0 ) {
 106                 $req = [
 107                         'method' => 'PUT',
 108                         'url' => $this->url . rawurlencode( $key ),
 109                         'body' => serialize( $value )
 110                 ];
 111                 list( $rcode, $rdesc, $rhdrs, $rbody, $rerr ) = $this->client->run( $req );
 112                 if ( $rcode === 200 || $rcode === 201 ) {
 113                         return true;
 114                 }
 115                 return $this->handleError( "Failed to store $key", $rcode, $rerr );
 116         }
 117
 118         /**
 119          * Delete an item.
 120          *
 121          * @param string $key
 122          * @return bool True if the item was deleted or not found, false on failure
 123          */
 124         public function delete( $key ) {
 125                 $req = [
 126                         'method' => 'DELETE',
 127                         'url' => $this->url . rawurlencode( $key ),
 128                 ];
 129                 list( $rcode, $rdesc, $rhdrs, $rbody, $rerr ) = $this->client->run( $req );
 130                 if ( $rcode === 200 || $rcode === 204 || $rcode === 205 ) {
 131                         return true;
 132                 }
 133                 return $this->handleError( "Failed to delete $key", $rcode, $rerr );
 134         }
 135 }