From 3eceb3c7190a054e9275da1e40b84e8259234b8f Mon Sep 17 00:00:00 2001
From: lishuang <lish516@126.com>
Date: Sat, 22 Jul 2023 18:09:25 +0800
Subject: [PATCH] fix: fix the bug for file size.

---
 code/rest/matter_dao.go     | 20 ++++++++++++++++++++
 code/rest/matter_service.go | 20 ++++++++------------
 code/rest/space_dao.go      |  5 +++++
 3 files changed, 33 insertions(+), 12 deletions(-)

diff --git a/code/rest/matter_dao.go b/code/rest/matter_dao.go
index b1cc724..cbe32f0 100644
--- a/code/rest/matter_dao.go
+++ b/code/rest/matter_dao.go
@@ -484,6 +484,26 @@ func (this *MatterDao) SizeByPuuidAndUserUuid(matterUuid string, userUuid string
 	return sumSize
 }
 
+func (this *MatterDao) SizeByPuuidAndSpaceUuid(matterUuid string, spaceUuid string) int64 {
+
+	var wp = &builder.WherePair{Query: "puuid = ? AND space_uuid = ?", Args: []interface{}{matterUuid, spaceUuid}}
+
+	var count int64
+	db := core.CONTEXT.GetDB().Model(&Matter{}).Where(wp.Query, wp.Args...).Count(&count)
+	if count == 0 {
+		return 0
+	}
+
+	var sumSize int64
+	db = core.CONTEXT.GetDB().Model(&Matter{}).Where(wp.Query, wp.Args...).Select("SUM(size)")
+	this.PanicError(db.Error)
+	row := db.Row()
+	err := row.Scan(&sumSize)
+	core.PanicError(err)
+
+	return sumSize
+}
+
 // delete a file from db and disk.
 func (this *MatterDao) Delete(matter *Matter) {
 
diff --git a/code/rest/matter_service.go b/code/rest/matter_service.go
index 5670950..82c146c 100644
--- a/code/rest/matter_service.go
+++ b/code/rest/matter_service.go
@@ -564,10 +564,9 @@ func (this *MatterService) ComputeRouteSize(matterUuid string, user *User, space
 	//if to root directory, then update to user's info.
 	if matterUuid == MATTER_ROOT {
 
-		size := this.matterDao.SizeByPuuidAndUserUuid(MATTER_ROOT, user.Uuid)
+		size := this.matterDao.SizeByPuuidAndSpaceUuid(MATTER_ROOT, space.Uuid)
 
-		db := core.CONTEXT.GetDB().Model(&Space{}).Where("uuid = ?", space.Uuid).Update("total_size", size)
-		this.PanicError(db.Error)
+		this.spaceDao.UpdateTotalSize(space.Uuid, size)
 
 		//update user total size info in cache.
 		space.TotalSize = size
@@ -580,12 +579,11 @@ func (this *MatterService) ComputeRouteSize(matterUuid string, user *User, space
 	//only compute dir
 	if matter.Dir {
 		//compute the total size.
-		size := this.matterDao.SizeByPuuidAndUserUuid(matterUuid, user.Uuid)
+		size := this.matterDao.SizeByPuuidAndSpaceUuid(matterUuid, space.Uuid)
 
 		//when changed, we update
 		if matter.Size != size {
-			db := core.CONTEXT.GetDB().Model(&Matter{}).Where("uuid = ?", matterUuid).Update("size", size)
-			this.PanicError(db.Error)
+			this.spaceDao.UpdateTotalSize(space.Uuid, size)
 		}
 
 	}
@@ -617,22 +615,20 @@ func (this *MatterService) ComputeDirSize(dirMatter *Matter, user *User, space *
 	//if to root directory, then update to user's info.
 	if dirMatter.Uuid == MATTER_ROOT {
 
-		size := this.matterDao.SizeByPuuidAndUserUuid(MATTER_ROOT, user.Uuid)
+		size := this.matterDao.SizeByPuuidAndSpaceUuid(MATTER_ROOT, space.Uuid)
 
-		db := core.CONTEXT.GetDB().Model(&User{}).Where("uuid = ?", user.Uuid).Update("total_size", size)
-		this.PanicError(db.Error)
+		this.spaceDao.UpdateTotalSize(space.Uuid, size)
 
 		//update user total size info in cache.
 		space.TotalSize = size
 	} else {
 
 		//compute self.
-		size := this.matterDao.SizeByPuuidAndUserUuid(dirMatter.Uuid, user.Uuid)
+		size := this.matterDao.SizeByPuuidAndSpaceUuid(dirMatter.Uuid, user.Uuid)
 
 		//when changed, we update
 		if dirMatter.Size != size {
-			db := core.CONTEXT.GetDB().Model(&Matter{}).Where("uuid = ?", dirMatter.Uuid).Update("size", size)
-			this.PanicError(db.Error)
+			this.spaceDao.UpdateTotalSize(space.Uuid, size)
 		}
 	}
 
diff --git a/code/rest/space_dao.go b/code/rest/space_dao.go
index 7d33269..efa5109 100644
--- a/code/rest/space_dao.go
+++ b/code/rest/space_dao.go
@@ -132,6 +132,11 @@ func (this *SpaceDao) Save(space *Space) *Space {
 	return space
 }
 
+func (this *SpaceDao) UpdateTotalSize(spaceUuid string, totalSize int64) {
+	db := core.CONTEXT.GetDB().Model(&Space{}).Where("uuid = ?", spaceUuid).Update("total_size", totalSize)
+	this.PanicError(db.Error)
+}
+
 func (this *SpaceDao) Delete(space *Space) {
 
 	db := core.CONTEXT.GetDB().Delete(&space)