Procházet zdrojové kódy

Fix attachment pruning logging; .mp4 extension issue

Philipp Heckel před 4 roky
rodič
revize
b5183612be
2 změnil soubory, kde provedl 7 přidání a 14 odebrání
  1. 5 14
      server/file_cache.go
  2. 2 0
      util/util.go

+ 5 - 14
server/file_cache.go

@@ -71,13 +71,12 @@ func (c *fileCache) Write(id string, in io.Reader, limiters ...*util.Limiter) (i
 }
 
 func (c *fileCache) Remove(ids ...string) error {
-	var firstErr error
 	for _, id := range ids {
-		if err := c.removeFile(id); err != nil {
-			if firstErr == nil {
-				firstErr = err // Continue despite error; we want to delete as many as we can
-			}
+		if !fileIDRegex.MatchString(id) {
+			return errInvalidFileID
 		}
+		file := filepath.Join(c.dir, id)
+		_ = os.Remove(file) // Best effort delete
 	}
 	size, err := dirSize(c.dir)
 	if err != nil {
@@ -86,15 +85,7 @@ func (c *fileCache) Remove(ids ...string) error {
 	c.mu.Lock()
 	c.totalSizeCurrent = size
 	c.mu.Unlock()
-	return firstErr
-}
-
-func (c *fileCache) removeFile(id string) error {
-	if !fileIDRegex.MatchString(id) {
-		return errInvalidFileID
-	}
-	file := filepath.Join(c.dir, id)
-	return os.Remove(file)
+	return nil
 }
 
 func (c *fileCache) Size() int64 {

+ 2 - 0
util/util.go

@@ -173,6 +173,8 @@ func ExtensionByType(contentType string) string {
 	switch contentType {
 	case "image/jpeg":
 		return ".jpg"
+	case "video/mp4":
+		return ".mp4"
 	default:
 		exts, err := mime.ExtensionsByType(contentType)
 		if err == nil && len(exts) > 0 && extRegex.MatchString(exts[0]) {