tuonina 6 years ago
parent
commit
59373a79f7

+ 2 - 1
fastdfs-client/src/main/java/cn/gygxzc/cloud/tina/fastdfs/client/controller/FdfsUploadController.java

@@ -2,6 +2,7 @@ package cn.gygxzc.cloud.tina.fastdfs.client.controller;
 
 import cn.gygxzc.cloud.tina.fastdfs.client.mata.CommonFileMata;
 import cn.gygxzc.cloud.tina.fastdfs.client.mata.FailedStorePath;
+import cn.gygxzc.cloud.tina.fastdfs.client.utils.FileUtils;
 import com.github.tobato.fastdfs.domain.MataData;
 import com.github.tobato.fastdfs.domain.StorePath;
 import com.github.tobato.fastdfs.service.FastFileStorageClient;
@@ -42,7 +43,7 @@ public class FdfsUploadController {
     public Object upload(@RequestParam("file") MultipartFile file, CommonFileMata fileMata) {
 
         Set<MataData> mataData = fileMata.mataData(file);
-        String fileExtName = file.getOriginalFilename();
+        String fileExtName = FileUtils.getFileExtName(file.getOriginalFilename());
         try {
             return storageClient.uploadFile(file.getInputStream(), file.getSize(), fileExtName, mataData);
         } catch (IOException e) {

+ 28 - 0
fastdfs-client/src/main/java/cn/gygxzc/cloud/tina/fastdfs/client/utils/FileUtils.java

@@ -0,0 +1,28 @@
+package cn.gygxzc.cloud.tina.fastdfs.client.utils;
+
+import org.apache.commons.lang.StringUtils;
+
+/**
+ * Created by niantuo on 2018/10/29.
+ * 文件相关的方法
+ */
+
+public class FileUtils {
+    private final static String UNKNOWN_EXT = "unknown";
+
+    /**
+     * 获取文件的后缀名称
+     *
+     * @return 文件的后缀
+     */
+    public static String getFileExtName(String fileName) {
+        if (StringUtils.isBlank(fileName)) {
+            return UNKNOWN_EXT;
+        }
+        int index = fileName.lastIndexOf(".");
+        if (index == 0) {
+            return UNKNOWN_EXT;
+        }
+        return fileName.substring(index + 1);
+    }
+}

+ 14 - 0
fastdfs-client/src/test/java/cn/gygxzc/cloud/tina/fdfs/utils/FileUtilsTest.java

@@ -0,0 +1,14 @@
+package cn.gygxzc.cloud.tina.fdfs.utils;
+
+import cn.gygxzc.cloud.tina.fastdfs.client.utils.FileUtils;
+
+/**
+ * Created by niantuo on 2018/10/29.
+ */
+
+public class FileUtilsTest {
+    public static void main(String[] args) {
+        String fileName="123.png";
+        System.out.printf("extName:%s", FileUtils.getFileExtName(fileName));
+    }
+}