plugins/DiskCachePlugin.php

   1 <?php
   2 /**
   3  * StatusNet - the distributed open-source microblogging tool
   4  * Copyright (C) 2009, StatusNet, Inc.
   5  *
   6  * Plugin to implement cache interface with disk files
   7  *
   8  * PHP version 5
   9  *
  10  * This program is free software: you can redistribute it and/or modify
  11  * it under the terms of the GNU Affero General Public License as published by
  12  * the Free Software Foundation, either version 3 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU Affero General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU Affero General Public License
  21  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22  *
  23  * @category  Cache
  24  * @package   StatusNet
  25  * @author    Evan Prodromou <evan@status.net>
  26  * @copyright 2009 StatusNet, Inc.
  27  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  28  * @link      http://status.net/
  29  */
  30
  31 if (!defined('STATUSNET')) {
  32     // This check helps protect against security problems;
  33     // your code file can't be executed directly from the web.
  34     exit(1);
  35 }
  36
  37 /**
  38  * A plugin to cache data on local disk
  39  *
  40  * @category  Cache
  41  * @package   StatusNet
  42  * @author    Evan Prodromou <evan@status.net>
  43  * @copyright 2009 StatusNet, Inc.
  44  * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html GNU Affero General Public License version 3.0
  45  * @link      http://status.net/
  46  */
  47
  48 class DiskCachePlugin extends Plugin
  49 {
  50     var $root = '/tmp';
  51
  52     function keyToFilename($key)
  53     {
  54         return $this->root . '/' . str_replace(':', '/', $key);
  55     }
  56
  57     /**
  58      * Get a value associated with a key
  59      *
  60      * The value should have been set previously.
  61      *
  62      * @param string &$key   in; Lookup key
  63      * @param mixed  &$value out; value associated with key
  64      *
  65      * @return boolean hook success
  66      */
  67
  68     function onStartCacheGet(&$key, &$value)
  69     {
  70         $filename = $this->keyToFilename($key);
  71         if (file_exists($filename)) {
  72             $data = file_get_contents($filename);
  73             $value = unserialize($data);
  74         }
  75
  76         Event::handle('EndCacheGet', array($key, &$value));
  77         return false;
  78     }
  79
  80     /**
  81      * Associate a value with a key
  82      *
  83      * @param string  &$key     in; Key to use for lookups
  84      * @param mixed   &$value   in; Value to associate
  85      * @param integer &$flag    in; Flag (passed through to Memcache)
  86      * @param integer &$expiry  in; Expiry (passed through to Memcache)
  87      * @param boolean &$success out; Whether the set was successful
  88      *
  89      * @return boolean hook success
  90      */
  91
  92     function onStartCacheSet(&$key, &$value, &$flag, &$expiry, &$success)
  93     {
  94         $filename = $this->keyToFilename($key);
  95         $parent = dirname($filename);
  96
  97         $sofar = '';
  98
  99         foreach (explode('/', $parent) as $part) {
 100             if (empty($part)) {
 101                 continue;
 102             }
 103             $sofar .= '/' . $part;
 104             if (!is_dir($sofar)) {
 105                 $this->debug("Creating new directory '$sofar'");
 106                 $success = mkdir($sofar, 0750);
 107                 if (!$success) {
 108                     $this->log(LOG_ERR, "Can't create directory '$sofar'");
 109                     return false;
 110                 }
 111             }
 112         }
 113
 114         if (is_dir($filename)) {
 115             $success = false;
 116             return false;
 117         }
 118
 119         file_put_contents($filename, serialize($value));
 120
 121         Event::handle('EndCacheSet', array($key, $value, $flag,
 122                                            $expiry));
 123
 124         return false;
 125     }
 126
 127     /**
 128      * Delete a value associated with a key
 129      *
 130      * @param string  &$key     in; Key to lookup
 131      * @param boolean &$success out; whether it worked
 132      *
 133      * @return boolean hook success
 134      */
 135
 136     function onStartCacheDelete(&$key, &$success)
 137     {
 138         $filename = $this->keyToFilename($key);
 139
 140         if (file_exists($filename) && !is_dir($filename)) {
 141             unlink($filename);
 142         }
 143
 144         Event::handle('EndCacheDelete', array($key));
 145         return false;
 146     }
 147 }
 148