Function createTiDBDriver

backend/plugin/schema/tidb/testutil_test.go:14–39

createTiDBDriver creates and opens a TiDB driver connection

(ctx context.Context, host, port, database string)

// createTiDBDriver creates and opens a TiDB driver connection
func createTiDBDriver(ctx context.Context, host, port, database string) (*tidbdb.Driver, error) {
	driver := &tidbdb.Driver{}
	config := db.ConnectionConfig{
		DataSource: &storepb.DataSource{
			Type:     storepb.DataSourceType_ADMIN,
			Username: "root",
			Host:     host,
			Port:     port,
			Database: database,
		},
		Password: "",
		ConnectionContext: db.ConnectionContext{
			EngineVersion: "8.5.0",
			DatabaseName:  database,
		},
	}
	d, err := driver.Open(ctx, storepb.Engine_TIDB, config)
	if err != nil {
		return nil, err
	}
	tidbDriver, ok := d.(*tidbdb.Driver)
	if !ok {
		return nil, errors.Errorf("failed to cast to TiDB driver")
	}
	return tidbDriver, nil
}