mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-06 17:18:35 +08:00
53e8f150e8
issue: #44014 - On standalone, the query node inside need to load segment and watch channel, so the querynode is not a embeded querynode in streamingnode without `LabelStreamingNodeEmbeddedQueryNode`. The channel dist manager can not confirm a standalone node is a embededStreamingNode. Bug is introduced by #44099 Signed-off-by: chyezh <chyezh@outlook.com>
29 lines
895 B
Go
29 lines
895 B
Go
package streamingutil
|
|
|
|
import (
|
|
"os"
|
|
)
|
|
|
|
const MilvusStreamingServiceEnabled = "MILVUS_STREAMING_SERVICE_ENABLED"
|
|
|
|
// IsStreamingServiceEnabled returns whether the streaming service is enabled.
|
|
func IsStreamingServiceEnabled() bool {
|
|
// TODO: check if the environment variable MILVUS_STREAMING_SERVICE_ENABLED is set
|
|
return os.Getenv(MilvusStreamingServiceEnabled) == "1"
|
|
}
|
|
|
|
// SetStreamingServiceEnabled set the env that indicates whether the streaming service is enabled.
|
|
func SetStreamingServiceEnabled() {
|
|
err := os.Setenv(MilvusStreamingServiceEnabled, "1")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// MustEnableStreamingService panics if the streaming service is not enabled.
|
|
func MustEnableStreamingService() {
|
|
if !IsStreamingServiceEnabled() {
|
|
panic("start a streaming node without enabling streaming service, please set environment variable MILVUS_STREAMING_SERVICE_ENABLED = 1")
|
|
}
|
|
}
|