feature: 增加参数校验

cmd/webpprof/main.go

@@ -3,6 +3,7 @@ package main
 import (
 	"crypto/md5"
 	"encoding/hex"
+	"errors"
 	"fmt"
 	"io/ioutil"
 	"net/http"
@@ -37,6 +38,10 @@ func mapping(c *gin.Context) {
 	filename := fmt.Sprintf("%s.pprof", hash)
 	f, err := os.OpenFile(filename, os.O_RDONLY, os.ModePerm)
 	if err != nil {
+		if os.IsNotExist(err) {
+			c.AbortWithStatus(http.StatusNotFound)
+			return
+		}
 		c.AbortWithError(http.StatusInternalServerError, err)
 		return
 	}
@@ -94,6 +99,11 @@ func upload(c *gin.Context) {
 		return
 	}
 
+	if _, err := profile.ParseData(fdata); err != nil {
+		c.AbortWithError(http.StatusBadRequest, errors.New("unrecongnized profile format"))
+		return
+	}
+
 	fileDst := fmt.Sprintf("%s.pprof", rhash)
 	file, err := os.Create(fileDst)
 	if err != nil {

web/src/components/Upload.vue

@@ -4,11 +4,13 @@
         class="upload-demo"
         drag
         action="/upload"
-        :on-success="uploadSuccess"
+        :on-progress="handleUploadProgress"
+        :on-success="handleUploadSuccess"
+        :on-error="handleUploadError"
     >
       <i class="el-icon-upload"></i>
       <div class="el-upload__text">将文件拖到此处，或<em>点击上传</em></div>
-      <div class="el-upload__tip" slot="tip">上传文件不超过500kb</div>
+      <div class="el-upload__tip" slot="tip">拖动文件至此即可生成火焰图</div>
     </el-upload>
   </div>
 </template>
@@ -16,10 +18,28 @@
 <script>
 export default {
   name: "Upload",
-  methods:{
-    uploadSuccess(response,file,fileList){
-      console.log(response,file,fileList)
-      window.location.href="http://localhost:8090/pprof/"+response.data+"/dot"
+  data() {
+    return {
+      src: 'http://localhost:8090/pprof/bd5001ffcea7949ff66b7368b1912aa7/dot',
+      loading: null
+    }
+  },
+  methods: {
+    handleUploadSuccess(response) {
+      this.loading.close()
+      window.location.href = "http://localhost:8090/pprof/" + response.data + "/dot"
+    },
+    handleUploadError() {
+      this.loading.close()
+      this.$message.error("文件上传失败或文件格式错误")
+    },
+    handleUploadProgress() {
+      this.loading = this.$loading({
+        lock: true,
+        text: 'Loading',
+        spinner: 'el-icon-loading',
+        background: 'rgba(0, 0, 0, 0.7)'
+      });
     }
   }
 }