schema: More partition key types

scylladb · Mar 8, 2019 · 305ac63 · 305ac63
1 parent f4c1eaf
commit 305ac63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/schema.go b/schema.go
@@ -99,7 +99,7 @@ func GenSchema() *Schema {
 	partitionKeys := []ColumnDef{}
 	numPartitionKeys := rand.Intn(MaxPartitionKeys-1) + 1
 	for i := 0; i < numPartitionKeys; i++ {
-		partitionKeys = append(partitionKeys, ColumnDef{Name: genColumnName("pk", i), Type: "int"})
+		partitionKeys = append(partitionKeys, ColumnDef{Name: genColumnName("pk", i), Type: genColumnType()})
 	}
 	clusteringKeys := []ColumnDef{}
 	numClusteringKeys := rand.Intn(MaxClusteringKeys)