From 1c25232b06a16392099a491e595f22dfb58ad485 Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Sat, 29 Oct 2022 11:08:16 +0800
Subject: [PATCH] feat(source link): record downloads for redirected source
 link

---
 middleware/file.go    | 1 +
 models/source_link.go | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/middleware/file.go b/middleware/file.go
index 69165a8..995637e 100644
--- a/middleware/file.go
+++ b/middleware/file.go
@@ -23,6 +23,7 @@ func ValidateSourceLink() gin.HandlerFunc {
 			return
 		}
 
+		sourceLink.Downloaded()
 		c.Set("source_link", sourceLink)
 		c.Next()
 	}
diff --git a/models/source_link.go b/models/source_link.go
index c1c1812..49dfea2 100644
--- a/models/source_link.go
+++ b/models/source_link.go
@@ -39,3 +39,9 @@ func GetSourceLinkByID(id interface{}) (*SourceLink, error) {
 
 	return link, result.Error
 }
+
+// Viewed 增加访问次数
+func (s *SourceLink) Downloaded() {
+	s.Downloads++
+	DB.Model(s).UpdateColumn("downloads", gorm.Expr("downloads + ?", 1))
+}