initial commit
This commit is contained in:
60
internal/metal/options.go
Normal file
60
internal/metal/options.go
Normal file
@@ -0,0 +1,60 @@
|
||||
package metal
|
||||
|
||||
import (
|
||||
provider "go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers"
|
||||
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers/emapi"
|
||||
"go.equinixmetal.net/infra9-metal-bridge/internal/metal/providers/emgql"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
// Option is a Client configuration Option definition.
|
||||
type Option func(c *Client) error
|
||||
|
||||
// WithProvider sets the provider on the client.
|
||||
func WithProvider(provider provider.Provider) Option {
|
||||
return func(c *Client) error {
|
||||
c.provider = provider
|
||||
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
// WithLogger sets the logger for the client.
|
||||
func WithLogger(logger *zap.Logger) Option {
|
||||
return func(c *Client) error {
|
||||
c.logger = logger
|
||||
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
// WithConfig applies all configurations defined in the config.
|
||||
func WithConfig(config Config) Option {
|
||||
return func(c *Client) error {
|
||||
var options []Option
|
||||
|
||||
if config.EMGQL.Populated() {
|
||||
client, err := emgql.New(emgql.WithConfig(config.EMGQL))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
options = append(options, WithProvider(client))
|
||||
} else if config.EMAPI.Populated() {
|
||||
client, err := emapi.New(emapi.WithConfig(config.EMAPI))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
options = append(options, WithProvider(client))
|
||||
}
|
||||
|
||||
for _, opt := range options {
|
||||
if err := opt(c); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user