mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 13:28:49 +08:00
18df2ba6fd
Support Database(#23742) Fix db nonexists error for FlushAll (#24222) Fix check collection limits fails (#24235) backward compatibility with empty DB name (#24317) Fix GetFlushAllState with DB (#24347) Remove db from global meta cache after drop database (#24474) Fix db name is empty for describe collection response (#24603) Add RBAC for Database API (#24653) Fix miss load the same name collection during recover stage (#24941) RBAC supports Database validation (#23609) Fix to list grant with db return empty (#23922) Optimize PrivilegeAll permission check (#23972) Add the default db value for the rbac request (#24307) Signed-off-by: jaime <yun.zhang@zilliz.com> Co-authored-by: SimFG <bang.fu@zilliz.com> Co-authored-by: longjiquan <jiquan.long@zilliz.com>
112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
package rootcoord
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"golang.org/x/exp/maps"
|
|
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
type nameDb struct {
|
|
db2Name2ID map[string]map[string]UniqueID // database -> collection name -> collection id
|
|
totalCollectionNum int
|
|
totalPartitionNum int
|
|
}
|
|
|
|
func (n *nameDb) exist(dbName string) bool {
|
|
_, ok := n.db2Name2ID[dbName]
|
|
return ok
|
|
}
|
|
|
|
func (n *nameDb) empty(dbName string) bool {
|
|
db, ok := n.db2Name2ID[dbName]
|
|
return ok && len(db) == 0
|
|
}
|
|
|
|
func (n *nameDb) createDbIfNotExist(dbName string) {
|
|
if !n.exist(dbName) {
|
|
n.db2Name2ID[dbName] = make(map[string]UniqueID)
|
|
}
|
|
}
|
|
|
|
func (n *nameDb) dropDb(dbName string) {
|
|
delete(n.db2Name2ID, dbName)
|
|
}
|
|
|
|
func (n *nameDb) insert(dbName string, collectionName string, collectionID UniqueID) {
|
|
n.createDbIfNotExist(dbName)
|
|
n.db2Name2ID[dbName][collectionName] = collectionID
|
|
}
|
|
|
|
func (n *nameDb) get(dbName string, collectionName string) (collectionID UniqueID, ok bool) {
|
|
if !n.exist(dbName) {
|
|
return 0, false
|
|
}
|
|
collectionID, ok = n.db2Name2ID[dbName][collectionName]
|
|
return collectionID, ok
|
|
}
|
|
|
|
func (n *nameDb) listDB() []string {
|
|
dbs := make([]string, 0, len(n.db2Name2ID))
|
|
for db := range n.db2Name2ID {
|
|
dbs = append(dbs, db)
|
|
}
|
|
return dbs
|
|
}
|
|
|
|
func (n *nameDb) listCollectionID(dbName string) ([]typeutil.UniqueID, error) {
|
|
name2ID, ok := n.db2Name2ID[dbName]
|
|
if !ok {
|
|
return nil, fmt.Errorf("database not exist: %s", dbName)
|
|
}
|
|
return maps.Values(name2ID), nil
|
|
}
|
|
|
|
func (n *nameDb) removeIf(selector func(db string, collection string, id UniqueID) bool) {
|
|
type union struct {
|
|
db string
|
|
collection string
|
|
id UniqueID
|
|
}
|
|
|
|
matches := make([]union, 0, len(n.db2Name2ID))
|
|
for dbName, db := range n.db2Name2ID {
|
|
for collection, id := range db {
|
|
if selector(dbName, collection, id) {
|
|
matches = append(matches, union{
|
|
db: dbName,
|
|
collection: collection,
|
|
id: id,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, match := range matches {
|
|
delete(n.db2Name2ID[match.db], match.collection)
|
|
}
|
|
}
|
|
|
|
func (n *nameDb) remove(db, collection string) {
|
|
if n.exist(db) {
|
|
delete(n.db2Name2ID[db], collection)
|
|
}
|
|
}
|
|
|
|
func (n *nameDb) iterate(exitOnFalse func(db string, collection string, id UniqueID) bool) {
|
|
for dbName, db := range n.db2Name2ID {
|
|
for collection, id := range db {
|
|
if !exitOnFalse(dbName, collection, id) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func newNameDb() *nameDb {
|
|
return &nameDb{
|
|
db2Name2ID: make(map[string]map[string]typeutil.UniqueID),
|
|
}
|
|
}
|