[实战]MVC5+EF6+MySql企业网盘实战(14)——逻辑重构

时间:2021-11-09 06:16:32

写在前面

上篇文章关于修改文件夹和文件名称导致的找不到物理文件的问题,这篇文章将对其进行逻辑的修改。

系列文章

[EF]vs15+ef6+mysql code first方式

[实战]MVC5+EF6+MySql企业网盘实战(1)

[实战]MVC5+EF6+MySql企业网盘实战(2)——用户注册

[实战]MVC5+EF6+MySql企业网盘实战(3)——验证码

[实战]MVC5+EF6+MySql企业网盘实战(4)——上传头像

[Bootstrap]modal弹出框

[实战]MVC5+EF6+MySql企业网盘实战(5)——登录界面,头像等比例压缩

[实战]MVC5+EF6+MySql企业网盘实战(5)——页面模板

[实战]MVC5+EF6+MySql企业网盘实战(5)——ajax方式注册

[实战]MVC5+EF6+MySql企业网盘实战(6)——ajax方式登录

[实战]MVC5+EF6+MySql企业网盘实战(7)——文件上传

[实战]MVC5+EF6+MySql企业网盘实战(8)——文件下载、删除

[实战]MVC5+EF6+MySql企业网盘实战(9)——编辑文件名

[实战]MVC5+EF6+MySql企业网盘实战(10)——新建文件夹

[实战]MVC5+EF6+MySql企业网盘实战(11)——新建文件夹2

[实战]MVC5+EF6+MySql企业网盘实战(12)——新建文件夹和上传文件

[实战]MVC5+EF6+MySql企业网盘实战(13)——编辑文件夹

[实战]MVC5+EF6+MySql企业网盘实战(14)——逻辑重构

上传文件

将上传的文件保存在网站的根目录下的NetDisk文件夹中。根据文件的md5值判断文件是否已经存在了,如果存在则不再保存文件,只是在数据库中插入一条新的文件信息记录,用于区别该文件所在路径以及名称。

        [HttpPost]
public void UploadFile(string filePath)
{
UserInfo userInfo
= Session["user"] as UserInfo;
//保证路径格式一直以/结束
filePath = filePath.TrimEnd('/') + "/";
if (userInfo == null)
{
RedirectToAction(
"Login", "UserInfo");
}
filePath
= filePath == "/" ? "/NetDisk/" : filePath;
var files = Request.Files;
MyFile myFile
= null;
if (files.Count > 0)
{
var file = files[0];
string fileName = file.FileName;
Stream inputStream
= file.InputStream;
string fileSaveFolder = Server.MapPath("~/NetDisk");
//如果目标不存在,则创建
if (!Directory.Exists(fileSaveFolder))
{
Directory.CreateDirectory(fileSaveFolder);
}
byte[] buffer = new byte[inputStream.Length];
//判断是否已经超出个人网盘大小
var myFiles = _myFileServiceRepository.FindAll(x => x.User.Id == userInfo.Id && x.IsDelete == false);
int myDiskSize = 0;
if (myFiles.Count() > 0)
{
myDiskSize
= myFiles.Sum(x => x.FileSize);
}
//如果已经超出网盘大小,则给出提示
if (myDiskSize + buffer.Length > userInfo.NetDiskSize)
{
AlertMsg(
"对不起,您的网盘空间不足,请清理后再次上传,或联系管理员进行扩容。", "");
return;
}
inputStream.Read(buffer,
0, buffer.Length);
string strFileMd5 = MD5Helper.GetMD5FromFile(buffer);
string fileSavePath = Path.Combine(fileSaveFolder, filePath);
//名称格式一律以md5命名。
string strNewName = strFileMd5 + Path.GetExtension(file.FileName);
fileSavePath
= Path.Combine(fileSaveFolder, strNewName);
//如果文件不存在,则保存物理文件。
if (!System.IO.File.Exists(fileSavePath))
{
file.SaveAs(fileSavePath);
}
//判断在数据库中同一路径下,md5相同,名称相同的文件是否存在,不存在则直接插入,否则对文件进行自动重命名保存。
var fileIsExist = _myFileServiceRepository.Find(x => x.FileMd5 == strFileMd5 && x.User.Id == userInfo.Id && x.FileName == fileName);
//如果找到的文件对象不为空,并且是删除状态,则修改删除状态,不再新增文件信息,否则添加。
//查询导航属性
var currentUser = _userInfoServiceRepository.Find(x => x.Id == userInfo.Id);
if (fileIsExist != null)
{
myFile
= fileIsExist;
//文件被逻辑删除了,则更新状态,否则重命名后新增文件。
if (fileIsExist.IsDelete)
{
fileIsExist.IsDelete
= false;
_myFileServiceRepository.Update(fileIsExist);
}
else
{
//查找该用户下类似文件名的文件并且没有被逻辑删除的个数
var count = _myFileServiceRepository.FindAll(x => x.User.Id == userInfo.Id && x.IsDelete == false && x.FileName.Contains(fileName)).Count();
fileName
= count > 0 ? Path.GetFileNameWithoutExtension(fileName) + "(" + (count + 1).ToString() + ")" + Path.GetExtension(fileName) : fileName;
myFile
= new MyFile()
{

FileMd5
= strFileMd5,
ModifyDt
= DateTime.Now,
IsDelete
= false,
FileSize
= buffer.Length,
FolderPath
= filePath,
FileExt
= Path.GetExtension(fileSavePath),
CreateDt
= DateTime.Now,
FileName
= fileName,
FileIcon
= GetFileIcon(Path.GetExtension(fileSavePath)),
User
= currentUser,
IsFolder
= 0
};
//保存数据库
_myFileServiceRepository.Add(myFile);
}

}
else
{

//新增文件信息
myFile = new MyFile()
{
FileMd5
= strFileMd5,
ModifyDt
= DateTime.Now,
IsDelete
= false,
FileSize
= buffer.Length,
FolderPath
= filePath,
FileExt
= Path.GetExtension(fileSavePath),
CreateDt
= DateTime.Now,
FileName
= fileName,
FileIcon
= GetFileIcon(Path.GetExtension(fileSavePath)),
User
= currentUser,
IsFolder
= 0
};
//保存数据库
_myFileServiceRepository.Add(myFile);
}
_myFileServiceRepository.SaveChanges();
string json = new JavaScriptSerializer().Serialize(myFile);
AlertMsg(
"上传成功", json);
}
}

[实战]MVC5+EF6+MySql企业网盘实战(14)——逻辑重构

总结

这里对上传文件的逻辑进行了一下修改。