|
1 | 1 | import qiniu from 'qiniu'
|
| 2 | +import * as url from 'url' |
2 | 3 |
|
3 |
| -import { |
4 |
| - BadRequestException, |
5 |
| - Injectable, |
6 |
| - InternalServerErrorException, |
7 |
| -} from '@nestjs/common' |
| 4 | +import { Injectable, InternalServerErrorException } from '@nestjs/common' |
8 | 5 |
|
9 | 6 | import { QINIU_SECRET } from '~/app.config'
|
10 | 7 |
|
11 |
| -const mac = new qiniu.auth.digest.Mac(QINIU_SECRET.qn_ak, QINIU_SECRET.qn_sk) |
12 |
| -const putPolicy = new qiniu.rs.PutPolicy({ |
13 |
| - scope: QINIU_SECRET.qn_scope, |
14 |
| -}) |
15 |
| -const uploadToken = putPolicy.uploadToken(mac) |
16 |
| - |
17 |
| -const config = new qiniu.conf.Config() |
18 |
| -const bucketManager = new qiniu.rs.BucketManager(mac, config) |
19 |
| - |
20 | 8 | @Injectable()
|
21 | 9 | export class UploadService {
|
22 |
| - upload(file: Express.Multer.File): Promise<string> { |
23 |
| - if (!file) { |
24 |
| - throw new BadRequestException('请传入图片') |
25 |
| - } |
26 |
| - const filename = `${Date.now()}-${file.originalname}` |
| 10 | + async uploadPhoto(file: Express.Multer.File) { |
| 11 | + const mac = new qiniu.auth.digest.Mac( |
| 12 | + QINIU_SECRET.qn_ak, |
| 13 | + QINIU_SECRET.qn_sk, |
| 14 | + ) |
| 15 | + const putPolicy = new qiniu.rs.PutPolicy({ |
| 16 | + scope: QINIU_SECRET.qn_scope, |
| 17 | + }) |
| 18 | + const uploadToken = putPolicy.uploadToken(mac) |
| 19 | + |
| 20 | + // upload |
27 | 21 | const formUploader = new qiniu.form_up.FormUploader(
|
28 | 22 | new qiniu.conf.Config({
|
29 | 23 | zone: qiniu.zone.Zone_z2,
|
30 | 24 | }),
|
31 | 25 | )
|
32 |
| - const putExtra = new qiniu.form_up.PutExtra() |
33 |
| - |
34 |
| - return new Promise((resolve) => { |
| 26 | + const img = (await new Promise((_res, _rej) => { |
35 | 27 | formUploader.put(
|
36 | 28 | uploadToken,
|
37 |
| - filename, |
| 29 | + `${Date.now()}-${file.originalname}`, |
38 | 30 | file.buffer,
|
39 |
| - putExtra, |
| 31 | + new qiniu.form_up.PutExtra(), |
40 | 32 | (respErr, respBody, respInfo) => {
|
41 | 33 | if (respErr) {
|
| 34 | + console.error(respErr) |
42 | 35 | throw new InternalServerErrorException(respErr.message)
|
43 | 36 | }
|
| 37 | + |
44 | 38 | if (respInfo.statusCode == 200) {
|
45 |
| - resolve(`${QINIU_SECRET.qn_host}/${respBody.key}`) |
| 39 | + _res({ |
| 40 | + url: new url.URL(respBody.key, QINIU_SECRET.qn_host).href, |
| 41 | + }) |
46 | 42 | } else {
|
47 |
| - throw new InternalServerErrorException(respErr.message) |
| 43 | + console.error(respInfo.statusCode, respBody) |
| 44 | + throw new InternalServerErrorException(respInfo) |
48 | 45 | }
|
49 | 46 | },
|
50 | 47 | )
|
51 |
| - }) |
52 |
| - } |
| 48 | + })) as { url: string } |
53 | 49 |
|
54 |
| - remove(fileName: string) { |
55 |
| - return new Promise((resolve) => { |
56 |
| - bucketManager.delete( |
57 |
| - QINIU_SECRET.qn_scope, |
58 |
| - fileName, |
59 |
| - (err, respBody, respInfo) => { |
60 |
| - if (err) { |
61 |
| - throw new InternalServerErrorException(err.message) |
62 |
| - } else { |
63 |
| - resolve('删除成功') |
64 |
| - } |
65 |
| - }, |
66 |
| - ) |
67 |
| - }) |
| 50 | + return img.url |
68 | 51 | }
|
69 | 52 | }
|
0 commit comments