support global index (#82)

framework/mainloop.go

@@ -5,12 +5,6 @@ import (
 	"database/sql"
 	"encoding/json"
 	"fmt"
-	"github.com/PingCAP-QE/schrddl/dump"
-	"github.com/PingCAP-QE/schrddl/norec"
-	"github.com/PingCAP-QE/schrddl/reduce"
-	"github.com/pingcap/tidb/pkg/parser"
-	"github.com/pingcap/tidb/pkg/util/logutil"
-	"go.uber.org/zap"
 	crc322 "hash/crc32"
 	"math/rand"
 	"os"
@@ -22,6 +16,13 @@ import (
 	"time"
 	"unsafe"
 
+	"github.com/PingCAP-QE/schrddl/dump"
+	"github.com/PingCAP-QE/schrddl/norec"
+	"github.com/PingCAP-QE/schrddl/reduce"
+	"github.com/pingcap/tidb/pkg/parser"
+	"github.com/pingcap/tidb/pkg/util/logutil"
+	"go.uber.org/zap"
+
 	"github.com/PingCAP-QE/schrddl/sqlgenerator"
 	"github.com/juju/errors"
 	"github.com/ngaut/log"

sqlgenerator/rule_index.go

@@ -63,12 +63,12 @@ var IndexDefinitionColumn = NewFn(func(state *State) Fn {
 	tbl := state.env.Table
 	idx := state.env.Index
 	partCol := state.env.PartColumn
-	if partCol != nil && idx.IsUnique() && !idx.Columns.Contain(partCol) {
+	// Global index supports primary key with non-clustered,
+	// but we don't decide the index is clustered or not at this time,
+	// so only check is primary or not here.
+	if partCol != nil && idx.Tp == IndexTypePrimary && !idx.Columns.Contain(partCol) {
 		state.env.IdxColumn = partCol
-		return Or(
-			IndexDefinitionColumnNoPrefix,
-			IndexDefinitionColumnPrefix.P(IndexColumnPrefixable),
-		)
+		return IndexDefinitionColumnNoPrefix
 	}
 	totalCols := tbl.Columns.Filter(func(c *Column) bool {
 		return !idx.HasColumn(c) && !state.env.MultiObjs.SameObject(c.Name)

sqlgenerator/rule_partition.go

@@ -28,8 +28,7 @@ var PartitionDefinitionHash = NewFn(func(state *State) Fn {
 })
 
 var PartitionDefinitionKey = NewFn(func(state *State) Fn {
-	partitionedCol := state.env.Table.Columns.Filter(func(c *Column) bool { return c.Tp.IsKeyPartitionType() }).Rand()
-	state.env.PartColumn = partitionedCol
+	partitionedCol := state.env.PartColumn
 	partitionNum := RandomNum(1, 6)
 	return And(
 		Str("partition by key ("),