diff --git a/lts-spring/src/main/java/com/github/ltsopensource/spring/tasktracker/JobRunnerBuilder.java b/lts-spring/src/main/java/com/github/ltsopensource/spring/tasktracker/JobRunnerBuilder.java index 29bfbdae8..9767c101b 100644 --- a/lts-spring/src/main/java/com/github/ltsopensource/spring/tasktracker/JobRunnerBuilder.java +++ b/lts-spring/src/main/java/com/github/ltsopensource/spring/tasktracker/JobRunnerBuilder.java @@ -18,25 +18,32 @@ public static JobRunner build(final Object targetObject, final Method targetMeth return new JobRunner() { @Override public Result run(JobContext jobContext) throws Throwable { + Object resultObj; if (pTypes == null || pTypes.length == 0) { - return (Result) targetMethod.invoke(targetObject); - } - Object[] pTypeValues = new Object[pTypes.length]; - - for (int i = 0; i < pTypes.length; i++) { - if (pTypes[i] == Job.class) { - pTypeValues[i] = jobContext.getJob(); - } else if (pTypes[i] == JobContext.class) { - pTypeValues[i] = jobContext; - } else { - pTypeValues[i] = null; + resultObj = targetMethod.invoke(targetObject); + }else{ + + Object[] pTypeValues = new Object[pTypes.length]; + + for (int i = 0; i < pTypes.length; i++) { + if (pTypes[i] == Job.class) { + pTypeValues[i] = jobContext.getJob(); + } else if (pTypes[i] == JobContext.class) { + pTypeValues[i] = jobContext; + } else { + pTypeValues[i] = null; + } } + + resultObj = targetMethod.invoke(targetObject, pTypeValues); } + Class returnType = targetMethod.getReturnType(); if (returnType != Result.class) { return new Result(Action.EXECUTE_SUCCESS); } - return (Result) targetMethod.invoke(targetObject, pTypeValues); + + return (Result) resultObj; } }; }