mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-02 08:55:56 +08:00
5669016af0
issue: #38399 - Make the wal scanner interface same with streaming scanner. - Use wal if the wal is located at current node. - Otherwise fallback the old logic. Signed-off-by: chyezh <chyezh@outlook.com>
45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/cockroachdb/errors"
|
|
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal"
|
|
"github.com/milvus-io/milvus/pkg/log"
|
|
"github.com/milvus-io/milvus/pkg/streaming/util/types"
|
|
"github.com/milvus-io/milvus/pkg/util/paramtable"
|
|
"github.com/milvus-io/milvus/pkg/util/syncutil"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
var (
|
|
registry = syncutil.NewFuture[WALManager]()
|
|
ErrNoStreamingNodeDeployed = errors.New("no streaming node deployed")
|
|
)
|
|
|
|
// RegisterLocalWALManager registers the local wal manager.
|
|
// When the streaming node is started, it should call this function to register the wal manager.
|
|
func RegisterLocalWALManager(manager WALManager) {
|
|
if !paramtable.IsLocalComponentEnabled(typeutil.StreamingNodeRole) {
|
|
panic("unreachable: streaming node is not enabled but wal setup")
|
|
}
|
|
registry.Set(manager)
|
|
log.Ctx(context.Background()).Info("register local wal manager done")
|
|
}
|
|
|
|
// GetAvailableWAL returns a available wal instance for the channel.
|
|
func GetAvailableWAL(channel types.PChannelInfo) (wal.WAL, error) {
|
|
if !paramtable.IsLocalComponentEnabled(typeutil.StreamingNodeRole) {
|
|
return nil, ErrNoStreamingNodeDeployed
|
|
}
|
|
return registry.Get().GetAvailableWAL(channel)
|
|
}
|
|
|
|
// WALManager is a hint type for wal manager at streaming node.
|
|
type WALManager interface {
|
|
// GetAvailableWAL returns a available wal instance for the channel.
|
|
// Return nil if the wal instance is not found.
|
|
GetAvailableWAL(channel types.PChannelInfo) (wal.WAL, error)
|
|
}
|