feat: skip index creation during loading MySQL snapshot

catalog/database.go

@@ -171,10 +171,8 @@ func (d *Database) createAllTable(ctx *sql.Context, name string, schema sql.Prim
 	}
 
 	// https://github.com/apecloud/myduckserver/issues/272
-	if !(mycontext.IsReplicationQuery(ctx) && configuration.IsReplicationWithoutIndex()) {
-		if len(primaryKeys) > 0 {
-			b.WriteString(fmt.Sprintf(", PRIMARY KEY (%s)", strings.Join(primaryKeys, ", ")))
-		}
+	if len(primaryKeys) > 0 && !isIndexCreationDisabled(ctx) {
+		b.WriteString(fmt.Sprintf(", PRIMARY KEY (%s)", strings.Join(primaryKeys, ", ")))
 	}
 
 	b.WriteString(")")
@@ -205,6 +203,23 @@ func (d *Database) createAllTable(ctx *sql.Context, name string, schema sql.Prim
 	return nil
 }
 
+func isIndexCreationDisabled(ctx *sql.Context) bool {
+	if !configuration.IsReplicationWithoutIndex() {
+		return false
+	}
+	if mycontext.IsReplicationQuery(ctx) {
+		return true
+	}
+	_, vv, ok := sql.SystemVariables.GetGlobal("replica_is_loading_snapshot")
+	if !ok {
+		return false
+	}
+	if b, ok := vv.(int8); ok {
+		return b != 0
+	}
+	return false
+}
+
 // CreateTable implements sql.TableCreator.
 func (d *Database) CreateTable(ctx *sql.Context, name string, schema sql.PrimaryKeySchema, collation sql.CollationID, comment string) error {
 	d.mu.Lock()

catalog/table.go

@@ -8,7 +8,6 @@ import (
 
 	"github.com/apecloud/myduckserver/adapter"
 	"github.com/apecloud/myduckserver/configuration"
-	"github.com/apecloud/myduckserver/mycontext"
 	"github.com/dolthub/go-mysql-server/sql"
 	"github.com/dolthub/go-mysql-server/sql/expression"
 	"github.com/marcboeker/go-duckdb"
@@ -362,7 +361,7 @@ func (t *Table) CreateIndex(ctx *sql.Context, indexDef sql.IndexDef) error {
 	defer t.mu.Unlock()
 
 	// https://github.com/apecloud/myduckserver/issues/272
-	if mycontext.IsReplicationQuery(ctx) && configuration.IsReplicationWithoutIndex() {
+	if isIndexCreationDisabled(ctx) {
 		return nil
 	}
 

configuration/env.go

@@ -11,7 +11,7 @@ const (
 
 func IsReplicationWithoutIndex() bool {
 	switch strings.ToLower(os.Getenv(replicationWithoutIndex)) {
-	case "", "t", "1", "true":
+	case "", "y", "t", "1", "on", "yes", "true":
 		return true
 	}
 	return false

devtools/replica-setup-mysql/prepare.sh

@@ -17,4 +17,5 @@ echo "Setting local_infile and server_id..."
 mysqlsh --sql --host=${MYDUCK_HOST} --port=${MYDUCK_PORT}  --user=root --no-password <<EOF
 SET GLOBAL local_infile = 1;
 SET GLOBAL server_id = ${MYDUCK_SERVER_ID};
+SET GLOBAL replica_is_loading_snapshot = ON;
 EOF

devtools/replica-setup-mysql/snapshot.sh

@@ -71,4 +71,9 @@ else
 fi
 
 
-echo "Snapshot completed successfully."
+echo "Snapshot completed successfully."
+
+echo "Reset replica_is_loading_snapshot..."
+mysqlsh --sql --host=${MYDUCK_HOST} --port=${MYDUCK_PORT}  --user=root --no-password <<EOF
+SET GLOBAL replica_is_loading_snapshot = OFF;
+EOF

replica/options.go

@@ -48,5 +48,14 @@ func RegisterReplicaOptions(options *ReplicaOptions) {
 			Type:              types.NewSystemStringType("report_password"),
 			Default:           options.ReportPassword,
 		},
+		// MyDuck-specific system variables
+		&sql.MysqlSystemVariable{
+			Name:              "replica_is_loading_snapshot",
+			Scope:             sql.GetMysqlScope(sql.SystemVariableScope_Global),
+			Dynamic:           true,
+			SetVarHintApplies: false,
+			Type:              types.NewSystemBoolType("replica_is_loading_snapshot"),
+			Default:           false,
+		},
 	})
 }