增加文件上传功能

Aaron.Liu
1 parent 82b13adb
Showing 5 changed files with 43 additions and 2 deletions Show diff stats
app/controller/file.js
app/router.js
app/service/contract.js
config/config.default.js
package.json
@@ -0,0 +1,28 @@
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const Controller = require('egg').Controller;
+const awaitWriteStream = require('await-stream-ready').write;
+const sendToWormhole = require('stream-wormhole');
+const uuid = require('node-uuid');
+
+class FileController extends Controller {
+
+  async upload() {
+    const stream = await this.ctx.getFileStream();
+    const filename = uuid.v1() + path.extname(stream.filename).toLowerCase();
+    const target = path.join(this.config.baseDir, 'app/public', filename);
+    const writeStream = fs.createWriteStream(target);
+    try {
+      await awaitWriteStream(stream.pipe(writeStream));
+    } catch (err) {
+      await sendToWormhole(stream);
+      throw err;
+    }
+
+    this.ctx.body = { url: '/public/' + filename };
+  }
+}
+
+module.exports = FileController;
@@ -8,6 +8,8 @@ module.exports = app =&gt; {
   router.get('/', controller.home.index);
   // 首页统计
   router.get('/statistic/count', controller.statistic.count);
+  // 文件上传
+  router.post('/file/upload', controller.file.upload);
   // 用户管理
   router.post('/user/password', controller.user.password);
   router.post('/user/validate', controller.user.validate);
@@ -11,7 +11,6 @@ class ContractService extends Service {
       select contract.*, user.name as manager_name from contract, user where contract.manager_id = user.id
       ${where !== '' ? `and ${where}` : ''}
     `;
-    console.log(sql);
     const result = await this.app.mysql.query(sql);
     return result;
   }
@@ -39,5 +39,15 @@ module.exports = appInfo =&gt; {
     credentials: true,
   };
+  config.multipart = {
+    // will append to whilelist
+    fileExtensions: [
+      '.doc',
+      '.docx',
+      '.xls',
+      '.xlsx',
+    ],
+  };
+
   return config;
 };
@@ -9,7 +9,9 @@
     "egg-mysql": "^3.0.0",
     "egg-scripts": "^2.5.0",
     "moment": "^2.22.1",
-    "node-uuid": "^1.4.8"
+    "node-uuid": "^1.4.8",
+    "await-stream-ready": "^1.0.1",
+    "stream-wormhole": "^1.0.3"
   },
   "devDependencies": {
     "autod": "^3.0.1",
@@ -0,0 +1,28 @@		@@ -0,0 +1,28 @@
	1	+'use strict';
	2	+
	3	+const fs = require('fs');
	4	+const path = require('path');
	5	+const Controller = require('egg').Controller;
	6	+const awaitWriteStream = require('await-stream-ready').write;
	7	+const sendToWormhole = require('stream-wormhole');
	8	+const uuid = require('node-uuid');
	9	+
	10	+class FileController extends Controller {
	11	+
	12	+ async upload() {
	13	+ const stream = await this.ctx.getFileStream();
	14	+ const filename = uuid.v1() + path.extname(stream.filename).toLowerCase();
	15	+ const target = path.join(this.config.baseDir, 'app/public', filename);
	16	+ const writeStream = fs.createWriteStream(target);
	17	+ try {
	18	+ await awaitWriteStream(stream.pipe(writeStream));
	19	+ } catch (err) {
	20	+ await sendToWormhole(stream);
	21	+ throw err;
	22	+ }
	23	+
	24	+ this.ctx.body = { url: '/public/' + filename };
	25	+ }
	26	+}
	27	+
	28	+module.exports = FileController;
	@@ -39,5 +39,15 @@ module.exports = appInfo => {		@@ -39,5 +39,15 @@ module.exports = appInfo => {
39	credentials: true,	39	credentials: true,
40	};	40	};
41		41
		42	+ config.multipart = {
		43	+ // will append to whilelist
		44	+ fileExtensions: [
		45	+ '.doc',
		46	+ '.docx',
		47	+ '.xls',
		48	+ '.xlsx',
		49	+ ],
		50	+ };
		51	+
42	return config;	52	return config;
43	};	53	};