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
71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
package pulsar
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/apache/pulsar-client-go/pulsar"
|
|
"github.com/cockroachdb/errors"
|
|
|
|
"github.com/milvus-io/milvus/pkg/v3/metrics"
|
|
"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/impls/pulsar/pulsarlog"
|
|
"github.com/milvus-io/milvus/pkg/v3/streaming/walimpls/registry"
|
|
"github.com/milvus-io/milvus/pkg/v3/util/merr"
|
|
"github.com/milvus-io/milvus/pkg/v3/util/paramtable"
|
|
)
|
|
|
|
func init() {
|
|
// register the builder to the wal registry.
|
|
registry.RegisterBuilder(&builderImpl{})
|
|
// register the unmarshaler to the message registry.
|
|
message.RegisterMessageIDUnmsarshaler(message.WALNamePulsar, UnmarshalMessageID)
|
|
}
|
|
|
|
// builderImpl is the builder for pulsar wal.
|
|
type builderImpl struct{}
|
|
|
|
// Name returns the name of the wal.
|
|
func (b *builderImpl) Name() message.WALName {
|
|
return message.WALNamePulsar
|
|
}
|
|
|
|
// Build build a wal instance.
|
|
func (b *builderImpl) Build() (walimpls.OpenerImpls, error) {
|
|
options, tenant, err := b.getPulsarClientOptions()
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "build pulsar client options failed")
|
|
}
|
|
c, err := pulsar.NewClient(options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &openerImpl{
|
|
tenant: tenant,
|
|
c: c,
|
|
}, nil
|
|
}
|
|
|
|
// getPulsarClientOptions gets the pulsar client options from the config.
|
|
func (b *builderImpl) getPulsarClientOptions() (pulsar.ClientOptions, tenant, error) {
|
|
cfg := ¶mtable.Get().PulsarCfg
|
|
auth, err := pulsar.NewAuthentication(cfg.AuthPlugin.GetValue(), cfg.AuthParams.GetValue())
|
|
if err != nil {
|
|
return pulsar.ClientOptions{}, tenant{}, merr.WrapErrParameterInvalidMsg("build authencation from config failed")
|
|
}
|
|
options := pulsar.ClientOptions{
|
|
URL: cfg.Address.GetValue(),
|
|
OperationTimeout: cfg.RequestTimeout.GetAsDuration(time.Second),
|
|
Authentication: auth,
|
|
Logger: pulsarlog.NewLogger(),
|
|
}
|
|
if cfg.EnableClientMetrics.GetAsBool() {
|
|
// Enable client metrics if config.EnableClientMetrics is true, use pkg-defined registerer.
|
|
options.MetricsRegisterer = metrics.GetRegisterer()
|
|
}
|
|
return options, tenant{
|
|
namespace: cfg.Namespace.GetValue(),
|
|
tenant: cfg.Tenant.GetValue(),
|
|
}, nil
|
|
}
|