refactor: 接口增加基础的controller

Author: zhongchubing

geekape-nav-main/components/AppSearch.vue

@@ -26,10 +26,10 @@ export default {
   methods: {
     async querySearchAsync(query, cb) {
       if (query !== '') {
-        const res = await axios.get(API_NAV + `?keyword=${query}`)
-        if (Array.isArray(res.data)) {
-          res.data = res.data.map(item=> (item.value = item.name, item))
-          cb(res.data)
+        const { data } = await axios.get(API_NAV + `?keyword=${query}`)
+        if (Array.isArray(data.data)) {
+          const finalData = data.data.map(item=> (item.value = item.name, item))
+          cb(finalData)
         }
       } else {
         cb([])

geekape-nav-server/app/controller/category.ts

@@ -1,6 +1,10 @@
-import Controller from './common';
+import Controller from '../core/base_controller';
 
 export default class CategoryController extends Controller {
+  tableName(): string {
+    return 'Category'
+  }
+
   async list() {
     const { ctx } = this
     const { showInMenu = true } = ctx.query
@@ -17,17 +21,13 @@ export default class CategoryController extends Controller {
       this.error(error.message)
     }
   }
+
   async add() {
-    const { ctx } = this
-    let data = await ctx.service.common.add(ctx.request.body, 'Category');
-    this.success(data)
+    await super.add()
   }
 
   async edit() {
-    const { ctx } = this
-    const { id } = ctx.request.body;
-    let data = await ctx.service.common.update(ctx.request.body, 'Category');
-    this.success(data)
+    await super.update()
   }
 
   async del() {

geekape-nav-server/app/controller/common.ts

@@ -1,4 +1,4 @@
-import Controller from './common';
+import Controller from '../core/base_controller';
 const request = require('request');
 const cheerio = require('cheerio');
 
@@ -9,63 +9,45 @@ enum NAV_STATUS {
 }
 
 export default class NavController extends Controller {
+  tableName(): string {
+    return 'Nav'
+  }
+
   async list() {
     const { ctx } = this
     const { model } = ctx
-    try {
-      let { pageSize = 10, pageNumber = 1, status = 0, categoryId, name } = ctx.query
-      pageSize = Number(pageSize)
-      pageNumber = Number(pageNumber)
-      const skipNumber = pageSize * pageNumber - pageSize
+    let { status = 0, categoryId, name } = ctx.query
 
-      let findParam: any = {
-        status,
-      }
-      if (!status) {
-        findParam = {
-          $or: [
-            { status: { $exists: false } },
-            { status: 0 },
-          ]
-        }
+    let findParam: any = {
+      status,
+    }
+    if (!status) {
+      findParam = {
+        $or: [
+          { status: { $exists: false } },
+          { status: 0 },
+        ]
       }
-      if (categoryId) {
-        findParam.categoryId = {
-          $eq: categoryId
-        }
+    }
+    if (categoryId) {
+      findParam.categoryId = {
+        $eq: categoryId
       }
-      if (name) {
-        let reg = new RegExp(name,'i');
-        findParam.name = {
-          $regex: reg
-        }
+    }
+    if (name) {
+      let reg = new RegExp(name,'i');
+      findParam.name = {
+        $regex: reg
       }
-
-      const [resData, total] = await Promise.all([
-        model.Nav.find(findParam).skip(skipNumber).limit(pageSize).sort({_id: -1}),
-        model.Nav.find(findParam).count(),
-      ])
-      this.success({
-        data: resData,
-        total,
-        pageNumber: Math.ceil(total / pageSize),
-      })
-    } catch (error) {
-      this.error(error.message)
     }
+
+    await super.getList(findParam);
   }
 
   async add() {
-    const { ctx } = this
-    const { request: req, model } = ctx
-    try {
-      req.body.status = NAV_STATUS.wait
-      req.body.createTime = new Date()
-      const res = await model.Nav.create(req.body);
-      this.success(res)
-    } catch (error) {
-      this.error(error.message)
-    }
+    this.ctx.request.body.status = NAV_STATUS.wait
+    this.ctx.request.body.createTime = new Date()
+    await super.add()
   }
 
   async reptile() {
@@ -94,23 +76,17 @@ export default class NavController extends Controller {
   }
 
   async del() {
-    const { id } = this.ctx.request.body
-    const res = await this.ctx.service.common.remove(id, 'Nav');
-    this.success(res)
+    await super.remove();
   }
 
   async edit() {
-    const { body } = this.ctx.request
-    body.updateTime = new Date()
-    const res = this.ctx.service.common.update(body, 'Nav')
-    this.success(res);
+    this.ctx.request.body.updateTime = new Date()
+    await super.update();
   }
 
   async audit() {
-    const { body } = this.ctx.request
-    body.auditTime = new Date()
-    const res = this.ctx.service.common.update(body, 'Nav')
-    this.success(res);
+    this.ctx.request.body.auditTime = new Date()
+    await super.update();
   }
 
   /**
@@ -150,21 +126,21 @@ export default class NavController extends Controller {
 
   async get() {
     const { ctx } = this
-    const { id, keyword, limit = 10 } = ctx.query
+    const { id, keyword } = ctx.query
 
     let res
 
     if (id) {
-      res = await ctx.service.common.get(ctx.query.id, 'Nav');
+      res = await super.get();
     } else if(keyword) {
-      res = await ctx.service.nav.find(keyword, limit);
+      let reg = new RegExp(keyword,'i');
+      await super.getList({
+        name: { $regex: reg },
+      }, (table)=> table.limit(10));
     }
-    this.success(res)
   }
 
   async random() {
-    const { ctx } = this
-    const res = await ctx.service.common.getRandomData(10, 'Nav');
-    this.success(res)
+    await super.getRandomList();
   }
 }

geekape-nav-server/app/controller/nav.ts

@@ -0,0 +1,7 @@
+import Controller from '../core/base_controller';
+
+export default class CategoryController extends Controller {
+    tableName(): string {
+        return 'Tag';
+    }
+}

geekape-nav-server/app/controller/tag.ts

@@ -1,4 +1,4 @@
-import Controller from './common'
+import Controller from '../core/base_controller'
 
 export default class UserController extends Controller {
   public async login() {

geekape-nav-server/app/controller/user.ts

@@ -0,0 +1,111 @@
+import { Controller } from 'egg';
+
+export default class CommonController extends Controller {
+  tableName() {
+    return ''
+  }
+  success(data) {
+    this.ctx.body = {
+      code: 1,
+      msg: 'ok',
+      data,
+    };
+  }
+  error(msg) {
+    this.ctx.body = {
+      code: 0,
+      msg,
+      data: null,
+    };
+  }
+
+  //添加
+  async add() {
+    const { request } = this.ctx
+    const tableName = this.tableName()
+    try {
+      const res = await this.ctx.model[tableName].create(request.body);
+      this.success(res)
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+
+  //删除
+  async remove() {
+    const { request } = this.ctx
+    const tableName = this.tableName()
+    try {
+      const id = request.body.id
+      const res = await this.ctx.model[tableName].remove({ _id: id })
+      this.success(res)
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+
+  //更新
+  async update() {
+    const { request } = this.ctx
+    const tableName = this.tableName()
+    try {
+      const id = request.body.id
+      delete request.body.id
+      const res = await this.ctx.model[tableName].update({ _id: id }, request.body)
+      this.success(res)
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+
+  //查找一个
+  async get() {
+    const { request, query } = this.ctx
+    const tableName = this.tableName()
+    try {
+      const id = query.id
+      const res = await this.ctx.model[tableName].findOne({ _id: id })
+      this.success(res)
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+
+  //查找多个
+  async getList(findObj = {}, otherCMD = (_table: any) => _table) {
+    const { request, query } = this.ctx
+    const tableName = this.tableName()
+    try {
+      let { pageSize = 10, pageNumber = 1 } = query
+      pageSize = Number(pageSize)
+      pageNumber = Number(pageNumber)
+      const skipNumber = pageSize * pageNumber - pageSize
+      const table = this.ctx.model[tableName]
+
+      const [data, total] = await Promise.all([
+        otherCMD(table.find(findObj).skip(skipNumber).limit(pageSize).sort({_id: -1})),
+        table.find(findObj).count()
+      ])
+
+      this.success({
+        data,
+        total,
+        pageNumber: Math.ceil(total / pageSize),
+      })
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+
+  //查找随机数量列表
+  async getRandomList(randomNumber: number = 10) {
+    const { request, query } = this.ctx
+    const tableName = this.tableName()
+    try {
+      const res = await this.ctx.model[tableName].aggregate( [ { $sample: { size: randomNumber } } ] )
+      this.success(res)
+    } catch (e) {
+      this.error(e.message)
+    }
+  }
+}

geekape-nav-server/app/core/base_controller.ts

@@ -0,0 +1,10 @@
+module.exports = app => {
+  const mongoose = app.mongoose;
+  const Schema = mongoose.Schema;
+
+  const TagSchema = new Schema({
+    name: String,
+    category: String,
+  }, { collection: 'tag' });
+  return mongoose.model('Tag', TagSchema);
+};