Merge branch 'complete' of github.com:ClSlaid/loadgen into complete

ClSlaid · Aug 20, 2021 · b47fe8a · b47fe8a
2 parents 578fef7 + 8ba13bb
commit b47fe8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/data/load_data.go b/data/load_data.go
@@ -98,6 +98,9 @@ func (c *LoadDataSuite) splitTableRegion(db *sql.DB, t *TableInfo, rows, regionR
 		return
 	}
 	regionNum := rows / regionRowNum
+	if regionNum <= 1 {
+		return
+	}
 	fmt.Printf("split %v regions for table %v\n", regionNum, t.DBTableName())
 
 	split := fmt.Sprintf("split table %v between (0) and (%v) regions %v;", t.DBTableName(), rows, regionNum)

diff --git a/payload/basic_query.go b/payload/basic_query.go
@@ -162,8 +162,8 @@ func (c *basicQuerySuite) prepare() error {
 		return err
 	}
 	c.tblInfo = tblInfo
-	load := data.NewLoadDataSuite(c.cfg)
-	return load.Prepare(tblInfo, c.rows, c.rows/100000)
+	load := data.NewLoadDataSuit(c.cfg)
+	return load.Prepare(tblInfo, c.rows, 100000)
 }
 
 func (c *basicQuerySuite) Run() error {