1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201 |
- var session = require('./session');
- var Async = require('./async');
- var EventProxy = require('./event').EventProxy;
- var util = require('./util');
- // 文件分块上传全过程,暴露的分块上传接口
- function sliceUploadFile(params, callback) {
- var self = this;
- // 如果小程序版本不支持获取文件分片内容,统一转到 postObject 接口上传
- if (!util.canFileSlice()) {
- params.SkipTask = true;
- self.postObject(params, callback);
- return;
- }
- var ep = new EventProxy();
- var TaskId = params.TaskId;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var FilePath = params.FilePath;
- var ChunkSize = params.ChunkSize || params.SliceSize || self.options.ChunkSize;
- var AsyncLimit = params.AsyncLimit;
- var StorageClass = params.StorageClass;
- var ServerSideEncryption = params.ServerSideEncryption;
- var FileSize;
- var onProgress;
- var onHashProgress = params.onHashProgress;
- // 上传过程中出现错误,返回错误
- ep.on('error', function (err) {
- if (!self._isRunningTask(TaskId)) return;
- var _err = {
- UploadId: params.UploadData.UploadId || '',
- err: err,
- };
- return callback(_err);
- });
- // 上传分块完成,开始 uploadSliceComplete 操作
- ep.on('upload_complete', function (UploadCompleteData) {
- var _UploadCompleteData = util.extend({
- UploadId: params.UploadData.UploadId || ''
- }, UploadCompleteData);
- callback(null, _UploadCompleteData);
- });
- // 上传分块完成,开始 uploadSliceComplete 操作
- ep.on('upload_slice_complete', function (UploadData) {
- uploadSliceComplete.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- UploadId: UploadData.UploadId,
- SliceList: UploadData.SliceList,
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- session.removeUsing(UploadData.UploadId);
- if (err) {
- onProgress(null, true);
- return ep.emit('error', err);
- }
- session.removeUploadId(UploadData.UploadId);
- onProgress({loaded: FileSize, total: FileSize}, true);
- ep.emit('upload_complete', data);
- });
- });
- // 获取 UploadId 完成,开始上传每个分片
- ep.on('get_upload_data_finish', function (UploadData) {
- // 处理 UploadId 缓存
- var uuid = session.getFileId(params.FileStat, params.ChunkSize, Bucket, Key);
- uuid && session.saveUploadId(uuid, UploadData.UploadId, self.options.UploadIdCacheLimit); // 缓存 UploadId
- session.setUsing(UploadData.UploadId); // 标记 UploadId 为正在使用
- // 获取 UploadId
- onProgress(null, true); // 任务状态开始 uploading
- uploadSliceList.call(self, {
- TaskId: TaskId,
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- FilePath: FilePath,
- FileSize: FileSize,
- SliceSize: ChunkSize,
- AsyncLimit: AsyncLimit,
- ServerSideEncryption: ServerSideEncryption,
- UploadData: UploadData,
- onProgress: onProgress
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- onProgress(null, true);
- return ep.emit('error', err);
- }
- ep.emit('upload_slice_complete', data);
- });
- });
- // 开始获取文件 UploadId,里面会视情况计算 ETag,并比对,保证文件一致性,也优化上传
- ep.on('get_file_size_finish', function () {
- onProgress = util.throttleOnProgress.call(self, FileSize, params.onProgress);
- if (params.UploadData.UploadId) {
- ep.emit('get_upload_data_finish', params.UploadData);
- } else {
- var _params = util.extend({
- TaskId: TaskId,
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- Headers: params.Headers,
- StorageClass: StorageClass,
- FilePath: FilePath,
- FileSize: FileSize,
- SliceSize: ChunkSize,
- onHashProgress: onHashProgress,
- }, params);
- getUploadIdAndPartList.call(self, _params, function (err, UploadData) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) return ep.emit('error', err);
- params.UploadData.UploadId = UploadData.UploadId;
- params.UploadData.PartList = UploadData.PartList;
- ep.emit('get_upload_data_finish', params.UploadData);
- });
- }
- });
- // 获取上传文件大小
- FileSize = params.ContentLength;
- delete params.ContentLength;
- !params.Headers && (params.Headers = {});
- util.each(params.Headers, function (item, key) {
- if (key.toLowerCase() === 'content-length') {
- delete params.Headers[key];
- }
- });
- // 控制分片大小
- (function () {
- var SIZE = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 1024 * 2, 1024 * 4, 1024 * 5];
- var AutoChunkSize = 1024 * 1024;
- for (var i = 0; i < SIZE.length; i++) {
- AutoChunkSize = SIZE[i] * 1024 * 1024;
- if (FileSize / AutoChunkSize <= self.options.MaxPartNumber) break;
- }
- params.ChunkSize = params.SliceSize = ChunkSize = Math.max(ChunkSize, AutoChunkSize);
- })();
- // 开始上传
- if (FileSize === 0) {
- params.Body = '';
- params.ContentLength = 0;
- params.SkipTask = true;
- self.putObject(params, function (err, data) {
- if (err) {
- return callback(err);
- }
- callback(null, data);
- });
- } else {
- ep.emit('get_file_size_finish');
- }
- }
- // 获取上传任务的 UploadId
- function getUploadIdAndPartList(params, callback) {
- var TaskId = params.TaskId;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var StorageClass = params.StorageClass;
- var self = this;
- // 计算 ETag
- var ETagMap = {};
- var FileSize = params.FileSize;
- var SliceSize = params.SliceSize;
- var SliceCount = Math.ceil(FileSize / SliceSize);
- var FinishSliceCount = 0;
- var FinishSize = 0;
- var onHashProgress = util.throttleOnProgress.call(self, FileSize, params.onHashProgress);
- var getChunkETag = function (PartNumber, callback) {
- var start = SliceSize * (PartNumber - 1);
- var end = Math.min(start + SliceSize, FileSize);
- var ChunkSize = end - start;
- if (ETagMap[PartNumber]) {
- callback(null, {
- PartNumber: PartNumber,
- ETag: ETagMap[PartNumber],
- Size: ChunkSize
- });
- } else {
- util.fileSlice(params.FilePath, start, end, function (chunkItem) {
- try {
- var md5 = util.getFileMd5(chunkItem);
- } catch (err) {
- return callback(err);
- }
- var ETag = '"' + md5 + '"';
- ETagMap[PartNumber] = ETag;
- FinishSliceCount += 1;
- FinishSize += ChunkSize;
- callback(null, {
- PartNumber: PartNumber,
- ETag: ETag,
- Size: ChunkSize
- });
- onHashProgress({loaded: FinishSize, total: FileSize});
- });
- }
- };
- // 通过和文件的 md5 对比,判断 UploadId 是否可用
- var isAvailableUploadList = function (PartList, callback) {
- var PartCount = PartList.length;
- // 如果没有分片,通过
- if (PartCount === 0) {
- return callback(null, true);
- }
- // 检查分片数量
- if (PartCount > SliceCount) {
- return callback(null, false);
- }
- // 检查分片大小
- if (PartCount > 1) {
- var PartSliceSize = Math.max(PartList[0].Size, PartList[1].Size);
- if (PartSliceSize !== SliceSize) {
- return callback(null, false);
- }
- }
- // 逐个分片计算并检查 ETag 是否一致
- var next = function (index) {
- if (index < PartCount) {
- var Part = PartList[index];
- getChunkETag(Part.PartNumber, function (err, chunk) {
- if (chunk && chunk.ETag === Part.ETag && chunk.Size === Part.Size) {
- next(index + 1);
- } else {
- callback(null, false);
- }
- });
- } else {
- callback(null, true);
- }
- };
- next(0);
- };
- var ep = new EventProxy();
- ep.on('error', function (errData) {
- if (!self._isRunningTask(TaskId)) return;
- return callback(errData);
- });
- // 存在 UploadId
- ep.on('upload_id_available', function (UploadData) {
- // 转换成 map
- var map = {};
- var list = [];
- util.each(UploadData.PartList, function (item) {
- map[item.PartNumber] = item;
- });
- for (var PartNumber = 1; PartNumber <= SliceCount; PartNumber++) {
- var item = map[PartNumber];
- if (item) {
- item.PartNumber = PartNumber;
- item.Uploaded = true;
- } else {
- item = {
- PartNumber: PartNumber,
- ETag: null,
- Uploaded: false
- };
- }
- list.push(item);
- }
- UploadData.PartList = list;
- callback(null, UploadData);
- });
- // 不存在 UploadId, 初始化生成 UploadId
- ep.on('no_available_upload_id', function () {
- if (!self._isRunningTask(TaskId)) return;
- var _params = util.extend({
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- Headers: util.clone(params.Headers),
- Query: util.clone(params.Query),
- StorageClass: StorageClass,
- }, params);
- self.multipartInit(_params, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) return ep.emit('error', err);
- var UploadId = data.UploadId;
- if (!UploadId) {
- return callback({Message: 'no upload id'});
- }
- ep.emit('upload_id_available', {UploadId: UploadId, PartList: []});
- });
- });
- // 如果已存在 UploadId,找一个可以用的 UploadId
- ep.on('has_and_check_upload_id', function (UploadIdList) {
- // 串行地,找一个内容一致的 UploadId
- UploadIdList = UploadIdList.reverse();
- Async.eachLimit(UploadIdList, 1, function (UploadId, asyncCallback) {
- if (!self._isRunningTask(TaskId)) return;
- // 如果正在上传,跳过
- if (session.using[UploadId]) {
- asyncCallback(); // 检查下一个 UploadId
- return;
- }
- // 判断 UploadId 是否可用
- wholeMultipartListPart.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- UploadId: UploadId,
- }, function (err, PartListData) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- session.removeUsing(UploadId);
- return ep.emit('error', err);
- }
- var PartList = PartListData.PartList;
- PartList.forEach(function (item) {
- item.PartNumber *= 1;
- item.Size *= 1;
- item.ETag = item.ETag || '';
- });
- isAvailableUploadList(PartList, function (err, isAvailable) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) return ep.emit('error', err);
- if (isAvailable) {
- asyncCallback({
- UploadId: UploadId,
- PartList: PartList
- }); // 马上结束
- } else {
- asyncCallback(); // 检查下一个 UploadId
- }
- });
- });
- }, function (AvailableUploadData) {
- if (!self._isRunningTask(TaskId)) return;
- onHashProgress(null, true);
- if (AvailableUploadData && AvailableUploadData.UploadId) {
- ep.emit('upload_id_available', AvailableUploadData);
- } else {
- ep.emit('no_available_upload_id');
- }
- });
- });
- // 在本地缓存找可用的 UploadId
- ep.on('seek_local_avail_upload_id', function (RemoteUploadIdList) {
- // 在本地找可用的 UploadId
- var uuid = session.getFileId(params.FileStat, params.ChunkSize, Bucket, Key);
- var LocalUploadIdList = session.getUploadIdList(uuid);
- if (!uuid || !LocalUploadIdList) {
- ep.emit('has_and_check_upload_id', RemoteUploadIdList);
- return;
- }
- var next = function (index) {
- // 如果本地找不到可用 UploadId,再一个个遍历校验远端
- if (index >= LocalUploadIdList.length) {
- ep.emit('has_and_check_upload_id', RemoteUploadIdList);
- return;
- }
- var UploadId = LocalUploadIdList[index];
- // 如果不在远端 UploadId 列表里,跳过并删除
- if (!util.isInArray(RemoteUploadIdList, UploadId)) {
- session.removeUploadId(UploadId);
- next(index + 1);
- return;
- }
- // 如果正在上传,跳过
- if (session.using[UploadId]) {
- next(index + 1);
- return;
- }
- // 判断 UploadId 是否存在线上
- wholeMultipartListPart.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- UploadId: UploadId,
- }, function (err, PartListData) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- // 如果 UploadId 获取会出错,跳过并删除
- session.removeUploadId(UploadId);
- next(index + 1);
- } else {
- // 找到可用 UploadId
- ep.emit('upload_id_available', {
- UploadId: UploadId,
- PartList: PartListData.PartList,
- });
- }
- });
- };
- next(0);
- });
- // 获取线上 UploadId 列表
- ep.on('get_remote_upload_id_list', function () {
- // 获取符合条件的 UploadId 列表,因为同一个文件可以有多个上传任务。
- wholeMultipartList.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- return ep.emit('error', err);
- }
- // 整理远端 UploadId 列表
- var RemoteUploadIdList = util.filter(data.UploadList, function (item) {
- return item.Key === Key && (!StorageClass || item.StorageClass.toUpperCase() === StorageClass.toUpperCase());
- }).reverse().map(function (item) {
- return item.UploadId || item.UploadID;
- });
- if (RemoteUploadIdList.length) {
- ep.emit('seek_local_avail_upload_id', RemoteUploadIdList);
- } else {
- // 远端没有 UploadId,清理缓存的 UploadId
- var uuid = session.getFileId(params.FileStat, params.ChunkSize, Bucket, Key), LocalUploadIdList;
- if (uuid && (LocalUploadIdList = session.getUploadIdList(uuid))) {
- util.each(LocalUploadIdList, function (UploadId) {
- session.removeUploadId(UploadId);
- });
- }
- ep.emit('no_available_upload_id');
- }
- });
- });
- // 开始找可用 UploadId
- ep.emit('get_remote_upload_id_list');
- }
- // 获取符合条件的全部上传任务 (条件包括 Bucket, Region, Prefix)
- function wholeMultipartList(params, callback) {
- var self = this;
- var UploadList = [];
- var sendParams = {
- Bucket: params.Bucket,
- Region: params.Region,
- Prefix: params.Key
- };
- var next = function () {
- self.multipartList(sendParams, function (err, data) {
- if (err) return callback(err);
- UploadList.push.apply(UploadList, data.Upload || []);
- if (data.IsTruncated === 'true') { // 列表不完整
- sendParams.KeyMarker = data.NextKeyMarker;
- sendParams.UploadIdMarker = data.NextUploadIdMarker;
- next();
- } else {
- callback(null, {UploadList: UploadList});
- }
- });
- };
- next();
- }
- // 获取指定上传任务的分块列表
- function wholeMultipartListPart(params, callback) {
- var self = this;
- var PartList = [];
- var sendParams = {
- Bucket: params.Bucket,
- Region: params.Region,
- Key: params.Key,
- UploadId: params.UploadId
- };
- var next = function () {
- self.multipartListPart(sendParams, function (err, data) {
- if (err) return callback(err);
- PartList.push.apply(PartList, data.Part || []);
- if (data.IsTruncated === 'true') { // 列表不完整
- sendParams.PartNumberMarker = data.NextPartNumberMarker;
- next();
- } else {
- callback(null, {PartList: PartList});
- }
- });
- };
- next();
- }
- // 上传文件分块,包括
- /*
- UploadId (上传任务编号)
- AsyncLimit (并发量),
- SliceList (上传的分块数组),
- FilePath (本地文件的位置),
- SliceSize (文件分块大小)
- FileSize (文件大小)
- onProgress (上传成功之后的回调函数)
- */
- function uploadSliceList(params, cb) {
- var self = this;
- var TaskId = params.TaskId;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var UploadData = params.UploadData;
- var FileSize = params.FileSize;
- var SliceSize = params.SliceSize;
- var ChunkParallel = Math.min(params.AsyncLimit || self.options.ChunkParallelLimit || 1, 256);
- var FilePath = params.FilePath;
- var SliceCount = Math.ceil(FileSize / SliceSize);
- var FinishSize = 0;
- var ServerSideEncryption = params.ServerSideEncryption;
- var needUploadSlices = util.filter(UploadData.PartList, function (SliceItem) {
- if (SliceItem['Uploaded']) {
- FinishSize += SliceItem['PartNumber'] >= SliceCount ? (FileSize % SliceSize || SliceSize) : SliceSize;
- }
- return !SliceItem['Uploaded'];
- });
- var onProgress = params.onProgress;
- Async.eachLimit(needUploadSlices, ChunkParallel, function (SliceItem, asyncCallback) {
- if (!self._isRunningTask(TaskId)) return;
- var PartNumber = SliceItem['PartNumber'];
- var currentSize = Math.min(FileSize, SliceItem['PartNumber'] * SliceSize) - (SliceItem['PartNumber'] - 1) * SliceSize;
- var preAddSize = 0;
- uploadSliceItem.call(self, {
- TaskId: TaskId,
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- SliceSize: SliceSize,
- FileSize: FileSize,
- PartNumber: PartNumber,
- ServerSideEncryption: ServerSideEncryption,
- FilePath: FilePath,
- UploadData: UploadData,
- onProgress: function (data) {
- FinishSize += data.loaded - preAddSize;
- preAddSize = data.loaded;
- onProgress({loaded: FinishSize, total: FileSize});
- },
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- FinishSize -= preAddSize;
- } else {
- FinishSize += currentSize - preAddSize;
- SliceItem.ETag = data.ETag;
- }
- onProgress({loaded: FinishSize, total: FileSize});
- asyncCallback(err || null, data);
- });
- }, function (err) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) return cb(err);
- cb(null, {
- UploadId: UploadData.UploadId,
- SliceList: UploadData.PartList
- });
- });
- }
- // 上传指定分片
- function uploadSliceItem(params, callback) {
- var self = this;
- var TaskId = params.TaskId;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var FileSize = params.FileSize;
- var FilePath = params.FilePath;
- var PartNumber = params.PartNumber * 1;
- var SliceSize = params.SliceSize;
- var ServerSideEncryption = params.ServerSideEncryption;
- var UploadData = params.UploadData;
- var ChunkRetryTimes = self.options.ChunkRetryTimes + 1;
- var Headers = params.Headers || {};
- var start = SliceSize * (PartNumber - 1);
- var ContentLength = SliceSize;
- var end = start + SliceSize;
- if (end > FileSize) {
- end = FileSize;
- ContentLength = end - start;
- }
- var headersWhiteList = ['x-cos-traffic-limit', 'x-cos-mime-limit'];
- var headers = {};
- util.each(Headers, function(v, k) {
- if (headersWhiteList.indexOf(k) > -1) {
- headers[k] = v;
- }
- });
- util.fileSlice(FilePath, start, end, function (Body) {
- var md5 = util.getFileMd5(Body);
- var contentMd5 = md5 ? util.binaryBase64(md5) : null;
- var PartItem = UploadData.PartList[PartNumber - 1];
- Async.retry(ChunkRetryTimes, function (tryCallback) {
- if (!self._isRunningTask(TaskId)) return;
- self.multipartUpload({
- TaskId: TaskId,
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- ContentLength: ContentLength,
- PartNumber: PartNumber,
- UploadId: UploadData.UploadId,
- ServerSideEncryption: ServerSideEncryption,
- Body: Body,
- Headers: headers,
- onProgress: params.onProgress,
- ContentMD5: contentMd5,
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- if (err) {
- return tryCallback(err);
- } else {
- PartItem.Uploaded = true;
- return tryCallback(null, data);
- }
- });
- }, function (err, data) {
- if (!self._isRunningTask(TaskId)) return;
- return callback(err, data);
- });
- });
- }
- // 完成分块上传
- function uploadSliceComplete(params, callback) {
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var UploadId = params.UploadId;
- var SliceList = params.SliceList;
- var self = this;
- var ChunkRetryTimes = this.options.ChunkRetryTimes + 1;
- var Parts = SliceList.map(function (item) {
- return {
- PartNumber: item.PartNumber,
- ETag: item.ETag
- };
- });
- // 完成上传的请求也做重试
- Async.retry(ChunkRetryTimes, function (tryCallback) {
- self.multipartComplete({
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- UploadId: UploadId,
- Parts: Parts
- }, tryCallback);
- }, function (err, data) {
- callback(err, data);
- });
- }
- // 抛弃分块上传任务
- /*
- AsyncLimit (抛弃上传任务的并发量),
- UploadId (上传任务的编号,当 Level 为 task 时候需要)
- Level (抛弃分块上传任务的级别,task : 抛弃指定的上传任务,file : 抛弃指定的文件对应的上传任务,其他值 :抛弃指定Bucket 的全部上传任务)
- */
- function abortUploadTask(params, callback) {
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var UploadId = params.UploadId;
- var Level = params.Level || 'task';
- var AsyncLimit = params.AsyncLimit;
- var self = this;
- var ep = new EventProxy();
- ep.on('error', function (errData) {
- return callback(errData);
- });
- // 已经获取到需要抛弃的任务列表
- ep.on('get_abort_array', function (AbortArray) {
- abortUploadTaskArray.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- Headers: params.Headers,
- AsyncLimit: AsyncLimit,
- AbortArray: AbortArray
- }, function (err, data) {
- if (err) {
- return callback(err);
- }
- callback(null, data);
- });
- });
- if (Level === 'bucket') {
- // Bucket 级别的任务抛弃,抛弃该 Bucket 下的全部上传任务
- wholeMultipartList.call(self, {
- Bucket: Bucket,
- Region: Region
- }, function (err, data) {
- if (err) {
- return callback(err);
- }
- ep.emit('get_abort_array', data.UploadList || []);
- });
- } else if (Level === 'file') {
- // 文件级别的任务抛弃,抛弃该文件的全部上传任务
- if (!Key) return callback({error: 'abort_upload_task_no_key'});
- wholeMultipartList.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key
- }, function (err, data) {
- if (err) {
- return callback(err);
- }
- ep.emit('get_abort_array', data.UploadList || []);
- });
- } else if (Level === 'task') {
- // 单个任务级别的任务抛弃,抛弃指定 UploadId 的上传任务
- if (!UploadId) return callback({error: 'abort_upload_task_no_id'});
- if (!Key) return callback({error: 'abort_upload_task_no_key'});
- ep.emit('get_abort_array', [{
- Key: Key,
- UploadId: UploadId
- }]);
- } else {
- return callback({error: 'abort_unknown_level'});
- }
- }
- // 批量抛弃分块上传任务
- function abortUploadTaskArray(params, callback) {
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var AbortArray = params.AbortArray;
- var AsyncLimit = params.AsyncLimit || 1;
- var self = this;
- var index = 0;
- var resultList = new Array(AbortArray.length);
- Async.eachLimit(AbortArray, AsyncLimit, function (AbortItem, callback) {
- var eachIndex = index;
- if (Key && Key !== AbortItem.Key) {
- resultList[eachIndex] = {error: {KeyNotMatch: true}};
- callback(null);
- return;
- }
- var UploadId = AbortItem.UploadId || AbortItem.UploadID;
- self.multipartAbort({
- Bucket: Bucket,
- Region: Region,
- Key: AbortItem.Key,
- Headers: params.Headers,
- UploadId: UploadId
- }, function (err) {
- var task = {
- Bucket: Bucket,
- Region: Region,
- Key: AbortItem.Key,
- UploadId: UploadId
- };
- resultList[eachIndex] = {error: err, task: task};
- callback(null);
- });
- index++;
- }, function (err) {
- if (err) {
- return callback(err);
- }
- var successList = [];
- var errorList = [];
- for (var i = 0, len = resultList.length; i < len; i++) {
- var item = resultList[i];
- if (item['task']) {
- if (item['error']) {
- errorList.push(item['task']);
- } else {
- successList.push(item['task']);
- }
- }
- }
- return callback(null, {
- successList: successList,
- errorList: errorList
- });
- });
- }
- // 高级上传
- function uploadFile(params, callback) {
- var self = this;
- // 判断多大的文件使用分片上传
- var SliceSize = params.SliceSize === undefined ? self.options.SliceSize : params.SliceSize;
- var taskList = [];
- var FileSize = params.FileSize;
- var fileInfo = {TaskId: ''};
- // 整理 option,用于返回给回调
- util.each(params, function (v, k) {
- if (typeof v !== 'object' && typeof v !== 'function') {
- fileInfo[k] = v;
- }
- });
- // 处理文件 TaskReady
- var _onTaskReady = params.onTaskReady;
- params.onTaskReady = function (tid) {
- fileInfo.TaskId = tid;
- _onTaskReady && _onTaskReady(tid);
- };
- // 处理文件完成
- var _onFileFinish = params.onFileFinish;
- var onFileFinish = function (err, data) {
- _onFileFinish && _onFileFinish(err, data, fileInfo);
- callback && callback(err, data);
- };
- // 添加上传任务
- var api = FileSize > SliceSize ? 'sliceUploadFile' : 'postObject';
- taskList.push({
- api: api,
- params: params,
- callback: onFileFinish,
- });
- self._addTasks(taskList);
- }
- // 批量上传文件
- function uploadFiles(params, callback) {
- var self = this;
- // 判断多大的文件使用分片上传
- var SliceSize = params.SliceSize === undefined ? self.options.SliceSize : params.SliceSize;
- // 汇总返回进度
- var TotalSize = 0;
- var TotalFinish = 0;
- var onTotalProgress = util.throttleOnProgress.call(self, TotalFinish, params.onProgress);
- // 汇总返回回调
- var unFinishCount = params.files.length;
- var _onTotalFileFinish = params.onFileFinish;
- var resultList = Array(unFinishCount);
- var onTotalFileFinish = function (err, data, options) {
- onTotalProgress(null, true);
- _onTotalFileFinish && _onTotalFileFinish(err, data, options);
- resultList[options.Index] = {
- options: options,
- error: err,
- data: data
- };
- if (--unFinishCount <= 0 && callback) {
- callback(null, {
- files: resultList,
- });
- }
- };
- // 开始处理每个文件
- var taskList = [];
- util.each(params.files, function (fileParams, index) {
- var FileSize = fileParams.FileSize;
- var fileInfo = {Index: index, TaskId: ''};
- // 更新文件总大小
- TotalSize += FileSize;
- // 整理 option,用于返回给回调
- util.each(fileParams, function (v, k) {
- if (typeof v !== 'object' && typeof v !== 'function') {
- fileInfo[k] = v;
- }
- });
- // 处理单个文件 TaskReady
- var _onTaskReady = fileParams.onTaskReady;
- fileParams.onTaskReady = function (tid) {
- fileInfo.TaskId = tid;
- _onTaskReady && _onTaskReady(tid);
- };
- // 处理单个文件进度
- var PreAddSize = 0;
- var _onProgress = fileParams.onProgress;
- fileParams.onProgress = function (info) {
- TotalFinish = TotalFinish - PreAddSize + info.loaded;
- PreAddSize = info.loaded;
- _onProgress && _onProgress(info);
- onTotalProgress({loaded: TotalFinish, total: TotalSize});
- };
- // 处理单个文件完成
- var _onFileFinish = fileParams.onFileFinish;
- var onFileFinish = function (err, data) {
- _onFileFinish && _onFileFinish(err, data);
- onTotalFileFinish && onTotalFileFinish(err, data, fileInfo);
- };
- // 添加上传任务
- var api = FileSize > SliceSize ? 'sliceUploadFile' : 'postObject';
- taskList.push({
- api: api,
- params: fileParams,
- callback: onFileFinish,
- });
- });
- self._addTasks(taskList);
- }
- // 分片复制文件
- function sliceCopyFile(params, callback) {
- var ep = new EventProxy();
- var self = this;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var CopySource = params.CopySource;
- var m = CopySource.match(/^([^.]+-\d+)\.cos(v6)?\.([^.]+)\.[^/]+\/(.+)$/);
- if (!m) {
- callback({error: 'CopySource format error'});
- return;
- }
- var SourceBucket = m[1];
- var SourceRegion = m[3];
- var SourceKey = decodeURIComponent(m[4]);
- var CopySliceSize = params.CopySliceSize === undefined ? self.options.CopySliceSize : params.CopySliceSize;
- CopySliceSize = Math.max(0, CopySliceSize);
- var ChunkSize = params.CopyChunkSize || this.options.CopyChunkSize;
- var ChunkParallel = this.options.CopyChunkParallelLimit;
- var FinishSize = 0;
- var FileSize;
- var onProgress;
- // 分片复制完成,开始 multipartComplete 操作
- ep.on('copy_slice_complete', function (UploadData) {
- self.multipartComplete({
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- UploadId: UploadData.UploadId,
- Parts: UploadData.PartList,
- },function (err, data) {
- if (err) {
- onProgress(null, true);
- return callback(err);
- }
- onProgress({loaded: FileSize, total: FileSize}, true);
- callback(null, data);
- });
- });
- ep.on('get_copy_data_finish',function (UploadData) {
- Async.eachLimit(UploadData.PartList, ChunkParallel, function (SliceItem, asyncCallback) {
- var PartNumber = SliceItem.PartNumber;
- var CopySourceRange = SliceItem.CopySourceRange;
- var currentSize = SliceItem.end - SliceItem.start;
- var preAddSize = 0;
- copySliceItem.call(self, {
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- CopySource: CopySource,
- UploadId: UploadData.UploadId,
- PartNumber: PartNumber,
- CopySourceRange: CopySourceRange,
- onProgress: function (data) {
- FinishSize += data.loaded - preAddSize;
- preAddSize = data.loaded;
- onProgress({loaded: FinishSize, total: FileSize});
- }
- },function (err,data) {
- if (err) {
- return asyncCallback(err);
- }
- onProgress({loaded: FinishSize, total: FileSize});
- FinishSize += currentSize - preAddSize;
- SliceItem.ETag = data.ETag;
- asyncCallback(err || null, data);
- });
- }, function (err) {
- if (err) {
- onProgress(null, true);
- return callback(err);
- }
- ep.emit('copy_slice_complete', UploadData);
- });
- });
- ep.on('get_file_size_finish', function (SourceHeaders) {
- // 控制分片大小
- (function () {
- var SIZE = [1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 1024 * 2, 1024 * 4, 1024 * 5];
- var AutoChunkSize = 1024 * 1024;
- for (var i = 0; i < SIZE.length; i++) {
- AutoChunkSize = SIZE[i] * 1024 * 1024;
- if (FileSize / AutoChunkSize <= self.options.MaxPartNumber) break;
- }
- params.ChunkSize = ChunkSize = Math.max(ChunkSize, AutoChunkSize);
- var ChunkCount = Math.ceil(FileSize / ChunkSize);
- var list = [];
- for (var partNumber = 1; partNumber <= ChunkCount; partNumber++) {
- var start = (partNumber - 1) * ChunkSize;
- var end = partNumber * ChunkSize < FileSize ? (partNumber * ChunkSize - 1) : FileSize - 1;
- var item = {
- PartNumber: partNumber,
- start: start,
- end: end,
- CopySourceRange: "bytes=" + start + "-" + end,
- };
- list.push(item);
- }
- params.PartList = list;
- })();
- var TargetHeader;
- if (params.Headers['x-cos-metadata-directive'] === 'Replaced') {
- TargetHeader = params.Headers;
- } else {
- TargetHeader = SourceHeaders;
- }
- TargetHeader['x-cos-storage-class'] = params.Headers['x-cos-storage-class'] || SourceHeaders['x-cos-storage-class'];
- TargetHeader = util.clearKey(TargetHeader);
- /**
- * 对于归档存储的对象,如果未恢复副本,则不允许 Copy
- */
- if (SourceHeaders['x-cos-storage-class'] === 'ARCHIVE' || SourceHeaders['x-cos-storage-class'] === 'DEEP_ARCHIVE') {
- var restoreHeader = SourceHeaders['x-cos-restore'];
- if (!restoreHeader || restoreHeader === 'ongoing-request="true"') {
- callback({ error: 'Unrestored archive object is not allowed to be copied' });
- return;
- }
- }
- /**
- * 去除一些无用的头部,规避 multipartInit 出错
- * 这些头部通常是在 putObjectCopy 时才使用
- */
- delete TargetHeader['x-cos-copy-source'];
- delete TargetHeader['x-cos-metadata-directive'];
- delete TargetHeader['x-cos-copy-source-If-Modified-Since'];
- delete TargetHeader['x-cos-copy-source-If-Unmodified-Since'];
- delete TargetHeader['x-cos-copy-source-If-Match'];
- delete TargetHeader['x-cos-copy-source-If-None-Match'];
- self.multipartInit({
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- Headers: TargetHeader,
- },function (err,data) {
- if (err) {
- return callback(err);
- }
- params.UploadId = data.UploadId;
- ep.emit('get_copy_data_finish', params);
- });
- });
- // 获取远端复制源文件的大小
- self.headObject({
- Bucket: SourceBucket,
- Region: SourceRegion,
- Key: SourceKey,
- },function(err, data) {
- if (err) {
- if (err.statusCode && err.statusCode === 404) {
- callback({ErrorStatus: SourceKey + ' Not Exist'});
- } else {
- callback(err);
- }
- return;
- }
- FileSize = params.FileSize = data.headers['content-length'];
- if (FileSize === undefined || !FileSize) {
- callback({error: 'get Content-Length error, please add "Content-Length" to CORS ExposeHeader setting.'});
- return;
- }
- onProgress = util.throttleOnProgress.call(self, FileSize, params.onProgress);
- // 开始上传
- if (FileSize <= CopySliceSize) {
- if (!params.Headers['x-cos-metadata-directive']) {
- params.Headers['x-cos-metadata-directive'] = 'Copy';
- }
- self.putObjectCopy(params, function (err, data) {
- if (err) {
- onProgress(null, true);
- return callback(err);
- }
- onProgress({loaded: FileSize, total: FileSize}, true);
- callback(err, data);
- });
- } else {
- var resHeaders = data.headers;
- var SourceHeaders = {
- 'Cache-Control': resHeaders['cache-control'],
- 'Content-Disposition': resHeaders['content-disposition'],
- 'Content-Encoding': resHeaders['content-encoding'],
- 'Content-Type': resHeaders['content-type'],
- 'Expires': resHeaders['expires'],
- 'x-cos-storage-class': resHeaders['x-cos-storage-class'],
- };
- util.each(resHeaders, function (v, k) {
- var metaPrefix = 'x-cos-meta-';
- if (k.indexOf(metaPrefix) === 0 && k.length > metaPrefix.length) {
- SourceHeaders[k] = v;
- }
- });
- ep.emit('get_file_size_finish', SourceHeaders);
- }
- });
- }
- // 复制指定分片
- function copySliceItem(params, callback) {
- var TaskId = params.TaskId;
- var Bucket = params.Bucket;
- var Region = params.Region;
- var Key = params.Key;
- var CopySource = params.CopySource;
- var UploadId = params.UploadId;
- var PartNumber = params.PartNumber * 1;
- var CopySourceRange = params.CopySourceRange;
- var ChunkRetryTimes = this.options.ChunkRetryTimes + 1;
- var self = this;
- Async.retry(ChunkRetryTimes, function (tryCallback) {
- self.uploadPartCopy({
- TaskId: TaskId,
- Bucket: Bucket,
- Region: Region,
- Key: Key,
- CopySource: CopySource,
- UploadId: UploadId,
- PartNumber:PartNumber,
- CopySourceRange:CopySourceRange,
- onProgress:params.onProgress,
- },function (err,data) {
- tryCallback(err || null, data);
- })
- }, function (err, data) {
- return callback(err, data);
- });
- }
- var API_MAP = {
- sliceUploadFile: sliceUploadFile,
- abortUploadTask: abortUploadTask,
- uploadFile: uploadFile,
- uploadFiles: uploadFiles,
- sliceCopyFile: sliceCopyFile,
- };
- module.exports.init = function (COS, task) {
- task.transferToTaskMethod(API_MAP, 'sliceUploadFile');
- util.each(API_MAP, function (fn, apiName) {
- COS.prototype[apiName] = util.apiWrapper(apiName, fn);
- });
- };
|