@atlaskit/media-client 17.1.2 → 17.1.4

package/dist/esm/client/media-store/index.js

@@ -11,7 +11,7 @@ function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { va
 import { FILE_CACHE_MAX_AGE, MAX_RESOLUTION } from '../../constants';
 import { getArtifactUrl } from '../../models/artifacts';
 import { request as _request } from '../../utils/request';
-import { createUrl, createMapResponseToJson, createMapResponseToBlob } from '../../utils/request/helpers';
+import { createUrl, createMapResponseToJson, createMapResponseToBlob, getRandomHex } from '../../utils/request/helpers';
 import { resolveAuth, resolveInitialAuth } from './resolveAuth';
 export { MediaStoreError, isMediaStoreError } from './error';
 var MEDIA_API_REGION = 'media-api-region';
@@ -59,7 +59,7 @@ export var MediaStore = /*#__PURE__*/function () {
   _createClass(MediaStore, [{
     key: "getCollectionItems",
     value: function () {
-      var _getCollectionItems = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee(collectionName, params) {
+      var _getCollectionItems = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee(collectionName, params, traceContext) {
         var metadata, options, response, _yield$createMapRespo, _yield$createMapRespo2, contents, nextInclusiveStartKey, contentsWithoutEmptyFiles;
 
         return _regeneratorRuntime.wrap(function _callee$(_context) {
@@ -77,7 +77,8 @@ export var MediaStore = /*#__PURE__*/function () {
                   params: _objectSpread(_objectSpread({}, defaultGetCollectionItems), params),
                   headers: {
                     Accept: 'application/json'
-                  }
+                  },
+                  traceContext: traceContext
                 });
                 _context.next = 4;
                 return this.request("/collection/".concat(collectionName, "/items"), options);
@@ -112,7 +113,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee, this);
       }));
 
-      function getCollectionItems(_x, _x2) {
+      function getCollectionItems(_x, _x2, _x3) {
         return _getCollectionItems.apply(this, arguments);
       }
 
@@ -121,7 +122,7 @@ export var MediaStore = /*#__PURE__*/function () {
   }, {
     key: "removeCollectionFile",
     value: function () {
-      var _removeCollectionFile = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2(id, collectionName, occurrenceKey) {
+      var _removeCollectionFile = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2(id, collectionName, occurrenceKey, traceContext) {
         var metadata, body, options;
         return _regeneratorRuntime.wrap(function _callee2$(_context2) {
           while (1) {
@@ -149,7 +150,8 @@ export var MediaStore = /*#__PURE__*/function () {
                     Accept: 'application/json',
                     'Content-Type': 'application/json'
                   },
-                  body: JSON.stringify(body)
+                  body: JSON.stringify(body),
+                  traceContext: traceContext
                 });
                 _context2.next = 5;
                 return this.request("/collection/".concat(collectionName), options);
@@ -162,7 +164,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee2, this);
       }));
 
-      function removeCollectionFile(_x3, _x4, _x5) {
+      function removeCollectionFile(_x4, _x5, _x6, _x7) {
         return _removeCollectionFile.apply(this, arguments);
       }
 
@@ -173,6 +175,7 @@ export var MediaStore = /*#__PURE__*/function () {
     value: function createUpload() {
       var createUpTo = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 1;
       var collectionName = arguments.length > 1 ? arguments[1] : undefined;
+      var traceContext = arguments.length > 2 ? arguments[2] : undefined;
       var metadata = {
         method: 'POST',
         endpoint: '/upload'
@@ -187,7 +190,8 @@ export var MediaStore = /*#__PURE__*/function () {
         },
         headers: {
           Accept: 'application/json'
-        }
+        },
+        traceContext: traceContext
       });
 
       return this.request("/upload", options).then(createMapResponseToJson(metadata));
@@ -200,6 +204,7 @@ export var MediaStore = /*#__PURE__*/function () {
             collectionName,
             uploadId,
             partNumber,
+            traceContext,
             metadata,
             options,
             _args3 = arguments;
@@ -209,6 +214,7 @@ export var MediaStore = /*#__PURE__*/function () {
             switch (_context3.prev = _context3.next) {
               case 0:
                 _ref = _args3.length > 2 && _args3[2] !== undefined ? _args3[2] : {}, collectionName = _ref.collectionName, uploadId = _ref.uploadId, partNumber = _ref.partNumber;
+                traceContext = _args3.length > 3 ? _args3[3] : undefined;
                 metadata = {
                   method: 'PUT',
                   endpoint: '/chunk/{etag}'
@@ -221,12 +227,13 @@ export var MediaStore = /*#__PURE__*/function () {
                   authContext: {
                     collectionName: collectionName
                   },
-                  body: blob
+                  body: blob,
+                  traceContext: traceContext
                 });
-                _context3.next = 5;
+                _context3.next = 6;
                 return this.request("/chunk/".concat(etag), options);
 
-              case 5:
+              case 6:
               case "end":
                 return _context3.stop();
             }
@@ -234,7 +241,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee3, this);
       }));
 
-      function uploadChunk(_x6, _x7) {
+      function uploadChunk(_x8, _x9) {
         return _uploadChunk.apply(this, arguments);
       }
 
@@ -247,6 +254,7 @@ export var MediaStore = /*#__PURE__*/function () {
           collectionName = _ref2.collectionName,
           uploadId = _ref2.uploadId;
 
+      var traceContext = arguments.length > 2 ? arguments[2] : undefined;
       var metadata = {
         method: 'POST',
         endpoint: '/chunk/probe'
@@ -262,7 +270,8 @@ export var MediaStore = /*#__PURE__*/function () {
         headers: jsonHeaders,
         body: JSON.stringify({
           chunks: chunks
-        })
+        }),
+        traceContext: traceContext
       });
 
       return this.request("/chunk/probe", options).then(createMapResponseToJson(metadata));
@@ -271,6 +280,7 @@ export var MediaStore = /*#__PURE__*/function () {
     key: "createFileFromUpload",
     value: function createFileFromUpload(body) {
       var params = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+      var traceContext = arguments.length > 2 ? arguments[2] : undefined;
       var metadata = {
         method: 'POST',
         endpoint: '/file/upload'
@@ -282,7 +292,8 @@ export var MediaStore = /*#__PURE__*/function () {
         },
         params: params,
         headers: jsonHeaders,
-        body: JSON.stringify(body)
+        body: JSON.stringify(body),
+        traceContext: traceContext
       });
 
       return this.request('/file/upload', options).then(createMapResponseToJson(metadata));
@@ -291,6 +302,7 @@ export var MediaStore = /*#__PURE__*/function () {
     key: "touchFiles",
     value: function touchFiles(body) {
       var params = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+      var traceContext = arguments.length > 2 ? arguments[2] : undefined;
       var metadata = {
         method: 'POST',
         endpoint: '/upload/createWithFiles'
@@ -301,7 +313,8 @@ export var MediaStore = /*#__PURE__*/function () {
           collectionName: params.collection
         },
         headers: jsonHeaders,
-        body: JSON.stringify(body)
+        body: JSON.stringify(body),
+        traceContext: traceContext
       });
 
       return this.request('/upload/createWithFiles', options).then(createMapResponseToJson(metadata));
@@ -310,6 +323,7 @@ export var MediaStore = /*#__PURE__*/function () {
     key: "getFile",
     value: function getFile(fileId) {
       var params = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : {};
+      var traceContext = arguments.length > 2 ? arguments[2] : undefined;
       var metadata = {
         method: 'GET',
         endpoint: '/file/{fileId}'
@@ -319,7 +333,8 @@ export var MediaStore = /*#__PURE__*/function () {
         authContext: {
           collectionName: params.collection
         },
-        params: params
+        params: params,
+        traceContext: traceContext
       });
 
       return this.request("/file/".concat(fileId), options).then(createMapResponseToJson(metadata));
@@ -352,12 +367,13 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee4, this);
       }));
 
-      function getFileImageURL(_x8, _x9) {
+      function getFileImageURL(_x10, _x11) {
         return _getFileImageURL.apply(this, arguments);
       }
 
       return getFileImageURL;
-    }()
+    }() // TODO Create ticket in case Trace Id can be supported through query params
+
   }, {
     key: "getFileImageURLSync",
     value: function getFileImageURLSync(id, params) {
@@ -407,7 +423,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee5, this);
       }));
 
-      function getFileBinaryURL(_x10, _x11) {
+      function getFileBinaryURL(_x12, _x13) {
         return _getFileBinaryURL.apply(this, arguments);
       }
 
@@ -456,7 +472,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee6, this);
       }));
 
-      function getArtifactURL(_x12, _x13, _x14) {
+      function getArtifactURL(_x14, _x15, _x16) {
         return _getArtifactURL.apply(this, arguments);
       }
 
@@ -465,7 +481,7 @@ export var MediaStore = /*#__PURE__*/function () {
   }, {
     key: "getImage",
     value: function () {
-      var _getImage = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee7(id, params, controller, fetchMaxRes) {
+      var _getImage = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee7(id, params, controller, fetchMaxRes, traceContext) {
         var isWebpSupported, headers, metadata, options;
         return _regeneratorRuntime.wrap(function _callee7$(_context7) {
           while (1) {
@@ -473,11 +489,10 @@ export var MediaStore = /*#__PURE__*/function () {
               case 0:
                 // TODO add checkWebpSupport() back https://product-fabric.atlassian.net/browse/MPT-584
                 isWebpSupported = false;
+                headers = {};
 
                 if (isWebpSupported) {
-                  headers = {
-                    accept: 'image/webp,image/*,*/*;q=0.8'
-                  };
+                  headers.accept = 'image/webp,image/*,*/*;q=0.8';
                 }
 
                 metadata = {
@@ -489,11 +504,12 @@ export var MediaStore = /*#__PURE__*/function () {
                     collectionName: params && params.collection
                   },
                   params: extendImageParams(params, fetchMaxRes),
-                  headers: headers
+                  headers: headers,
+                  traceContext: traceContext
                 });
                 return _context7.abrupt("return", this.request("/file/".concat(id, "/image"), options, controller).then(createMapResponseToBlob(metadata)));
 
-              case 5:
+              case 6:
               case "end":
                 return _context7.stop();
             }
@@ -501,7 +517,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee7, this);
       }));
 
-      function getImage(_x15, _x16, _x17, _x18) {
+      function getImage(_x17, _x18, _x19, _x20, _x21) {
         return _getImage.apply(this, arguments);
       }
 
@@ -510,7 +526,7 @@ export var MediaStore = /*#__PURE__*/function () {
   }, {
     key: "getItems",
     value: function () {
-      var _getItems = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee8(ids, collectionName) {
+      var _getItems = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee8(ids, collectionName, traceContext) {
         var descriptors, metadata, options;
         return _regeneratorRuntime.wrap(function _callee8$(_context8) {
           while (1) {
@@ -534,7 +550,8 @@ export var MediaStore = /*#__PURE__*/function () {
                   headers: jsonHeaders,
                   body: JSON.stringify({
                     descriptors: descriptors
-                  })
+                  }),
+                  traceContext: traceContext
                 });
                 return _context8.abrupt("return", this.request('/items', options).then(createMapResponseToJson(metadata)));
 
@@ -546,7 +563,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee8, this);
       }));
 
-      function getItems(_x19, _x20) {
+      function getItems(_x22, _x23, _x24) {
         return _getItems.apply(this, arguments);
       }
 
@@ -555,7 +572,7 @@ export var MediaStore = /*#__PURE__*/function () {
   }, {
     key: "getImageMetadata",
     value: function () {
-      var _getImageMetadata = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee9(id, params) {
+      var _getImageMetadata = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee9(id, params, traceContext) {
         var metadata, options;
         return _regeneratorRuntime.wrap(function _callee9$(_context9) {
           while (1) {
@@ -569,7 +586,8 @@ export var MediaStore = /*#__PURE__*/function () {
                   authContext: {
                     collectionName: params && params.collection
                   },
-                  params: params
+                  params: params,
+                  traceContext: traceContext
                 });
                 return _context9.abrupt("return", this.request("/file/".concat(id, "/image/metadata"), options).then(createMapResponseToJson(metadata)));
 
@@ -581,7 +599,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee9, this);
       }));
 
-      function getImageMetadata(_x21, _x22) {
+      function getImageMetadata(_x25, _x26, _x27) {
         return _getImageMetadata.apply(this, arguments);
       }
 
@@ -590,7 +608,7 @@ export var MediaStore = /*#__PURE__*/function () {
   }, {
     key: "appendChunksToUpload",
     value: function () {
-      var _appendChunksToUpload = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee10(uploadId, body, collectionName) {
+      var _appendChunksToUpload = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee10(uploadId, body, collectionName, traceContext) {
         var metadata, options;
         return _regeneratorRuntime.wrap(function _callee10$(_context10) {
           while (1) {
@@ -605,7 +623,8 @@ export var MediaStore = /*#__PURE__*/function () {
                     collectionName: collectionName
                   },
                   headers: jsonHeaders,
-                  body: JSON.stringify(body)
+                  body: JSON.stringify(body),
+                  traceContext: traceContext
                 });
                 _context10.next = 4;
                 return this.request("/upload/".concat(uploadId, "/chunks"), options);
@@ -618,7 +637,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee10, this);
       }));
 
-      function appendChunksToUpload(_x23, _x24, _x25) {
+      function appendChunksToUpload(_x28, _x29, _x30, _x31) {
         return _appendChunksToUpload.apply(this, arguments);
       }
 
@@ -626,7 +645,7 @@ export var MediaStore = /*#__PURE__*/function () {
     }()
   }, {
     key: "copyFileWithToken",
-    value: function copyFileWithToken(body, params) {
+    value: function copyFileWithToken(body, params, traceContext) {
       var metadata = {
         method: 'POST',
         endpoint: '/file/copy/withToken'
@@ -640,8 +659,9 @@ export var MediaStore = /*#__PURE__*/function () {
         params: params,
         // Contains collection name to write to
         headers: jsonHeaders,
-        body: JSON.stringify(body) // Contains collection name to read from
-
+        body: JSON.stringify(body),
+        // Contains collection name to read from
+        traceContext: traceContext
       });
 
       return this.request('/file/copy/withToken', options).then(createMapResponseToJson(metadata));
@@ -659,7 +679,9 @@ export var MediaStore = /*#__PURE__*/function () {
             headers,
             body,
             clientOptions,
+            traceContext,
             auth,
+            extendedTraceContext,
             response,
             _args11 = arguments;
         return _regeneratorRuntime.wrap(function _callee11$(_context11) {
@@ -672,13 +694,16 @@ export var MediaStore = /*#__PURE__*/function () {
                   authContext: {}
                 };
                 controller = _args11.length > 2 ? _args11[2] : undefined;
-                method = options.method, endpoint = options.endpoint, authContext = options.authContext, params = options.params, headers = options.headers, body = options.body, clientOptions = options.clientOptions;
+                method = options.method, endpoint = options.endpoint, authContext = options.authContext, params = options.params, headers = options.headers, body = options.body, clientOptions = options.clientOptions, traceContext = options.traceContext;
                 _context11.next = 5;
                 return this.resolveAuth(authContext);
 
               case 5:
                 auth = _context11.sent;
-                _context11.next = 8;
+                extendedTraceContext = traceContext ? _objectSpread(_objectSpread({}, traceContext), {}, {
+                  spanId: (traceContext === null || traceContext === void 0 ? void 0 : traceContext.spanId) || getRandomHex(16)
+                }) : undefined;
+                _context11.next = 9;
                 return _request("".concat(auth.baseUrl).concat(path), {
                   method: method,
                   endpoint: endpoint,
@@ -686,16 +711,17 @@ export var MediaStore = /*#__PURE__*/function () {
                   params: params,
                   headers: headers,
                   body: body,
-                  clientOptions: clientOptions
+                  clientOptions: clientOptions,
+                  traceContext: extendedTraceContext
                 }, controller);
 
-              case 8:
+              case 9:
                 response = _context11.sent;
                 setKeyValueInSessionStorage(MEDIA_API_REGION, response.headers.get('x-media-region'));
                 setKeyValueInSessionStorage(MEDIA_API_ENVIRONMENT, response.headers.get('x-media-env'));
                 return _context11.abrupt("return", response);
 
-              case 12:
+              case 13:
               case "end":
                 return _context11.stop();
             }
@@ -703,7 +729,7 @@ export var MediaStore = /*#__PURE__*/function () {
         }, _callee11, this);
       }));
 
-      function request(_x26) {
+      function request(_x32) {
         return _request2.apply(this, arguments);
       }
 

package/dist/esm/index.js

@@ -8,7 +8,7 @@ export { isUploadingFileState, isProcessingFileState, isProcessedFileState, isEr
 export { StreamsCache } from './file-streams-cache';
 export { uploadFile } from './uploader';
 export { request, RequestError, isRequestError, isRateLimitedError } from './utils/request';
-export { isAbortedRequestError, mapResponseToJson, mapResponseToBlob, mapResponseToVoid, createUrl } from './utils/request/helpers';
+export { isAbortedRequestError, mapResponseToJson, mapResponseToBlob, mapResponseToVoid, createUrl, getRandomHex } from './utils/request/helpers';
 export { PollingFunction } from './utils/polling';
 export { isPollingError, PollingError } from './utils/polling/errors';
 export { imageResizeModeToFileImageMode } from './utils/imageResizeModeToFileImageMode';

package/dist/esm/uploader/index.js

@@ -13,7 +13,7 @@ import { createHasher } from '../utils/hashing/hasherCreator';
 import { UploaderError } from './error';
 import { CHUNK_SIZE, PROCESSING_BATCH_SIZE } from '../constants';
 import { calculateChunkSize, fileSizeError } from './calculateChunkSize';
-import { getMediaFeatureFlag } from '@atlaskit/media-common'; // TODO: Allow to pass multiple files
+import { getMediaFeatureFlag } from '@atlaskit/media-common';
 
 var hashingFunction = /*#__PURE__*/function () {
   var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee(blob) {
@@ -42,7 +42,7 @@ var hashingFunction = /*#__PURE__*/function () {
   };
 }();
 
-var createProbingFunction = function createProbingFunction(store, deferredUploadId, collectionName) {
+var createProbingFunction = function createProbingFunction(store, deferredUploadId, collectionName, traceContext) {
   return /*#__PURE__*/function () {
     var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2(chunks) {
       var response, results;
@@ -76,17 +76,18 @@ var createProbingFunction = function createProbingFunction(store, deferredUpload
                 collectionName: _context2.t2,
                 uploadId: _context2.t4
               };
-              _context2.next = 14;
-              return _context2.t0.probeChunks.call(_context2.t0, _context2.t1, _context2.t5);
+              _context2.t6 = traceContext;
+              _context2.next = 15;
+              return _context2.t0.probeChunks.call(_context2.t0, _context2.t1, _context2.t5, _context2.t6);
 
-            case 14:
+            case 15:
               response = _context2.sent;
               results = response.data.results;
               return _context2.abrupt("return", Object.values(results).map(function (result) {
                 return result.exists;
               }));
 
-            case 17:
+            case 18:
             case "end":
               return _context2.stop();
           }
@@ -100,7 +101,7 @@ var createProbingFunction = function createProbingFunction(store, deferredUpload
   }();
 };
 
-var createUploadingFunction = function createUploadingFunction(store, deferredUploadId, collectionName) {
+var createUploadingFunction = function createUploadingFunction(store, deferredUploadId, collectionName, traceContext) {
   return /*#__PURE__*/function () {
     var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee3(chunk) {
       var options;
@@ -134,7 +135,7 @@ var createUploadingFunction = function createUploadingFunction(store, deferredUp
               _context3.next = 12;
               return store.uploadChunk(chunk.hash, chunk.blob, _objectSpread({
                 collectionName: collectionName
-              }, options));
+              }, options), traceContext);
 
             case 12:
               return _context3.abrupt("return", _context3.sent);
@@ -153,7 +154,7 @@ var createUploadingFunction = function createUploadingFunction(store, deferredUp
   }();
 };
 
-var createProcessingFunction = function createProcessingFunction(store, deferredUploadId, collection) {
+var createProcessingFunction = function createProcessingFunction(store, deferredUploadId, collection, traceContext) {
   var offset = 0;
   return /*#__PURE__*/function () {
     var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee4(chunks) {
@@ -172,13 +173,14 @@ var createProcessingFunction = function createProcessingFunction(store, deferred
                 offset: offset
               };
               _context4.t3 = collection;
-              _context4.next = 8;
-              return _context4.t0.appendChunksToUpload.call(_context4.t0, _context4.t1, _context4.t2, _context4.t3);
+              _context4.t4 = traceContext;
+              _context4.next = 9;
+              return _context4.t0.appendChunksToUpload.call(_context4.t0, _context4.t1, _context4.t2, _context4.t3, _context4.t4);
 
-            case 8:
+            case 9:
               offset += chunks.length;
 
-            case 9:
+            case 10:
             case "end":
               return _context4.stop();
           }
@@ -193,7 +195,7 @@ var createProcessingFunction = function createProcessingFunction(store, deferred
 };
 
 var createFileFromUpload = /*#__PURE__*/function () {
-  var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee5(file, store, uploadableFileUpfrontIds, uploadId) {
+  var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee5(file, store, uploadableFileUpfrontIds, uploadId, traceContext) {
     var collection, name, mimeType, id, occurrenceKey;
     return _regeneratorRuntime.wrap(function _callee5$(_context5) {
       while (1) {
@@ -209,7 +211,7 @@ var createFileFromUpload = /*#__PURE__*/function () {
               occurrenceKey: occurrenceKey,
               collection: collection,
               replaceFileId: id
-            }));
+            }, traceContext));
 
           case 3:
           case "end":
@@ -219,12 +221,12 @@ var createFileFromUpload = /*#__PURE__*/function () {
     }, _callee5);
   }));
 
-  return function createFileFromUpload(_x5, _x6, _x7, _x8) {
+  return function createFileFromUpload(_x5, _x6, _x7, _x8, _x9) {
     return _ref5.apply(this, arguments);
   };
 }();
 
-export var uploadFile = function uploadFile(file, store, uploadableFileUpfrontIds, callbacks) {
+export var uploadFile = function uploadFile(file, store, uploadableFileUpfrontIds, callbacks, traceContext) {
   var content = file.content,
       collection = file.collection;
   var deferredUploadId = uploadableFileUpfrontIds.deferredUploadId,
@@ -257,10 +259,10 @@ export var uploadFile = function uploadFile(file, store, uploadableFileUpfrontId
     probingBatchSize: 100,
     chunkSize: chunkSize,
     uploadingConcurrency: 3,
-    uploadingFunction: createUploadingFunction(store, deferredUploadId, collection),
-    probingFunction: createProbingFunction(store, deferredUploadId, collection),
+    uploadingFunction: createUploadingFunction(store, deferredUploadId, collection, traceContext),
+    probingFunction: createProbingFunction(store, deferredUploadId, collection, traceContext),
     processingBatchSize: PROCESSING_BATCH_SIZE,
-    processingFunction: createProcessingFunction(store, deferredUploadId, collection)
+    processingFunction: createProcessingFunction(store, deferredUploadId, collection, traceContext)
   }, {
     onProgress: function onProgress(progress) {
       if (callbacks) {
@@ -273,7 +275,7 @@ export var uploadFile = function uploadFile(file, store, uploadableFileUpfrontId
 
   var subscription = from(deferredUploadId).pipe(concatMap(function (uploadId) {
     return chunkinatorObservable.pipe(concatMap(function () {
-      return from(createFileFromUpload(file, store, uploadableFileUpfrontIds, uploadId));
+      return from(createFileFromUpload(file, store, uploadableFileUpfrontIds, uploadId, traceContext));
     }));
   })).subscribe({
     error: function error(err) {