Merge branch 'feature/文件模块增加根据url上传文件' into 'wrdp'

[云文件]file模块增加根据url上传文件

See merge request o2oa/o2oa!5421

o2server/x_cms_assemble_control/src/main/java/com/x/cms/assemble/control/jaxrs/fileinfo/ActionFileUploadWithUrl.java

@@ -23,7 +23,6 @@ import com.x.cms.assemble.control.ThisApplication;
 import com.x.cms.core.entity.Document;
 import com.x.cms.core.entity.FileInfo;
 import com.x.processplatform.core.entity.content.Attachment;
-import com.x.processplatform.core.entity.content.Work;
 import org.apache.commons.io.FilenameUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.tika.Tika;
@@ -43,7 +42,7 @@ public class ActionFileUploadWithUrl extends BaseAction {
 
 		Wi wi = this.convertToWrapIn(jsonElement, Wi.class);
 		if(StringUtils.isEmpty(wi.getDocId())){
-			throw new ExceptionEntityFieldEmpty(Work.class, wi.getDocId());
+			throw new ExceptionEntityFieldEmpty(Document.class, wi.getDocId());
 		}
 		if(StringUtils.isEmpty(wi.getFileName())){
 			throw new ExceptionEntityFieldEmpty(Attachment.class, wi.getFileName());

o2server/x_file_assemble_control/src/main/java/com/x/file/assemble/control/jaxrs/file/ActionUploadWithUrl.java

+package com.x.file.assemble.control.jaxrs.file;
+
+import com.google.gson.JsonElement;
+import com.x.base.core.container.EntityManagerContainer;
+import com.x.base.core.container.factory.EntityManagerContainerFactory;
+import com.x.base.core.entity.annotation.CheckPersistType;
+import com.x.base.core.project.annotation.AuditLog;
+import com.x.base.core.project.annotation.FieldDescribe;
+import com.x.base.core.project.cache.CacheManager;
+import com.x.base.core.project.config.StorageMapping;
+import com.x.base.core.project.connection.CipherConnectionAction;
+import com.x.base.core.project.exception.ExceptionEntityFieldEmpty;
+import com.x.base.core.project.gson.GsonPropertyObject;
+import com.x.base.core.project.http.ActionResult;
+import com.x.base.core.project.http.EffectivePerson;
+import com.x.base.core.project.logger.Logger;
+import com.x.base.core.project.logger.LoggerFactory;
+import com.x.file.assemble.control.ThisApplication;
+import com.x.file.core.entity.open.File;
+import com.x.file.core.entity.open.ReferenceType;
+import org.apache.commons.io.FilenameUtils;
+import org.apache.commons.io.output.ByteArrayOutputStream;
+import org.apache.commons.lang3.ArrayUtils;
+import org.apache.commons.lang3.EnumUtils;
+import org.apache.commons.lang3.StringUtils;
+import org.imgscalr.Scalr;
+
+import javax.imageio.ImageIO;
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayInputStream;
+
+public class ActionUploadWithUrl extends BaseAction {
+
+	private static Logger logger = LoggerFactory.getLogger(ActionUploadWithUrl.class);
+
+	@AuditLog(operation = "上传附件")
+	protected ActionResult<Wo> execute(EffectivePerson effectivePerson,
+										JsonElement jsonElement) throws Exception {
+		logger.debug("ActionFileUploadWithUrl receive:{}.", jsonElement.toString());
+		ActionResult<Wo> result = new ActionResult<>();
+
+		Wi wi = this.convertToWrapIn(jsonElement, Wi.class);
+		if(StringUtils.isEmpty(wi.getReference())){
+			throw new ExceptionEntityFieldEmpty(File.class, "reference");
+		}
+		if(StringUtils.isEmpty(wi.getFileName())){
+			throw new ExceptionEntityFieldEmpty(File.class, "fileName");
+		}
+		if(StringUtils.isEmpty(wi.getFileUrl())){
+			throw new ExceptionEntityFieldEmpty(File.class, "fileUrl");
+		}
+		if(wi.getScale()==null){
+			throw new ExceptionEntityFieldEmpty(File.class, "scale");
+		}
+		String fileName = wi.getFileName();
+		if (StringUtils.isEmpty(FilenameUtils.getExtension(fileName))) {
+			throw new ExceptionEmptyExtension(fileName);
+		}
+
+		File file = null;
+		ReferenceType type = EnumUtils.getEnum(ReferenceType.class, wi.getReferenceType());
+		StorageMapping mapping = ThisApplication.context().storageMappings().random( File.class );
+		if (null == mapping) {
+			throw new ExceptionAllocateStorageMaaping();
+		}
+		byte[] bytes = CipherConnectionAction.getBinary(false, wi.getFileUrl());
+		if(bytes==null || bytes.length==0){
+			throw new ExceptionEntityFieldEmpty(File.class, "bytes");
+		}
+		Wo wo = new Wo();
+		try (EntityManagerContainer emc = EntityManagerContainerFactory.instance().create();
+			 ByteArrayInputStream in = new ByteArrayInputStream(bytes)) {
+			/* 先保存原图 */
+			file = new File(mapping.getName(), fileName, effectivePerson.getDistinguishedName(), type, wi.getReference());
+			emc.check(file, CheckPersistType.all);
+			file.saveContent(mapping, in, fileName);
+			emc.beginTransaction(File.class);
+			emc.persist(file);
+			emc.commit();
+			wo.setOrigId(file.getId());
+		}
+
+		/*保存压缩图*/
+		try (EntityManagerContainer emc = EntityManagerContainerFactory.instance().create();
+			 ByteArrayInputStream in = new ByteArrayInputStream(bytes)) {
+			if ((wi.getScale() > 0) && ArrayUtils.contains(IMAGE_EXTENSIONS, file.getExtension())) {
+				String fileNameThumbnail = fileName.substring(0, fileName.lastIndexOf(".")) + "_t" + "."+ file.getExtension();
+				File fileThumbnail = new File(mapping.getName(), fileNameThumbnail,effectivePerson.getDistinguishedName(), type, wi.getReference());
+				/** 如果是需要压缩的附件 */
+				try (ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
+					BufferedImage image = ImageIO.read(in);
+					if (image.getWidth() > wi.getScale()) {
+						/** 图像的实际大小比scale大的要进行压缩 */
+						BufferedImage scalrImage = Scalr.resize(image, Scalr.Method.QUALITY, Scalr.Mode.FIT_TO_WIDTH, wi.getScale());
+						ImageIO.write(scalrImage, fileThumbnail.getExtension(), baos);
+					} else {
+						/** 图像的实际大小比scale小,保存原图不进行压缩 */
+						ImageIO.write(image, fileThumbnail.getExtension(), baos);
+					}
+					try (ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray())) {
+						fileThumbnail.saveContent(mapping, bais, fileNameThumbnail);
+					}
+
+					emc.beginTransaction(File.class);
+					emc.persist(fileThumbnail);
+					emc.commit();
+					wo.setId(fileThumbnail.getId());
+				}
+
+			} else {
+				//不进行压缩,保存原图id
+				wo.setId(file.getId());
+			}
+		}
+
+		CacheManager.notify(File.class);
+
+		result.setData(wo);
+
+		return result;
+	}
+
+	public static class Wi extends GsonPropertyObject {
+
+		private static final long serialVersionUID = -3707551996175419386L;
+
+		@FieldDescribe("文件类型")
+		private String referenceType;
+
+		@FieldDescribe("文件名称,带扩展名的文件名.")
+		private String fileName;
+
+		@FieldDescribe("*附件来源url地址.")
+		private String fileUrl;
+
+		@FieldDescribe("*关联id.")
+		private String reference;
+
+		@FieldDescribe("缩放,如果指定宽度<=0,则不进行压缩")
+		private Integer scale;
+
+		public String getReferenceType() {
+			return referenceType;
+		}
+
+		public void setReferenceType(String referenceType) {
+			this.referenceType = referenceType;
+		}
+
+		public String getFileName() {
+			return fileName;
+		}
+
+		public void setFileName(String fileName) {
+			this.fileName = fileName;
+		}
+
+		public String getFileUrl() {
+			return fileUrl;
+		}
+
+		public void setFileUrl(String fileUrl) {
+			this.fileUrl = fileUrl;
+		}
+
+		public String getReference() {
+			return reference;
+		}
+
+		public void setReference(String reference) {
+			this.reference = reference;
+		}
+
+		public Integer getScale() {
+			return scale;
+		}
+
+		public void setScale(Integer scale) {
+			this.scale = scale;
+		}
+	}
+
+	public static class Wo extends GsonPropertyObject {
+
+		public Wo() {
+
+		}
+
+		public Wo(String id) throws Exception {
+			this.id = id;
+		}
+
+		@FieldDescribe("缩略图id")
+		private String id;
+
+		@FieldDescribe("原图id")
+		private String origId;
+
+		public String getId() {
+			return id;
+		}
+
+		public void setId(String id) {
+			this.id = id;
+		}
+
+		public String getOrigId() {
+			return origId;
+		}
+
+		public void setOrigId(String origId) {
+			this.origId = origId;
+		}
+	}
+}

o2server/x_file_assemble_control/src/main/java/com/x/file/assemble/control/jaxrs/file/FileAction.java

@@ -385,4 +385,22 @@ public class FileAction extends StandardJaxrsAction {
 		}
 		asyncResponse.resume(ResponseFactory.getEntityTagActionResultResponse(request, result));
 	}
+
+	@JaxrsMethodDescribe(value = "根据url上传附件", action = ActionUploadWithUrl.class)
+	@POST
+	@Path("upload/with/url")
+	@Produces(HttpMediaType.APPLICATION_JSON_UTF_8)
+	@Consumes(MediaType.APPLICATION_JSON)
+	public void uploadWithUrl(@Suspended final AsyncResponse asyncResponse, @Context HttpServletRequest request,
+							  JsonElement jsonElement) {
+		ActionResult<ActionUploadWithUrl.Wo> result = new ActionResult<>();
+		EffectivePerson effectivePerson = this.effectivePerson(request);
+		try {
+			result = new ActionUploadWithUrl().execute(effectivePerson, jsonElement);
+		} catch (Exception e) {
+			logger.error(e, effectivePerson, request, null);
+			result.error(e);
+		}
+		asyncResponse.resume(ResponseFactory.getEntityTagActionResultResponse(request, result));
+	}
 }
\ No newline at end of file