From 27ca8cc0092d482444422435db0eb06a115b07f7 Mon Sep 17 00:00:00 2001
From: Ming-Hung Tsai <mtsai@redhat.com>
Date: Mon, 17 Aug 2020 01:32:55 +0800
Subject: [PATCH] [block_counter] Add block_counter::clear()

---
 persistent-data/block_counter.h | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/persistent-data/block_counter.h b/persistent-data/block_counter.h
index ac9c68f..a7bd666 100644
--- a/persistent-data/block_counter.h
+++ b/persistent-data/block_counter.h
@@ -62,6 +62,10 @@ namespace persistent_data {
 			return stop_on_error_;
 		}
 
+		virtual void clear() {
+			counts_.clear();
+		}
+
 	private:
 		count_map counts_;
 		bool stop_on_error_;
@@ -86,6 +90,11 @@ namespace persistent_data {
 		base::run_set<block_address> const& get_visited() const {
 			return visited_;
 		}
+
+		virtual void clear() {
+			visited_.clear();
+		}
+
 	private:
 		base::run_set<block_address> visited_;
 	};