modified: models/policy.go (#1718)

modified:   models/policy_test.go
pull/1741/head
Arkylin 2 years ago committed by GitHub
parent 00d56d6d07
commit 6358740cc9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -170,6 +170,7 @@ func (policy *Policy) GenerateFileName(uid uint, origin string) string {
"{second}": time.Now().Format("05"), "{second}": time.Now().Format("05"),
"{originname}": origin, "{originname}": origin,
"{ext}": filepath.Ext(origin), "{ext}": filepath.Ext(origin),
"{originname_without_ext}": strings.TrimSuffix(origin, filepath.Ext(origin)),
"{uuid}": uuid.Must(uuid.NewV4()).String(), "{uuid}": uuid.Must(uuid.NewV4()).String(),
} }

@ -134,6 +134,12 @@ func TestPolicy_GenerateFileName(t *testing.T) {
testPolicy.FileNameRule = "123{date}ss{datetime}" testPolicy.FileNameRule = "123{date}ss{datetime}"
asserts.Len(testPolicy.GenerateFileName(1, "123.txt"), 27) asserts.Len(testPolicy.GenerateFileName(1, "123.txt"), 27)
testPolicy.FileNameRule = "{originname_without_ext}"
asserts.Len(testPolicy.GenerateFileName(1, "123.txt"), 3)
testPolicy.FileNameRule = "{originname_without_ext}_{randomkey8}{ext}"
asserts.Len(testPolicy.GenerateFileName(1, "123.txt"), 16)
// 支持{originname}的策略 // 支持{originname}的策略
testPolicy.Type = "local" testPolicy.Type = "local"
testPolicy.FileNameRule = "123{originname}" testPolicy.FileNameRule = "123{originname}"

Loading…
Cancel
Save