POST /api/resource reconstruction

Author: lovehunter9

pkg/http/resource.go

@@ -12,7 +12,6 @@ import (
 	"os"
 	"path"
 	"path/filepath"
-	"strconv"
 	"strings"
 	"time"
 
@@ -62,37 +61,21 @@ func resourceDeleteHandler(fileCache fileutils.FileCache) handleFunc {
 }
 
 func resourcePostHandler(w http.ResponseWriter, r *http.Request, d *common.Data) (int, error) {
-	srcType := r.URL.Query().Get("src")
-	if srcType == "google" {
-		_, status, err := drives.ResourcePostGoogle("", w, r, false)
-		return status, err
-	} else if srcType == "cloud" || srcType == "awss3" || srcType == "tencent" || srcType == "dropbox" {
-		_, status, err := drives.ResourcePostCloudDrive("", w, r, false)
-		return status, err
-	}
+	start := time.Now()
+	klog.Infoln("Function resourcePostHandler starts at", start)
+	defer func() {
+		elapsed := time.Since(start)
+		klog.Infof("Function resourcePostHandler execution time: %v\n", elapsed)
+	}()
 
-	modeParam := r.URL.Query().Get("mode")
+	srcType := r.URL.Query().Get("src")
 
-	mode, err := strconv.ParseUint(modeParam, 8, 32)
-	if err != nil || modeParam == "" {
-		mode = 0775
+	handler, err := drives.GetResourceService(srcType)
+	if err != nil {
+		return http.StatusBadRequest, err
 	}
 
-	fileMode := os.FileMode(mode)
-
-	// Directories creation on POST.
-	if strings.HasSuffix(r.URL.Path, "/") {
-		if err = files.DefaultFs.MkdirAll(r.URL.Path, fileMode); err != nil {
-			klog.Errorln(err)
-			return common.ErrToStatus(err), err
-		}
-		if err = fileutils.Chown(files.DefaultFs, r.URL.Path, 1000, 1000); err != nil {
-			klog.Errorf("can't chown directory %s to user %d: %s", r.URL.Path, 1000, err)
-			return common.ErrToStatus(err), err
-		}
-		return http.StatusOK, nil
-	}
-	return http.StatusBadRequest, fmt.Errorf("%s is not a valid directory path", r.URL.Path)
+	return handler.PostHandler(w, r, d)
 }
 
 func resourcePutHandler(w http.ResponseWriter, r *http.Request, d *common.Data) (int, error) {