mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
bef06e5acf
issue: #33285 - register streaming coord service into datacoord. - add new streaming node role. - add global static switch to enable streaming service or not. Signed-off-by: chyezh <chyezh@outlook.com>
49 lines
1.2 KiB
Go
49 lines
1.2 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/util/componentutil"
|
|
"github.com/milvus-io/milvus/internal/util/sessionutil"
|
|
"github.com/milvus-io/milvus/pkg/kv"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
type ServerBuilder struct {
|
|
etcdClient *clientv3.Client
|
|
metaKV kv.MetaKv
|
|
session sessionutil.SessionInterface
|
|
}
|
|
|
|
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) 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)),
|
|
)
|
|
return &Server{
|
|
session: s.session,
|
|
componentStateService: componentutil.NewComponentStateService(typeutil.StreamingCoordRole),
|
|
}
|
|
}
|