mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
f9febe3bae
Merge RootCoord, DataCoord And QueryCoord into MixCoord Make Session into one issue : https://github.com/milvus-io/milvus/issues/37764 --------- Signed-off-by: Xianhui.Lin <xianhui.lin@zilliz.com>
52 lines
1.7 KiB
Go
52 lines
1.7 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.
|
|
|
|
//go:build test
|
|
// +build test
|
|
|
|
package grpcmixcoord
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/milvus-io/milvus/internal/querycoordv2"
|
|
"github.com/milvus-io/milvus/pkg/v2/proto/querypb"
|
|
)
|
|
|
|
func (s *Server) StopCheckerForTestOnly() {
|
|
if qc, ok := s.mixCoord.(querypb.QueryCoordServer); ok {
|
|
if server, ok := qc.(*querycoordv2.Server); ok {
|
|
server.StopCheckerForTestOnly()
|
|
} else {
|
|
log.Println("mixCoord does not hold a *querycoordv2.Server instance")
|
|
}
|
|
} else {
|
|
log.Println("mixCoord does not implement querypb.QueryCoordServer")
|
|
}
|
|
}
|
|
|
|
func (s *Server) StartCheckerForTestOnly() {
|
|
if qc, ok := s.mixCoord.(querypb.QueryCoordServer); ok {
|
|
if server, ok := qc.(*querycoordv2.Server); ok {
|
|
server.StartCheckerForTestOnly()
|
|
} else {
|
|
log.Println("mixCoord does not hold a *querycoordv2.Server instance")
|
|
}
|
|
} else {
|
|
log.Println("mixCoord does not implement querypb.QueryCoordServer")
|
|
}
|
|
}
|