diff --git a/src/main/java/cn/lihongjie/coal/file/service/FileService.java b/src/main/java/cn/lihongjie/coal/file/service/FileService.java index 5821598d..2095cc8d 100644 --- a/src/main/java/cn/lihongjie/coal/file/service/FileService.java +++ b/src/main/java/cn/lihongjie/coal/file/service/FileService.java @@ -174,7 +174,8 @@ public class FileService extends BaseService { extension = config.getMimeRepository() .getRegisteredMimeType(fileEntity.getMimeType()) - .getExtension().replace(".", ""); + .getExtension() + .replace(".", ""); } } catch (Exception e) { log.warn("获取文件扩展名失败", e); @@ -224,11 +225,8 @@ public class FileService extends BaseService { } @SneakyThrows - public FileDto upload( - InputStream file, - String name, - String dir) { - return upload(file, name, dir, null, null, null); + public FileDto upload(InputStream file, String name, String dir, String organizationId) { + return upload(file, name, dir, null, null, null, organizationId); } @SneakyThrows @@ -239,11 +237,27 @@ public class FileService extends BaseService { String sortKey, String code, String remarks) { + return upload(file, name, dir, sortKey, code, remarks, null); + } + + @SneakyThrows + public FileDto upload( + InputStream file, + String name, + String dir, + String sortKey, + String code, + String remarks, + String organizationId) { FileEntity fileEntity = new FileEntity(); fileEntity.setFileName(name); fileEntity.setFileSize(Long.valueOf(file.available())); fileEntity.setCode(code); fileEntity.setRemarks(remarks); + if (organizationId != null) { + + fileEntity.setOrganizationId(organizationId); + } try { fileEntity.setSortKey(Integer.parseInt(sortKey)); diff --git a/src/main/java/cn/lihongjie/coal/invoice/service/InvoiceService.java b/src/main/java/cn/lihongjie/coal/invoice/service/InvoiceService.java index c4daa744..4fe61f77 100644 --- a/src/main/java/cn/lihongjie/coal/invoice/service/InvoiceService.java +++ b/src/main/java/cn/lihongjie/coal/invoice/service/InvoiceService.java @@ -39,6 +39,7 @@ import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.io.InputStream; +import java.util.Optional; @Service @Slf4j @@ -120,7 +121,7 @@ public class InvoiceService extends BaseService one = this.repository.findOne( new Specification() { @Override public Predicate toPredicate( @@ -138,6 +139,12 @@ public class InvoiceService extends BaseService