Skip to content

Commit

Permalink
FMWK-521-as-client-interface
Browse files Browse the repository at this point in the history
- reiview fixes
  • Loading branch information
filkeith committed Aug 5, 2024
1 parent 87aaba5 commit fa4a410
Show file tree
Hide file tree
Showing 5 changed files with 13 additions and 15 deletions.
10 changes: 5 additions & 5 deletions client.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,10 +33,10 @@ const (
MaxPartitions = 4096
)

// AsClient describes aerospike client interface for easy mocking.
// AerospikeClient describes aerospike client interface for easy mocking.
//
// go:generate mockery --name AsClient
type AsClient interface {
// go:generate mockery --name AerospikeClient
type AerospikeClient interface {
GetDefaultScanPolicy() *a.ScanPolicy
GetDefaultInfoPolicy() *a.InfoPolicy
GetDefaultWritePolicy() *a.WritePolicy
Expand Down Expand Up @@ -88,7 +88,7 @@ type AsClient interface {
// // handle error
// }
type Client struct {
aerospikeClient AsClient
aerospikeClient AerospikeClient
logger *slog.Logger
id string
}
Expand All @@ -97,7 +97,7 @@ type Client struct {
// - ac is the aerospike client to use for backup and restore operations.
// - id is an identifier for the client.
// - logger is the logger that this client will log to.
func NewClient(ac AsClient, id string, logger *slog.Logger) (*Client, error) {
func NewClient(ac AerospikeClient, id string, logger *slog.Logger) (*Client, error) {
if ac == nil {
return nil, errors.New("aerospike client pointer is nil")
}
Expand Down
4 changes: 2 additions & 2 deletions handler_backup.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ type BackupHandler struct {
writer Writer
encoder Encoder
config *BackupConfig
aerospikeClient AsClient
aerospikeClient AerospikeClient
logger *slog.Logger
firstFileHeaderWritten *atomic.Bool
limiter *rate.Limiter
Expand All @@ -64,7 +64,7 @@ type BackupHandler struct {
// newBackupHandler creates a new BackupHandler.
func newBackupHandler(
config *BackupConfig,
ac AsClient,
ac AerospikeClient,
logger *slog.Logger,
writer Writer,
) *BackupHandler {
Expand Down
6 changes: 2 additions & 4 deletions handler_backup_records.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,13 @@ import (

type backupRecordsHandler struct {
config *BackupConfig
aerospikeClient AsClient
aerospikeClient AerospikeClient
logger *slog.Logger
}

func newBackupRecordsHandler(
config *BackupConfig,
ac AsClient,
ac AerospikeClient,
logger *slog.Logger,
) *backupRecordsHandler {
logger.Debug("created new backup records handler")
Expand Down Expand Up @@ -139,8 +139,6 @@ func (bh *backupRecordsHandler) makeAerospikeReadWorkers(

readWorkers := make([]pipeline.Worker[*models.Token], n)

// counter inject

for i := 0; i < n; i++ {
recordReader := aerospike.NewRecordReader(
bh.aerospikeClient,
Expand Down
4 changes: 2 additions & 2 deletions handler_restore.go
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ type StreamingReader interface {
type RestoreHandler struct {
reader StreamingReader
config *RestoreConfig
aerospikeClient AsClient
aerospikeClient AerospikeClient
logger *slog.Logger
limiter *rate.Limiter
errors chan error
Expand All @@ -58,7 +58,7 @@ type RestoreHandler struct {
// newRestoreHandler creates a new RestoreHandler.
func newRestoreHandler(
config *RestoreConfig,
ac AsClient,
ac AerospikeClient,
logger *slog.Logger,
reader StreamingReader,
) *RestoreHandler {
Expand Down
4 changes: 2 additions & 2 deletions internal/asinfo/info_client.go
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ type infoGetter interface {
RequestInfo(infoPolicy *a.InfoPolicy, commands ...string) (map[string]string, a.Error)
}

type asClient interface {
type aerospikeClient interface {
Cluster() *a.Cluster
}

Expand All @@ -84,7 +84,7 @@ type InfoClient struct {
cluster *a.Cluster
}

func NewInfoClientFromAerospike(aeroClient asClient, policy *a.InfoPolicy) *InfoClient {
func NewInfoClientFromAerospike(aeroClient aerospikeClient, policy *a.InfoPolicy) *InfoClient {
return &InfoClient{
cluster: aeroClient.Cluster(),
policy: policy,
Expand Down

0 comments on commit fa4a410

Please sign in to comment.