FSLockManager.php

Go to the documentation of this file.

<?php
use Wikimedia\AtEase\AtEase;
 
class FSLockManager extends LockManager {
    protected $lockTypeMap = [
        self::LOCK_SH => self::LOCK_SH,
        self::LOCK_UW => self::LOCK_SH,
        self::LOCK_EX => self::LOCK_EX
    ];
 
    protected $lockDir;
 
    protected $handles = [];
 
    protected $isWindows;
 
    public function __construct( array $config ) {
        parent::__construct( $config );
 
        $this->lockDir = $config['lockDirectory'];
        $this->isWindows = ( PHP_OS_FAMILY === 'Windows' );
    }
 
    protected function doLock( array $paths, $type ) {
        $status = StatusValue::newGood();
 
        $lockedPaths = []; // files locked in this attempt
        foreach ( $paths as $path ) {
            $status->merge( $this->doSingleLock( $path, $type ) );
            if ( $status->isOK() ) {
                $lockedPaths[] = $path;
            } else {
                // Abort and unlock everything
                $status->merge( $this->doUnlock( $lockedPaths, $type ) );
 
                return $status;
            }
        }
 
        return $status;
    }
 
    protected function doUnlock( array $paths, $type ) {
        $status = StatusValue::newGood();
 
        foreach ( $paths as $path ) {
            $status->merge( $this->doSingleUnlock( $path, $type ) );
        }
 
        return $status;
    }
 
    protected function doSingleLock( $path, $type ) {
        $status = StatusValue::newGood();
 
        if ( isset( $this->locksHeld[$path][$type] ) ) {
            ++$this->locksHeld[$path][$type];
        } elseif ( isset( $this->locksHeld[$path][self::LOCK_EX] ) ) {
            $this->locksHeld[$path][$type] = 1;
        } else {
            if ( isset( $this->handles[$path] ) ) {
                $handle = $this->handles[$path];
            } else {
                AtEase::suppressWarnings();
                $handle = fopen( $this->getLockPath( $path ), 'a+' );
                if ( !$handle && !is_dir( $this->lockDir ) ) {
                    // Create the lock directory in case it is missing
                    if ( mkdir( $this->lockDir, 0777, true ) ) {
                        $handle = fopen( $this->getLockPath( $path ), 'a+' ); // try again
                    } else {
                        $this->logger->error( "Cannot create directory '{$this->lockDir}'." );
                    }
                }
                AtEase::restoreWarnings();
            }
            if ( $handle ) {
                // Either a shared or exclusive lock
                $lock = ( $type == self::LOCK_SH ) ? LOCK_SH : LOCK_EX;
                if ( flock( $handle, $lock | LOCK_NB ) ) {
                    // Record this lock as active
                    $this->locksHeld[$path][$type] = 1;
                    $this->handles[$path] = $handle;
                } else {
                    fclose( $handle );
                    $status->fatal( 'lockmanager-fail-conflict' );
                }
            } else {
                $status->fatal( 'lockmanager-fail-openlock', $path );
            }
        }
 
        return $status;
    }
 
    protected function doSingleUnlock( $path, $type ) {
        $status = StatusValue::newGood();
 
        if ( !isset( $this->locksHeld[$path] ) ) {
            $status->warning( 'lockmanager-notlocked', $path );
        } elseif ( !isset( $this->locksHeld[$path][$type] ) ) {
            $status->warning( 'lockmanager-notlocked', $path );
        } else {
            $handlesToClose = [];
            --$this->locksHeld[$path][$type];
            if ( $this->locksHeld[$path][$type] <= 0 ) {
                unset( $this->locksHeld[$path][$type] );
            }
            if ( $this->locksHeld[$path] === [] ) {
                unset( $this->locksHeld[$path] ); // no locks on this path
                if ( isset( $this->handles[$path] ) ) {
                    $handlesToClose[] = $this->handles[$path];
                    unset( $this->handles[$path] );
                }
            }
            // Unlock handles to release locks and delete
            // any lock files that end up with no locks on them...
            if ( $this->isWindows ) {
                // Windows: for any process, including this one,
                // calling unlink() on a locked file will fail
                $status->merge( $this->closeLockHandles( $path, $handlesToClose ) );
                $status->merge( $this->pruneKeyLockFiles( $path ) );
            } else {
                // Unix: unlink() can be used on files currently open by this
                // process and we must do so in order to avoid race conditions
                $status->merge( $this->pruneKeyLockFiles( $path ) );
                $status->merge( $this->closeLockHandles( $path, $handlesToClose ) );
            }
        }
 
        return $status;
    }
 
    private function closeLockHandles( $path, array $handlesToClose ) {
        $status = StatusValue::newGood();
        foreach ( $handlesToClose as $handle ) {
            if ( !flock( $handle, LOCK_UN ) ) {
                $status->fatal( 'lockmanager-fail-releaselock', $path );
            }
            if ( !fclose( $handle ) ) {
                $status->warning( 'lockmanager-fail-closelock', $path );
            }
        }
 
        return $status;
    }
 
    private function pruneKeyLockFiles( $path ) {
        $status = StatusValue::newGood();
        if ( !isset( $this->locksHeld[$path] ) ) {
            # No locks are held for the lock file anymore
            if ( !unlink( $this->getLockPath( $path ) ) ) {
                $status->warning( 'lockmanager-fail-deletelock', $path );
            }
            unset( $this->handles[$path] );
        }
 
        return $status;
    }
 
    protected function getLockPath( $path ) {
        return "{$this->lockDir}/{$this->sha1Base36Absolute( $path )}.lock";
    }
 
    public function __destruct() {
        while ( count( $this->locksHeld ) ) {
            foreach ( $this->locksHeld as $path => $locks ) {
                $this->doSingleUnlock( $path, self::LOCK_EX );
                $this->doSingleUnlock( $path, self::LOCK_SH );
            }
        }
    }
}