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
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package rmq
|
|
|
|
import (
|
|
"github.com/milvus-io/milvus/pkg/v3/mq/mqimpl/rocksmq/client"
|
|
"github.com/milvus-io/milvus/pkg/v3/mq/mqimpl/rocksmq/server"
|
|
"github.com/milvus-io/milvus/pkg/v3/streaming/util/message"
|
|
"github.com/milvus-io/milvus/pkg/v3/streaming/walimpls"
|
|
"github.com/milvus-io/milvus/pkg/v3/streaming/walimpls/registry"
|
|
)
|
|
|
|
func init() {
|
|
// register the builder to the registry.
|
|
registry.RegisterBuilder(&builderImpl{})
|
|
// register the unmarshaler to the message registry.
|
|
message.RegisterMessageIDUnmsarshaler(message.WALNameRocksmq, UnmarshalMessageID)
|
|
}
|
|
|
|
// builderImpl is the builder for rmq opener.
|
|
type builderImpl struct{}
|
|
|
|
// Name of the wal builder, should be a lowercase string.
|
|
func (b *builderImpl) Name() message.WALName {
|
|
return message.WALNameRocksmq
|
|
}
|
|
|
|
// Build build a wal instance.
|
|
func (b *builderImpl) Build() (walimpls.OpenerImpls, error) {
|
|
c, err := client.NewClient(client.Options{
|
|
Server: server.Rmq,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &openerImpl{
|
|
c: c,
|
|
}, nil
|
|
}
|