add test case

Signed-off-by: joccau <[email protected]>
pingcap · Nov 2, 2022 · 96eb444 · 96eb444
1 parent f250079
commit 96eb444
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 3 deletions.
diff --git a/br/pkg/streamhelper/client.go b/br/pkg/streamhelper/client.go
@@ -404,7 +404,11 @@ func (t *Task) GetGlobalCheckPointTS(ctx context.Context) (uint64, error) {
 	initialized := false
 	checkpoint := t.Info.StartTs
 	for _, cp := range checkPointMap {
-		if !initialized || cp.TS < checkpoint {
+		if cp.Type() == CheckpointTypeGlobal {
+			return cp.TS, nil
+		}
+
+		if cp.Type() == CheckpointTypeStore && (!initialized || cp.TS < checkpoint) {
 			initialized = true
 			checkpoint = cp.TS
 		}

diff --git a/br/pkg/streamhelper/integration_test.go b/br/pkg/streamhelper/integration_test.go
@@ -13,6 +13,7 @@ import (
 	"testing"
 
 	"github.com/pingcap/errors"
+	backup "github.com/pingcap/kvproto/pkg/brpb"
 	"github.com/pingcap/log"
 	berrors "github.com/pingcap/tidb/br/pkg/errors"
 	"github.com/pingcap/tidb/br/pkg/logutil"
@@ -138,7 +139,8 @@ func TestIntegration(t *testing.T) {
 	metaCli := streamhelper.MetaDataClient{Client: cli}
 	t.Run("TestBasic", func(t *testing.T) { testBasic(t, metaCli, etcd) })
 	t.Run("TestForwardProgress", func(t *testing.T) { testForwardProgress(t, metaCli, etcd) })
-	t.Run("testGetStorageCheckpoint", func(t *testing.T) { testGetStorageCheckpoint(t, metaCli, etcd) })
+	t.Run("testGetStorageCheckpoint", func(t *testing.T) { testGetStorageCheckpoint(t, metaCli) })
+	t.Run("testGetGlobalCheckPointTS", func(t *testing.T) { testGetGlobalCheckPointTS(t, metaCli) })
 	t.Run("TestStreamListening", func(t *testing.T) { testStreamListening(t, streamhelper.AdvancerExt{MetaDataClient: metaCli}) })
 	t.Run("TestStreamCheckpoint", func(t *testing.T) { testStreamCheckpoint(t, streamhelper.AdvancerExt{MetaDataClient: metaCli}) })
 }
@@ -233,7 +235,7 @@ func testForwardProgress(t *testing.T, metaCli streamhelper.MetaDataClient, etcd
 	require.Equal(t, store2Checkpoint, uint64(40))
 }
 
-func testGetStorageCheckpoint(t *testing.T, metaCli streamhelper.MetaDataClient, etcd *embed.Etcd) {
+func testGetStorageCheckpoint(t *testing.T, metaCli streamhelper.MetaDataClient) {
 	var (
 		taskName = "my_task"
 		ctx      = context.Background()
@@ -264,6 +266,44 @@ func testGetStorageCheckpoint(t *testing.T, metaCli streamhelper.MetaDataClient,
 	ts, err := task.GetStorageCheckpoint(ctx)
 	require.NoError(t, err)
 	require.Equal(t, uint64(10002), ts)
+
+	ts, err = task.GetGlobalCheckPointTS(ctx)
+	require.NoError(t, err)
+	require.Equal(t, uint64(10002), ts)
+}
+
+func testGetGlobalCheckPointTS(t *testing.T, metaCli streamhelper.MetaDataClient) {
+	var (
+		taskName = "my_task"
+		ctx      = context.Background()
+		value    = make([]byte, 8)
+	)
+
+	cases := []struct {
+		storeID           string
+		storageCheckPoint uint64
+	}{
+		{
+			"1",
+			10001,
+		}, {
+			"2",
+			10002,
+		},
+	}
+	for _, c := range cases {
+		key := path.Join(streamhelper.StorageCheckpointOf(taskName), c.storeID)
+		binary.BigEndian.PutUint64(value, c.storageCheckPoint)
+		_, err := metaCli.Put(ctx, key, string(value))
+		require.NoError(t, err)
+	}
+
+	task := streamhelper.NewTask(&metaCli, backup.StreamBackupTaskInfo{Name: taskName})
+	task.UploadGlobalCheckpoint(ctx, 1003)
+
+	globalTS, err := task.GetGlobalCheckPointTS(ctx)
+	require.NoError(t, err)
+	require.Equal(t, globalTS, uint64(1003))
 }
 
 func testStreamListening(t *testing.T, metaCli streamhelper.AdvancerExt) {