better

opendx · Nov 23, 2020 · 7425bbe · 7425bbe
1 parent cfdebc6
commit 7425bbe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/java/com/daxiang/utils/UUIDUtil.java b/src/main/java/com/daxiang/utils/UUIDUtil.java
@@ -1,5 +1,6 @@
 package com.daxiang.utils;
 
+import org.apache.commons.io.FilenameUtils;
 import org.springframework.util.Assert;
 import org.springframework.util.StringUtils;
 
@@ -16,6 +17,8 @@ public static String getUUID() {
 
     public static String getUUIDFilename(String filename) {
         Assert.hasText(filename, "filename不能为空");
-        return filename.contains(".") ? getUUID() + "." + StringUtils.unqualify(filename) : getUUID();
+        String fileExtension = FilenameUtils.getExtension(filename);
+        String uuid = getUUID();
+        return StringUtils.isEmpty(fileExtension) ? uuid : uuid + "." + fileExtension;
     }
 }