fix: instrumentation

Author: not-matthias

google_benchmark/include/benchmark/benchmark.h

@@ -1065,9 +1065,16 @@ struct State::StateIterator {
   bool operator!=(StateIterator const&) const {
     if (BENCHMARK_BUILTIN_EXPECT(cached_ != 0, true)) return true;
 
+
 #if defined(CODSPEED_INSTRUMENTATION) || defined(CODSPEED_WALLTIME)
     measurement_stop();
 #endif
+#ifdef CODSPEED_INSTRUMENTATION
+    if (parent_->codspeed_ != NULL) {
+      parent_->codspeed_->end_benchmark();
+    }
+#endif
+
     parent_->FinishKeepRunning();
     return false;
   }
@@ -1083,6 +1090,11 @@ inline BENCHMARK_ALWAYS_INLINE State::StateIterator State::begin() {
 inline BENCHMARK_ALWAYS_INLINE State::StateIterator State::end() {
   StartKeepRunning();
 
+#ifdef CODSPEED_INSTRUMENTATION
+    if (parent_->codspeed_ != NULL) {
+      parent_->codspeed_->start_benchmark();
+    }
+#endif
 #if defined(CODSPEED_INSTRUMENTATION) || defined(CODSPEED_WALLTIME)
   measurement_start();
 #endif