Repository: reef Updated Branches: refs/heads/master a44cd58ea -> b4595db95
[REEF-1495] Fix suspend example JIRA: [REEF-1495](https://issues.apache.org/jira/browse/REEF-1495) Pull Request: Closes #1076 Project: http://git-wip-us.apache.org/repos/asf/reef/repo Commit: http://git-wip-us.apache.org/repos/asf/reef/commit/b4595db9 Tree: http://git-wip-us.apache.org/repos/asf/reef/tree/b4595db9 Diff: http://git-wip-us.apache.org/repos/asf/reef/diff/b4595db9 Branch: refs/heads/master Commit: b4595db9574eacaff3244a63eada77d1d5a98703 Parents: a44cd58 Author: bellatoris <[email protected]> Authored: Wed Jul 20 11:52:03 2016 +0900 Committer: Sergey Dudoladov <[email protected]> Committed: Mon Jul 25 13:33:50 2016 -0700 ---------------------------------------------------------------------- .../java/org/apache/reef/examples/suspend/SuspendDriver.java | 4 ++++ .../java/org/apache/reef/examples/suspend/SuspendTestTask.java | 6 ++---- 2 files changed, 6 insertions(+), 4 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/reef/blob/b4595db9/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendDriver.java ---------------------------------------------------------------------- diff --git a/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendDriver.java b/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendDriver.java index f7ed1c2..731eb06 100644 --- a/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendDriver.java +++ b/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendDriver.java @@ -248,6 +248,7 @@ public class SuspendDriver { .set(TaskConfiguration.IDENTIFIER, context.getId() + "_task") .set(TaskConfiguration.TASK, SuspendTestTask.class) .set(TaskConfiguration.ON_SUSPEND, SuspendTestTask.SuspendHandler.class) + .set(TaskConfiguration.ON_SEND_MESSAGE, SuspendTestTask.class) .build()); } catch (final BindException ex) { LOG.log(Level.SEVERE, "Bad Task configuration for context: " + context.getId(), ex); @@ -295,6 +296,9 @@ public class SuspendDriver { try { suspendedTask.getActiveContext().submitTask(TaskConfiguration.CONF .set(TaskConfiguration.IDENTIFIER, taskId) + .set(TaskConfiguration.TASK, SuspendTestTask.class) + .set(TaskConfiguration.ON_SUSPEND, SuspendTestTask.SuspendHandler.class) + .set(TaskConfiguration.ON_SEND_MESSAGE, SuspendTestTask.class) .set(TaskConfiguration.MEMENTO, DatatypeConverter.printBase64Binary(suspendedTask.get())) .build()); http://git-wip-us.apache.org/repos/asf/reef/blob/b4595db9/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendTestTask.java ---------------------------------------------------------------------- diff --git a/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendTestTask.java b/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendTestTask.java index 1222f01..8c667d7 100644 --- a/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendTestTask.java +++ b/lang/java/reef-examples/src/main/java/org/apache/reef/examples/suspend/SuspendTestTask.java @@ -169,13 +169,11 @@ public class SuspendTestTask implements Task, TaskMessageSource { * Handler for suspend event. */ public class SuspendHandler implements EventHandler<SuspendEvent> { - @Override public void onNext(final SuspendEvent suspendEvent) { - final byte[] message = suspendEvent.get().get(); synchronized (SuspendTestTask.this) { - LOG.log(Level.INFO, "Suspend: {0} with: {1} bytes; counter: {2}", - new Object[]{this, message.length, SuspendTestTask.this.counter}); + LOG.log(Level.INFO, "Suspend: {0}; counter: {1}", + new Object[]{this, SuspendTestTask.this.counter}); SuspendTestTask.this.suspended = true; SuspendTestTask.this.notify(); }
