@smart100/spu-web-plugin 0.0.1

package/src/oss/multiUpload.ts

@@ -0,0 +1,129 @@
+interface IMultiUploadOptions {
+    bucket: string
+    key: string
+    partSize?: number
+    parallel: number
+    onProgress?: (number: number) => void
+}
+interface IUploadPartRes {
+    PartNumber: number
+    ETag: string
+}
+
+interface ITask<T> {
+    (): Promise<T>
+}
+
+export const obsMultiUpload = async (obs: any, file: File | Blob, options: IMultiUploadOptions) => {
+  const partSize = options.partSize || 1024 * 1024 // 1M
+  const parallel = options.parallel || 3
+  const initialRes = await obs.initiateMultipartUpload({
+    Bucket: options.bucket,
+    Key: options.key
+  })
+  if (initialRes.CommonMsg.Status < 300) {
+    const { UploadId } = initialRes.InterfaceResult
+    const count = Math.ceil(file.size / partSize)
+    // const dones: Array<IUploadPartRes> = []
+    const tasks: Array<ITask<IUploadPartRes>> = []
+    const transferredArr = new Array(count).fill(0)
+    for (let i = 1; i <= count; i++) {
+      const uploadPart = async () => {
+        const partRes = await obs.uploadPart({
+          Bucket: options.bucket,
+          Key: options.key,
+          // 设置分段号，范围是1~10000
+          PartNumber: i,
+          // 设置Upload ID
+          UploadId: UploadId,
+          // 设置将要上传的大文件
+          SourceFile: file, // 设置分段大小
+          PartSize: partSize,
+          // 设置分段的起始偏移大小
+          Offset: partSize * (i - 1),
+          ProgressCallback: (transferredAmount: number, totalAmount: number) => {
+            // TODO 不能直接加，transferredAmount有重复的
+            transferredArr[i - 1] = transferredAmount
+            const transferred = transferredArr.reduce((item, prev) => {
+              return item + prev
+            }, 0)
+            let percent = transferred / file.size
+            if (percent === 1) {
+              percent = 0.9999999
+            }
+            options.onProgress && options.onProgress(percent)
+          }
+        })
+        if (partRes.CommonMsg.Status < 300) {
+          // dones.push({
+          //     part: i,
+          //     etag: partRes.InterfaceResult.ETag
+          // })
+          return {
+            PartNumber: i,
+            ETag: partRes.InterfaceResult.ETag
+          }
+        } else {
+          throw partRes.CommonMsg.Message
+        }
+      }
+      tasks.push(uploadPart)
+    }
+    const partsRes = await executeTasks(tasks, parallel)
+    // complete
+    const completeRes = await obs.completeMultipartUpload({
+      Bucket: options.bucket,
+      Key: options.key,
+      UploadId: UploadId,
+      Parts: partsRes
+    })
+    if (completeRes.CommonMsg.Status < 300) {
+      // 进度条
+      options.onProgress && options.onProgress(1)
+      return true
+    } else {
+      throw completeRes.CommonMsg.Message
+    }
+  } else {
+    console.error(initialRes)
+    throw initialRes.CommonMsg.Message
+  }
+}
+
+const executeTasks = async <T>(tasks: Array<ITask<T>>, parallel = 3): Promise<Array<T>> => {
+  return new Promise((resolve, reject) => {
+    const todos = tasks.slice(0, parallel)
+    const pendings = tasks.slice(parallel)
+    let success = 0
+    const res: Array<T> = []
+    const dealPending = (todoRes: any, index?: number): Promise<any> | undefined => {
+      success = success + 1
+      if (index !== undefined && index !== null) {
+        res[index] = todoRes
+      }
+      if (success === tasks.length) {
+        resolve(res)
+      }
+      if (!pendings.length) return
+      const todo = pendings.shift()
+      const todoIndex = tasks.findIndex((task) => {
+        return task === todo
+      })
+      return todo && todo()
+        .then((todoRes) => dealPending(todoRes, todoIndex))
+        .catch(reject)
+    }
+    if (!pendings.length) {
+      const running = todos.map((todo) => {
+        return todo()
+      })
+      return Promise.all(running).then(resolve).catch(reject)
+    }
+    for (let i = 0; i < todos.length; i++) {
+      const todo = todos[i]
+      todo()
+        .then((todoRes) => dealPending(todoRes, i))
+        .catch(reject)
+    }
+  })
+}

package/src/oss/servtoken.ts

@@ -0,0 +1,131 @@
+// import { noBaseRequest } from '@/service/http'
+import { apaasAxios } from '../axios'
+
+export type ServToken = {
+  accesskeyid: string
+  accesskeysecret: string
+  expiration: string
+  securitytoken: string
+}
+
+export type Response<T> = {
+  // eslint-disable-next-line camelcase
+  resp_data: T
+}
+
+// 一个小时的超时能力 1000 * 60 * 60
+let last = 0
+let servtoken: ServToken | null = null
+let isGetting = false
+let xhr: Promise<ServToken> | null = null
+
+const getServToken = async () => {
+  try {
+    const response: ServToken = await apaasAxios.get('/api/teapi/auth/servtoken', {}).then((res: any) => res.data)
+    // debugger
+    return response
+  } catch (e) {
+    throw new Error((e as any).errorMsg || 'get servtoken error')
+  }
+}
+
+export const initServToken = () => {
+  return new Promise<ServToken>((resolve, reject) => {
+    // 在请求中,排队列
+    if (isGetting && xhr) {
+      xhr
+        .then((res) => {
+          if (res) {
+            servtoken = res
+            last = new Date(res.expiration).getTime()
+            isGetting = false
+            xhr = null
+            resolve(servtoken)
+          } else {
+            // eslint-disable-next-line prefer-promise-reject-errors
+            reject(null)
+          }
+          // return res
+        })
+        .catch(() => {
+          xhr = null
+          isGetting = false
+          // return null
+          // eslint-disable-next-line prefer-promise-reject-errors
+          reject(null)
+        })
+    } else {
+      if (!last || last - Date.now() < 1000 * 60 * 5) {
+        // 过期了 重新请求
+        isGetting = true
+        xhr = getServToken()
+        xhr
+          .then((res) => {
+            servtoken = res
+            last = new Date(res.expiration).getTime()
+            isGetting = false
+            xhr = null
+            // return res
+            resolve(servtoken)
+          })
+          .catch(() => {
+            xhr = null
+            isGetting = false
+            // return null
+            // eslint-disable-next-line prefer-promise-reject-errors
+            reject(null)
+          })
+      } else {
+        // 没过期
+        resolve(servtoken as ServToken)
+      }
+    }
+  })
+}
+
+
+
+
+
+
+
+
+
+// // let last = 0
+// // let servtoken
+// let isRequesting = false
+// const getServToken = function () {
+//   let servtoken = wx.getStorageSync('servtoken')
+//   let last
+//   if (servtoken) {
+//     last = new Date(servtoken.expiration).getTime()
+//   }
+//   const now = Date.now()
+//   if (!servtoken || (last - now < (1000 * 60 * 5))) {
+//     wx.removeStorageSync('servtoken')
+//     if (!isRequesting) {
+//       isRequesting = true
+//       return request({
+//         url: `${api.HOST}/api/teapi/auth/servtoken`,
+//         method: 'get',
+//         isShowLoading: false
+//       }).then((res) => {
+//         wx.setStorageSync('servtoken', res.data.resp_data)
+//         isRequesting = false
+//         return res.data.resp_data
+//       })
+//     } else {
+//       return new Promise((resolve, reject) => {
+//         const fn = setInterval(() => {
+//           let servtoken = wx.getStorageSync('servtoken')
+//           if (servtoken) {
+//             resolve(servtoken)
+//             clearInterval(fn)
+//           }
+//         }, 200)
+//       })
+//     }
+//   } else {
+//     return Promise.resolve(servtoken)
+//   }
+// }

package/src/oss/uploadService.ts

@@ -0,0 +1,145 @@
+import CloudServ from '../cloudServ'
+import { initServToken } from './servtoken'
+import { v4 as uuidv4 } from 'uuid'
+import OSS from '../package/ali-oss/aliyun-oss-sdk.apaas.min.js'
+// const OSS = require('../package/ali-oss/aliyun-oss-sdk.apaas.min.js')
+import ObsClient from '../package/huaweicloud-obs/esdk-obs-browserjs.3.22.3.min.js'
+// const ObsClient = require('../package/huaweicloud-obs/esdk-obs-browserjs.3.22.3.min.js')
+import dayjs from 'dayjs'
+import co from 'co'
+import login from '../login'
+import { obsMultiUpload } from './multiUpload'
+
+
+
+interface IUpload {
+  type?: 'att' | 'img',
+  file: File,
+  source?: string,
+  datetime?: string | number,
+  storagetype?: StorageType,
+  onprogress?: (p: number, _checkpoint?: IAny) => void
+}
+
+const upload = async ({
+  type = 'img',
+  file,
+  source = '',
+  datetime = '',
+  storagetype = 'storage',
+  onprogress,
+}: IUpload) => {
+  if (!file) throw Error('请传入文件')
+  const cloudServStorage = CloudServ.get(storagetype)
+  if (!cloudServStorage) throw Error('无可用存储设置')
+  const servToken = await initServToken()
+  if (!servToken) throw Error('无可用servToken')
+
+  const isAliYun = CloudServ.isAliyun(storagetype)
+  const isHuawei = CloudServ.isHuawei(storagetype)
+  const tenantCode = login.getUser('tenantcode')
+  const suffix = '.' + file.name.substring(file.name.lastIndexOf('.') + 1)
+  source = source ? source : (uuidv4() + suffix)
+  datetime = datetime ? datetime : Date.now()
+  const date = dayjs(+datetime).format('YYYYMMDD')
+  const objectKey = `${source.slice(0, 3)}/${type}/${date}/${tenantCode}/${source}`
+
+  // console.log(file)
+  // console.log(source)
+  // console.log(datetime)
+  const promise = new Promise(async (resolve, reject) => {
+    if (isAliYun) {
+      const ossClient = new OSS({
+        // region: cloudServ.cloudserv_storage_storageendpoint,
+        endpoint: cloudServStorage.cloudserv_storage_storageendpoint,
+        accessKeyId: servToken.accesskeyid,
+        accessKeySecret: servToken.accesskeysecret,
+        stsToken: servToken.securitytoken,
+        bucket: cloudServStorage.cloudserv_storage_storagebucket,
+        secure: true
+      })
+      // promise.cancel = () => {
+      //   ossClient.cancel()
+      //   reject(new Error('Cancel'))
+      // }
+      co(function* () {
+        yield ossClient.multipartUpload(objectKey, file, {
+          headers: {
+            'Content-Disposition': 'filename="' + (file.name && encodeURIComponent(file.name)) // 阿里云提供的下载名字
+          },
+          progress: function* (p: number, _checkpoint: IAny): any {
+            // console.log('httpUploadProgress', p)
+            onprogress && onprogress(p, _checkpoint)
+            if (p === 1) {
+              resolve({
+                source,
+                filename: file.name,
+                type: file.type,
+                date: date,
+                datetime: datetime,
+                file
+              })
+            }
+          }
+        })
+      }).catch((error: any) => {
+        console.error(error)
+        reject(error)
+      })
+    } else if (isHuawei) {
+        const obs = new ObsClient({
+          access_key_id: servToken.accesskeyid,
+          secret_access_key: servToken.accesskeysecret,
+          server: cloudServStorage.cloudserv_storage_storageendpoint,
+          security_token: servToken.securitytoken
+        })
+        // const obs = new ObsClient({
+        //   access_key_id: '75ISL4GWAOOO1USWUMRG',
+        //   secret_access_key: 'srn6eJ1BpFbjxoFrJgiQjeS65Z3eKC3rnqeyBBlL',
+        //   server: cloudServ.cloudserv_storage_storageendpoint,
+        //   security_token: 'gQ5jbi1zb3V0aHdlc3QtMogAsKjYcI5CQldRAvQQJysdkZ7tAt6arLsoWHFGCaco8s8FPk6wtbslHWhvg2SZh3QMM4aUA4FhPEWQt9A7gXoC_Lh4DpF6hhHIxUyACKgakNHNPdPegy5G9-sibBXkGueIY1X3K12tzpjbyd08esLKEEu-M_QmfDoDdkgOcyidITc-lOg5EzXb27f91Ym26u2mAMTaNjCLRulJ4rziLSW6IAprSx8LUkuQQE-wUz-tMzVL9oFiVykHz980o0Y0CNdCIwn98Y-xbMdslZ3U8raydy6Wnf2LchXc0ajvMix0gg-CV0tpA4cgiZFqPxwEDXSv42hvfccboWlpGmOVR3llHLUirrlgFod8rhm-Rmk6MIfQw4NA8rddow1Gx6g-MugFV5arMDLfsOhqeSFQRJWizb3q50zk6GcUFulewitxP8HSkXMGt_rDDYCcCEmdu15D3imX5431Mbdt0qCgxH80OPCDmFXw0xMOsggxE0PBVexVY2x3wHGeql3JNyevUZAhqlskNNu77ui2Vnp-ZbHMcxgDLcPuAULINId4V_QGdhAkDaxQk53AE237DAFXtlyWWaBRMsTNVnpq9mCXJup9pdBbjLRVAO4OxfYVnwwvl-w_mb-xCgOf5EPHqA_zbZF8z-ad6JjWgLOQCHaawE7kNGHIQwAgzneik33wP2jPlG1ak9KEWyXr1n2QMJCmDM3bIrRit5_8LLFoPjXcwurBjZ-AomM_ztOe34sdr357atlQcPD7a2xWtrvn9mnqqndg12m7QxnwkmJsOYK4ZO5Hoyb2vd_NX2Nd9PaSIVAcFw=='
+        // })
+        try {
+          const uploadRes = await obsMultiUpload(obs, file, {
+            bucket: cloudServStorage.cloudserv_storage_storagebucket,
+            key: objectKey,
+            parallel: 3,
+            onProgress: (percent: number) => {
+              onprogress && onprogress(percent)
+            }
+          })
+          if (uploadRes) {
+            resolve({
+              source,
+              filename: file.name,
+              type: file.type,
+              date: date,
+              datetime: datetime,
+              file
+            })
+          }
+        } catch (e) {
+          console.error(e)
+          reject(e)
+        }
+        // obs.putObject({
+        //   Bucket: cloudServ.cloudserv_storage_storagebucket,
+        //   Key: objectKey,
+        //   SourceFile: file
+        // }, (err, result) => {
+        //   result.CommonMsg.Status
+        //   console.error(err)
+        //   console.log(result)
+        //   debugger
+        // })
+    } else {
+      throw Error('暂不支持非阿里云OSS/华为云OBS存储类型')
+    }
+  })
+
+  return promise
+}
+
+export default {
+  upload
+}