From e7fa117ff8d9c9f6b82aa94dcfbef3ac05cbe68a Mon Sep 17 00:00:00 2001
From: Oliver Schneider <git-spam-no-reply9815368754983@oli-obk.de>
Date: Fri, 4 Mar 2016 16:27:03 +0100
Subject: [PATCH] simplify cyclomatic complexity auxiliarly value computation

previously the HIR was unnecessarily traversed twice
---
 src/cyclomatic_complexity.rs | 49 +++++++++++++++++-------------------
 1 file changed, 23 insertions(+), 26 deletions(-)

diff --git a/src/cyclomatic_complexity.rs b/src/cyclomatic_complexity.rs
index 7034f87f59fa..05f633ee0e40 100644
--- a/src/cyclomatic_complexity.rs
+++ b/src/cyclomatic_complexity.rs
@@ -48,18 +48,22 @@ impl CyclomaticComplexity {
         let n = cfg.graph.len_nodes() as u64;
         let e = cfg.graph.len_edges() as u64;
         let cc = e + 2 - n;
-        let mut arm_counter = MatchArmCounter(0);
-        arm_counter.visit_block(block);
-        let narms = arm_counter.0;
+        let mut helper = CCHelper {
+            match_arms: 0,
+            divergence: 0,
+            tcx: &cx.tcx,
+        };
+        helper.visit_block(block);
+        let CCHelper {
+            match_arms,
+            divergence,
+            ..
+        } = helper;
 
-        let mut diverge_counter = DivergenceCounter(0, &cx.tcx);
-        diverge_counter.visit_block(block);
-        let divergence = diverge_counter.0;
-
-        if cc + divergence < narms {
-            report_cc_bug(cx, cc, narms, divergence, span);
+        if cc + divergence < match_arms {
+            report_cc_bug(cx, cc, match_arms, divergence, span);
         } else {
-            let rust_cc = cc + divergence - narms;
+            let rust_cc = cc + divergence - match_arms;
             if rust_cc > self.limit.limit() {
                 span_help_and_lint(cx,
                                    CYCLOMATIC_COMPLEXITY,
@@ -98,35 +102,28 @@ impl LateLintPass for CyclomaticComplexity {
     }
 }
 
-struct MatchArmCounter(u64);
+struct CCHelper<'a, 'tcx: 'a> {
+    match_arms: u64,
+    divergence: u64,
+    tcx: &'a ty::TyCtxt<'tcx>,
+}
 
-impl<'a> Visitor<'a> for MatchArmCounter {
+impl<'a, 'b, 'tcx> Visitor<'a> for CCHelper<'b, 'tcx> {
     fn visit_expr(&mut self, e: &'a Expr) {
         match e.node {
             ExprMatch(_, ref arms, _) => {
                 walk_expr(self, e);
                 let arms_n: u64 = arms.iter().map(|arm| arm.pats.len() as u64).sum();
                 if arms_n > 1 {
-                    self.0 += arms_n - 2;
+                    self.match_arms += arms_n - 2;
                 }
             }
-            ExprClosure(..) => {}
-            _ => walk_expr(self, e),
-        }
-    }
-}
-
-struct DivergenceCounter<'a, 'tcx: 'a>(u64, &'a ty::TyCtxt<'tcx>);
-
-impl<'a, 'b, 'tcx> Visitor<'a> for DivergenceCounter<'b, 'tcx> {
-    fn visit_expr(&mut self, e: &'a Expr) {
-        match e.node {
             ExprCall(ref callee, _) => {
                 walk_expr(self, e);
-                let ty = self.1.node_id_to_type(callee.id);
+                let ty = self.tcx.node_id_to_type(callee.id);
                 if let ty::TyBareFn(_, ty) = ty.sty {
                     if ty.sig.skip_binder().output.diverges() {
-                        self.0 += 1;
+                        self.divergence += 1;
                     }
                 }
             }