From dc8ee89c853efefdcecb3c5410f1bc928d8c7040 Mon Sep 17 00:00:00 2001
From: link <a624669980@163.com>
Date: Mon, 13 Mar 2023 18:30:22 +0800
Subject: [PATCH] Statistics on the number of files added (#951)

---
 route/v1.go      |  1 +
 route/v1/file.go | 12 ++++++++++++
 2 files changed, 13 insertions(+)

diff --git a/route/v1.go b/route/v1.go
index 6d30ffc..4f20ee5 100644
--- a/route/v1.go
+++ b/route/v1.go
@@ -106,6 +106,7 @@ func InitV1Router() *gin.Engine {
 			v1FolderGroup.GET("", v1.DirPath)   ///file/dirpath
 			v1FolderGroup.POST("", v1.MkdirAll) ///file/mkdir
 			v1FolderGroup.GET("/size", v1.GetSize)
+			v1FolderGroup.GET("/count", v1.GetFileCount)
 		}
 		v1BatchGroup := v1Group.Group("/batch")
 		v1BatchGroup.Use()
diff --git a/route/v1/file.go b/route/v1/file.go
index 5a1cd56..e9bec36 100644
--- a/route/v1/file.go
+++ b/route/v1/file.go
@@ -807,6 +807,18 @@ func GetSize(c *gin.Context) {
 	c.JSON(common_err.SUCCESS, model.Result{Success: common_err.SUCCESS, Message: common_err.GetMsg(common_err.SUCCESS), Data: size})
 }
 
+func GetFileCount(c *gin.Context) {
+	json := make(map[string]string)
+	c.ShouldBind(&json)
+	path := json["path"]
+	list, err := ioutil.ReadDir(path)
+	if err != nil {
+		c.JSON(common_err.SERVICE_ERROR, model.Result{Success: common_err.SERVICE_ERROR, Message: common_err.GetMsg(common_err.SERVICE_ERROR), Data: err.Error()})
+		return
+	}
+	c.JSON(common_err.SERVICE_ERROR, model.Result{Success: common_err.SERVICE_ERROR, Message: common_err.GetMsg(common_err.SERVICE_ERROR), Data: len(list)})
+}
+
 type CenterHandler struct {
 	// 广播通道，有数据则循环每个用户广播出去
 	broadcast chan []byte