Find methods that have been removed

avantgardnerio · Oct 31, 2018 · e433694 · e433694
1 parent d66c04d
commit e433694
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/src/main/kotlin/net/squarelabs/depends/Reporter.kt b/src/main/kotlin/net/squarelabs/depends/Reporter.kt
@@ -0,0 +1,28 @@
+package net.squarelabs.depends
+
+import net.squarelabs.depends.models.Artifact
+
+fun fqmns(a: Artifact): HashSet<String> = HashSet(a.classes.values.flatMap { c -> c.methods.values.map { m -> "${c.name}:${m.name}${m.descriptor}" } })
+
+fun allMethods(artifacts: Collection<Artifact>): HashSet<String> {
+    val reducer = { acc: HashSet<String>, cur: Artifact ->
+        acc.addAll(fqmns(cur))
+        acc
+    }
+    return artifacts.fold(HashSet(), reducer)
+}
+
+fun commonMethods(artifacts: Collection<Artifact>, all: Set<String>): HashSet<String> {
+    val reducer = { acc: HashSet<String>, cur: Artifact ->
+        acc.retainAll(fqmns(cur))
+        acc
+    }
+    return artifacts.fold(HashSet(all), reducer)
+}
+
+fun removedMethods(artifacts: Collection<Artifact>): Set<String> {
+    val accumulator: HashSet<String> = allMethods(artifacts)
+    val common: Set<String> = commonMethods(artifacts, accumulator)
+    accumulator.removeAll(common)
+    return accumulator
+}
diff --git a/src/test/kotlin/net/squarelabs/depends/ResolverTest.kt b/src/test/kotlin/net/squarelabs/depends/ResolverTest.kt
@@ -19,8 +19,11 @@ class ResolverTest {
         println("conflicts:\n\t $conflicts")
 
         // find removals
-        state.artifactsByGa.keys
+        val removedMethods = state.artifactsByGa.keys
                 .filter { key -> state.artifactsByGa[key]!!.size > 1 }
+                .map { key -> state.artifactsByGa[key]!!.values }
+                .flatMap { artifacts -> removedMethods(artifacts) }
+        println("methods removed: ${removedMethods.size}")
 
         // print memory
         Runtime.getRuntime().gc()