Fix download stats (#1256)

* fix download stats * fix typo
0chain · Sep 19, 2023 · c494c6f · c494c6f
1 parent d10e14b
commit c494c6f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/code/go/0chain.net/blobbercore/handler/object_operation_handler.go b/code/go/0chain.net/blobbercore/handler/object_operation_handler.go
@@ -425,7 +425,7 @@ func (fsh *StorageHandler) DownloadFile(ctx context.Context, r *http.Request) (i
 	}
 
 	fileDownloadResponse.Data = chunkData
-	reference.FileBlockDownloaded(ctx, fileref.ID, dr.NumBlocks)
+	reference.FileBlockDownloaded(ctx, fileref, dr.NumBlocks)
 	addDailyBlocks(clientID, dr.NumBlocks)
 	return fileDownloadResponse, nil
 }

diff --git a/code/go/0chain.net/blobbercore/reference/filestats.go b/code/go/0chain.net/blobbercore/reference/filestats.go
@@ -86,9 +86,9 @@ func FileUpdated(ctx context.Context, refID, newRefID int64) {
 	// }
 }
 
-func FileBlockDownloaded(ctx context.Context, refID, blocks int64) {
+func FileBlockDownloaded(ctx context.Context, ref *Ref, blocks int64) {
 	db := datastore.GetStore().GetTransaction(ctx)
-	db.Model(&Ref{}).Where("id=?", refID).Update("num_of_block_downloads", gorm.Expr("num_of_block_downloads + ?", blocks))
+	db.Model(ref).Update("num_of_block_downloads", gorm.Expr("num_of_block_downloads + ?", blocks))
 }
 
 func GetFileStats(ctx context.Context, ref *Ref) (*FileStats, error) {