wehub-resource-sync
498b235461
Build and test / Build and test AMD64 Ubuntu 22.04 (push) Failing after 0s
Publish Builder / amazonlinux2023 (push) Failing after 1s
Build and test / UT for Go (push) Has been skipped
Publish KRTE Images / KRTE (push) Failing after 1s
Build and test / Integration Test (push) Has been skipped
Build and test / Upload Code Coverage (push) Has been skipped
Publish Builder / rockylinux9 (push) Failing after 1s
Publish Builder / ubuntu22.04 (push) Failing after 0s
Publish Builder / ubuntu24.04 (push) Failing after 0s
Publish Gpu Builder / publish-gpu-builder (push) Failing after 1s
Publish Test Images / PyTest (push) Failing after 0s
Build and test / UT for Cpp (push) Has been cancelled
61 lines
1.7 KiB
Go
61 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/v3/kv"
|
|
"github.com/milvus-io/milvus/pkg/v3/mlog"
|
|
"github.com/milvus-io/milvus/pkg/v3/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),
|
|
resource.OptSession(s.session),
|
|
)
|
|
return &Server{
|
|
logger: resource.Resource().Logger().With(mlog.FieldComponent("server")),
|
|
session: s.session,
|
|
assignmentService: service.NewAssignmentService(),
|
|
broadcastService: service.NewBroadcastService(),
|
|
}
|
|
}
|