support Source config

batch.go

@@ -3,6 +3,7 @@ package objects
 import "encoding/json"
 
 type batch struct {
+	Source     string          `json:"source"`
 	Collection string          `json:"collection"`
 	WriteKey   string          `json:"write_key"`
 	Objects    json.RawMessage `json:"objects"`

client.go

@@ -25,6 +25,9 @@ const (
 
 	// Endpoint for Segment Objects API.
 	DefaultBaseEndpoint = "https://objects.segment.com"
+
+	// Default source
+	DefaultSource = "project"
 )
 
 var (
@@ -36,6 +39,8 @@ type Config struct {
 	Logger       *log.Logger
 	Client       *http.Client
 
+	Source string
+
 	MaxBatchBytes    int
 	MaxBatchCount    int
 	MaxBatchInterval time.Duration
@@ -89,6 +94,10 @@ func getFinalConfig(c Config) Config {
 		c.MaxBatchInterval = 10 * time.Second
 	}
 
+	if c.Source == "" {
+		c.Source = DefaultSource
+	}
+
 	return c
 }
 
@@ -107,6 +116,7 @@ func (c *Client) flush(b *buffer) {
 	rm := b.marshalArray()
 	c.semaphore.Run(func() {
 		batchRequest := &batch{
+			Source:     c.Source,
 			Collection: b.collection,
 			WriteKey:   c.writeKey,
 			Objects:    rm,

client_test.go

@@ -58,6 +58,7 @@ func (c *ClientTestSuite) TestNewClient() {
 	c.NotNil(client.Logger)
 	c.NotNil(client.semaphore)
 	c.NotNil(client.wg)
+	c.NotNil(client.Source)
 	c.Equal("writeKey", client.writeKey)
 	c.Equal(0, client.cmap.Count())
 }