Add new constructor for client with log config as parameter (#117)

* NewClientWithLogsConfig * NewClientWithLogsConfig
0xPolygon · Mar 21, 2024 · 0c30017 · 0c30017
1 parent ebf261c
commit 0c30017
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/datastreamer/streamclient.go b/datastreamer/streamclient.go
@@ -69,6 +69,12 @@ func NewClient(server string, streamType StreamType) (*StreamClient, error) {
 	return &c, nil
 }
 
+// NewClientWithLogsConfig creates a new data stream client with logs configuration
+func NewClientWithLogsConfig(server string, streamType StreamType, logsConfig log.Config) (*StreamClient, error) {
+	log.Init(logsConfig)
+	return NewClient(server, streamType)
+}
+
 // Start connects to the data stream server and starts getting data from the server
 func (c *StreamClient) Start() error {
 	// Connect to server