milvus/internal/rootcoord/ddl_callbacks_database_test.go
Zhen Ye 02e2170601
enhance: cherry pick patch of new DDL framework and CDC 2 (#45241)
issue: #43897, #44123
pr: #45224
also pick pr: #45216,#45154,#45033,#45145,#45092,#45058,#45029

enhance: Close channel replicator more gracefully (#45029)

issue: https://github.com/milvus-io/milvus/issues/44123

enhance: Show create time for import job (#45058)

issue: https://github.com/milvus-io/milvus/issues/45056

fix: wal state may be unconsistent after recovering from crash (#45092)

issue: #45088, #45086

- Message on control channel should trigger the checkpoint update.
- LastConfrimedMessageID should be recovered from the minimum of
checkpoint or the LastConfirmedMessageID of uncommitted txn.
- Add more log info for wal debugging.

fix: make ack of broadcaster cannot canceled by client (#45145)

issue: #45141

- make ack of broadcaster cannot canceled by rpc.
- make clone for assignment snapshot of wal balancer.
- add server id for GetReplicateCheckpoint to avoid failure.

enhance: support collection and index with WAL-based DDL framework
(#45033)

issue: #43897

- Part of collection/index related DDL is implemented by WAL-based DDL
framework now.
- Support following message type in wal, CreateCollection,
DropCollection, CreatePartition, DropPartition, CreateIndex, AlterIndex,
DropIndex.
- Part of collection/index related DDL can be synced by new CDC now.
- Refactor some UT for collection/index DDL.
- Add Tombstone scheduler to manage the tombstone GC for collection or
partition meta.
- Move the vchannel allocation into streaming pchannel manager.

enhance: support load/release collection/partition with WAL-based DDL
framework (#45154)

issue: #43897

- Load/Release collection/partition is implemented by WAL-based DDL
framework now.
- Support AlterLoadConfig/DropLoadConfig in wal now.
- Load/Release operation can be synced by new CDC now.
- Refactor some UT for load/release DDL.

enhance: Don't start cdc by default (#45216)

issue: https://github.com/milvus-io/milvus/issues/44123


fix: unrecoverable when replicate from old (#45224)

issue: #44962

---------

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
Signed-off-by: chyezh <chyezh@outlook.com>
Co-authored-by: yihao.dai <yihao.dai@zilliz.com>
2025-11-04 01:35:33 +08:00

85 lines
2.9 KiB
Go

// Licensed to the LF AI & Data foundation under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package rootcoord
import (
"context"
"testing"
"github.com/stretchr/testify/require"
"github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
"github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
"github.com/milvus-io/milvus/pkg/v2/proto/rootcoordpb"
"github.com/milvus-io/milvus/pkg/v2/util/merr"
"github.com/milvus-io/milvus/pkg/v2/util/typeutil"
)
func TestDDLCallbacksDatabaseDDL(t *testing.T) {
core := initStreamingSystemAndCore(t)
// Create a new database
status, err := core.CreateDatabase(context.Background(), &milvuspb.CreateDatabaseRequest{
DbName: "test",
})
require.NoError(t, merr.CheckRPCCall(status, err))
db, err := core.meta.GetDatabaseByName(context.Background(), "test", typeutil.MaxTimestamp)
require.NoError(t, merr.CheckRPCCall(status, err))
require.Equal(t, db.Name, "test")
require.Empty(t, db.Properties)
// Alter a database to add properties
status, err = core.AlterDatabase(context.Background(), &rootcoordpb.AlterDatabaseRequest{
DbName: "test",
Properties: []*commonpb.KeyValuePair{
{
Key: "key",
Value: "value",
},
{
Key: "key2",
Value: "value2",
},
},
})
require.NoError(t, merr.CheckRPCCall(status, err))
db, err = core.meta.GetDatabaseByName(context.Background(), "test", typeutil.MaxTimestamp)
require.NoError(t, merr.CheckRPCCall(status, err))
require.Equal(t, db.Name, "test")
require.Len(t, db.Properties, 2)
// Drop a property
status, err = core.AlterDatabase(context.Background(), &rootcoordpb.AlterDatabaseRequest{
DbName: "test",
DeleteKeys: []string{"key"},
})
require.NoError(t, merr.CheckRPCCall(status, err))
db, err = core.meta.GetDatabaseByName(context.Background(), "test", typeutil.MaxTimestamp)
require.NoError(t, merr.CheckRPCCall(status, err))
require.Equal(t, db.Name, "test")
require.Len(t, db.Properties, 1)
// Drop a database
status, err = core.DropDatabase(context.Background(), &milvuspb.DropDatabaseRequest{
DbName: "test",
})
require.NoError(t, merr.CheckRPCCall(status, err))
db, err = core.meta.GetDatabaseByName(context.Background(), "test", typeutil.MaxTimestamp)
require.Error(t, err)
require.Nil(t, db)
}