diff --git a/util/plancodec/id.go b/util/plancodec/id.go index 2b2e5e7e97..8b17660bfa 100644 --- a/util/plancodec/id.go +++ b/util/plancodec/id.go @@ -394,6 +394,8 @@ func PhysicalIDToTypeString(id int) string { return TypeBatchPointGet case typeClusterMemTableReader: return TypeClusterMemTableReader + case typeDataSourceID: + return TypeDataSource case typeLoadDataID: return TypeLoadData case typeTableSampleID: diff --git a/util/plancodec/id_test.go b/util/plancodec/id_test.go index 8a7addd23f..fe275d81b2 100644 --- a/util/plancodec/id_test.go +++ b/util/plancodec/id_test.go @@ -87,3 +87,9 @@ func TestPlanIDChanged(t *testing.T) { require.Equal(t, testcase.Expected, testcase.Value) } } + +func TestReverse(t *testing.T) { + for i := 1; i <= 55; i++ { + require.Equal(t, TypeStringToPhysicalID(PhysicalIDToTypeString(i)), i) + } +}