add setUserAgent for Client and producer

client.go

@@ -116,6 +116,11 @@ func convertLocked(c *Client, projName string) *LogProject {
 	return p
 }
 
+// SetUserAgent set a custom userAgent
+func (c *Client) SetUserAgent(userAgent string) {
+	c.UserAgent = userAgent
+}
+
 // SetHTTPClient set a custom http client, all request will send to sls by this client
 func (c *Client) SetHTTPClient(client *http.Client) {
 	c.HTTPClient = client

client_interface.go

@@ -41,6 +41,8 @@ func CreateTokenAutoUpdateClient(endpoint string, tokenUpdateFunc UpdateTokenFun
 
 // ClientInterface for all log's open api
 type ClientInterface interface {
+	// SetUserAgent set userAgent for sls client
+	SetUserAgent(userAgent string)
 	// SetHTTPClient set a custom http client, all request will send to sls by this client
 	SetHTTPClient(client *http.Client)
 	// #################### Client Operations #####################

producer/producer.go

@@ -44,6 +44,9 @@ func InitProducer(producerConfig *ProducerConfig) *Producer {
 	if producerConfig.HTTPClient != nil {
 		client.SetHTTPClient(producerConfig.HTTPClient)
 	}
+	if producerConfig.UserAgent != "" {
+		client.SetUserAgent(producerConfig.UserAgent)
+	}
 	finalProducerConfig := validateProducerConfig(producerConfig)
 	retryQueue := initRetryQueue()
 	errorStatusMap := func() map[int]*string {

producer/producer_config.go

@@ -33,6 +33,7 @@ type ProducerConfig struct {
 	UpdateStsToken        func() (accessKeyID, accessKeySecret, securityToken string, expireTime time.Time, err error)
 	StsTokenShutDown      chan struct{}
 	HTTPClient            *http.Client
+	UserAgent             string
 }
 
 func GetDefaultProducerConfig() *ProducerConfig {

token_auto_update_client.go

@@ -137,6 +137,10 @@ func (c *TokenAutoUpdateClient) processError(err error) (retry bool) {
 
 }
 
+func (c *TokenAutoUpdateClient) SetUserAgent(userAgent string) {
+	c.logClient.SetUserAgent(userAgent)
+}
+
 // SetHTTPClient set a custom http client, all request will send to sls by this client
 func (c *TokenAutoUpdateClient) SetHTTPClient(client *http.Client) {
 	c.logClient.SetHTTPClient(client)