mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
19e5e9f910
issue: #43897 - Return LastConfirmedMessageID when wal append operation. - Add resource-key-based locker for broadcast-ack operation to protect the coord state when executing ddl. - Resource-key-based locker is held until the broadcast operation is acked. - ResourceKey support shared and exclusive lock. - Add FastAck execute ack right away after the broadcast done to speed up ddl. - Ack callback will support broadcast message result now. - Add tombstone for broadcaster to avoid to repeatedly commit DDL and ABA issue. --------- Signed-off-by: chyezh <chyezh@outlook.com>
60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
|
|
"github.com/milvus-io/milvus/internal/metastore/kv/streamingcoord"
|
|
"github.com/milvus-io/milvus/internal/streamingcoord/server/resource"
|
|
"github.com/milvus-io/milvus/internal/streamingcoord/server/service"
|
|
"github.com/milvus-io/milvus/internal/types"
|
|
"github.com/milvus-io/milvus/internal/util/sessionutil"
|
|
"github.com/milvus-io/milvus/pkg/v2/kv"
|
|
"github.com/milvus-io/milvus/pkg/v2/log"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/syncutil"
|
|
)
|
|
|
|
type ServerBuilder struct {
|
|
etcdClient *clientv3.Client
|
|
metaKV kv.MetaKv
|
|
session sessionutil.SessionInterface
|
|
mixCoordClient *syncutil.Future[types.MixCoordClient]
|
|
}
|
|
|
|
func NewServerBuilder() *ServerBuilder {
|
|
return &ServerBuilder{}
|
|
}
|
|
|
|
func (b *ServerBuilder) WithETCD(e *clientv3.Client) *ServerBuilder {
|
|
b.etcdClient = e
|
|
return b
|
|
}
|
|
|
|
func (b *ServerBuilder) WithMetaKV(metaKV kv.MetaKv) *ServerBuilder {
|
|
b.metaKV = metaKV
|
|
return b
|
|
}
|
|
|
|
func (b *ServerBuilder) WithMixCoordClient(mixCoordClient *syncutil.Future[types.MixCoordClient]) *ServerBuilder {
|
|
b.mixCoordClient = mixCoordClient
|
|
return b
|
|
}
|
|
|
|
func (b *ServerBuilder) WithSession(session sessionutil.SessionInterface) *ServerBuilder {
|
|
b.session = session
|
|
return b
|
|
}
|
|
|
|
func (s *ServerBuilder) Build() *Server {
|
|
resource.Init(
|
|
resource.OptETCD(s.etcdClient),
|
|
resource.OptStreamingCatalog(streamingcoord.NewCataLog(s.metaKV)),
|
|
resource.OptMixCoordClient(s.mixCoordClient),
|
|
)
|
|
return &Server{
|
|
logger: resource.Resource().Logger().With(log.FieldComponent("server")),
|
|
session: s.session,
|
|
assignmentService: service.NewAssignmentService(),
|
|
broadcastService: service.NewBroadcastService(),
|
|
}
|
|
}
|