public/docs
1
0
Fork 0
mirror of https://github.com/sismics/docs.git synced 2025-12-26 08:01:45 +00:00
Code Issues Releases Wiki Activity

Closes #308: update user quota when deleting a document

Browse Source
This commit is contained in:
Benjamin Gamard
2019-05-02 11:13:16 +02:00
parent 2235a0498b
commit d5e73ecd8b
2 changed files with 60 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
docs-web/src/main/java/com/sismics/docs/rest/resource/DocumentResource.java
View File

@@ -960,15 +960,30 @@ public class DocumentResource extends BaseResource {
// Delete the document
documentDao.delete(id, principal.getId());
// Raise file deleted events (don't bother sending document updated event)
long totalSize = 0L;
for (File file : fileList) {
// Store the file size to update the quota
java.nio.file.Path storedFile = DirectoryUtil.getStorageDirectory().resolve(file.getId());
try {
totalSize += Files.size(storedFile);
} catch (IOException e) {
// The file doesn't exists on disk, which is weird, but not fatal
}
// Raise file deleted event
FileDeletedAsyncEvent fileDeletedAsyncEvent = new FileDeletedAsyncEvent();
fileDeletedAsyncEvent.setUserId(principal.getId());
fileDeletedAsyncEvent.setFile(file);
ThreadLocalContext.get().addAsyncEvent(fileDeletedAsyncEvent);
}
// Update the user quota
UserDao userDao = new UserDao();
User user = userDao.getById(principal.getId());
user.setStorageCurrent(user.getStorageCurrent() - totalSize);
userDao.updateQuota(user);
// Raise a document deleted event
DocumentDeletedAsyncEvent documentDeletedAsyncEvent = new DocumentDeletedAsyncEvent();
documentDeletedAsyncEvent.setUserId(principal.getId());