Function TestCreateTableConsistent

br/pkg/restore/internal/prealloc_db/db_test.go:567–680

(t *testing.T)

}

func TestCreateTableConsistent(t *testing.T) {
	ctx := context.Background()
	allocator := testAllocator(0)
	s := utiltest.CreateRestoreSchemaSuite(t)
	tk := testkit.NewTestKit(t, s.Mock.Storage)
	tk.MustExec("use test")
	tk.MustExec("set @@sql_mode=''")

	db, supportPolicy, err := preallocdb.NewDB(gluetidb.New(), s.Mock.Storage, "STRICT")
	require.NoError(t, err)
	require.True(t, supportPolicy)
	defer db.Close()

	getTableInfo := func(name string) (*model.DBInfo, *model.TableInfo) {
		info, err := s.Mock.Domain.GetSnapshotInfoSchema(math.MaxUint64)
		require.NoError(t, err)
		dbInfo, exists := info.SchemaByName(pmodel.NewCIStr("test"))
		require.True(t, exists)
		tableInfo, err := info.TableByName(context.Background(), pmodel.NewCIStr("test"), pmodel.NewCIStr(name))
		require.NoError(t, err)
		return dbInfo, tableInfo.Meta()
	}
	tk.MustExec("create sequence test.s increment by 1 minvalue = 10;")
	dbInfo, seqInfo := getTableInfo("s")
	tk.MustExec("drop sequence test.s;")

	preallocId := func(tables []*metautil.Table) {
		ids, allocErr := prealloctableid.New(tables)
		require.NoError(t, allocErr)
		ids.PreallocIDs(&allocator)
		db.RegisterPreallocatedIDs(ids)
		allocator += testAllocator(len(tables))
	}

	newSeqInfo := seqInfo.Clone()
	newSeqInfo.ID += 100
	newTables := []*metautil.Table{
		{
			DB:   dbInfo.Clone(),
			Info: newSeqInfo,
		},
	}

	preallocId(newTables)
	err = db.CreateTables(ctx, newTables, nil, false, nil)
	require.NoError(t, err)
	r11 := tk.MustQuery("select nextval(s)").Rows()
	r12 := tk.MustQuery("show create table test.s").Rows()

	tk.MustExec("drop sequence test.s;")

	newSeqInfo = seqInfo.Clone()
	newSeqInfo.ID += 100
	newTable := &metautil.Table{DB: dbInfo.Clone(), Info: newSeqInfo}
	preallocId([]*metautil.Table{newTable})
	err = db.CreateTable(ctx, newTable, nil, false, nil)
	require.NoError(t, err)
	r21 := tk.MustQuery("select nextval(s)").Rows()