fix downlod api

casson · Dec 12, 2019 · deb614f · deb614f
1 parent 3aa894b
commit deb614f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/routers/download.go b/routers/download.go
@@ -24,6 +24,14 @@ func DownloadStructFile(c *gin.Context) {
 		return
 	}
 
+	if downLoadInfo.StructName == "" {
+		downLoadInfo.StructName = "new_struct"
+	}
+	if downLoadInfo.Content == "" {
+		services.HandleError(http.StatusBadRequest, c, err)
+		return
+	}
+
 	var filename = filepath.Join(viper.GetString("genFilePath"), downLoadInfo.StructName) + ".go"
 	if !utils.CheckFileIsExist(filename) {
 		file, err := os.Create(filename) //创建文件