animeic-cd 1 year ago
parent
commit
693addf47f
2 changed files with 7 additions and 2 deletions
  1. 6 1
      src/api/upload.go
  2. 1 1
      src/db/db.go

+ 6 - 1
src/api/upload.go

@@ -3,7 +3,9 @@ package api
 import (
 	"cr-svc/db/model"
 	"cr-svc/db/repo"
+	"errors"
 	"fmt"
+	"path"
 	"path/filepath"
 	"time"
 
@@ -24,9 +26,12 @@ func UploadPkg(c *gin.Context, apictx *ApiSession) (interface{}, error) {
 	if err != nil {
 		return nil, err
 	}
-
+	fmt.Println(file.Size)
 	size := fmt.Sprintf("%.2fM", float64(file.Size)/float64(1024*1024))
 	fmt.Println(size)
+	if path.Ext(file.Filename) != ".zip" {
+		return nil, errors.New("仅支持zip文件")
+	}
 
 	basePath := "./package/"
 	filename := basePath + filepath.Base(file.Filename)

+ 1 - 1
src/db/db.go

@@ -36,7 +36,7 @@ func (db *MongoDB) GetOrCreateDatabase(name string) *mongo.Database {
 
 func NewMongoDB(bus *comm.NatsBus) *MongoDB {
 
-	inst, err := bus.NewMongoDBFromConfig("cr-mongo")
+	inst, err := bus.NewMongoDBFromConfigDev("cr-mongo")
 
 	if err != nil {
 		panic(err)