DumpFileOutput.php

Go to the documentation of this file.

<?php
use MediaWiki\Exception\MWException;
 
class DumpFileOutput extends DumpOutput {
    protected $handle = false;
    protected $filename;
 
    public function __construct( $file ) {
        $this->handle = fopen( $file, "wt" );
        $this->filename = $file;
    }
 
    public function writeCloseStream( $string ) {
        parent::writeCloseStream( $string );
        if ( $this->handle ) {
            fclose( $this->handle );
            $this->handle = false;
        }
    }
 
    public function write( $string ) {
        fputs( $this->handle, $string );
    }
 
    public function closeRenameAndReopen( $newname ) {
        $this->closeAndRename( $newname, true );
    }
 
    protected function renameOrException( $newname ) {
        if ( !rename( $this->filename, $newname ) ) {
            throw new RuntimeException( __METHOD__ . ": rename of file {$this->filename} to $newname failed\n" );
        }
    }
 
    protected function checkRenameArgCount( $newname ) {
        if ( is_array( $newname ) ) {
            if ( count( $newname ) > 1 ) {
                throw new MWException( __METHOD__ . ": passed multiple arguments for rename of single file\n" );
            }
            $newname = $newname[0];
        }
        return $newname;
    }
 
    public function closeAndRename( $newname, $open = false ) {
        $newname = $this->checkRenameArgCount( $newname );
        if ( $newname ) {
            if ( $this->handle ) {
                fclose( $this->handle );
                $this->handle = false;
            }
            $this->renameOrException( $newname );
            if ( $open ) {
                $this->handle = fopen( $this->filename, "wt" );
            }
        }
    }
 
    public function getFilenames() {
        return $this->filename;
    }
}