mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
Remove not used QueryService (#16973)
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
parent
5b658349c4
commit
23f7ef931c
@ -1,160 +0,0 @@
|
||||
// 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 querynode
|
||||
|
||||
import "C"
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"github.com/milvus-io/milvus/internal/log"
|
||||
"github.com/milvus-io/milvus/internal/mq/msgstream"
|
||||
"github.com/milvus-io/milvus/internal/storage"
|
||||
)
|
||||
|
||||
type queryService struct {
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
|
||||
historical *historical
|
||||
streaming *streaming
|
||||
|
||||
queryCollectionMu sync.Mutex // guards queryCollections
|
||||
queryCollections map[UniqueID]*queryCollection
|
||||
|
||||
factory msgstream.Factory
|
||||
|
||||
sessionManager *SessionManager
|
||||
|
||||
cacheStorage storage.ChunkManager
|
||||
vectorStorage storage.ChunkManager
|
||||
}
|
||||
|
||||
type qsOpt func(*queryService)
|
||||
|
||||
func qsOptWithSessionManager(s *SessionManager) qsOpt {
|
||||
return func(qs *queryService) {
|
||||
qs.sessionManager = s
|
||||
}
|
||||
}
|
||||
|
||||
func newQueryService(ctx context.Context,
|
||||
historical *historical,
|
||||
streaming *streaming,
|
||||
vectorStorage storage.ChunkManager,
|
||||
cacheStorage storage.ChunkManager,
|
||||
factory msgstream.Factory,
|
||||
opts ...qsOpt,
|
||||
) *queryService {
|
||||
|
||||
queryServiceCtx, queryServiceCancel := context.WithCancel(ctx)
|
||||
|
||||
qs := &queryService{
|
||||
ctx: queryServiceCtx,
|
||||
cancel: queryServiceCancel,
|
||||
|
||||
historical: historical,
|
||||
streaming: streaming,
|
||||
|
||||
queryCollections: make(map[UniqueID]*queryCollection),
|
||||
|
||||
vectorStorage: vectorStorage,
|
||||
cacheStorage: cacheStorage,
|
||||
factory: factory,
|
||||
}
|
||||
|
||||
for _, opt := range opts {
|
||||
opt(qs)
|
||||
}
|
||||
|
||||
return qs
|
||||
}
|
||||
|
||||
func (q *queryService) close() {
|
||||
log.Info("search service closed")
|
||||
q.queryCollectionMu.Lock()
|
||||
for collectionID, sc := range q.queryCollections {
|
||||
sc.close()
|
||||
sc.cancel()
|
||||
delete(q.queryCollections, collectionID)
|
||||
}
|
||||
q.queryCollections = make(map[UniqueID]*queryCollection)
|
||||
q.queryCollectionMu.Unlock()
|
||||
q.cancel()
|
||||
}
|
||||
|
||||
func (q *queryService) addQueryCollection(collectionID UniqueID) error {
|
||||
q.queryCollectionMu.Lock()
|
||||
defer q.queryCollectionMu.Unlock()
|
||||
if _, ok := q.queryCollections[collectionID]; ok {
|
||||
log.Warn("query collection already exists", zap.Any("collectionID", collectionID))
|
||||
err := errors.New(fmt.Sprintln("query collection already exists, collectionID = ", collectionID))
|
||||
return err
|
||||
}
|
||||
ctx1, cancel := context.WithCancel(q.ctx)
|
||||
qc, err := newQueryCollection(ctx1,
|
||||
cancel,
|
||||
collectionID,
|
||||
q.historical,
|
||||
q.streaming,
|
||||
q.factory,
|
||||
q.cacheStorage,
|
||||
q.vectorStorage,
|
||||
qcOptWithSessionManager(q.sessionManager),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
q.queryCollections[collectionID] = qc
|
||||
return nil
|
||||
}
|
||||
|
||||
func (q *queryService) hasQueryCollection(collectionID UniqueID) bool {
|
||||
q.queryCollectionMu.Lock()
|
||||
defer q.queryCollectionMu.Unlock()
|
||||
_, ok := q.queryCollections[collectionID]
|
||||
return ok
|
||||
}
|
||||
|
||||
func (q *queryService) getQueryCollection(collectionID UniqueID) (*queryCollection, error) {
|
||||
q.queryCollectionMu.Lock()
|
||||
defer q.queryCollectionMu.Unlock()
|
||||
_, ok := q.queryCollections[collectionID]
|
||||
if ok {
|
||||
return q.queryCollections[collectionID], nil
|
||||
}
|
||||
return nil, errors.New(fmt.Sprintln("queryCollection not exists, collectionID = ", collectionID))
|
||||
}
|
||||
|
||||
func (q *queryService) stopQueryCollection(collectionID UniqueID) {
|
||||
q.queryCollectionMu.Lock()
|
||||
defer q.queryCollectionMu.Unlock()
|
||||
sc, ok := q.queryCollections[collectionID]
|
||||
if !ok {
|
||||
log.Warn("stopQueryCollection failed, collection doesn't exist", zap.Int64("collectionID", collectionID))
|
||||
return
|
||||
}
|
||||
sc.close()
|
||||
sc.cancel()
|
||||
// for not blocking waitNewTsafe, which will block doUnsolvedMsg quit.
|
||||
sc.watcherCond.Broadcast()
|
||||
delete(q.queryCollections, collectionID)
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user