[proxy]文件缓存增加“是否自动创建目录”选项

Author: iwind

teacache/file.go

@@ -32,6 +32,7 @@ type FileManager struct {
 
 	looper      *timers.Looper
 	dir         string
+	autoCreate  bool // 是否自动创建
 	writeLocker sync.RWMutex
 }
 
@@ -89,13 +90,18 @@ func (this *FileManager) SetOptions(options map[string]interface{}) {
 		this.Life = 1800 * time.Second
 	}
 
-	dir, found := options["dir"]
-	if found {
+	dir, ok := options["dir"]
+	if ok {
 		this.dir = types.String(dir)
 		if !filepath.IsAbs(this.dir) {
 			this.dir = Tea.Root + Tea.DS + this.dir
 		}
 	}
+
+	autoCreate, ok := options["autoCreate"]
+	if ok {
+		this.autoCreate = types.Bool(autoCreate)
+	}
 }
 
 // 写入
@@ -110,7 +116,15 @@ func (this *FileManager) Write(key string, data []byte) error {
 
 	dirFile := files.NewFile(this.dir)
 	if !dirFile.IsDir() {
-		return errors.New("cache dir should be a valid dir")
+		// 自动创建
+		if this.autoCreate {
+			err := dirFile.MkdirAll()
+			if err != nil {
+				return errors.New("can not create cache dir: " + err.Error())
+			}
+		} else {
+			return errors.New("cache dir should be a valid dir")
+		}
 	}
 
 	md5 := stringutil.Md5(key)

teaweb/actions/default/cache/createPolicy.go

@@ -42,7 +42,8 @@ func (this *CreatePolicyAction) RunPost(params struct {
 	SkipSetCookie             bool
 	EnableRequestCachePragma  bool
 
-	FileDir string
+	FileDir        string
+	FileAutoCreate bool
 
 	RedisNetwork  string
 	RedisHost     string
@@ -88,7 +89,8 @@ func (this *CreatePolicyAction) RunPost(params struct {
 			Field("fileDir", params.FileDir).
 			Require("请输入缓存存放目录")
 		policy.Options = map[string]interface{}{
-			"dir": params.FileDir,
+			"dir":        params.FileDir,
+			"autoCreate": params.FileAutoCreate,
 		}
 	case "redis":
 		params.Must.

teaweb/actions/default/cache/updatePolicy.go

@@ -80,7 +80,8 @@ func (this *UpdatePolicyAction) RunPost(params struct {
 	SkipSetCookie             bool
 	EnableRequestCachePragma  bool
 
-	FileDir string
+	FileDir        string
+	FileAutoCreate bool
 
 	RedisNetwork  string
 	RedisHost     string
@@ -129,7 +130,8 @@ func (this *UpdatePolicyAction) RunPost(params struct {
 			Field("fileDir", params.FileDir).
 			Require("请输入缓存存放目录")
 		policy.Options = map[string]interface{}{
-			"dir": params.FileDir,
+			"dir":        params.FileDir,
+			"autoCreate": params.FileAutoCreate,
 		}
 	case "redis":
 		params.Must.