增加oss上传度量

xincao9 · Sep 28, 2020 · fda026a · fda026a
1 parent 36b92c3
commit fda026a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/api/oss/oss.go b/api/oss/oss.go
@@ -46,6 +46,7 @@ func Route(engine *gin.Engine) {
 				"code":    http.StatusBadRequest,
 				"message": constant.InvalidArgument,
 			})
+			metrics.UploadCount.WithLabelValues("failure", "bad_request").Inc()
 			return
 		}
 		files := form.File["file[]"]
@@ -61,12 +62,14 @@ func Route(engine *gin.Engine) {
 			if err != nil {
 				items[i].Status = false
 				logger.L.Error(err)
+				metrics.UploadCount.WithLabelValues("failure", "server_error").Inc()
 				continue
 			}
 			val, err := ioutil.ReadAll(f)
 			if err != nil {
 				items[i].Status = false
 				logger.L.Error(err)
+				metrics.UploadCount.WithLabelValues("failure", "server_error").Inc()
 				continue
 			}
 			h := md5.New()
@@ -79,8 +82,10 @@ func Route(engine *gin.Engine) {
 			if err != nil {
 				items[i].Status = false
 				logger.L.Error(err)
+				metrics.UploadCount.WithLabelValues("failure", "server_error").Inc()
 				continue
 			}
+			metrics.UploadCount.WithLabelValues("success", "disk").Inc()
 			items[i].Oid = string(key)
 		}
 		c.JSON(http.StatusOK,

diff --git a/component/metrics/metrics.go b/component/metrics/metrics.go
@@ -10,7 +10,8 @@ var (
 	ObjectCurrentCount prometheus.Gauge
 	PutCount           *prometheus.GaugeVec
 	GetCount           *prometheus.GaugeVec
-	DownloadCount           *prometheus.GaugeVec
+	DownloadCount      *prometheus.GaugeVec
+	UploadCount        *prometheus.GaugeVec
 )
 
 func init() {
@@ -28,8 +29,12 @@ func init() {
 	}, []string{"status", "source"})
 	DownloadCount = prometheus.NewGaugeVec(prometheus.GaugeOpts{
 		Name: "oss_download_count",
-		Help: "下载操作数",
+		Help: "download操作数",
 	}, []string{"status", "source"})
+	UploadCount = prometheus.NewGaugeVec(prometheus.GaugeOpts{
+		Name: "oss_upload_count",
+		Help: "upload操作数",
+	}, []string{"status"})
 	prometheus.MustRegister(ObjectCurrentCount, PutCount, GetCount)
 }