@Override
protected CloudWatchWorkflowClientExternal createClientInstance(
WorkflowExecution workflowExecution,
StartWorkflowOptions options,
DataConverter dataConverter,
GenericWorkflowClientExternal genericClient) {
WorkflowType workflowType = new WorkflowType();
workflowType.setName("CloudWatchWorkflow.sendMetrics");
workflowType.setVersion("2.0");
return new CloudWatchWorkflowClientExternalImpl(
workflowExecution, workflowType, options, dataConverter, genericClient);
}
@Override
protected DeleteLoadBalancerWorkflowClient createClientInstance(
WorkflowExecution execution,
StartWorkflowOptions options,
DataConverter dataConverter,
GenericWorkflowClient genericClient) {
WorkflowType workflowType = new WorkflowType();
workflowType.setName("DeleteLoadBalancer");
workflowType.setVersion("1.0");
return new DeleteLoadBalancerWorkflowClientImpl(
execution, workflowType, options, dataConverter, genericClient);
}
public WorkflowTestBase(DecisionContext decisionContext) {
this.decisionContext = decisionContext;
workflowContext = (TestWorkflowContext) decisionContext.getWorkflowContext();
workflowClock = (TestWorkflowClock) decisionContext.getWorkflowClock();
WorkflowExecution we = new WorkflowExecution();
we.setWorkflowId("testWorkflowId");
we.setRunId("testRunId");
workflowContext.setWorkflowExecution(we);
WorkflowType wt = new WorkflowType();
wt.setName("testWorkflow");
wt.setVersion("0.0");
workflowContext.setWorkflowType(wt);
}
public void sendGCM(String registration_id, String message, int badge) throws IOException {
Object[] workflowInput = new Object[] {registration_id, message, badge};
DataConverter converter = new JsonDataConverter();
StartWorkflowExecutionRequest startWorkflowExecutionRequest =
new StartWorkflowExecutionRequest();
startWorkflowExecutionRequest.setInput(converter.toData(workflowInput));
startWorkflowExecutionRequest.setDomain("MobMonkey");
TaskList tasks = new TaskList();
tasks.setName("Gcm");
startWorkflowExecutionRequest.setTaskList(tasks);
WorkflowType workflowType = new WorkflowType();
workflowType.setName("GcmWorkflow.sendNotification");
workflowType.setVersion("1.12");
startWorkflowExecutionRequest.setWorkflowType(workflowType);
startWorkflowExecutionRequest.setWorkflowId(UUID.randomUUID().toString());
this.swfClient().startWorkflowExecution(startWorkflowExecutionRequest);
}
public void sendAPNS(String eMailAddress, String[] deviceIds, String message, int badge)
throws IOException {
Object[] workflowInput = new Object[] {eMailAddress, deviceIds, message, badge};
DataConverter converter = new JsonDataConverter();
StartWorkflowExecutionRequest startWorkflowExecutionRequest =
new StartWorkflowExecutionRequest();
startWorkflowExecutionRequest.setInput(converter.toData(workflowInput));
startWorkflowExecutionRequest.setDomain("MobMonkey");
TaskList tasks = new TaskList();
tasks.setName("Apns");
startWorkflowExecutionRequest.setTaskList(tasks);
WorkflowType workflowType = new WorkflowType();
workflowType.setName("ApnsWorkflow.sendNotification");
workflowType.setVersion("1.10");
startWorkflowExecutionRequest.setWorkflowType(workflowType);
startWorkflowExecutionRequest.setWorkflowId(UUID.randomUUID().toString());
this.swfClient().startWorkflowExecution(startWorkflowExecutionRequest);
}
protected WorkflowType getWorkflowType(
String interfaceName, Method method, Execute executeAnnotation) {
assert (method != null);
assert (executeAnnotation != null);
WorkflowType workflowType = new WorkflowType();
String workflowName = null;
if (executeAnnotation.name() != null && !executeAnnotation.name().isEmpty()) {
workflowName = executeAnnotation.name();
} else {
workflowName = interfaceName + "." + method.getName();
}
if (executeAnnotation.version().isEmpty()) {
throw new IllegalArgumentException(
"Empty value of the required \"version\" parameter of the @Execute annotation found on "
+ getMethodFullName(method));
}
workflowType.setName(workflowName);
workflowType.setVersion(executeAnnotation.version());
return workflowType;
}
public Request<StartWorkflowExecutionRequest> marshall(
StartWorkflowExecutionRequest startWorkflowExecutionRequest) {
if (startWorkflowExecutionRequest == null) {
throw new AmazonClientException("Invalid argument passed to marshall(...)");
}
Request<StartWorkflowExecutionRequest> request =
new DefaultRequest<StartWorkflowExecutionRequest>(
startWorkflowExecutionRequest, "AmazonSimpleWorkflow");
String target = "SimpleWorkflowService.StartWorkflowExecution";
request.addHeader("X-Amz-Target", target);
request.setHttpMethod(HttpMethodName.POST);
request.setResourcePath("");
try {
StringWriter stringWriter = new StringWriter();
JSONWriter jsonWriter = new JSONWriter(stringWriter);
jsonWriter.object();
if (startWorkflowExecutionRequest.getDomain() != null) {
jsonWriter.key("domain").value(startWorkflowExecutionRequest.getDomain());
}
if (startWorkflowExecutionRequest.getWorkflowId() != null) {
jsonWriter.key("workflowId").value(startWorkflowExecutionRequest.getWorkflowId());
}
WorkflowType workflowType = startWorkflowExecutionRequest.getWorkflowType();
if (workflowType != null) {
jsonWriter.key("workflowType");
jsonWriter.object();
if (workflowType.getName() != null) {
jsonWriter.key("name").value(workflowType.getName());
}
if (workflowType.getVersion() != null) {
jsonWriter.key("version").value(workflowType.getVersion());
}
jsonWriter.endObject();
}
TaskList taskList = startWorkflowExecutionRequest.getTaskList();
if (taskList != null) {
jsonWriter.key("taskList");
jsonWriter.object();
if (taskList.getName() != null) {
jsonWriter.key("name").value(taskList.getName());
}
jsonWriter.endObject();
}
if (startWorkflowExecutionRequest.getInput() != null) {
jsonWriter.key("input").value(startWorkflowExecutionRequest.getInput());
}
if (startWorkflowExecutionRequest.getExecutionStartToCloseTimeout() != null) {
jsonWriter
.key("executionStartToCloseTimeout")
.value(startWorkflowExecutionRequest.getExecutionStartToCloseTimeout());
}
com.amazonaws.internal.ListWithAutoConstructFlag<String> tagListList =
(com.amazonaws.internal.ListWithAutoConstructFlag<String>)
(startWorkflowExecutionRequest.getTagList());
if (tagListList != null && !(tagListList.isAutoConstruct() && tagListList.isEmpty())) {
jsonWriter.key("tagList");
jsonWriter.array();
for (String tagListListValue : tagListList) {
if (tagListListValue != null) {
jsonWriter.value(tagListListValue);
}
}
jsonWriter.endArray();
}
if (startWorkflowExecutionRequest.getTaskStartToCloseTimeout() != null) {
jsonWriter
.key("taskStartToCloseTimeout")
.value(startWorkflowExecutionRequest.getTaskStartToCloseTimeout());
}
if (startWorkflowExecutionRequest.getChildPolicy() != null) {
jsonWriter.key("childPolicy").value(startWorkflowExecutionRequest.getChildPolicy());
}
jsonWriter.endObject();
String snippet = stringWriter.toString();
byte[] content = snippet.getBytes(UTF8);
request.setContent(new StringInputStream(snippet));
request.addHeader("Content-Length", Integer.toString(content.length));
request.addHeader("Content-Type", "application/x-amz-json-1.0");
} catch (Throwable t) {
throw new AmazonClientException("Unable to marshall request to JSON: " + t.getMessage(), t);
}
return request;
}
public Request<StartWorkflowExecutionRequest> marshall(
StartWorkflowExecutionRequest startWorkflowExecutionRequest) {
if (startWorkflowExecutionRequest == null) {
throw new AmazonClientException("Invalid argument passed to marshall(...)");
}
Request<StartWorkflowExecutionRequest> request =
new DefaultRequest<StartWorkflowExecutionRequest>(
startWorkflowExecutionRequest, "AmazonSimpleWorkflow");
String target = "SimpleWorkflowService.StartWorkflowExecution";
request.addHeader("X-Amz-Target", target);
request.addHeader("Content-Type", "application/x-amz-json-1.0");
request.setHttpMethod(HttpMethodName.POST);
String uriResourcePath = "";
uriResourcePath = uriResourcePath.replaceAll("//", "/");
if (uriResourcePath.contains("?")) {
String queryString = uriResourcePath.substring(uriResourcePath.indexOf("?") + 1);
uriResourcePath = uriResourcePath.substring(0, uriResourcePath.indexOf("?"));
for (String s : queryString.split("[;&]")) {
String[] nameValuePair = s.split("=");
if (nameValuePair.length == 2) {
request.addParameter(nameValuePair[0], nameValuePair[1]);
} else {
request.addParameter(s, null);
}
}
}
request.setResourcePath(uriResourcePath);
try {
StringWriter stringWriter = new StringWriter();
JSONWriter jsonWriter = new JSONWriter(stringWriter);
jsonWriter.object();
if (startWorkflowExecutionRequest.getDomain() != null) {
jsonWriter.key("domain").value(startWorkflowExecutionRequest.getDomain());
}
if (startWorkflowExecutionRequest.getWorkflowId() != null) {
jsonWriter.key("workflowId").value(startWorkflowExecutionRequest.getWorkflowId());
}
WorkflowType workflowType = startWorkflowExecutionRequest.getWorkflowType();
if (workflowType != null) {
jsonWriter.key("workflowType");
jsonWriter.object();
if (workflowType.getName() != null) {
jsonWriter.key("name").value(workflowType.getName());
}
if (workflowType.getVersion() != null) {
jsonWriter.key("version").value(workflowType.getVersion());
}
jsonWriter.endObject();
}
TaskList taskList = startWorkflowExecutionRequest.getTaskList();
if (taskList != null) {
jsonWriter.key("taskList");
jsonWriter.object();
if (taskList.getName() != null) {
jsonWriter.key("name").value(taskList.getName());
}
jsonWriter.endObject();
}
if (startWorkflowExecutionRequest.getInput() != null) {
jsonWriter.key("input").value(startWorkflowExecutionRequest.getInput());
}
if (startWorkflowExecutionRequest.getExecutionStartToCloseTimeout() != null) {
jsonWriter
.key("executionStartToCloseTimeout")
.value(startWorkflowExecutionRequest.getExecutionStartToCloseTimeout());
}
java.util.List<String> tagListList = startWorkflowExecutionRequest.getTagList();
if (tagListList != null && tagListList.size() > 0) {
jsonWriter.key("tagList");
jsonWriter.array();
for (String tagListListValue : tagListList) {
if (tagListListValue != null) {
jsonWriter.value(tagListListValue);
}
}
jsonWriter.endArray();
}
if (startWorkflowExecutionRequest.getTaskStartToCloseTimeout() != null) {
jsonWriter
.key("taskStartToCloseTimeout")
.value(startWorkflowExecutionRequest.getTaskStartToCloseTimeout());
}
if (startWorkflowExecutionRequest.getChildPolicy() != null) {
jsonWriter.key("childPolicy").value(startWorkflowExecutionRequest.getChildPolicy());
}
jsonWriter.endObject();
String snippet = stringWriter.toString();
byte[] content = snippet.getBytes("UTF-8");
request.setContent(new StringInputStream(snippet));
request.addHeader("Content-Length", Integer.toString(content.length));
} catch (Throwable t) {
throw new AmazonClientException("Unable to marshall request to JSON: " + t.getMessage(), t);
}
return request;
}
