为 Model 基类定义一个静态块 defineAttributes

Author: yetrun

lib/defines/buildModelBase.js

@@ -21,6 +21,11 @@ function buildModelBase ({
         value: Model
       })
 
+      if (!this.constructor.definedAttributes) {
+        // 如果尚未定义 definedAttributes，调用模型类的静态方法调用
+        this.constructor.defineAttributes && this.constructor.defineAttributes()
+      }
+
       if (defineAttributesIn === 'object') {
         defineAttributes(this, attributesDefinition || this.constructor.definedAttributes)
         // 计算属性还是考虑定义在原型
@@ -94,6 +99,13 @@ function buildModelBase ({
         deleter.call(this, key)
       }
     }
+
+    static defineAttribute (name, options) {
+      defineAttributes(this, { [name]: options })
+
+      this.definedAttributes = this.definedAttributes || {}
+      Object.assign(this.definedAttributes, { [name]: options })
+    }
   }
 
   Model.defaultConfig = {

test/inherit-style/modelDefineAttributes.test.js

@@ -0,0 +1,22 @@
+const test = require('ava')
+const Model = require('../../lib/model')
+
+class User extends Model {
+  static defineAttributes () {
+    this.defineAttribute('name', {})
+    this.defineAttribute('age', {})
+  }
+}
+
+test('测试 defineAttributes 成功设置了属性定义', t => {
+  const user = new User()
+  user.attributes = {
+    name: 'Jim',
+    age: 18,
+    foo: 'foo'
+  }
+
+  t.is(user.name, 'Jim')
+  t.is(user.age, 18)
+  t.is(user.foo, undefined)
+})