264 lines
8.6 KiB
PHP
264 lines
8.6 KiB
PHP
<?php
|
||
|
||
namespace mon\util;
|
||
|
||
use mon\util\exception\UploadException;
|
||
|
||
/**
|
||
* 大文件分片上传
|
||
*
|
||
* @author Mon <985558837@qq.com>
|
||
* @version 1.0.1 优化代码 2022-07-08
|
||
*/
|
||
class UploadSlice
|
||
{
|
||
/**
|
||
* 配置信息
|
||
*
|
||
* @var array
|
||
*/
|
||
protected $config = [
|
||
// 允许上传的文件后缀
|
||
'exts' => [],
|
||
// 分片文件大小限制
|
||
'sliceSize' => 0,
|
||
// 保存根路径
|
||
'rootPath' => '',
|
||
// 临时文件存储路径，基于rootPath
|
||
'tmpPath' => 'tmp'
|
||
];
|
||
|
||
/**
|
||
* 错误的分片序号
|
||
*
|
||
* @var array
|
||
*/
|
||
protected $error_chunk = [];
|
||
|
||
/**
|
||
* 构造方法
|
||
*
|
||
* @param array $config 自定义配置信息
|
||
*/
|
||
public function __construct(array $config = [])
|
||
{
|
||
$this->config = array_merge($this->config, $config);
|
||
}
|
||
|
||
/**
|
||
* 获取配置信息
|
||
*
|
||
* @return array
|
||
*/
|
||
public function getConfig()
|
||
{
|
||
return $this->config;
|
||
}
|
||
|
||
/**
|
||
* 设置配置信息
|
||
*
|
||
* @param array|string $config 配置信息或配置节点
|
||
* @param mixed $value 值
|
||
* @return UploadSlice
|
||
*/
|
||
public function setConifg($config, $value = null)
|
||
{
|
||
if (is_array($config)) {
|
||
$this->config = array_merge($this->config, $config);
|
||
} else {
|
||
$this->config[$config] = $value;
|
||
}
|
||
|
||
return $this;
|
||
}
|
||
|
||
/**
|
||
* 获取错误的分片序号
|
||
*
|
||
* @return array
|
||
*/
|
||
public function getErrorChunk()
|
||
{
|
||
return $this->error_chunk;
|
||
}
|
||
|
||
/**
|
||
* 保存上传的文件分片到临时文件目录
|
||
*
|
||
* @param string $fileID 文件唯一ID
|
||
* @param integer $chunk 文件分片序号，从0递增到N
|
||
* @param array $files 文件流，默认 $_FILES
|
||
* @param string $name 文件流索引，默认 file
|
||
* @throws UploadException
|
||
* @return false|array 文件保存路径
|
||
*/
|
||
public function upload($fileID, $chunk = 0, $name = 'file', $files = null)
|
||
{
|
||
if (is_null($files)) {
|
||
$files = $_FILES;
|
||
}
|
||
if (empty($files) || !isset($files[$name])) {
|
||
throw new UploadException('未上传文件', UploadException::ERROR_UPLOAD_FAILD);
|
||
}
|
||
// 检测上传保存路径
|
||
if (!$this->checkPath()) {
|
||
return false;
|
||
}
|
||
// 文件信息
|
||
$file = $files[$name];
|
||
// 校验文件
|
||
if (!$this->checkFile($file)) {
|
||
return false;
|
||
}
|
||
// 保存临时文件
|
||
$fileName = md5($fileID) . '_' . $chunk;
|
||
$tmpPath = $this->config['rootPath'] . DIRECTORY_SEPARATOR . $this->config['tmpPath'] . DIRECTORY_SEPARATOR . $fileID;
|
||
if (!File::instance()->createDir($tmpPath)) {
|
||
throw new UploadException('创建临时文件存储目录失败', UploadException::ERROR_UPLOAD_DIR_NOT_FOUND);
|
||
}
|
||
$savePath = $tmpPath . DIRECTORY_SEPARATOR . $fileName;
|
||
if (!move_uploaded_file($file['tmp_name'], $savePath)) {
|
||
throw new UploadException('临时文件保存失败', UploadException::ERROR_UPLOAD_SAVE_FAILD);
|
||
}
|
||
|
||
return ['savePath' => $savePath, 'saveDir' => $tmpPath, 'fileName' => $fileName];
|
||
}
|
||
|
||
/**
|
||
* 合并分片临时文件，生成上传文件
|
||
*
|
||
* @param string $fileID 文件唯一ID
|
||
* @param integer $chunkLength 文件分片长度
|
||
* @param string $fileName 保存文件名
|
||
* @param string $saveDir 基于 rootPath 路径下的多级目录存储路径
|
||
* @throws UploadException
|
||
* @return array 文件保存路径
|
||
*/
|
||
public function merge($fileID, $chunkLength, $fileName, $saveDir = '')
|
||
{
|
||
// 分片临时文件存储目录
|
||
$tmpPath = $this->config['rootPath'] . DIRECTORY_SEPARATOR . $this->config['tmpPath'] . DIRECTORY_SEPARATOR . $fileID;
|
||
if (!is_dir($tmpPath)) {
|
||
throw new UploadException('临时文件不存在', UploadException::ERROR_UPLOAD_DIR_NOT_FOUND);
|
||
}
|
||
// 验证文件名
|
||
$ext = File::instance()->getExt($fileName);
|
||
if (!empty($this->config['exts']) && !in_array($ext, $this->config['exts'])) {
|
||
throw new UploadException('不支持文件保存类型', UploadException::ERROR_UPLOAD_EXT_FAILD);
|
||
}
|
||
// 多级目录存储
|
||
$savePath = $this->config['rootPath'] . DIRECTORY_SEPARATOR . $saveDir;
|
||
if (!empty($saveDir) && !is_dir($savePath)) {
|
||
if (!File::instance()->createDir($savePath)) {
|
||
throw new UploadException('创建文件存储目录失败', UploadException::ERROR_UPLOAD_DIR_NOT_FOUND);
|
||
}
|
||
}
|
||
// 验证分片文件完整性
|
||
$this->error_chunk = [];
|
||
$chunkName = md5($fileID);
|
||
for ($i = 0; $i < $chunkLength; $i++) {
|
||
$checkName = $chunkName . '_' . $i;
|
||
$chunkPath = $tmpPath . DIRECTORY_SEPARATOR . $checkName;
|
||
if (!file_exists($chunkPath)) {
|
||
$this->error_chunk[] = $i;
|
||
throw new UploadException('分片文件不完整', UploadException::ERROR_CHUNK_FAILD);
|
||
}
|
||
}
|
||
// 合并文件
|
||
$saveFile = $savePath . DIRECTORY_SEPARATOR . $fileName;
|
||
// 打开保存文件句柄
|
||
$writerFp = fopen($saveFile, "ab");
|
||
for ($k = 0; $k < $chunkLength; $k++) {
|
||
$checkName = $chunkName . '_' . $k;
|
||
$chunkPath = $tmpPath . DIRECTORY_SEPARATOR . $checkName;
|
||
// 读取临时文件
|
||
$readerFp = fopen($chunkPath, "rb");
|
||
// 写入
|
||
fwrite($writerFp, fread($readerFp, filesize($chunkPath)));
|
||
// 关闭句柄
|
||
fclose($readerFp);
|
||
unset($readerFp);
|
||
// 删除临时文件
|
||
File::instance()->removeFile($chunkPath);
|
||
}
|
||
// 关闭保存文件句柄
|
||
fclose($writerFp);
|
||
// 删除临时目录
|
||
File::instance()->removeDir($tmpPath);
|
||
|
||
return ['savePath' => $saveFile, 'saveDir' => $savePath, 'fileName' => $fileName];
|
||
}
|
||
|
||
/**
|
||
* 校验文件
|
||
*
|
||
* @param array $file 文件信息
|
||
* @return boolean
|
||
*/
|
||
protected function checkFile($file)
|
||
{
|
||
if ($file['error']) {
|
||
throw new UploadException($this->uploadErrorMsg($file['error']), UploadException::ERROR_UPLOAD_CHECK_FAILD);
|
||
}
|
||
// 无效上传
|
||
if (empty($file['name'])) {
|
||
throw new UploadException('未知上传错误', UploadException::ERROR_UPLOAD_NOT_MESSAGE);
|
||
}
|
||
// 检查是否合法上传
|
||
if (!is_uploaded_file($file['tmp_name'])) {
|
||
throw new UploadException('非法上传文件', UploadException::ERROR_UPLOAD_ILLEGAL);
|
||
}
|
||
if ($file['size'] > $this->config['sliceSize'] && $this->config['sliceSize'] > 0) {
|
||
throw new UploadException('分片文件大小不符', UploadException::ERROR_UPLOAD_SIZE_FAILD);
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 检测上传根目录
|
||
*
|
||
* @throws UploadException
|
||
* @return boolean
|
||
*/
|
||
protected function checkPath()
|
||
{
|
||
$rootPath = $this->config['rootPath'];
|
||
if ((!is_dir($rootPath) && !File::instance()->createDir($rootPath)) || (is_dir($rootPath) && !is_writable($rootPath))) {
|
||
throw new UploadException('上传文件保存目录不可写入：' . $rootPath, UploadException::ERROR_UPLOAD_DIR_NOT_FOUND);
|
||
}
|
||
$tmpPath = $rootPath . DIRECTORY_SEPARATOR . $this->config['tmpPath'];
|
||
if ((!is_dir($tmpPath) && !File::instance()->createDir($tmpPath)) || (is_dir($tmpPath) && !is_writable($tmpPath))) {
|
||
throw new UploadException('上传文件临时保存目录不可写入：' . $tmpPath, UploadException::ERROR_UPLOAD_DIR_NOT_FOUND);
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 获取错误代码信息
|
||
*
|
||
* @param integer $errorNo 错误号
|
||
* @return string
|
||
*/
|
||
protected function uploadErrorMsg($errorNo)
|
||
{
|
||
switch ($errorNo) {
|
||
case 1:
|
||
return '上传的文件超过了 php.ini 中 upload_max_filesize 选项限制的值！';
|
||
case 2:
|
||
return '上传文件的大小超过了 HTML 表单中 MAX_FILE_SIZE 选项指定的值！';
|
||
case 3:
|
||
return '文件只有部分被上传！';
|
||
case 4:
|
||
return '没有文件被上传！';
|
||
case 6:
|
||
return '找不到临时文件夹！';
|
||
case 7:
|
||
return '文件写入失败！';
|
||
default:
|
||
return '未知上传错误！';
|
||
}
|
||
}
|
||
}
|