mirror of https://github.com/milvus-io/milvus.git
42 lines
1003 B
Go
42 lines
1003 B
Go
package rmq
|
|
|
|
import (
|
|
"github.com/milvus-io/milvus/pkg/mq/mqimpl/rocksmq/client"
|
|
"github.com/milvus-io/milvus/pkg/mq/mqimpl/rocksmq/server"
|
|
"github.com/milvus-io/milvus/pkg/streaming/util/message"
|
|
"github.com/milvus-io/milvus/pkg/streaming/walimpls"
|
|
"github.com/milvus-io/milvus/pkg/streaming/walimpls/registry"
|
|
)
|
|
|
|
const (
|
|
walName = "rocksmq"
|
|
)
|
|
|
|
func init() {
|
|
// register the builder to the registry.
|
|
registry.RegisterBuilder(&builderImpl{})
|
|
// register the unmarshaler to the message registry.
|
|
message.RegisterMessageIDUnmsarshaler(walName, 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() string {
|
|
return walName
|
|
}
|
|
|
|
// 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
|
|
}
|