snap2/core/SnapPathCache.php5

<?php

class SnapPathCache extends SObject {
    public static function addPathSet($paths, $objId, $objType) {
        if(!SnapDBI::startTransaction())
            return false;

        $dirId = 'NULL';
        $resId = 'NULL';
        $versId = 'NULL';
        if($objType == 'Directory') {
            $dirId = $objId;
            $type = 'DIR';
        }
        else if($objType == 'Resource') {
            $resId = $objId;
            $type = 'RES';
        }
        else {
            $versId = $objId;
            $type = 'VERS';
        }

        $query = 'INSERT INTO PathCache (path, directoryId, resourceId, versionId, type) VALUES ';
        foreach($paths as $path)
            $query .= "('$path', $dirId, $resId, $versId, '$type'), ";
        $query = substr($query, 0, -2);

        if(!SnapDBI::query($query)) {
            SnapDBI::cancelTransaction();
            return false;
        }

        SnapDBI::commitTransaction();

        return true;
    }

    public static function removePathSet($paths) {
        if(!SnapDBI::startTransaction())
            return false;

        foreach($paths as $path) {
            $query = 'DELETE FROM PathCache WHERE path LIKE BINARY \'' . $path . '%\'';
            if(!SnapDBI::query($query)) {
                SnapDBI::cancelTransaction();
                return false;
            }
        }

        SnapDBI::commitTransaction();

        return true;
    }

    public static function retrievePath($type, $id) {
        $query = 'SELECT path FROM PathCache WHERE ' . strtolower($type) . 'Id = ' . $id;
        $results = SnapDBI::query($query, true);
        if(!$results)
            return false;
        if(count($results) == 1)
            return $results[0]['path'];
        else
            return false;
    }

    // one newPath for each oldPath
    public static function updatePathSet($newPaths, $oldPaths) {
        if(!SnapDBI::startTransaction())
            return false;

        foreach($newPaths as $i => $newPath) {
            $oldPath = $oldPaths[$i];
            $query = 'UPDATE PathCache SET path = CONCAT(\'' . $newPath . '\', SUBSTR(path, ' . (strlen($oldPath) + 1)
                        . ')) WHERE path LIKE BINARY \'' . $oldPath . '%\'';
            if(!SnapDBI::query($query)) {
                SnapDBI::cancelTransaction();
                return false;
            }
        }

        SnapDBI::commitTransaction();

        return true;
    }

    public static function regenPathCache() {
        // XXX: hopefully, this will be enough
        set_time_limit(450);

        SnapDBI::startTransaction();

        if(!SnapDBI::query('TRUNCATE PathCache')) {
            SnapDBI::cancelTransaction();
            return false;
        }

        $query = 'INSERT INTO PathCache (path, directoryId, type) VALUES (\'//\', 1, \'DIR\'), (\'/DELETED/\', 2, \'DIR\')';
        if(!SnapDBI::query($query)) {
            SnapDBI::cancelTransaction();
            return false;
        }

        if(!self::recRegenDirectory('//', 1)) {
            SnapDBI::cancelTransaction();
            return false;
        }

        if(!self::recRegenDirectory('/DELETED/', 2)) {
            SnapDBI::cancelTransaction();
            return false;
        }

        SnapDBI::commitTransaction();

        return true;
    }

    private static function recRegenDirectory($pathBase, $dirId) {
        $query = 'SELECT childId, shortName FROM DirectoryLink WHERE parentId = ' . $dirId . ' AND childId <> parentId';
        $results = SnapDBI::query($query, true);
        if($results === false)
            return false;

        if(count($results) > 0) {
            $query = 'INSERT INTO PathCache (path, directoryId, type) VALUES ';
            foreach($results as $row) {
                $query .= "('$pathBase$row[shortName]/',$row[childId],'DIR'),";
                if(!self::recRegenDirectory($pathBase . $row['shortName'] . '/', $row['childId']))
                    return false;
            }
            $query = substr($query, 0, -1);
            if(!SnapDBI::query($query))
                return false;
        }

        $query = 'SELECT childId, shortName FROM ResourceLink WHERE parentId = ' . $dirId;
        $results = SnapDBI::query($query, true);
        if($results === false)
            return false;

        if(count($results) > 0) {
            $query = 'INSERT INTO PathCache (path, resourceId, type) VALUES ';
            foreach($results as $row) {
                $query .= "('$pathBase$row[shortName]',$row[childId],'RES'),";
                if(!self::recRegenResource($pathBase . $row['shortName'], $row['childId']))
                    return false;
            }
            $query = substr($query, 0, -1);
            if(!SnapDBI::query($query))
                return false;
        }

        return true;
    }

    private static function recRegenResource($pathBase, $resId) {
        $query = 'SELECT id, ordinal FROM Version WHERE resourceId = ' . $resId;
        $results = SnapDBI::query($query, true);
        if($results === false)
            return false;

        if(count($results) > 0) {
            $query = 'INSERT INTO PathCache (path, versionId, type) VALUES ';
            foreach($results as $row)
                $query .= "('$pathBase@$row[ordinal]',$row[id],'VERS'),";
            $query = substr($query, 0, -1);
            if(!SnapDBI::query($query))
                return false;
        }

        return true;
    }
}

?>

---