Copy tracing (#240)

Co-authored-by: alonadam <[email protected]>

kusto/ingest/ingest.go

@@ -37,6 +37,9 @@ type Ingestion struct {
 
 	bufferSize int
 	maxBuffers int
+
+	applicationForTracing   string
+	clientVersionForTracing string
 }
 
 // Option is an optional argument to New().
@@ -63,12 +66,14 @@ func New(client QueryClient, db, table string, options ...Option) (*Ingestion, e
 		db:     db,
 		table:  table,
 	}
+	i.applicationForTracing = client.ClientDetails().ApplicationForTracing()
+	i.clientVersionForTracing = client.ClientDetails().ClientVersionForTracing()
 
 	for _, option := range options {
 		option(i)
 	}
 
-	fs, err := queued.New(db, table, mgr, client.HttpClient(), queued.WithStaticBuffer(i.bufferSize, i.maxBuffers))
+	fs, err := queued.New(db, table, mgr, client.HttpClient(), i.applicationForTracing, i.clientVersionForTracing, queued.WithStaticBuffer(i.bufferSize, i.maxBuffers))
 	if err != nil {
 		return nil, err
 	}

kusto/ingest/internal/properties/properties.go

@@ -273,6 +273,10 @@ type Ingestion struct {
 	Additional Additional `json:"AdditionalProperties"`
 	// TableEntryRef points to the staus table entry used to report the status of this ingestion.
 	TableEntryRef StatusTableDescription `json:"IngestionStatusInTable,omitempty"`
+	// ApplicationForTracing is the application name that is used for tracing.
+	ApplicationForTracing string `json:",omitempty"`
+	// ClientVersionForTracing is the client version that is used for tracing.
+	ClientVersionForTracing string `json:",omitempty"`
 }
 
 // Additional is additional properites.

kusto/ingest/internal/queued/queued.go

@@ -66,6 +66,9 @@ type Ingestion struct {
 
 	bufferSize int
 	maxBuffers int
+
+	applicationForTracing   string
+	clientVersionForTracing string
 }
 
 // Option is an optional argument to New().
@@ -80,7 +83,7 @@ func WithStaticBuffer(bufferSize int, maxBuffers int) Option {
 }
 
 // New is the constructor for Ingestion.
-func New(db, table string, mgr *resources.Manager, http *http.Client, options ...Option) (*Ingestion, error) {
+func New(db, table string, mgr *resources.Manager, http *http.Client, applicationForTracing string, clientVersionForTracing string, options ...Option) (*Ingestion, error) {
 	i := &Ingestion{
 		db:    db,
 		table: table,
@@ -94,6 +97,8 @@ func New(db, table string, mgr *resources.Manager, http *http.Client, options ..
 			options *azblob.UploadFileOptions) (azblob.UploadFileResponse, error) {
 			return client.UploadFile(ctx, container, blob, file, options)
 		},
+		applicationForTracing:   applicationForTracing,
+		clientVersionForTracing: clientVersionForTracing,
 	}
 
 	for _, opt := range options {
@@ -246,6 +251,8 @@ func (i *Ingestion) Blob(ctx context.Context, from string, fileSize int64, props
 	}
 
 	props.Ingestion.RetainBlobOnSuccess = !props.Source.DeleteLocalSource
+	props.Ingestion.ApplicationForTracing = i.applicationForTracing
+	props.Ingestion.ClientVersionForTracing = i.clientVersionForTracing
 
 	err := CompleteFormatFromFileName(&props, from)
 	if err != nil {