diff --git a/v1/client.go b/v1/client.go index d7cd6c1..afb1860 100644 --- a/v1/client.go +++ b/v1/client.go @@ -14,16 +14,46 @@ import ( "github.com/google/go-querystring/query" ) +type Option func(*MgClient) + +// OptionHTTPClient set custom http.Client for MgClient +func OptionHTTPClient(client *http.Client) func(*MgClient) { + return func(c *MgClient) { + c.httpClient = client + } +} + +// OptionLogger sets the provided logger instance into the MgClient +func OptionLogger(logger BasicLogger) func(*MgClient) { + return func(c *MgClient) { + c.logger = logger + } +} + +// OptionDebug enables debug mode for MgClient +func OptionDebug() func(*MgClient) { + return func(c *MgClient) { + c.Debug = true + } +} + // New initialize client -func New(url string, token string) *MgClient { - return &MgClient{ +func New(url string, token string, opts ...Option) *MgClient { + c := &MgClient{ URL: url, Token: token, httpClient: &http.Client{Timeout: time.Minute}, } + + for _, opt := range opts { + opt(c) + } + + return c } // WithLogger sets the provided logger instance into the Client. +// Deprecated: Use functional option OptionLogger instead func (c *MgClient) WithLogger(logger BasicLogger) *MgClient { c.logger = logger return c diff --git a/v1/client_test.go b/v1/client_test.go index 2cad797..5965543 100644 --- a/v1/client_test.go +++ b/v1/client_test.go @@ -44,14 +44,12 @@ var ( debug, _ = strconv.ParseBool(os.Getenv("DEBUG")) ) -func client() *MgClient { - c := New(mgURL, mgToken) - +func client(opts ...Option) *MgClient { if debug != false { - c.Debug = true + opts = append(opts, OptionDebug()) } - return c + return New(mgURL, mgToken, opts...) } func TestMgClient_Bots(t *testing.T) { @@ -951,9 +949,7 @@ func TestMgClient_DebugWithLogger(t *testing.T) { var buf bytes.Buffer logger := log.New(&buf, "Custom log prefix ", 0) - c := client() - c.Debug = true - c.WithLogger(logger) + c := client(OptionDebug(), OptionLogger(logger)) c.writeLog("Test log string")