mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 21:39:19 +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>
143 lines
5.0 KiB
Go
143 lines
5.0 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"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
|
|
"github.com/milvus-io/milvus/internal/proto/datapb"
|
|
"github.com/milvus-io/milvus/internal/proto/indexpb"
|
|
"github.com/milvus-io/milvus/pkg/log"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
type GetCollectionNameFunc func(dbName string, collID, partitionID UniqueID) (string, string, error)
|
|
|
|
type IDAllocator func(count uint32) (UniqueID, UniqueID, error)
|
|
|
|
type ImportFunc func(ctx context.Context, req *datapb.ImportTaskRequest) (*datapb.ImportTaskResponse, error)
|
|
|
|
type GetSegmentStatesFunc func(ctx context.Context, req *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error)
|
|
|
|
type DescribeIndexFunc func(ctx context.Context, colID UniqueID) (*indexpb.DescribeIndexResponse, error)
|
|
|
|
type GetSegmentIndexStateFunc func(ctx context.Context, collID UniqueID, indexName string, segIDs []UniqueID) ([]*indexpb.SegmentIndexState, error)
|
|
|
|
type UnsetIsImportingStateFunc func(context.Context, *datapb.UnsetIsImportingStateRequest) (*commonpb.Status, error)
|
|
|
|
type ImportFactory interface {
|
|
NewGetCollectionNameFunc() GetCollectionNameFunc
|
|
NewIDAllocator() IDAllocator
|
|
NewImportFunc() ImportFunc
|
|
NewGetSegmentStatesFunc() GetSegmentStatesFunc
|
|
NewDescribeIndexFunc() DescribeIndexFunc
|
|
NewGetSegmentIndexStateFunc() GetSegmentIndexStateFunc
|
|
NewUnsetIsImportingStateFunc() UnsetIsImportingStateFunc
|
|
}
|
|
|
|
type ImportFactoryImpl struct {
|
|
c *Core
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewGetCollectionNameFunc() GetCollectionNameFunc {
|
|
return GetCollectionNameWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewIDAllocator() IDAllocator {
|
|
return IDAllocatorWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewImportFunc() ImportFunc {
|
|
return ImportFuncWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewGetSegmentStatesFunc() GetSegmentStatesFunc {
|
|
return GetSegmentStatesWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewDescribeIndexFunc() DescribeIndexFunc {
|
|
return DescribeIndexWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewGetSegmentIndexStateFunc() GetSegmentIndexStateFunc {
|
|
return GetSegmentIndexStateWithCore(f.c)
|
|
}
|
|
|
|
func (f ImportFactoryImpl) NewUnsetIsImportingStateFunc() UnsetIsImportingStateFunc {
|
|
return UnsetIsImportingStateWithCore(f.c)
|
|
}
|
|
|
|
func NewImportFactory(c *Core) ImportFactory {
|
|
return &ImportFactoryImpl{c: c}
|
|
}
|
|
|
|
func GetCollectionNameWithCore(c *Core) GetCollectionNameFunc {
|
|
return func(dbName string, collID, partitionID UniqueID) (string, string, error) {
|
|
colInfo, err := c.meta.GetCollectionByID(c.ctx, dbName, collID, typeutil.MaxTimestamp, false)
|
|
if err != nil {
|
|
log.Error("Core failed to get collection name by id", zap.Int64("ID", collID), zap.Error(err))
|
|
return "", "", err
|
|
}
|
|
partName, err := c.meta.GetPartitionNameByID(collID, partitionID, 0)
|
|
if err != nil {
|
|
log.Error("Core failed to get partition name by id", zap.Int64("ID", partitionID), zap.Error(err))
|
|
return colInfo.Name, "", err
|
|
}
|
|
|
|
return colInfo.Name, partName, nil
|
|
}
|
|
}
|
|
|
|
func IDAllocatorWithCore(c *Core) IDAllocator {
|
|
return func(count uint32) (UniqueID, UniqueID, error) {
|
|
return c.idAllocator.Alloc(count)
|
|
}
|
|
}
|
|
|
|
func ImportFuncWithCore(c *Core) ImportFunc {
|
|
return func(ctx context.Context, req *datapb.ImportTaskRequest) (*datapb.ImportTaskResponse, error) {
|
|
return c.broker.Import(ctx, req)
|
|
}
|
|
}
|
|
|
|
func GetSegmentStatesWithCore(c *Core) GetSegmentStatesFunc {
|
|
return func(ctx context.Context, req *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error) {
|
|
return c.broker.GetSegmentStates(ctx, req)
|
|
}
|
|
}
|
|
|
|
func DescribeIndexWithCore(c *Core) DescribeIndexFunc {
|
|
return func(ctx context.Context, colID UniqueID) (*indexpb.DescribeIndexResponse, error) {
|
|
return c.broker.DescribeIndex(ctx, colID)
|
|
}
|
|
}
|
|
|
|
func GetSegmentIndexStateWithCore(c *Core) GetSegmentIndexStateFunc {
|
|
return func(ctx context.Context, collID UniqueID, indexName string, segIDs []UniqueID) ([]*indexpb.SegmentIndexState, error) {
|
|
return c.broker.GetSegmentIndexState(ctx, collID, indexName, segIDs)
|
|
}
|
|
}
|
|
|
|
func UnsetIsImportingStateWithCore(c *Core) UnsetIsImportingStateFunc {
|
|
return func(ctx context.Context, req *datapb.UnsetIsImportingStateRequest) (*commonpb.Status, error) {
|
|
return c.broker.UnsetIsImportingState(ctx, req)
|
|
}
|
|
}
|