diff --git a/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java b/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java index 6e8df1c..3076433 100644 --- a/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java +++ b/tamguo-bms/src/main/java/com/tamguo/web/member/DocumentController.java @@ -4,9 +4,11 @@ import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.util.Date; +import java.util.List; import javax.servlet.http.HttpServletRequest; +import org.apache.commons.collections.CollectionUtils; import org.apache.commons.io.FileUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -25,8 +27,12 @@ import com.tamguo.common.utils.DateUtil; import com.tamguo.common.utils.Result; import com.tamguo.modules.book.model.DocumentEntity; import com.tamguo.modules.book.model.FileEntity; +import com.tamguo.modules.book.model.FileUploadEntity; +import com.tamguo.modules.book.model.enums.BizTypeEnum; +import com.tamguo.modules.book.model.enums.FileUploadStatusEnum; import com.tamguo.modules.book.service.IDocumentService; import com.tamguo.modules.book.service.IFileEntityService; +import com.tamguo.modules.book.service.IFileUploadService; import com.tamguo.utils.FileMd5Utils; @Controller @@ -39,18 +45,32 @@ public class DocumentController { IDocumentService iDocumentService; @Autowired IFileEntityService iFileEntityService; + @Autowired + IFileUploadService iFileUploadService; @Value("${file.storage.path}") private String fileStoragePath; @Value("${domain.name}") private String domainName; + @SuppressWarnings("unchecked") @RequestMapping(value = "{id}" , method = RequestMethod.GET) @ResponseBody public Result getDocument(@PathVariable String id) { DocumentEntity document = null; try { document = iDocumentService.selectById(id); + // 查询附件 + List fileUploads = iFileUploadService.selectList(Condition.create().eq("biz_key", document.getId()).eq("biz_type", BizTypeEnum.DOCUMENT.getValue())); + if(!CollectionUtils.isEmpty(fileUploads)) { + for(int i=0 ; i