milvus/internal/util/testutil/reset_env.go

16 lines
524 B
Go

package testutil
import (
"github.com/milvus-io/milvus/internal/coordinator/snmanager"
"github.com/milvus-io/milvus/internal/streamingcoord/server/balancer/channel"
"github.com/milvus-io/milvus/internal/streamingcoord/server/broadcaster/registry"
registry2 "github.com/milvus-io/milvus/internal/streamingnode/client/handler/registry"
)
func ResetEnvironment() {
channel.ResetStaticPChannelStatsManager()
registry.ResetRegistration()
snmanager.ResetStreamingNodeManager()
registry2.ResetRegisterLocalWALManager()
}