update - 处理流程已结束后添加评论的逻辑
This commit is contained in:
parent
f8f385055b
commit
86cf230fcc
@ -0,0 +1,170 @@
|
||||
package org.flowable.engine.impl.cmd;
|
||||
|
||||
import org.flowable.common.engine.api.FlowableException;
|
||||
import org.flowable.common.engine.api.FlowableObjectNotFoundException;
|
||||
import org.flowable.common.engine.api.delegate.event.FlowableEngineEventType;
|
||||
import org.flowable.common.engine.api.delegate.event.FlowableEventDispatcher;
|
||||
import org.flowable.common.engine.impl.identity.Authentication;
|
||||
import org.flowable.common.engine.impl.interceptor.Command;
|
||||
import org.flowable.common.engine.impl.interceptor.CommandContext;
|
||||
import org.flowable.common.engine.impl.persistence.entity.ByteArrayEntity;
|
||||
import org.flowable.common.engine.impl.util.IoUtil;
|
||||
import org.flowable.engine.compatibility.Flowable5CompatibilityHandler;
|
||||
import org.flowable.engine.delegate.event.impl.FlowableEventBuilder;
|
||||
import org.flowable.engine.impl.cfg.ProcessEngineConfigurationImpl;
|
||||
import org.flowable.engine.impl.persistence.entity.AttachmentEntity;
|
||||
import org.flowable.engine.impl.persistence.entity.ExecutionEntity;
|
||||
import org.flowable.engine.impl.util.CommandContextUtil;
|
||||
import org.flowable.engine.impl.util.Flowable5Util;
|
||||
import org.flowable.engine.runtime.ProcessInstance;
|
||||
import org.flowable.engine.task.Attachment;
|
||||
import org.flowable.task.api.Task;
|
||||
import org.flowable.task.service.impl.persistence.entity.TaskEntity;
|
||||
|
||||
import java.io.InputStream;
|
||||
|
||||
/**
|
||||
* 覆盖 Flowable 自带的 CreateAttachmentCmd, 由于 Flowable 5 的实现导致异常，需要覆盖
|
||||
*
|
||||
* @author Tom Baeyens
|
||||
* @author Joram Barrez
|
||||
*/
|
||||
public class CreateAttachmentCmd implements Command<Attachment> {
|
||||
|
||||
protected String attachmentType;
|
||||
protected String taskId;
|
||||
protected String processInstanceId;
|
||||
protected String attachmentName;
|
||||
protected String attachmentDescription;
|
||||
protected InputStream content;
|
||||
protected String url;
|
||||
|
||||
public CreateAttachmentCmd(String attachmentType, String taskId, String processInstanceId, String attachmentName,
|
||||
String attachmentDescription, InputStream content, String url) {
|
||||
this.attachmentType = attachmentType;
|
||||
this.taskId = taskId;
|
||||
this.processInstanceId = processInstanceId;
|
||||
this.attachmentName = attachmentName;
|
||||
this.attachmentDescription = attachmentDescription;
|
||||
this.content = content;
|
||||
this.url = url;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Attachment execute(CommandContext commandContext) {
|
||||
|
||||
if (taskId != null) {
|
||||
TaskEntity task = verifyTaskParameters(commandContext);
|
||||
if (task.getProcessDefinitionId() != null && Flowable5Util.isFlowable5ProcessDefinitionId(commandContext,
|
||||
task.getProcessDefinitionId())) {
|
||||
Flowable5CompatibilityHandler compatibilityHandler = Flowable5Util.getFlowable5CompatibilityHandler();
|
||||
return compatibilityHandler.createAttachment(attachmentType, taskId, processInstanceId,
|
||||
attachmentName, attachmentDescription, content, url);
|
||||
}
|
||||
}
|
||||
|
||||
// 覆盖类的处理
|
||||
// if (processInstanceId != null) {
|
||||
// ExecutionEntity execution = verifyExecutionParameters(commandContext);
|
||||
// if (Flowable5Util.isFlowable5ProcessDefinitionId(commandContext, execution
|
||||
// .getProcessDefinitionId())) {
|
||||
// Flowable5CompatibilityHandler compatibilityHandler = Flowable5Util
|
||||
// .getFlowable5CompatibilityHandler();
|
||||
// return compatibilityHandler.createAttachment(attachmentType, taskId, processInstanceId,
|
||||
// attachmentName, attachmentDescription, content, url);
|
||||
// }
|
||||
// }
|
||||
|
||||
ProcessEngineConfigurationImpl processEngineConfiguration =
|
||||
CommandContextUtil.getProcessEngineConfiguration(commandContext);
|
||||
AttachmentEntity attachment = processEngineConfiguration.getAttachmentEntityManager().create();
|
||||
attachment.setName(attachmentName);
|
||||
attachment.setProcessInstanceId(processInstanceId);
|
||||
attachment.setTaskId(taskId);
|
||||
attachment.setDescription(attachmentDescription);
|
||||
attachment.setType(attachmentType);
|
||||
attachment.setUrl(url);
|
||||
attachment.setUserId(Authentication.getAuthenticatedUserId());
|
||||
attachment.setTime(processEngineConfiguration.getClock().getCurrentTime());
|
||||
|
||||
processEngineConfiguration.getAttachmentEntityManager().insert(attachment, false);
|
||||
|
||||
if (content != null) {
|
||||
byte[] bytes = IoUtil.readInputStream(content, attachmentName);
|
||||
ByteArrayEntity byteArray = processEngineConfiguration.getByteArrayEntityManager().create();
|
||||
byteArray.setBytes(bytes);
|
||||
processEngineConfiguration.getByteArrayEntityManager().insert(byteArray);
|
||||
attachment.setContentId(byteArray.getId());
|
||||
attachment.setContent(byteArray);
|
||||
}
|
||||
|
||||
ExecutionEntity processInstance = null;
|
||||
if (processInstanceId != null) {
|
||||
processInstance = processEngineConfiguration.getExecutionEntityManager().findById(processInstanceId);
|
||||
}
|
||||
|
||||
TaskEntity task = null;
|
||||
if (taskId != null) {
|
||||
task = processEngineConfiguration.getTaskServiceConfiguration().getTaskService().getTask(taskId);
|
||||
}
|
||||
|
||||
processEngineConfiguration.getHistoryManager().createAttachmentComment(task, processInstance, attachmentName,
|
||||
true);
|
||||
|
||||
FlowableEventDispatcher eventDispatcher = processEngineConfiguration.getEventDispatcher();
|
||||
if (eventDispatcher != null && eventDispatcher.isEnabled()) {
|
||||
// Forced to fetch the process-instance to associate the right
|
||||
// process definition
|
||||
String processDefinitionId = null;
|
||||
if (attachment.getProcessInstanceId() != null) {
|
||||
ExecutionEntity process =
|
||||
processEngineConfiguration.getExecutionEntityManager().findById(processInstanceId);
|
||||
if (process != null) {
|
||||
processDefinitionId = process.getProcessDefinitionId();
|
||||
}
|
||||
}
|
||||
|
||||
eventDispatcher.dispatchEvent(FlowableEventBuilder.createEntityEvent(FlowableEngineEventType.ENTITY_CREATED, attachment,
|
||||
processInstanceId, processInstanceId, processDefinitionId),
|
||||
processEngineConfiguration.getEngineCfgKey());
|
||||
eventDispatcher.dispatchEvent(FlowableEventBuilder.createEntityEvent(FlowableEngineEventType.ENTITY_INITIALIZED, attachment,
|
||||
processInstanceId, processInstanceId, processDefinitionId),
|
||||
processEngineConfiguration.getEngineCfgKey());
|
||||
}
|
||||
|
||||
return attachment;
|
||||
}
|
||||
|
||||
protected TaskEntity verifyTaskParameters(CommandContext commandContext) {
|
||||
ProcessEngineConfigurationImpl processEngineConfiguration =
|
||||
CommandContextUtil.getProcessEngineConfiguration(commandContext);
|
||||
TaskEntity task = processEngineConfiguration.getTaskServiceConfiguration().getTaskService().getTask(taskId);
|
||||
|
||||
if (task == null) {
|
||||
throw new FlowableObjectNotFoundException("Cannot find task with id " + taskId, Task.class);
|
||||
}
|
||||
|
||||
if (task.isSuspended()) {
|
||||
throw new FlowableException("It is not allowed to add an attachment to a suspended task");
|
||||
}
|
||||
|
||||
return task;
|
||||
}
|
||||
|
||||
protected ExecutionEntity verifyExecutionParameters(CommandContext commandContext) {
|
||||
ExecutionEntity execution =
|
||||
CommandContextUtil.getExecutionEntityManager(commandContext).findById(processInstanceId);
|
||||
|
||||
if (execution == null) {
|
||||
throw new FlowableObjectNotFoundException("Process instance " + processInstanceId + " doesn't exist",
|
||||
ProcessInstance.class);
|
||||
}
|
||||
|
||||
if (execution.isSuspended()) {
|
||||
throw new FlowableException("It is not allowed to add an attachment to a suspended process instance");
|
||||
}
|
||||
|
||||
return execution;
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user