首先,我正在使用带有 PDO 扩展的 PHP 5.4.11 和 MySQL 5.1.66(在 Debian Squeeze 上)开发共享托管服务。
目前我正在开发一项服务,用户在数据库中存储数据的配额有限。目前,只有一张表存储了用户数据,必须遵守有关配额的规定(但这可能会改变)。所有表都使用 InnoDB 存储引擎和文本列的 utf8_unicode_ci 排序规则。假设与配额相关的表具有以下列:
+--------------+-----------+
| Column name | Type |
+--------------+-----------+
| id | int |
| userId | int |
| created | timestamp |
| lastModified | timestamp |
| description | varchar |
| content | text |
+--------------+-----------+
现在我需要计算属于特定用户的所有行的字节大小。我搜索了文档并四处搜索,但只发现其他人提出了类似的问题而没有得到令人满意的答案。
我知道 MySQLLENGTH()
函数,但由于它是一个字符串函数,它不会返回(固定长度)数字和日期/时间字段占用的空间。如果只考虑字符串字段,用户可能只是用空字符串填满数据库,永远不会达到他的配额。我也知道 MySQL 中每行的描述都有一些开销,但我不想将其包含在计算中。(作为等价物,我想计算实际文件大小,而不是磁盘上的文件大小。)
此外,我不想依赖特定的表结构,因为这可能会改变,并且必须记住更新计算配额的函数。
由于缺乏现有的解决方案,我想出了自己的(见下文)。但它有一些缺点,例如:
- 它需要表中使用的数据类型及其各自大小的列表。
- 它无法准确处理、 和
FLOAT(p)
数据DECIMAL(M,D)
类型(尽管可以实现)。NUMERIC(M,D)
BIT(M)
- 它需要两个单独的查询。
所以现在,这就是我想出的:
$db = new PDO(...);
$tablename = 'users';
$userId = 1;
// Make a list of type sizes in bytes - null indicates string types of
// varying size. If there is a type used in the database which is not
// listed here, an exception will be thrown.
$typeSizes = array(
'int' => 4,
'timestamp' => 4,
'varchar' => null,
'text' => null
);
// Get datatypes used in the table.
$sql = 'SELECT COLUMN_NAME, DATA_TYPE FROM INFORMATION_SCHEMA.COLUMNS '
. 'WHERE TABLE_NAME=?';
$stmt = $db->prepare($sql);
$stmt->bindValue(1, $tablename);
$stmt->execute();
$colTypes = array_map('reset', array_map('reset',
$stmt->fetchAll(PDO::FETCH_GROUP|PDO::FETCH_ASSOC)));
// Iterate over the existing columns. Sum up sizes of fixed size columns to
// get a 'fixed-size-factor' for a row. Make a list of varying size columns.
$fixedSizeFactor = 0;
$varyingSizeCols = array();
foreach ($colTypes as $colName => $colType) {
if (array_key_exists($colType, $typeSizes)) {
if ($typeSizes[$colType] !== null) {
$fixedSizeFactor += $typeSizes[$colType];
} else {
$varyingSizeCols[] = $colName;
}
} else {
$msg = "Unhandled column type '$colType' - unable to calculate used "
. 'storage. Probably the $typeSizes array needs to be updated.';
throw new Exception($msg);
}
}
// Get number of all records of the user and the size of his data in
// varying size columns.
$sumArgument = 0;
if (count($varyingSizeCols) > 0) {
$sumArgument = 'LENGTH(' . implode(') + LENGTH(', $varyingSizeCols) . ')';
}
$sql = 'SELECT SUM(' . $sumArgument . ') AS size, COUNT(*) AS count FROM '
. $tablename . ' WHERE userId=?';
$stmt = $db->prepare($sql);
$stmt->bindValue(1, $userId);
$stmt->execute();
$result = $stmt->fetch(PDO::FETCH_ASSOC);
// Calculate used storage.
$usedStorage = $result['count'] * $fixedSizeFactor + $result['size'];
我的问题是:有没有更“原生”、不那么骇人听闻的方式来做到这一点?如果没有,您对性能优化有什么建议吗?