ci: refractor get fileparam from uri

Author: aby913

pkg/global/node.go

@@ -2,6 +2,7 @@ package global
 
 import (
 	"context"
+	"os"
 	"sync"
 	"time"
 
@@ -22,7 +23,8 @@ var (
 )
 
 var (
-	GlobalNode *Node
+	GlobalNode      *Node
+	CurrentNodeName string
 )
 
 type NodeInfo struct {
@@ -36,6 +38,8 @@ type Node struct {
 }
 
 func InitGlobalNodes(config *rest.Config) error {
+	CurrentNodeName = os.Getenv("NODE_NAME")
+
 	GlobalNode = &Node{
 		k8sClient: kubernetes.NewForConfigOrDie(config),
 		Nodes:     make(map[string]*v1.Node),

pkg/global/pvc.go

@@ -64,13 +64,58 @@ func (g *Data) GetPvcUser(user string) string {
 	return g.UserPvcMap[user]
 }
 
+func (g *Data) GetPvcUserName(pvcUser string) (string, error) {
+	g.mu.RLock()
+	defer g.mu.RUnlock()
+
+	var user string
+	for k, v := range g.UserPvcMap {
+		if v == pvcUser {
+			user = k
+			break
+		}
+	}
+
+	if user == "" {
+		return "", fmt.Errorf("userspace pvc not found, name: %s", pvcUser)
+	}
+
+	return user, nil
+}
+
+func (g *Data) GetPvcCacheName(pvcCache string) (string, error) {
+	g.mu.RLock()
+	defer g.mu.RUnlock()
+
+	var user string
+	for k, v := range g.CachePvcMap {
+		if v == pvcCache {
+			user = k
+			break
+		}
+	}
+
+	if user == "" {
+		return "", fmt.Errorf("appcache pvc not found, name: %s", pvcCache)
+	}
+
+	return user, nil
+}
+
 func (g *Data) GetPvcCache(user string) string {
 	g.mu.RLock()
 	defer g.mu.RUnlock()
 
 	return g.CachePvcMap[user]
 }
 
+func (g *Data) GetPvcCaches() map[string]string {
+	g.mu.RLock()
+	defer g.mu.RUnlock()
+
+	return g.CachePvcMap
+}
+
 func (g *Data) getGlobalData() error {
 	users := g.getUsers()
 	if users == nil || len(users) == 0 {

pkg/models/file_param.go

@@ -142,6 +142,70 @@ func (r *FileParam) GetResourceUri() (string, error) {
 
 }
 
+func (r *FileParam) GetFileParam(uri string) error {
+	var u = strings.TrimLeft(uri, "/")
+
+	var s = strings.Split(u, "/")
+	if len(s) < 2 {
+		return errors.New("url invalid")
+	}
+
+	if s[0] == constant.AwsS3 || s[0] == constant.DropBox || s[0] == constant.GoogleDrive {
+		r.Owner = ""
+		r.FileType = s[0]
+		r.Extend = s[1]
+		r.Path = r.joinPath(2, s)
+		return nil
+
+	}
+
+	if s[0] == constant.Sync {
+		r.Owner = ""
+		r.FileType = s[0]
+		r.Extend = s[1]
+		r.Path = r.joinPath(2, s)
+		return nil
+
+	}
+
+	if strings.HasPrefix(uri, constant.ROOT_PREFIX+"/") {
+		var p = strings.TrimPrefix(uri, constant.ROOT_PREFIX+"/")
+		s = strings.Split(p, "/")
+		pvcUser, err := global.GlobalData.GetPvcUserName(s[0])
+		if err == nil {
+			r.Owner = pvcUser
+			r.FileType = constant.Drive
+			r.Extend = s[1]
+			r.Path = r.joinPath(2, s)
+			return nil
+		}
+	}
+
+	if strings.HasPrefix(uri, constant.CACHE_PREFIX+"/") {
+		var p = strings.TrimPrefix(uri, constant.CACHE_PREFIX+"/")
+		s = strings.Split(p, "/")
+		pvcCache, err := global.GlobalData.GetPvcCacheName(s[0])
+		if err == nil {
+			r.Owner = pvcCache
+			r.FileType = constant.Cache
+			r.Extend = global.CurrentNodeName
+			r.Path = r.joinPath(1, s)
+			return nil
+		}
+	}
+
+	if strings.HasPrefix(uri, constant.EXTERNAL_PREFIX+"/") {
+		var p = strings.TrimPrefix(uri, constant.EXTERNAL_PREFIX+"/")
+		s = strings.Split(p, "/")
+		r.Owner = ""
+		r.FileType = constant.External
+		r.Extend = global.CurrentNodeName
+		r.Path = r.joinPath(0, s)
+		return nil
+	}
+	return nil
+}
+
 func (r *FileParam) IsFile() (string, bool) {
 	var p = strings.Split(r.Path, "/")
 	var fileName = p[len(p)-1]
@@ -150,3 +214,14 @@ func (r *FileParam) IsFile() (string, bool) {
 	}
 	return "", false
 }
+
+func (r *FileParam) joinPath(pos int, s []string) string {
+	var str string
+	for i := pos; i < len(s); i++ {
+		str = str + "/" + s[i]
+	}
+	if str == "" {
+		str = "/"
+	}
+	return str
+}