From 7fa9d75f9996f74755a59ca99087dcfde2d13d54 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 15 Jul 2016 22:58:24 +0100
Subject: Remove Guava functions from NodeTransformer

---
 .../java/org/zwobble/couscous/ast/visitors/NodeTransformer.java   | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/main/java/org/zwobble/couscous/ast')

diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
index fb344e0..a1bbda6 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -1,9 +1,7 @@
 package org.zwobble.couscous.ast.visitors;
 
-import com.google.common.base.Function;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
-import com.google.common.collect.Iterables;
 import org.zwobble.couscous.ast.*;
 import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.types.Type;
@@ -14,7 +12,9 @@ import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
+import java.util.function.Function;
 
+import static org.zwobble.couscous.util.ExtraIterables.lazyMap;
 import static org.zwobble.couscous.util.ExtraLists.*;
 
 public class NodeTransformer {
@@ -233,10 +233,10 @@ public class NodeTransformer {
     }
 
     private <T> Set<T> transformSet(Set<T> values, Function<T, T> function) {
-        return ImmutableSet.copyOf(Iterables.transform(values, function));
+        return ImmutableSet.copyOf(lazyMap(values, function));
     }
 
     private <T> List<T> transformList(List<? extends T> values, Function<T, T> function) {
-        return ImmutableList.copyOf(Iterables.transform(values, function));
+        return ImmutableList.copyOf(lazyMap(values, function));
     }
 }
-- 
cgit v1.2.3