feat:裁剪

src/modules/fund/fund.model.js

@@ -0,0 +1,68 @@
+'use strict';
+
+const mongoose = require('mongoose');
+const ASSETTYPES = ['gin', 'vic', 'usdtBTC', 'usdtETH', 'usdtTRX'];
+
+const FundSchema = new mongoose.Schema({
+    userId: {
+        type: String
+    },
+    tokenAddress: {
+        type: {},
+        default: {
+            'usdtBTC': '',
+            'usdtETH': '',
+            'usdtTRX': ''
+        }
+    },
+    // fund 为充值矿晶的一系列数值，用于计算收益
+    fund: [{
+        type: Number
+    }],
+    groupFundSum: {
+        type: Number
+    },
+    asset: {
+        type: {}
+    },
+    option: {
+        type: {}
+    },
+}, {
+    timestamps: {
+        createdAt: 'createdAt', 
+        updatedAt: 'updatedAt'
+    }
+});
+
+
+const FundModel = mongoose.model('Fund', FundSchema);
+
+FundModel.prototype.increase = async function (assetName, amount, option = {}) {
+    if (!assetName || !amount || amount <= 0 || !Number.isFinite(amount) || !ASSETTYPES.includes(assetName)) return null;
+    const asset = 'asset.' + assetName;
+    const res = await FundModel.findByIdAndUpdate(this.id, {
+        $inc: {
+            [asset]: amount
+        }
+    }, { ...option, returnOriginal: false }).exec().catch(err => null);
+    if (!res || !res.asset) throw new Error('用户资产修改[increase]失败！');
+    if (res.asset[assetName] < 0) throw new Error('Insufficient funds');
+    return res;
+};
+FundModel.prototype.decrease = async function (assetName, amount, option) {
+    // 传入的amount应该是大于0的正数
+    if (!assetName || !amount || amount <= 0 || !Number.isFinite(amount) || !ASSETTYPES.includes(assetName)) return null;
+    amount = 0 - amount;
+    const asset = 'asset.' + assetName;
+    const res = await FundModel.findByIdAndUpdate(this.id, {
+        $inc: {
+            [asset]: amount
+        }
+    }, { ...option, returnOriginal: false }).exec().catch(err => null);
+    if (!res || !res.asset) throw new Error('用户资产修改[decrease]失败！');
+    if (res.asset[assetName] < 0) throw new Error('Insufficient funds');
+    return res;
+};
+
+module.exports = FundModel;