address comment

Signed-off-by: bufferflies <1045931706@qq.com>
tikv · Aug 12, 2022 · 877c381 · 877c381
1 parent d226b87
commit 877c381
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/pd-simulator/simulator/task.go b/tools/pd-simulator/simulator/task.go
@@ -19,16 +19,16 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/docker/go-units"
 	"github.com/pingcap/kvproto/pkg/eraftpb"
 	"github.com/pingcap/kvproto/pkg/metapb"
 	"github.com/pingcap/kvproto/pkg/pdpb"
 	"github.com/tikv/pd/server/core"
 	"github.com/tikv/pd/tools/pd-analysis/analysis"
-	"github.com/tikv/pd/tools/pd-simulator/simulator/cases"
 )
 
 var (
-	chunkSize                = int64(4 * cases.KB)
+	chunkSize                = int64(4 * units.KiB)
 	maxSnapGeneratorPoolSize = uint32(2)
 	maxSnapReceivePoolSize   = uint32(4)
 	compressionRatio         = int64(2)