Move `addUse()` return statement to `keep()` (#243)

1 files changed, 4 insertions, 7 deletions

diff --git a/chapter24/src/main/java/com/seaofnodes/simple/node/Node.java b/chapter24/src/main/java/com/seaofnodes/simple/node/Node.java
index 1979dcd1..458a3407 100644
--- a/chapter24/src/main/java/com/seaofnodes/simple/node/Node.java
+++ b/chapter24/src/main/java/com/seaofnodes/simple/node/Node.java
@@ -229,8 +229,7 @@ public abstract class Node implements Cloneable {
     }
 
     // Breaks the edge invariants, used temporarily
-    @SuppressWarnings("unchecked")
-    protected <N extends Node> N addUse(Node n) { _outputs.add(n); return (N)this; }
+    protected <N extends Node> void addUse(N n) { _outputs.add(n); }
 
     // Remove node 'use' from 'def's (i.e. our) output list, by compressing the list in-place.
     // Return true if the output list is empty afterward.
@@ -285,13 +284,11 @@ public abstract class Node implements Cloneable {
 
     // Shortcuts to stop DCE mid-parse
     // Add bogus null use to keep node alive
-    public <N extends Node> N keep() { return addUse(null); }
+    @SuppressWarnings("unchecked")
+    public <N extends Node> N keep() { addUse(null); return (N)this; }
     // Remove bogus null.
     @SuppressWarnings("unchecked")
-    public <N extends Node> N unkeep() {
-        delUse(null);
-        return (N)this;
-    }
+    public <N extends Node> N unkeep() { delUse(null); return (N)this; }
     // Test "keep" status
     public boolean iskeep() { return _outputs.find(null) != -1; }
     public void unkill() {