Fixed comments and added a mutex for the feature map.

Author: mariuspot

client.go

@@ -20,6 +20,7 @@ import (
 	"net/url"
 	"runtime"
 	"strings"
+	"sync"
 	"time"
 )
 
@@ -46,17 +47,18 @@ var (
 type Feature int
 
 const (
-	// EnableConversationsAPIWhatsAppSandbox Enables the WhatsApp sandbox for conversations API
+	// FeatureConversationsAPIWhatsAppSandbox Enables the WhatsApp sandbox for conversations API.
 	FeatureConversationsAPIWhatsAppSandbox Feature = iota
 )
 
 // Client is used to access API with a given key.
 // Uses standard lib HTTP client internally, so should be reused instead of created as needed and it is safe for concurrent use.
 type Client struct {
-	AccessKey  string           // The API access key
-	HTTPClient *http.Client     // The HTTP client to send requests on
-	DebugLog   *log.Logger      // Optional logger for debugging purposes
-	features   map[Feature]bool // Enabled features
+	AccessKey     string           // The API access key.
+	HTTPClient    *http.Client     // The HTTP client to send requests on.
+	DebugLog      *log.Logger      // Optional logger for debugging purposes.
+	features      map[Feature]bool // Enabled features.
+	featuresMutex *sync.RWMutex    //Mutex for accessing feature map.
 }
 
 type contentType string
@@ -79,7 +81,8 @@ func New(accessKey string) *Client {
 		HTTPClient: &http.Client{
 			Timeout: httpClientTimeout,
 		},
-		features: make(map[Feature]bool),
+		features:      make(map[Feature]bool),
+		featuresMutex: &sync.RWMutex{},
 	}
 }
 
@@ -89,18 +92,24 @@ func SetVoiceErrorReader(r errorReader) {
 	voiceErrorReader = r
 }
 
-// EnableFeatures enables a feature
+// EnableFeatures enables a feature.
 func (c *Client) EnableFeatures(feature Feature) {
+	c.featuresMutex.Lock()
+	defer c.featuresMutex.Unlock()
 	c.features[feature] = true
 }
 
-// DisableFeatures enables a feature
+// DisableFeatures disables a feature.
 func (c *Client) DisableFeatures(feature Feature) {
+	c.featuresMutex.Lock()
+	defer c.featuresMutex.Unlock()
 	c.features[feature] = false
 }
 
-// IsFeatureEnabled check if a feature is enabled
+// IsFeatureEnabled checks if a feature is enabled.
 func (c *Client) IsFeatureEnabled(feature Feature) bool {
+	c.featuresMutex.RLock()
+	defer c.featuresMutex.RUnlock()
 	if enabled, ok := c.features[feature]; ok {
 		return enabled
 	}