CRM/Utils/Cache/SqlGroup.php

   1 <?php
   2 /*
   3  +--------------------------------------------------------------------+
   4  | Copyright CiviCRM LLC. All rights reserved.                        |
   5  |                                                                    |
   6  | This work is published under the GNU AGPLv3 license with some      |
   7  | permitted exceptions and without any warranty. For full license    |
   8  | and copyright information, see https://civicrm.org/licensing       |
   9  +--------------------------------------------------------------------+
  10  */
  11
  12 /**
  13  *
  14  * @package CRM
  15  * @copyright CiviCRM LLC https://civicrm.org/licensing
  16  */
  17
  18 /**
  19  * This caching provider stores all cached items as a "group" in the
  20  * "civicrm_cache" table. The entire 'group' may be prefetched when
  21  * instantiating the cache provider.
  22  */
  23 class CRM_Utils_Cache_SqlGroup implements CRM_Utils_Cache_Interface {
  24
  25   // 6*60*60
  26   const DEFAULT_TTL = 21600;
  27
  28   const TS_FMT = 'Y-m-d H:i:s';
  29   // TODO Consider native implementation.
  30   use CRM_Utils_Cache_NaiveMultipleTrait;
  31
  32   /**
  33    * The host name of the memcached server.
  34    *
  35    * @var string
  36    */
  37   protected $group;
  38
  39   /**
  40    * @var int
  41    */
  42   protected $componentID;
  43
  44   /**
  45    * In-memory cache to optimize redundant get()s.
  46    *
  47    * @var array
  48    */
  49   protected $valueCache;
  50
  51   /**
  52    * In-memory cache to optimize redundant get()s.
  53    *
  54    * @var array
  55    *   Note: expiresCache[$key]===NULL means cache-miss
  56    */
  57   protected $expiresCache;
  58
  59   /**
  60    * Table.
  61    *
  62    * @var string
  63    */
  64   protected $table;
  65
  66   /**
  67    * Constructor.
  68    *
  69    * @param array $config
  70    *   An array of configuration params.
  71    *   - group: string
  72    *   - componentID: int
  73    *   - prefetch: bool, whether to preemptively read the entire cache group; default: TRUE
  74    *
  75    * @throws RuntimeException
  76    * @return \CRM_Utils_Cache_SqlGroup
  77    */
  78   public function __construct($config) {
  79     $this->table = CRM_Core_DAO_Cache::getTableName();
  80     if (isset($config['group'])) {
  81       $this->group = $config['group'];
  82     }
  83     else {
  84       throw new RuntimeException("Cannot construct SqlGroup cache: missing group");
  85     }
  86     if (isset($config['componentID'])) {
  87       $this->componentID = $config['componentID'];
  88     }
  89     else {
  90       $this->componentID = NULL;
  91     }
  92     $this->valueCache = [];
  93     if (CRM_Utils_Array::value('prefetch', $config, TRUE)) {
  94       $this->prefetch();
  95     }
  96   }
  97
  98   /**
  99    * @param string $key
 100    * @param mixed $value
 101    * @param null|int|\DateInterval $ttl
 102    * @return bool
 103    */
 104   public function set($key, $value, $ttl = NULL) {
 105     CRM_Utils_Cache::assertValidKey($key);
 106
 107     $lock = Civi::lockManager()->acquire("cache.{$this->group}_{$key}._null");
 108     if (!$lock->isAcquired()) {
 109       throw new \CRM_Utils_Cache_CacheException("SqlGroup: Failed to acquire lock on cache key.");
 110     }
 111
 112     if (is_int($ttl) && $ttl <= 0) {
 113       return $this->delete($key);
 114     }
 115
 116     $dataExists = CRM_Core_DAO::singleValueQuery("SELECT COUNT(*) FROM {$this->table} WHERE {$this->where($key)}");
 117     $expires = round(microtime(1)) + CRM_Utils_Date::convertCacheTtl($ttl, self::DEFAULT_TTL);
 118
 119     $dataSerialized = CRM_Core_BAO_Cache::encode($value);
 120
 121     // This table has a wonky index, so we cannot use REPLACE or
 122     // "INSERT ... ON DUPE". Instead, use SELECT+(INSERT|UPDATE).
 123     if ($dataExists) {
 124       $sql = "UPDATE {$this->table} SET data = %1, created_date = FROM_UNIXTIME(%2), expired_date = FROM_UNIXTIME(%3) WHERE {$this->where($key)}";
 125       $args = [
 126         1 => [$dataSerialized, 'String'],
 127         2 => [time(), 'Positive'],
 128         3 => [$expires, 'Positive'],
 129       ];
 130       $dao = CRM_Core_DAO::executeQuery($sql, $args, FALSE, NULL, FALSE, FALSE);
 131     }
 132     else {
 133       $sql = "INSERT INTO {$this->table} (group_name,path,data,created_date,expired_date) VALUES (%1,%2,%3,FROM_UNIXTIME(%4),FROM_UNIXTIME(%5))";
 134       $args = [
 135         1 => [$this->group, 'String'],
 136         2 => [$key, 'String'],
 137         3 => [$dataSerialized, 'String'],
 138         4 => [time(), 'Positive'],
 139         5 => [$expires, 'Positive'],
 140       ];
 141       $dao = CRM_Core_DAO::executeQuery($sql, $args, FALSE, NULL, FALSE, FALSE);
 142     }
 143
 144     $lock->release();
 145
 146     $this->valueCache[$key] = CRM_Core_BAO_Cache::decode($dataSerialized);
 147     $this->expiresCache[$key] = $expires;
 148     return TRUE;
 149   }
 150
 151   /**
 152    * @param string $key
 153    * @param mixed $default
 154    *
 155    * @return mixed
 156    */
 157   public function get($key, $default = NULL) {
 158     CRM_Utils_Cache::assertValidKey($key);
 159     if (!isset($this->expiresCache[$key]) || time() >= $this->expiresCache[$key]) {
 160       $sql = "SELECT path, data, UNIX_TIMESTAMP(expired_date) as expires FROM {$this->table} WHERE " . $this->where($key);
 161       $dao = CRM_Core_DAO::executeQuery($sql);
 162       while ($dao->fetch()) {
 163         $this->expiresCache[$key] = $dao->expires;
 164         $this->valueCache[$key] = CRM_Core_BAO_Cache::decode($dao->data);
 165       }
 166     }
 167     return (isset($this->expiresCache[$key]) && time() < $this->expiresCache[$key]) ? $this->reobjectify($this->valueCache[$key]) : $default;
 168   }
 169
 170   private function reobjectify($value) {
 171     return is_object($value) ? unserialize(serialize($value)) : $value;
 172   }
 173
 174   /**
 175    * @param $key
 176    * @param null $default
 177    *
 178    * @return mixed
 179    */
 180   public function getFromFrontCache($key, $default = NULL) {
 181     if (isset($this->expiresCache[$key]) && time() < $this->expiresCache[$key] && $this->valueCache[$key]) {
 182       return $this->reobjectify($this->valueCache[$key]);
 183     }
 184     else {
 185       return $default;
 186     }
 187   }
 188
 189   public function has($key) {
 190     $this->get($key);
 191     return isset($this->expiresCache[$key]) && time() < $this->expiresCache[$key];
 192   }
 193
 194   /**
 195    * @param string $key
 196    * @return bool
 197    */
 198   public function delete($key) {
 199     CRM_Utils_Cache::assertValidKey($key);
 200     // If we are triggering a deletion of a prevNextCache key in the civicrm_cache tabl
 201     // Alssure that the relevant prev_next_cache values are also removed.
 202     if ($this->group == CRM_Utils_Cache::cleanKey('CiviCRM Search PrevNextCache')) {
 203       Civi::service('prevnext')->deleteItem(NULL, $key);
 204     }
 205     CRM_Core_DAO::executeQuery("DELETE FROM {$this->table} WHERE {$this->where($key)}");
 206     unset($this->valueCache[$key]);
 207     unset($this->expiresCache[$key]);
 208     return TRUE;
 209   }
 210
 211   public function flush() {
 212     if ($this->group == CRM_Utils_Cache::cleanKey('CiviCRM Search PrevNextCache') &&
 213       Civi::service('prevnext') instanceof CRM_Core_PrevNextCache_Sql) {
 214       // Use the standard PrevNextCache cleanup function here not just delete from civicrm_cache
 215       CRM_Core_BAO_PrevNextCache::cleanupCache();
 216     }
 217     else {
 218       CRM_Core_DAO::executeQuery("DELETE FROM {$this->table} WHERE {$this->where()}");
 219     }
 220     $this->valueCache = [];
 221     $this->expiresCache = [];
 222     return TRUE;
 223   }
 224
 225   public function clear() {
 226     return $this->flush();
 227   }
 228
 229   public function prefetch() {
 230     $dao = CRM_Core_DAO::executeQuery("SELECT path, data, UNIX_TIMESTAMP(expired_date) AS expires FROM {$this->table} WHERE " . $this->where(NULL));
 231     $this->valueCache = [];
 232     $this->expiresCache = [];
 233     while ($dao->fetch()) {
 234       $this->valueCache[$dao->path] = CRM_Core_BAO_Cache::decode($dao->data);
 235       $this->expiresCache[$dao->path] = $dao->expires;
 236     }
 237   }
 238
 239   protected function where($path = NULL) {
 240     $clauses = [];
 241     $clauses[] = ('group_name = "' . CRM_Core_DAO::escapeString($this->group) . '"');
 242     if ($path) {
 243       $clauses[] = ('path = "' . CRM_Core_DAO::escapeString($path) . '"');
 244     }
 245     return $clauses ? implode(' AND ', $clauses) : '(1)';
 246   }
 247
 248 }