add a helper function for new ru details (tikv#1078)

Signed-off-by: glorv <[email protected]>
pingyu · Dec 20, 2023 · 4ce1e45 · 4ce1e45
1 parent 6f9ba83
commit 4ce1e45
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/util/execdetails.go b/util/execdetails.go
@@ -703,6 +703,16 @@ func NewRUDetails() *RUDetails {
 	}
 }
 
+// NewRUDetails creates a new RUDetails with specifical values.
+// This function is used in tidb's unit test.
+func NewRUDetailsWith(rru, wru float64, waitDur time.Duration) *RUDetails {
+	return &RUDetails{
+		readRU:         uatomic.NewFloat64(rru),
+		writeRU:        uatomic.NewFloat64(wru),
+		ruWaitDuration: uatomic.NewDuration(waitDur),
+	}
+}
+
 // Clone implements the RuntimeStats interface.
 func (rd *RUDetails) Clone() *RUDetails {
 	return &RUDetails{