includes/utils/CdbDBA.php

   1 <?php
   2 /**
   3  * DBA-based CDB reader/writer
   4  *
   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.
   9  *
  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.
  14  *
  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
  19  *
  20  * @file
  21  */
  22
  23 /**
  24  * Reader class which uses the DBA extension
  25  */
  26 class CdbReaderDBA extends CdbReader {
  27         public function __construct( $fileName ) {
  28                 $this->handle = dba_open( $fileName, 'r-', 'cdb' );
  29                 if ( !$this->handle ) {
  30                         throw new CdbException( 'Unable to open CDB file "' . $fileName . '"' );
  31                 }
  32         }
  33
  34         public function close() {
  35                 if ( isset( $this->handle ) ) {
  36                         dba_close( $this->handle );
  37                 }
  38                 unset( $this->handle );
  39         }
  40
  41         public function get( $key ) {
  42                 return dba_fetch( $key, $this->handle );
  43         }
  44 }
  45
  46 /**
  47  * Writer class which uses the DBA extension
  48  */
  49 class CdbWriterDBA extends CdbWriter {
  50         public function __construct( $fileName ) {
  51                 $this->realFileName = $fileName;
  52                 $this->tmpFileName = $fileName . '.tmp.' . mt_rand( 0, 0x7fffffff );
  53                 $this->handle = dba_open( $this->tmpFileName, 'n', 'cdb_make' );
  54                 if ( !$this->handle ) {
  55                         throw new CdbException( 'Unable to open CDB file for write "' . $fileName . '"' );
  56                 }
  57         }
  58
  59         public function set( $key, $value ) {
  60                 return dba_insert( $key, $value, $this->handle );
  61         }
  62
  63         public function close() {
  64                 if ( isset( $this->handle ) ) {
  65                         dba_close( $this->handle );
  66                 }
  67                 if ( $this->isWindows() ) {
  68                         unlink( $this->realFileName );
  69                 }
  70                 if ( !rename( $this->tmpFileName, $this->realFileName ) ) {
  71                         throw new CdbException( 'Unable to move the new CDB file into place.' );
  72                 }
  73                 unset( $this->handle );
  74         }
  75 }