mirror/php-phpbb
1
0
Fork 0
mirror of https://github.com/phpbb/phpbb.git synced 2025-07-30 21:40:43 +02:00
Code Issues Releases Wiki Activity

[ticket/15342] Track storage files

Browse Source 
PHPBB3-15342
This commit is contained in:
Rubén Calvo
2018-02-11 11:33:04 +01:00
parent cd84919f1a
commit a02fa1170b
6 changed files with 199 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
phpBB/phpbb/db/migration/data/v330/storage_track.php Normal file
View File

@@ -0,0 +1,43 @@
<?php
/**
*
* This file is part of the phpBB Forum Software package.
*
* @copyright (c) phpBB Limited <https://www.phpbb.com>
* @license GNU General Public License, version 2 (GPL-2.0)
*
* For full copyright and license information, please see
* the docs/CREDITS.txt file.
*
*/
namespace phpbb\db\migration\data\v330;
class storage_track extends \phpbb\db\migration\migration
{
public function update_schema()
{
return array(
'add_tables' => array(
$this->table_prefix . 'storage' => array(
'COLUMNS' => array(
'file_id' => array('UINT', null, 'auto_increment'),
'file_path' => array('VCHAR', ''),
'storage' => array('VCHAR', ''),
'filesize' => array('UINT:20', 0),
),
'PRIMARY_KEY' => 'file_id',
),
),
);
}
public function revert_schema()
{
return array(
'drop_tables' => array(
$this->table_prefix . 'storage',
),
);
}
}

156
phpBB/phpbb/storage/storage.php
View File

@@ -13,16 +13,28 @@
namespace phpbb\storage;
use phpbb\db\driver\driver_interface;
/**
* @internal Experimental
*/
class storage
{
/**
* @var \phpbb\db\driver\driver_interface
*/
protected $db;
/**
* @var string
*/
protected $storage_name;
/**
* @var string
*/
protected $storage_table;
/**
* @var \phpbb\storage\adapter_factory
*/
@@ -36,13 +48,16 @@ class storage
/**
* Constructor
*
* @param \phpbb\db\driver\driver_interface $db
* @param \phpbb\storage\adapter_factory $factory
* @param string $storage_name
*/
public function __construct(adapter_factory $factory, $storage_name)
public function __construct(driver_interface $db, adapter_factory $factory, $storage_name, $storage_table)
{
$this->db = $db;
$this->factory = $factory;
$this->storage_name = $storage_name;
$this->storage_table = $storage_table;
}
/**
@@ -81,7 +96,23 @@ class storage
*/
public function put_contents($path, $content)
{
$this->get_adapter()->put_contents($path, $content);
try
{
$this->get_adapter()->put_contents($path, $content);
$sql_ary = array(
'file_path' => $path,
'storage' => $this->get_name(),
'filesize' => strlen($content),
);
$sql = 'INSERT INTO ' . $this->storage_table . $this->db->sql_build_array('INSERT', $sql_ary);
$this->db->sql_query($sql);
}
catch (\Exception $e)
{
throw $e;
}
}
/**
@@ -121,7 +152,23 @@ class storage
*/
public function delete($path)
{
$this->get_adapter()->delete($path);
try
{
$this->get_adapter()->delete($path);
$sql_ary = array(
'file_path' => $path,
'storage' => $this->get_name(),
);
$sql = 'DELETE FROM ' . $this->storage_table . '
WHERE ' . $this->db->sql_build_array('DELETE', $sql_ary);
$this->db->sql_query($sql);
}
catch (\Exception $e)
{
throw $e;
}
}
/**
@@ -135,7 +182,28 @@ class storage
*/
public function rename($path_orig, $path_dest)
{
$this->get_adapter()->rename($path_orig, $path_dest);
try
{
$this->get_adapter()->rename($path_orig, $path_dest);
$sql_ary1 = array(
'file_path' => $path_dest,
);
$sql_ary2 = array(
'file_path' => $path_orig,
'storage' => $this->get_name(),
);
$sql = 'UPDATE ' . $this->storage_table . '
SET ' . $this->db->sql_build_array('UPDATE', $sql_ary1) . '
WHERE ' . $this->db->sql_build_array('SELECT', $sql_ary2);
$this->db->sql_query($sql);
}
catch (\Exception $e)
{
throw $e;
}
}
/**
@@ -149,7 +217,34 @@ class storage
*/
public function copy($path_orig, $path_dest)
{
$this->get_adapter()->copy($path_orig, $path_dest);
try
{
$this->get_adapter()->copy($path_orig, $path_dest);
$sql_ary = array(
'file_path' => $path_orig,
'storage' => $this->get_name(),
);
$sql = 'SELECT filesize FROM ' . $this->storage_table . '
WHERE ' . $this->db->sql_build_array('SELECT', $sql_ary);
$result = $this->db->sql_query($sql);
$row = $this->db->sql_fetchrow($result);
$this->db->sql_freeresult($result);
$sql_ary = array(
'file_path' => $path_dest,
'storage' => $this->get_name(),
'filesize' => (int) $row['filesize'],
);
$sql = 'INSERT INTO ' . $this->storage_table . $this->db->sql_build_array('INSERT', $sql_ary);
$this->db->sql_query($sql);
}
catch (\Exception $e)
{
throw $e;
}
}
/**
@@ -158,7 +253,7 @@ class storage
* @param string $path File to read
*
* @throws \phpbb\storage\exception\exception When unable to open file
*
* @return resource Returns a file pointer
*/
public function read_stream($path)
@@ -186,7 +281,8 @@ class storage
*
* @param string $path The target file
* @param resource $resource The resource
* When target file cannot be created
*
* @throws \phpbb\storage\exception\exception When target file cannot be created
*/
public function write_stream($path, $resource)
{
@@ -195,6 +291,33 @@ class storage
if ($adapter instanceof stream_interface)
{
$adapter->write_stream($path, $resource);
$sql_ary = array(
'file_path' => $path,
'storage' => $this->get_name(),
);
// Get file, if exist update filesize, if not add new record
$sql = 'SELECT * FROM ' . $this->storage_table . '
WHERE ' . $this->db->sql_build_array('SELECT', $sql_ary);
$result = $this->db->sql_query($sql);
$row = $this->db->sql_fetchrow($result);
$this->db->sql_freeresult($result);
if ($row)
{
$sql = 'UPDATE ' . $this->storage_table . '
SET filesize = filesize + ' . strlen($content) . '
WHERE ' . $this->db->sql_build_array('SELECT', $sql_ary);
$this->db->sql_query($sql);
}
else
{
$sql_ary['filesize'] = strlen($content);
$sql = 'INSERT INTO ' . $this->storage_table . $this->db->sql_build_array('INSERT', $sql_ary);
$this->db->sql_query($sql);
}
}
else
{
@@ -229,4 +352,23 @@ class storage
{
return $this->get_adapter()->get_link($path);
}
/**
* Get total storage size.
*
* @param string $path The file
*
* @return int Size in bytes
*/
public function get_size()
{
$sql = 'SELECT SUM(filesize) AS total
FROM ' . $this->storage_table . '
WHERE storage = ' . $this->get_name();
$result = $this->db->sql_query($sql);
$row = $this->db->sql_fetchrow($result);
$this->db->sql_freeresult($result);
return $row['total'];
}
}