fork from bc4552c5a8

exts/yapi-plugin-advanced-mock/advMockModel.js

@@ -0,0 +1,61 @@
+const yapi = require('yapi.js');
+const baseModel = require('models/base.js');
+
+class advMockModel extends baseModel {
+  getName() {
+    return 'adv_mock';
+  }
+
+  getSchema() {
+    return {
+      interface_id: { type: Number, required: true },
+      project_id: {type: Number, required: true},
+      enable: {type: Boolean, default: false}, 
+      mock_script: String,
+      uid: String,
+      up_time: Number
+    };
+  }
+
+  get(interface_id) {
+
+    return this.model.findOne({
+      interface_id: interface_id
+    });
+  }
+
+  delByInterfaceId(interface_id) {
+    return this.model.remove({
+      interface_id: interface_id
+    });
+  }
+
+  delByProjectId(project_id){
+    return this.model.remove({
+      project_id: project_id
+    })
+  }
+
+  save(data) {
+    data.up_time = yapi.commons.time();
+    let m = new this.model(data);
+    return m.save();
+  }
+
+  up(data) {
+    data.up_time = yapi.commons.time();
+    return this.model.update({
+      interface_id: data.interface_id
+    }, {
+        uid: data.uid,
+        up_time: data.up_time,
+        mock_script: data.mock_script,
+        enable: data.enable
+      }, {
+        upsert: true
+      })
+  }
+
+}
+
+module.exports = advMockModel;