mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 09:38:39 +08:00
9871966415
#33285 - add segment alloc interceptor for streamingnode. - add add manual alloc segment rpc for datacoord. --------- Signed-off-by: chyezh <chyezh@outlook.com>
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package adaptor
|
|
|
|
import (
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal"
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal/interceptors"
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal/interceptors/ddl"
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal/interceptors/segment"
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal/interceptors/timetick"
|
|
"github.com/milvus-io/milvus/pkg/streaming/walimpls"
|
|
)
|
|
|
|
var _ wal.OpenerBuilder = (*builderAdaptorImpl)(nil)
|
|
|
|
func AdaptImplsToBuilder(builder walimpls.OpenerBuilderImpls) wal.OpenerBuilder {
|
|
return builderAdaptorImpl{
|
|
builder: builder,
|
|
}
|
|
}
|
|
|
|
type builderAdaptorImpl struct {
|
|
builder walimpls.OpenerBuilderImpls
|
|
}
|
|
|
|
func (b builderAdaptorImpl) Name() string {
|
|
return b.builder.Name()
|
|
}
|
|
|
|
func (b builderAdaptorImpl) Build() (wal.Opener, error) {
|
|
o, err := b.builder.Build()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Add all interceptor here.
|
|
return adaptImplsToOpener(o, []interceptors.InterceptorBuilder{
|
|
timetick.NewInterceptorBuilder(),
|
|
segment.NewInterceptorBuilder(),
|
|
ddl.NewInterceptorBuilder(),
|
|
}), nil
|
|
}
|