UNPKG

@aws-amplify/storage

Version:

Storage category of aws-amplify

60 lines (52 loc) 1.84 kB
// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. // SPDX-License-Identifier: Apache-2.0 import { createUploadTask } from '../../../utils'; import { assertValidationError } from '../../../../../errors/utils/assertValidationError'; import { StorageValidationErrorCode } from '../../../../../errors/types/validation'; import { DEFAULT_PART_SIZE, MAX_OBJECT_SIZE } from '../../../utils/constants'; import { byteLength } from './byteLength'; import { SinglePartUploadDataInput, putObjectJob } from './putObjectJob'; import { MultipartUploadDataInput, getMultipartUploadHandlers, } from './multipart'; export const uploadData = ( input: SinglePartUploadDataInput | MultipartUploadDataInput, ) => { const { data } = input; const dataByteLength = byteLength(data); // Using InvalidUploadSource error code because the input data must NOT be any // of permitted Blob, string, ArrayBuffer(View) if byteLength could not be determined. assertValidationError( dataByteLength !== undefined, StorageValidationErrorCode.InvalidUploadSource, ); assertValidationError( dataByteLength <= MAX_OBJECT_SIZE, StorageValidationErrorCode.ObjectIsTooLarge, ); if (dataByteLength <= DEFAULT_PART_SIZE) { // Single part upload const abortController = new AbortController(); return createUploadTask({ isMultipartUpload: false, job: putObjectJob(input, abortController.signal, dataByteLength), onCancel: (message?: string) => { abortController.abort(message); }, }); } else { // Multipart upload const { multipartUploadJob, onPause, onResume, onCancel } = getMultipartUploadHandlers(input, dataByteLength); return createUploadTask({ isMultipartUpload: true, job: multipartUploadJob, onCancel: (message?: string) => { onCancel(message); }, onPause, onResume, }); } };