72 lines
1.8 KiB
JavaScript
72 lines
1.8 KiB
JavaScript
const dayjs = require("dayjs");
|
||
let db = null;
|
||
exports.main = async (event, mongodb) => {
|
||
db = mongodb;
|
||
switch (event.type) {
|
||
case "upDatetransferTime":
|
||
return await exports.upDatetransferTime(event);
|
||
case "judgeTransferTime":
|
||
return await exports.judgeTransferTime(event);
|
||
}
|
||
};
|
||
// 更新transferTime字段
|
||
exports.upDatetransferTime = async (event) => {
|
||
const { corpId, userId } = event;
|
||
try {
|
||
const transferTime = dayjs().valueOf();
|
||
await db.collection("corp-member").updateOne(
|
||
{ corpId, userid: userId }, // 查询条件
|
||
{ $set: { transferTime } } // 更新内容
|
||
);
|
||
return {
|
||
success: true,
|
||
message: "更新成功",
|
||
};
|
||
} catch (error) {
|
||
return {
|
||
success: false,
|
||
message: error.message,
|
||
};
|
||
}
|
||
};
|
||
|
||
// 判断transferTime是否过期 期限为30天
|
||
exports.judgeTransferTime = async (event) => {
|
||
const { corpId, userId, env } = event;
|
||
try {
|
||
const user = await db.collection("corp-member").findOne(
|
||
{ corpId, userid: userId } // 查询条件
|
||
);
|
||
// 如果没有transferTime字段，返回false
|
||
const { transferTime = "" } = user || {};
|
||
if (!transferTime) {
|
||
return {
|
||
success: true,
|
||
message: "获取成功",
|
||
data: {
|
||
isAfterToday: true,
|
||
remainingTime: 0,
|
||
},
|
||
};
|
||
}
|
||
const isAfterToday = dayjs().isAfter(dayjs(transferTime).add(30, "day"));
|
||
// 获取剩余天数
|
||
const remainingTime = dayjs(transferTime)
|
||
.add(30, "day")
|
||
.diff(dayjs(), "day");
|
||
return {
|
||
success: true,
|
||
message: "获取成功",
|
||
data: {
|
||
isAfterToday,
|
||
remainingTime,
|
||
},
|
||
};
|
||
} catch (error) {
|
||
return {
|
||
success: false,
|
||
message: error.message,
|
||
};
|
||
}
|
||
};
|