'use strict'; const moment = require('moment') async function allDeps (ctx) { try { const models = ctx.fs.dc.models; const { redis } = ctx.app const start = moment() let depRes = await redis.get('allDepartments') console.log(`DurationCalc: 1 用时 ${moment().diff(start, 'milliseconds')}`); if (depRes) { depRes = JSON.parse(depRes) console.log(`DurationCalc: 2 用时 ${moment().diff(start, 'milliseconds')}`); depRes = depRes.departments console.log(`DurationCalc: 3 用时 ${moment().diff(start, 'milliseconds')}`); } console.log(`DurationCalc: 4 用时 ${moment().diff(start, 'milliseconds')}`); ctx.status = 200; ctx.body = depRes || [] } catch (error) { ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`); ctx.status = 400; ctx.body = { } } } async function editUser (ctx) { try { const models = ctx.fs.dc.models; const { pomsUserId, pepUserId, role = [], correlationProject = [] } = ctx.request.body const existUserRes = await models.User.findOne({ where: { pepUserId } }) if ( existUserRes && !existUserRes.deleted && !pomsUserId && ( (existUserRes.role.includes('admin') && role.includes('admin')) || (!existUserRes.role.includes('admin') && !role.includes('admin')) // 都有或都没有管理员 ) ) { // 新增已存在未删除 throw '人员账号已添加' } let storageData = { pepUserId, role, correlationProject, updateTime: moment().format() } if (existUserRes) { // 存在且传递id 或者 不传id也存在 // 修改 update storageData.deleted = false if ( role.includes('admin') ) { if (existUserRes.role.includes('admin')) { // 已是管理员 throw '当前人员已是管理员' } // 正在修改为管理员 storageData.disabled = true storageData.role = [...new Set([...existUserRes.role, ...role])] } else if (existUserRes.role.includes('admin')) { // 正在修改成员 但是此时还是管理员 storageData.role = [...role, 'admin'] } await models.User.update(storageData, { where: { pepUserId } }) } else { // 新增 await models.User.create(storageData) } ctx.status = 204 } catch (error) { ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`); ctx.status = 400; ctx.body = { message: typeof error == 'string' ? error : undefined } } } async function putUser (ctx) { try { const models = ctx.fs.dc.models; const { pomsUserId } = ctx.params const { disabled = undefined, deleted = undefined } = ctx.request.body const existUserRes = await models.User.findOne({ where: { id: pomsUserId } }) if (existUserRes && existUserRes.role.includes('admin') && disabled == false) { throw '成员不能既是管理员又是普通成员' } const updateData = { disabled, deleted, } for (let k in updateData) { if (updateData[k] == undefined) { delete updateData[k] } } await models.User.update(updateData, { where: { id: pomsUserId } }) ctx.status = 204 } catch (error) { ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`); ctx.status = 400; ctx.body = { message: typeof error == 'string' ? error : undefined } } } async function delAdmin (ctx) { try { const models = ctx.fs.dc.models; const { pomsUserId } = ctx.params const existUserRes = await models.User.findOne({ where: { id: pomsUserId } }) if (existUserRes) { let updateValues = existUserRes.dataValues let adminIndex = updateValues.role.findIndex(r => r == 'admin') if (adminIndex > -1) { updateValues.role.splice(adminIndex, 1) } await models.User.update(updateValues, { where: { id: pomsUserId } }) } ctx.status = 204; } catch (error) { ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`); ctx.status = 400; ctx.body = { message: typeof error == 'string' ? error : undefined } } } async function user (ctx) { try { const models = ctx.fs.dc.models; const { clickHouse } = ctx.app.fs const { role, limit, page, } = ctx.query const excludeField = ['lastInTime', 'inTimes', 'onlineDuration', 'lastInAddress', 'deleted', 'updateTime'] let findOption = { attributes: { exclude: excludeField, }, where: { deleted: false, $not: { role: { $contained: ['SuperAdmin', 'admin'] } } }, order: [['updateTime', 'DESC']] } if (role) { findOption.where.role = { $contains: [role] } } if (limit) { findOption.limit = limit } if (page && limit) { findOption.offset = page * limit } const userRes = await models.User.findAndCountAll(findOption) const adminRes = await models.User.findAll({ where: { role: { $contains: ['admin'] } }, attributes: { exclude: excludeField, }, order: [['updateTime', 'DESC']] }) let userIds = new Set() for (let u of userRes.rows.concat(adminRes)) { userIds.add(u.pepUserId) } let userPepRes = userIds.size ? await clickHouse.pepEmis.query(`SELECT DISTINCT user.id AS id, "user"."name" AS name, department.name AS depName, department.id AS depId FROM department_user LEFT JOIN user ON department_user.user=user.id LEFT JOIN department ON department.id=department_user.department WHERE user.id IN (${[...userIds].join(',')}) AND department.delete=false`).toPromise() : [] for (let u of userRes.rows.concat(adminRes)) { const corUsers = userPepRes.filter(up => up.id == u.pepUserId) u.dataValues.name = corUsers[0].name u.dataValues.departments = corUsers.map(cu => { return { name: cu.depName, id: cu.depId } }) } ctx.status = 200 ctx.body = { admin: adminRes, users: userRes } } catch (error) { ctx.fs.logger.error(`path: ${ctx.path}, error: ${error}`); ctx.status = 400; ctx.body = { } } } module.exports = { allDeps, editUser, putUser, delAdmin, user, };