|
|
@@ -1,5 +1,6 @@
|
|
|
package cn.tr.module.export.bean;
|
|
|
|
|
|
+import cn.hutool.core.util.StrUtil;
|
|
|
import lombok.AllArgsConstructor;
|
|
|
import lombok.Data;
|
|
|
import lombok.NoArgsConstructor;
|
|
|
@@ -16,4 +17,11 @@ import lombok.NoArgsConstructor;
|
|
|
public class ExportResult {
|
|
|
private String fileName;
|
|
|
private String base64;
|
|
|
+
|
|
|
+ public String getFileName() {
|
|
|
+ if (StrUtil.endWith(fileName, ".xlsx")) {
|
|
|
+ return fileName;
|
|
|
+ }
|
|
|
+ return fileName+".xlsx";
|
|
|
+ }
|
|
|
}
|