Added ability to stop generating values by plugins to benchmark faker

Author: iluhinsky

acronis-db-bench/engine/helpers.go

@@ -219,15 +219,15 @@ func (p *DataSetSourcePlugin) GenCommonFakeValue(columnType string, rz *benchmar
 	}
 
 	if row == nil {
-		return false, nil
+		return true, nil
 	}
 
 	p.currentValues = make(map[string]interface{}, len(row))
 	for i, value := range row {
 		p.currentValues[p.columns[i]] = value
 	}
 
-	return true, p.currentValues[columnType]
+	return false, p.currentValues[columnType]
 }
 
 func (p *DataSetSourcePlugin) GenFakeValue(columnType string, rz *benchmark.Randomizer, cardinality int, preGenerated map[string]interface{}) (bool, interface{}) {

acronis-db-bench/tenants-cache/tenant_generator.go

@@ -1003,7 +1003,7 @@ func (tc *TenantsCache) GenCommonFakeValue(columnType string, rz *benchmark.Rand
 		tc.benchmark.Exit(err.Error())
 	}
 
-	return true, tenantUUID
+	return false, tenantUUID
 }
 
 func (tc *TenantsCache) GenFakeValue(columnType string, rz *benchmark.Randomizer, cardinality int, preGenerated map[string]interface{}) (bool, interface{}) {

benchmark/faker.go

@@ -339,11 +339,16 @@ func (rz *Randomizer) GenFakeData(colConfs *[]DBFakeColumnConf, WithAutoInc bool
 	var preGenerated map[string]interface{}
 	for _, plugin := range rz.plugins {
 		for _, c := range *colConfs {
-			if exists, value := plugin.GenCommonFakeValue(c.ColumnType, rz, c.Cardinality); exists {
-				if preGenerated == nil {
-					preGenerated = make(map[string]interface{})
+			if shouldStop, value := plugin.GenCommonFakeValue(c.ColumnType, rz, c.Cardinality); !shouldStop {
+				if value != nil {
+					if preGenerated == nil {
+						preGenerated = make(map[string]interface{})
+					}
+					preGenerated[c.ColumnType] = value
 				}
-				preGenerated[c.ColumnType] = value
+			} else {
+				// End of generation
+				return nil, nil, nil
 			}
 		}
 	}
@@ -372,11 +377,16 @@ func (rz *Randomizer) GenFakeDataAsMap(colConfs *[]DBFakeColumnConf, WithAutoInc
 	var preGenerated map[string]interface{}
 	for _, plugin := range rz.plugins {
 		for _, c := range *colConfs {
-			if exists, value := plugin.GenCommonFakeValue(c.ColumnType, rz, c.Cardinality); exists {
-				if preGenerated == nil {
-					preGenerated = make(map[string]interface{})
+			if shouldStop, value := plugin.GenCommonFakeValue(c.ColumnType, rz, c.Cardinality); !shouldStop {
+				if value != nil {
+					if preGenerated == nil {
+						preGenerated = make(map[string]interface{})
+					}
+					preGenerated[c.ColumnType] = value
 				}
-				preGenerated[c.ColumnType] = value
+			} else {
+				// End of generation
+				return nil, nil
 			}
 		}
 	}