save

v2.0/compileServer/config.json

@@ -18,6 +18,13 @@
                             "params": [
                                 "session"
                             ]
+                        },
+                        {
+                            "type": "custom",
+                            "name": "test",
+                            "params": [
+                                "tt"
+                            ]
                         }
                     ],
                     "main": [

v2.0/compileServer/generate.js

@@ -17,7 +17,55 @@ const controllerImport = (controller = {}) => {
   return result
 }
 
+const customFileMap = (controllers = {}) => {
+  const result = {
+    decorators: {},
+    utils: {},
+    services: {}
+  }
+  for (let controllerType in controllers) {
+    for (let controller in controllers[controllerType]) {
+      for (let fn in controllers[controllerType][controller]) {
+        for (let decorator of controllers[controllerType][controller][fn].decorators) {
+          if (decorator.type === 'custom') {
+            result.decorators[decorator.name] = true
+          }
+        }
+      }
+    }
+  }
+  return result
+}
+
 const run = async (gulp, nunjucksRender, rename, nunjucksRenderConfig) => {
+  //编译自定义文件
+  const { decorators } = customFileMap(config.controllers)
+  for (let decorator in decorators) {
+    await gulp
+      .src(`${serverTemplatePath}/decorator.njk`)
+      .pipe(
+        nunjucksRender({
+          ...nunjucksRenderConfig,
+          data: {
+            name: decorator
+          }
+        })
+      )
+      .pipe(rename(`${decorator}.ts`))
+      .pipe(gulp.dest(nunjucksRenderConfig.ServerFullPath + '/decorator/custom'))
+  }
+  await gulp
+    .src(`${serverTemplatePath}/decorator.export.njk`)
+    .pipe(
+      nunjucksRender({
+        ...nunjucksRenderConfig,
+        data: {
+          decorators
+        }
+      })
+    )
+    .pipe(rename(`index.ts`))
+    .pipe(gulp.dest(nunjucksRenderConfig.ServerFullPath + '/decorator/custom'))
   //编译controller.common
   for (let commonController in config.controllers.common) {
     await gulp

v2.0/compileServer/templates/controller.njk

@@ -4,6 +4,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 {{-macro.genControllerService(service)-}}
 

v2.0/compileServer/templates/decorator.export.njk

+/** @format */
+
+{%for decorator, decorator_item in decorators-%}
+import {{decorator}} from './{{decorator}}'
+{%-endfor%}
+
+const custom = {
+{%-for decorator, decorator_item in decorators%}
+  {{decorator}}{%if loop.last%}{%else%},{%endif%}
+{%-endfor%}
+}
+
+export default custom

v2.0/compileServer/templates/decorator.njk

+/** @format */
+import { resultsModel } from '../../sdk'
+import { CODE_TYPES } from '../../constants'
+
+export default function {{name}}(params) {
+  return function (target: Object, name: string, descriptor: PropertyDescriptor) {
+    const method = descriptor.value
+
+    descriptor.value = function (...args: any[]) {
+      const [context, otherArgs = {}] = args
+      return method.apply(target, [context, { ...otherArgs }])
+    }
+  }
+}

v2.0/src/controller/common/stat.controller.ts

@@ -2,6 +2,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 import { StatService } from '../../service'
 export default class StatController {

v2.0/src/controller/common/task.controller.ts

@@ -2,6 +2,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 import { TaskService, BaseService } from '../../service'
 export default class TaskController {

v2.0/src/controller/common/user.controller.ts

@@ -2,6 +2,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 import { UserService } from '../../service'
 export default class UserController {
@@ -10,6 +11,7 @@ export default class UserController {
    */
   @commonDecorator.checkParams(['activityId'])
   @commonDecorator.registeInfos(['session'])
+  @customDecorator.test(['tt'])
   @services([UserService])
   async getVipInfo(
     context: IContext<IParams>,

v2.0/src/controller/custom/test1.controller.ts

@@ -2,6 +2,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 import { StatService } from '../../service'
 export default class Test1Controller {

v2.0/src/controller/custom/test2.controller.ts

@@ -2,6 +2,7 @@
 import commonDecorator, { services } from '../../decorator/common'
 import commonUtilCheck from '../../utils/common/check'
 import commonUtilUpdate from '../../utils/common/update'
+import customDecorator from '../../decorator/custom'
 import { resultsModel } from '../../sdk'
 import { StatService } from '../../service'
 export default class Test2Controller {

v2.0/src/decorator/custom/index.ts

+/** @format */
+
+import test from './test'
+
+const custom = {
+  test
+}
+
+export default custom

v2.0/src/decorator/custom/test.ts

+/** @format */
+import { resultsModel } from '../../sdk'
+import { CODE_TYPES } from '../../constants'
+
+export default function test(params) {
+  return function (target: Object, name: string, descriptor: PropertyDescriptor) {
+    const method = descriptor.value
+
+    descriptor.value = function (...args: any[]) {
+      const [context, otherArgs = {}] = args
+      return method.apply(target, [context, { ...otherArgs }])
+    }
+  }
+}