Compare commits

..

No commits in common. "main" and "feature/pinia" have entirely different histories.

334 changed files with 3702 additions and 12055 deletions

View File

@ -37,3 +37,6 @@ jobs:
- name: Lint - name: Lint
run: cd server && npm run lint run: cd server && npm run lint
- name: Format
run: cd server && npm run format

View File

@ -40,3 +40,6 @@ jobs:
- name: Lint - name: Lint
run: cd web && npm run lint run: cd web && npm run lint
- name: Format
run: cd web && npm run format

7
.gitignore vendored
View File

@ -25,10 +25,3 @@ pnpm-debug.log*
*.sw? *.sw?
.history .history
components.d.ts
# 默认的上传文件夹
userUpload
exportfile
yarn.lock

View File

@ -1,133 +0,0 @@
# Contributor Covenant Code of Conduct
## Our Pledge
We as members, contributors, and leaders pledge to make participation in our
community a harassment-free experience for everyone, regardless of age, body
size, visible or invisible disability, ethnicity, sex characteristics, gender
identity and expression, level of experience, education, socio-economic status,
nationality, personal appearance, race, caste, color, religion, or sexual
identity and orientation.
We pledge to act and interact in ways that contribute to an open, welcoming,
diverse, inclusive, and healthy community.
## Our Standards
Examples of behavior that contributes to a positive environment for our
community include:
* Demonstrating empathy and kindness toward other people
* Being respectful of differing opinions, viewpoints, and experiences
* Giving and gracefully accepting constructive feedback
* Accepting responsibility and apologizing to those affected by our mistakes,
and learning from the experience
* Focusing on what is best not just for us as individuals, but for the overall
community
Examples of unacceptable behavior include:
* The use of sexualized language or imagery, and sexual attention or advances of
any kind
* Trolling, insulting or derogatory comments, and personal or political attacks
* Public or private harassment
* Publishing others' private information, such as a physical or email address,
without their explicit permission
* Other conduct which could reasonably be considered inappropriate in a
professional setting
## Enforcement Responsibilities
Community leaders are responsible for clarifying and enforcing our standards of
acceptable behavior and will take appropriate and fair corrective action in
response to any behavior that they deem inappropriate, threatening, offensive,
or harmful.
Community leaders have the right and responsibility to remove, edit, or reject
comments, commits, code, wiki edits, issues, and other contributions that are
not aligned to this Code of Conduct, and will communicate reasons for moderation
decisions when appropriate.
## Scope
This Code of Conduct applies within all community spaces, and also applies when
an individual is officially representing the community in public spaces.
Examples of representing our community include using an official email address,
posting via an official social media account, or acting as an appointed
representative at an online or offline event.
## Enforcement
Instances of abusive, harassing, or otherwise unacceptable behavior may be
reported to the community leaders responsible for enforcement at
[xiaojusurvey@gmail.com](mailto:xiaojusurvey@gmail.com).
All complaints will be reviewed and investigated promptly and fairly.
All community leaders are obligated to respect the privacy and security of the
reporter of any incident.
## Enforcement Guidelines
Community leaders will follow these Community Impact Guidelines in determining
the consequences for any action they deem in violation of this Code of Conduct:
### 1. Correction
**Community Impact**: Use of inappropriate language or other behavior deemed
unprofessional or unwelcome in the community.
**Consequence**: A private, written warning from community leaders, providing
clarity around the nature of the violation and an explanation of why the
behavior was inappropriate. A public apology may be requested.
### 2. Warning
**Community Impact**: A violation through a single incident or series of
actions.
**Consequence**: A warning with consequences for continued behavior. No
interaction with the people involved, including unsolicited interaction with
those enforcing the Code of Conduct, for a specified period of time. This
includes avoiding interactions in community spaces as well as external channels
like social media. Violating these terms may lead to a temporary or permanent
ban.
### 3. Temporary Ban
**Community Impact**: A serious violation of community standards, including
sustained inappropriate behavior.
**Consequence**: A temporary ban from any sort of interaction or public
communication with the community for a specified period of time. No public or
private interaction with the people involved, including unsolicited interaction
with those enforcing the Code of Conduct, is allowed during this period.
Violating these terms may lead to a permanent ban.
### 4. Permanent Ban
**Community Impact**: Demonstrating a pattern of violation of community
standards, including sustained inappropriate behavior, harassment of an
individual, or aggression toward or disparagement of classes of individuals.
**Consequence**: A permanent ban from any sort of public interaction within the
community.
## Attribution
This Code of Conduct is adapted from the [Contributor Covenant][homepage],
version 2.1, available at
[https://www.contributor-covenant.org/version/2/1/code_of_conduct.html][v2.1].
Community Impact Guidelines were inspired by
[Mozilla's code of conduct enforcement ladder][Mozilla CoC].
For answers to common questions about this code of conduct, see the FAQ at
[https://www.contributor-covenant.org/faq][FAQ]. Translations are available at
[https://www.contributor-covenant.org/translations][translations].
[homepage]: https://www.contributor-covenant.org
[v2.1]: https://www.contributor-covenant.org/version/2/1/code_of_conduct.html
[Mozilla CoC]: https://github.com/mozilla/diversity
[FAQ]: https://www.contributor-covenant.org/faq
[translations]: https://www.contributor-covenant.org/translations

View File

@ -1,22 +0,0 @@
# Important Disclosure reXIAOJUSURVEY Data Collection
XIAOJUSURVEY is open-source software developed and maintained by XIAOJUSURVEY Team and available at https://github.com/didi/xiaoju-survey.
We hereby state the purpose and reason for collecting data.
## Purpose of data collection
Data collected is used to help improve XIAOJUSURVEY for all users. It is important that our team understands the usage patterns as soon as possible, so we can best decide how to design future features and prioritize current work.
## Types of data collected
XIAOJUSURVEY just collects data about version's information. The data collected is subsequently reported to the XIAOJUSURVEY's backend services.
All data collected will be used exclusively by the XIAOJUSURVEY team for analytical purposes only. The data will be neither accessible nor sold to any third party.
## Sensitive data
XIAOJUSURVEY will never collect and/or report sensitive information, such as private keys, API keys, or passwords.
## How do I opt-in to or opt-out of data sharing?
See [docs](https://xiaojusurvey.didi.cn/docs/next/community/%E6%95%B0%E6%8D%AE%E4%B8%8A%E6%8A%A5%E5%A3%B0%E6%98%8E) for information on configuring this functionality.

View File

@ -1,5 +1,5 @@
# 镜像集成 # 镜像集成
FROM node:18-slim FROM node:18
# 设置工作区间 # 设置工作区间
WORKDIR /xiaoju-survey WORKDIR /xiaoju-survey

View File

@ -186,7 +186,7 @@
same "printed page" as the copyright notice for easier same "printed page" as the copyright notice for easier
identification within third-party archives. identification within third-party archives.
Copyright (C) 2023 Beijing Didi Infinity Technology and Development Co.,Ltd. All rights reserved. Copyright [yyyy] [name of copyright owner]
Licensed under the Apache License, Version 2.0 (the "License"); Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License. you may not use this file except in compliance with the License.

166
README.md
View File

@ -29,53 +29,35 @@
<br /> <br />
&ensp;&ensp;**XIAOJUSURVEY**是一套轻量、安全的调研系统,提供面向个人和企业的一站式产品级解决方案,用于构建各类问卷、考试、测评和复杂表单,快速满足各类线上调研场景。 &ensp;&ensp;**XIAOJUSURVEY**是一套轻量、安全的**问卷系统基座**,提供面向个人和企业的一站式产品级解决方案,快速满足各类线上调研场景。
&ensp;&ensp;内部系统已沉淀 40+种题型,累积精选模板 100+,适用于市场调研、客户满意度调研、在线考试、投票、报道、测评等众多场景。数据能力上,经过上亿量级打磨,沉淀了分题统计、交叉分析、多渠道分析等在线报表能力,快速满足专业化分析。 &ensp;&ensp;内部系统已沉淀 40+种题型,累积精选模板 100+,适用于市场调研、客户满意度调研、在线考试、投票、报道、测评等众多场景。数据能力上,经过上亿量级打磨,沉淀了分题统计、交叉分析、多渠道分析等在线报表能力,快速满足专业化分析。
# 功能特性 &ensp;&ensp;开源项目以打造**调研基座**为核心,围绕**平台能力**、**工程架构**、**研发体系**进行建设,大家可以「快速」打造「专属」问卷系统:[快速了解生态发展理念](https://xiaojusurvey.didi.cn/docs/next/community/%E7%94%9F%E6%80%81%E5%BB%BA%E8%AE%BE)
**🌈 易用** # 功能简介
- 多类型数据采集,轻松创建调研表单:文本输入、数据选择、评分、投票、文件上传等。 - 问卷管理:创、编、投、收、数据分析
- 智能逻辑编排,设计多规则动态表单:显示逻辑、跳转逻辑、选项引用、题目引用等。 - 多样化题型:单行输入框、多行输入框、单项选择、多项选择、判断题、评分、投票、...
- 精细权限管理,支持高效团队协同:空间管理、多角色权限管理等。 - 用户管理:登录、注册、权限管理
- 数据在线分析和导出,洞察调研结果:数据导出、回收数据管理、分题统计、交叉分析等。 - 数据安全:传输加密、脱敏等
**🎨 好看** > 更全的建设请查阅 [官方 Feature](https://github.com/didi/xiaoju-survey/issues/45)
- 主题自由定制适配您的品牌自定义颜色、背景、图片、Logo、结果页规则等。
- 无缝嵌入各终端,满足不同场景需求:多端嵌入式小问卷 SDK。
**🚀 安全、可扩展**
- 安全能力可扩展,提供安全相关建设的经验指导:传输加密、敏感词库、发布审查等。
- 自定义 Hook 配置,轻松集成多方系统与各类工具:数据推送集成、消息推送集成等。
<img src="https://github.com/didi/xiaoju-survey/assets/16012672/dd427471-368d-49d9-bc44-13c34d84e3be" width="700" /> <img src="https://github.com/didi/xiaoju-survey/assets/16012672/dd427471-368d-49d9-bc44-13c34d84e3be" width="700" />
1、 **全部功能**请查看 [功能介绍](https://xiaojusurvey.didi.cn/docs/next/document/%E4%BA%A7%E5%93%81%E6%89%8B%E5%86%8C/%E5%8A%9F%E8%83%BD%E4%BB%8B%E7%BB%8D/%E5%9F%BA%E7%A1%80%E6%B5%81%E7%A8%8B)。 _**(个人和企业用户均可快速构建特定领域的调研类解决方案。)**_
2、**企业**和**个人**均可快速构建特定领域的调研类解决方案。
# 技术 # 技术
**1、Web 端Vue3 + ElementPlus** Web 端Vue3 + ElementPlusC 端多端渲染(规划中)
&ensp;&ensp;C 端多端渲染ReactNative SDK 建设中 Server 端Nestjs + MongoDBJava在建[欢迎加入共建](https://github.com/didi/xiaoju-survey/issues/306)
**2、Server 端NestJS + MongoDB** 智能化基座:(规划中)
&ensp;&ensp;Java 版:建设中,[欢迎加入共建](https://github.com/didi/xiaoju-survey/issues/306)
**3、能力增强**
&ensp;&ensp;在线平台:建设中、智能化问卷:规划中
# 项目优势 # 项目优势
@ -126,25 +108,103 @@
前后端分离,提供 Docker 化方案,提供了完善的部署指导手册。 前后端分离,提供 Docker 化方案,提供了完善的部署指导手册。
# 快速使用 # 快速启动
_在线平台建设中_ Node 版本 >= 18.x
[查看环境准备指导](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E5%BF%AB%E9%80%9F%E5%BC%80%E5%A7%8B)
# 本地开发 复制工程
请查看 [本地安装手册](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E5%BF%AB%E9%80%9F%E5%BC%80%E5%A7%8B) 来启动项目。 ```shell
git clone git@github.com:didi/xiaoju-survey.git
```
# 快速部署 ## 服务端启动
### 服务部署 ### 方案一、快速启动,无需安装数据库
请查看 [快速部署指导](https://xiaojusurvey.didi.cn/docs/next/document/%E5%B7%A5%E7%A8%8B%E9%83%A8%E7%BD%B2/Docker%E9%83%A8%E7%BD%B2) 。 > _便于快速预览工程对于正式项目需要使用方案二。_
### 一键部署 #### 1、安装依赖
_手册编写中_ ```shell
cd server
npm install
```
<br /> #### 2、启动
```shell
npm run local
```
> 服务运行依赖 [mongodb-memory-server](https://github.com/nodkz/mongodb-memory-server)
>
> 1、数据保存在内存中重启服务会更新数据。<br />2、启动内存服务器新实例时如果找不到 MongoDB 二进制文件会自动下载,因此首次可能需要一些时间。
### 方案二、(生产推荐)
#### 1、启动数据库
> 项目使用 MongoDB[MongoDB 安装指导](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E6%95%B0%E6%8D%AE%E5%BA%93#%E5%AE%89%E8%A3%85)
- 配置数据库,查看[MongoDB 配置](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E6%95%B0%E6%8D%AE%E5%BA%93)
- 启动本地数据库,查看[MongoDB 启动](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E6%95%B0%E6%8D%AE%E5%BA%93#%E4%BA%94%E5%90%AF%E5%8A%A8)
#### 2、安装依赖
```shell
cd server
npm install
```
#### 3、启动
```shell
npm run dev
```
## 前端启动
### 安装依赖
```shell
cd web
npm install
```
### 启动
```shell
npm run serve
```
## 访问
### 问卷管理端
[http://localhost:8080/management](http://localhost:8080)
### 问卷投放端
创建并发布问卷。
[http://localhost:8080/render/:surveyPath](http://localhost:8080/render/:surveyPath)
<br /><br />
## 微信交流群(推荐)
官方群会发布项目最新消息、建设计划和社区活动,欢迎你的加入。任何问题和合作可以联系小助手:
<img src="https://img-hxy021.didistatic.com/static/starimg/img/KXKvc7sjHz1700061188156.png" width="200" />
## QQ 交流群
官方群会发布项目最新消息、建设计划和社区活动,欢迎你的加入:
[<img src="https://img-hxy021.didistatic.com/static/starimg/img/iJUmLIHKV21700192846057.png" width="210" />](http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=P61UJI_q8AzizyBLGOm-bUvzNrUnSQq-&authKey=yZFtL9biGB5yiIME3%2Bi%2Bf6XMOdTNiuf0pCIaviEEAIryySNzVy6LJ4xl7uHdEcrM&noverify=0&group_code=920623419)
## Star ## Star
@ -152,26 +212,24 @@ _手册编写中_
[![Star History Chart](https://api.star-history.com/svg?repos=didi/xiaoju-survey&type=Date)](https://star-history.com/#didi/xiaoju-survey&Date) [![Star History Chart](https://api.star-history.com/svg?repos=didi/xiaoju-survey&type=Date)](https://star-history.com/#didi/xiaoju-survey&Date)
## 交流群 ## 记录
官方群会发布项目最新消息、建设计划和社区活动,欢迎你的加入。
<img src="https://img-hxy021.didistatic.com/static/starimg/img/KXKvc7sjHz1700061188156.png" width="200" />
_任何问题和合作可以联系小助手。_
## 案例
如果你使用了该项目,请记录反馈:[我在使用](https://github.com/didi/xiaoju-survey/issues/64),你的支持是我们最大的动力。 如果你使用了该项目,请记录反馈:[我在使用](https://github.com/didi/xiaoju-survey/issues/64),你的支持是我们最大的动力。
## Future Tasks
[欢迎了解项目发展和共建](https://github.com/didi/xiaoju-survey/issues/85),你的支持是我们最大的动力。
## 贡献 ## 贡献
如果你想成为贡献者或者扩展技术栈,请查看:[贡献者指南](https://xiaojusurvey.didi.cn/docs/next/share/%E5%A6%82%E4%BD%95%E5%8F%82%E4%B8%8E%E8%B4%A1%E7%8C%AE),你的加入使我们最大的荣幸。 如果你想成为贡献者或者扩展技术栈,请查看:[贡献者指南](https://xiaojusurvey.didi.cn/docs/next/share/%E5%A6%82%E4%BD%95%E5%8F%82%E4%B8%8E%E8%B4%A1%E7%8C%AE),你的加入使我们最大的荣幸。
## Feature
关注每周推出的建设:[官方 Feature](https://github.com/didi/xiaoju-survey/issues/45)
## CHANGELOG ## CHANGELOG
关注项目重大变更:[MAJOR CHANGELOG](https://github.com/didi/xiaoju-survey/issues/48)。 关注重大变更:[MAJOR CHANGELOG](https://github.com/didi/xiaoju-survey/issues/48)
## 文章分享
1、[掘金](https://juejin.cn/user/3705833332160473/posts)、2、[InfoQ](https://www.infoq.cn/profile/7E08AC616A07B2/publish)
[欢迎投稿](https://xiaojusurvey.didi.cn/docs/next/article/%E7%AE%80%E4%BB%8B)

View File

@ -29,47 +29,33 @@
<br /> <br />
&ensp;&ensp;XIAOJUSURVEY is an enterprises form builder and analytics platform to create questionnaires, exams, polls, quizzes, and analyze data online. &ensp;&ensp;XIAOJUSURVEY is a lightweight, secure questionnaire system foundation that provides one-stop product-level solutions for individuals and enterprises, quickly meeting various online survey scenarios.
&ensp;&ensp;The internal system has accumulated over 40 question types and more than 100 selected templates, suitable for market research, customer satisfaction surveys, online exams, voting, reporting, evaluations, and many other scenarios. In terms of data capabilities, it has been honed through hundreds of millions of iterations, resulting in the ability to provide online reports with per-question statistics, cross-analysis, and multi-channel analysis, quickly meeting professional analysis needs. &ensp;&ensp;The internal system has accumulated over 40 question types and more than 100 selected templates, suitable for market research, customer satisfaction surveys, online exams, voting, reporting, evaluations, and many other scenarios. In terms of data capabilities, it has been honed through hundreds of millions of iterations, resulting in the ability to provide online reports with per-question statistics, cross-analysis, and multi-channel analysis, quickly meeting professional analysis needs.
# Features &ensp;&ensp;The open-source project focuses on building a survey foundation, constructing around platform capabilities, engineering structure, and development systems, allowing everyone to 「quickly」 create their own 「exclusive」 questionnaire system: [quickly understanding the ecological development philosophy](https://xiaojusurvey.didi.cn/docs/next/community/%E7%94%9F%E6%80%81%E5%BB%BA%E8%AE%BE).
**🌈 Easy to use** # Function Overview
- Multi-type data collection, easy to create forms: text input, data selection, scoring, voting, file upload, etc. - Questionnaire Management: Create, edit, distribute, collect, data analysis.
- Smart logic arrangement, design multi-rule dynamic forms: display logic, jump logic, option reference, title reference, etc. - Diverse Question Types: Single-line input, multi-line input, single choice, multiple choice, true/false, rating, voting, etc.
- Multiple permission management, support efficient team collaboration: space management, multi-role permission management, etc. - User Management: Login, registration, permissions management.
- Online data analysis and export, insight into survey results: data export, recycled data management, sub-topic statistics, cross-analysis, etc. - Data Security: Encrypted transmission, data masking, etc.
**🎨 Good-looking** > For more comprehensive features, please refer to the official Feature documentation.
- Free customization of themes to adapt to your brand: custom colors, backgrounds, pictures, logos, result page rules, etc.
- Seamlessly embedded in various terminals to meet the needs of different scenarios: multi-terminal embedded small questionnaire SDK.
**🚀 Secure and scalable**
- Scalable security capabilities, providing experience guidance for security-related construction: encrypted transmission, data masking, etc.
- Customized Hook configuration, easy integration of multiple systems and various tools: data push, message push, etc.
<img src="https://github.com/didi/xiaoju-survey/assets/16012672/508ce30f-0ae8-4f5f-84a7-e96de8238a7f" width="700" /> <img src="https://github.com/didi/xiaoju-survey/assets/16012672/508ce30f-0ae8-4f5f-84a7-e96de8238a7f" width="700" />
1. For more comprehensive features, please refer to the [documentation](https://xiaojusurvey.didi.cn/docs/next/document/%E4%BA%A7%E5%93%81%E6%89%8B%E5%86%8C/%E5%8A%9F%E8%83%BD%E4%BB%8B%E7%BB%8D/%E5%9F%BA%E7%A1%80%E6%B5%81%E7%A8%8B). _**(Both individual and enterprise users can quickly build survey solutions specific to their fields.)**_
2. Both individual and enterprise users can quickly build survey solutions specific to their fields.
# Technology # Technology
Web: Vue3 + ElementPlus; Multi-end rendering for C-end (planning). Web: Vue3 + ElementPlus; Multi-end rendering for C-end (planning).
Server: NestJS + MongoDB; Java ([under construction](https://github.com/didi/xiaoju-survey/issues/306)). Server: Nestjs + MongoDB; Java ([under construction](https://github.com/didi/xiaoju-survey/issues/306)).
Online Platform: (under construction).
Intelligent Foundation: (planning). Intelligent Foundation: (planning).
@ -159,11 +145,12 @@ npm run local
### Option 2: (Recommended for Production) ### Option 2: (Recommended for Production)
#### 1.Configure Database #### 1.Start Database
> The project uses MongoDB: [MongoDB Guide](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E6%95%B0%E6%8D%AE%E5%BA%93#%E5%AE%89%E8%A3%85) > The project uses MongoDB: [MongoDB Installation Guide](https://xiaojusurvey.didi.cn/docs/next/document/%E6%A6%82%E8%BF%B0/%E6%95%B0%E6%8D%AE%E5%BA%93#%E5%AE%89%E8%A3%85)
Configure the database, check MongoDB configuration. - Configure the database, check MongoDB configuration.
- Start local database, check MongoDB startup.
#### 2.Install Dependencies #### 2.Install Dependencies
@ -207,18 +194,22 @@ Create and publish a questionnaire.
<br /><br /> <br /><br />
## Star
Open source is not easy. If this project helps you, please star it ❤️❤️❤️. Your support is our greatest motivation.
[![Star History Chart](https://api.star-history.com/svg?repos=didi/xiaoju-survey&type=Date)](https://star-history.com/#didi/xiaoju-survey&Date)
## WeChat Group ## WeChat Group
The official group will release the latest project news, construction plans, and community activities. Any questions and cooperation can contact the assistant: The official group will release the latest project news, construction plans, and community activities. Any questions and cooperation can contact the assistant:
<img src="https://img-hxy021.didistatic.com/static/starimg/img/KXKvc7sjHz1700061188156.png" width="200" /> <img src="https://img-hxy021.didistatic.com/static/starimg/img/KXKvc7sjHz1700061188156.png" width="200" />
## QQ Group
The official group will release the latest project news, construction plans, and community activities. Welcome to join:
[<img src="https://img-hxy021.didistatic.com/static/starimg/img/iJUmLIHKV21700192846057.png" width="210" />](http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=P61UJI_q8AzizyBLGOm-bUvzNrUnSQq-&authKey=yZFtL9biGB5yiIME3%2Bi%2Bf6XMOdTNiuf0pCIaviEEAIryySNzVy6LJ4xl7uHdEcrM&noverify=0&group_code=920623419)
## Star
Open source is not easy. If this project helps you, please star it ❤️❤️❤️. Your support is our greatest motivation.
## Feedback ## Feedback
If you use this project, please leave feedback:[I'm using](https://github.com/didi/xiaoju-survey/issues/64), Your support is our greatest. If you use this project, please leave feedback:[I'm using](https://github.com/didi/xiaoju-survey/issues/64), Your support is our greatest.
@ -227,11 +218,16 @@ If you use this project, please leave feedback:[I'm using](https://github.com/di
If you want to become a contributor or expand your technical stack, please check: [Contributor Guide](https://xiaojusurvey.didi.cn/docs/next/share/%E5%A6%82%E4%BD%95%E5%8F%82%E4%B8%8E%E8%B4%A1%E7%8C%AE). Your participation is our greatest honor. If you want to become a contributor or expand your technical stack, please check: [Contributor Guide](https://xiaojusurvey.didi.cn/docs/next/share/%E5%A6%82%E4%BD%95%E5%8F%82%E4%B8%8E%E8%B4%A1%E7%8C%AE). Your participation is our greatest honor.
## Future Tasks ## Feature
1. [Official Feature](https://github.com/didi/xiaoju-survey/issues/45) Pay attention to weekly construction updates: [Official Feature](https://github.com/didi/xiaoju-survey/issues/45)
2. [WIP](https://github.com/didi/xiaoju-survey/labels/WIP)
## CHANGELOG ## CHANGELOG
Follow major changes: [MAJOR CHANGELOG](https://github.com/didi/xiaoju-survey/issues/48) Follow major changes: [MAJOR CHANGELOG](https://github.com/didi/xiaoju-survey/issues/48)
## Article Sharing
1、[JueJin](https://juejin.cn/user/3705833332160473/posts)、2、[InfoQ](https://www.infoq.cn/profile/7E08AC616A07B2/publish)
[Welcome to contribute.](https://xiaojusurvey.didi.cn/docs/next/article/%E7%AE%80%E4%BB%8B)

View File

@ -15,7 +15,7 @@ services:
- xiaoju-survey - xiaoju-survey
xiaoju-survey: xiaoju-survey:
image: "xiaojusurvey/xiaoju-survey:1.3.0-slim" # 最新版本https://hub.docker.com/r/xiaojusurvey/xiaoju-survey/tags image: "xiaojusurvey/xiaoju-survey:1.1.2-slim"
container_name: xiaoju-survey container_name: xiaoju-survey
restart: always restart: always
ports: ports:

View File

@ -51,15 +51,6 @@ http {
location /api { location /api {
proxy_pass http://127.0.0.1:3000; proxy_pass http://127.0.0.1:3000;
} }
location /exportfile {
proxy_pass http://127.0.0.1:3000;
}
# 静态文件的默认存储文件夹
# 文件夹的配置在 server/src/modules/file/config/index.ts SERVER_LOCAL_CONFIG.FILE_KEY_PREFIX
location /userUpload {
proxy_pass http://127.0.0.1:3000;
}
error_page 500 502 503 504 /500.html; error_page 500 502 503 504 /500.html;
client_max_body_size 20M; client_max_body_size 20M;

View File

@ -1,13 +1,7 @@
XIAOJU_SURVEY_MONGO_DB_NAME=xiaojuSurvey XIAOJU_SURVEY_MONGO_DB_NAME=xiaojuSurvey
XIAOJU_SURVEY_MONGO_URL= XIAOJU_SURVEY_MONGO_URL=mongodb://localhost:27017
XIAOJU_SURVEY_MONGO_AUTH_SOURCE=admin XIAOJU_SURVEY_MONGO_AUTH_SOURCE=admin
XIAOJU_SURVEY_REDIS_HOST=
XIAOJU_SURVEY_REDIS_PORT=
XIAOJU_SURVEY_REDIS_USERNAME=
XIAOJU_SURVEY_REDIS_PASSWORD=
XIAOJU_SURVEY_REDIS_DB=
XIAOJU_SURVEY_RESPONSE_AES_ENCRYPT_SECRET_KEY=dataAesEncryptSecretKey XIAOJU_SURVEY_RESPONSE_AES_ENCRYPT_SECRET_KEY=dataAesEncryptSecretKey
XIAOJU_SURVEY_HTTP_DATA_ENCRYPT_TYPE=rsa XIAOJU_SURVEY_HTTP_DATA_ENCRYPT_TYPE=rsa

View File

@ -1,20 +0,0 @@
XIAOJU_SURVEY_MONGO_DB_NAME=xiaojuSurvey
XIAOJU_SURVEY_MONGO_URL=mongodb://127.0.0.1:27017
XIAOJU_SURVEY_MONGO_AUTH_SOURCE=admin
XIAOJU_SURVEY_REDIS_HOST=
XIAOJU_SURVEY_REDIS_PORT=
XIAOJU_SURVEY_REDIS_USERNAME=
XIAOJU_SURVEY_REDIS_PASSWORD=
XIAOJU_SURVEY_REDIS_DB=
XIAOJU_SURVEY_RESPONSE_AES_ENCRYPT_SECRET_KEY=dataAesEncryptSecretKey
XIAOJU_SURVEY_HTTP_DATA_ENCRYPT_TYPE=rsa
XIAOJU_SURVEY_JWT_SECRET=xiaojuSurveyJwtSecret
XIAOJU_SURVEY_JWT_EXPIRES_IN=8h
XIAOJU_SURVEY_LOGGER_FILENAME=./logs/app.log
XIAOJU_SURVEY_REPORT=true

View File

@ -1,17 +0,0 @@
XIAOJU_SURVEY_MONGO_DB_NAME=xiaojuSurvey
XIAOJU_SURVEY_MONGO_URL=
XIAOJU_SURVEY_MONGO_AUTH_SOURCE=admin
XIAOJU_SURVEY_REDIS_HOST=
XIAOJU_SURVEY_REDIS_PORT=
XIAOJU_SURVEY_REDIS_USERNAME=
XIAOJU_SURVEY_REDIS_PASSWORD=
XIAOJU_SURVEY_REDIS_DB=
XIAOJU_SURVEY_RESPONSE_AES_ENCRYPT_SECRET_KEY=dataAesEncryptSecretKey
XIAOJU_SURVEY_HTTP_DATA_ENCRYPT_TYPE=rsa
XIAOJU_SURVEY_JWT_SECRET=xiaojuSurveyJwtSecret
XIAOJU_SURVEY_JWT_EXPIRES_IN=8h
XIAOJU_SURVEY_LOGGER_FILENAME=./logs/app.log

5
server/.gitignore vendored
View File

@ -13,7 +13,6 @@ pnpm-debug.log*
yarn-debug.log* yarn-debug.log*
yarn-error.log* yarn-error.log*
lerna-debug.log* lerna-debug.log*
yarn.lock
# OS # OS
.DS_Store .DS_Store
@ -38,6 +37,4 @@ yarn.lock
!.vscode/launch.json !.vscode/launch.json
!.vscode/extensions.json !.vscode/extensions.json
tmp tmp
exportfile
userUpload

View File

@ -1,7 +1,7 @@
{ {
"name": "xiaoju-survey-server", "name": "server",
"version": "1.3.0", "version": "0.0.1",
"description": "XIAOJUSURVEY的server端", "description": "",
"author": "", "author": "",
"scripts": { "scripts": {
"build": "nest build", "build": "nest build",
@ -22,17 +22,15 @@
"@nestjs/common": "^10.0.0", "@nestjs/common": "^10.0.0",
"@nestjs/config": "^3.1.1", "@nestjs/config": "^3.1.1",
"@nestjs/core": "^10.0.0", "@nestjs/core": "^10.0.0",
"@nestjs/microservices": "^10.4.4",
"@nestjs/platform-express": "^10.0.0", "@nestjs/platform-express": "^10.0.0",
"@nestjs/serve-static": "^4.0.0", "@nestjs/serve-static": "^4.0.0",
"@nestjs/swagger": "^7.3.0", "@nestjs/swagger": "^7.3.0",
"@nestjs/typeorm": "^10.0.1", "@nestjs/typeorm": "^10.0.1",
"ali-oss": "^6.20.0", "ali-oss": "^6.20.0",
"cheerio": "1.0.0-rc.12", "cheerio": "^1.0.0-rc.12",
"crypto-js": "^4.2.0", "crypto-js": "^4.2.0",
"dotenv": "^16.3.2", "dotenv": "^16.3.2",
"fs-extra": "^11.2.0", "fs-extra": "^11.2.0",
"ioredis": "^5.4.1",
"joi": "^17.11.0", "joi": "^17.11.0",
"jsonwebtoken": "^9.0.2", "jsonwebtoken": "^9.0.2",
"lodash": "^4.17.21", "lodash": "^4.17.21",
@ -43,9 +41,7 @@
"nanoid": "^3.3.7", "nanoid": "^3.3.7",
"node-fetch": "^2.7.0", "node-fetch": "^2.7.0",
"node-forge": "^1.3.1", "node-forge": "^1.3.1",
"node-xlsx": "^0.24.0",
"qiniu": "^7.11.1", "qiniu": "^7.11.1",
"redlock": "^5.0.0-beta.2",
"reflect-metadata": "^0.1.13", "reflect-metadata": "^0.1.13",
"rxjs": "^7.8.1", "rxjs": "^7.8.1",
"svg-captcha": "^1.4.0", "svg-captcha": "^1.4.0",
@ -65,8 +61,8 @@
"@types/node": "^20.3.1", "@types/node": "^20.3.1",
"@types/node-forge": "^1.3.11", "@types/node-forge": "^1.3.11",
"@types/supertest": "^2.0.12", "@types/supertest": "^2.0.12",
"@typescript-eslint/eslint-plugin": "^7.16.0", "@typescript-eslint/eslint-plugin": "^6.0.0",
"@typescript-eslint/parser": "^7.16.0", "@typescript-eslint/parser": "^6.0.0",
"cross-env": "^7.0.3", "cross-env": "^7.0.3",
"eslint": "^8.42.0", "eslint": "^8.42.0",
"eslint-config-prettier": "^9.0.0", "eslint-config-prettier": "^9.0.0",
@ -74,14 +70,13 @@
"jest": "^29.5.0", "jest": "^29.5.0",
"mongodb-memory-server": "^9.1.4", "mongodb-memory-server": "^9.1.4",
"prettier": "^3.0.0", "prettier": "^3.0.0",
"redis-memory-server": "^0.11.0",
"source-map-support": "^0.5.21", "source-map-support": "^0.5.21",
"supertest": "^7.0.0", "supertest": "^6.3.3",
"ts-jest": "^29.1.0", "ts-jest": "^29.1.0",
"ts-loader": "^9.4.3", "ts-loader": "^9.4.3",
"ts-node": "^10.9.1", "ts-node": "^10.9.1",
"tsconfig-paths": "^4.2.0", "tsconfig-paths": "^4.2.0",
"typescript": "^5.5.3" "typescript": "^5.1.3"
}, },
"jest": { "jest": {
"moduleFileExtensions": [ "moduleFileExtensions": [
@ -95,9 +90,7 @@
"^.+\\.(t|j)s$": "ts-jest" "^.+\\.(t|j)s$": "ts-jest"
}, },
"collectCoverageFrom": [ "collectCoverageFrom": [
"**/*.(t|j)s", "**/*.(t|j)s"
"!**/*.module.ts",
"!**/upgrade.*.ts"
], ],
"coverageDirectory": "../coverage", "coverageDirectory": "../coverage",
"testEnvironment": "node", "testEnvironment": "node",

View File

@ -1,6 +1,5 @@
import { MongoMemoryServer } from 'mongodb-memory-server'; import { MongoMemoryServer } from 'mongodb-memory-server';
import { spawn } from 'child_process'; import { spawn } from 'child_process';
// import { RedisMemoryServer } from 'redis-memory-server';
async function startServerAndRunScript() { async function startServerAndRunScript() {
// 启动 MongoDB 内存服务器 // 启动 MongoDB 内存服务器
@ -9,19 +8,12 @@ async function startServerAndRunScript() {
console.log('MongoDB Memory Server started:', mongoUri); console.log('MongoDB Memory Server started:', mongoUri);
// const redisServer = new RedisMemoryServer();
// const redisHost = await redisServer.getHost();
// const redisPort = await redisServer.getPort();
// 通过 spawn 运行另一个脚本,并传递 MongoDB 连接 URL 作为环境变量 // 通过 spawn 运行另一个脚本,并传递 MongoDB 连接 URL 作为环境变量
const tsnode = spawn( const tsnode = spawn(
'cross-env', 'cross-env',
[ [
`XIAOJU_SURVEY_MONGO_URL=${mongoUri}`, `XIAOJU_SURVEY_MONGO_URL=${mongoUri}`,
// `XIAOJU_SURVEY_REDIS_HOST=${redisHost}`,
// `XIAOJU_SURVEY_REDIS_PORT=${redisPort}`,
'NODE_ENV=development', 'NODE_ENV=development',
'SERVER_ENV=local',
'npm', 'npm',
'run', 'run',
'start:dev', 'start:dev',
@ -39,10 +31,9 @@ async function startServerAndRunScript() {
console.error(data); console.error(data);
}); });
tsnode.on('close', async (code) => { tsnode.on('close', (code) => {
console.log(`Nodemon process exited with code ${code}`); console.log(`Nodemon process exited with code ${code}`);
await mongod.stop(); // 停止 MongoDB 内存服务器 mongod.stop(); // 停止 MongoDB 内存服务器
// await redisServer.stop();
}); });
} }

View File

@ -14,7 +14,6 @@ import { AuthModule } from './modules/auth/auth.module';
import { MessageModule } from './modules/message/message.module'; import { MessageModule } from './modules/message/message.module';
import { FileModule } from './modules/file/file.module'; import { FileModule } from './modules/file/file.module';
import { WorkspaceModule } from './modules/workspace/workspace.module'; import { WorkspaceModule } from './modules/workspace/workspace.module';
import { UpgradeModule } from './modules/upgrade/upgrade.module';
import { join } from 'path'; import { join } from 'path';
@ -36,21 +35,16 @@ import { MessagePushingLog } from './models/messagePushingLog.entity';
import { WorkspaceMember } from './models/workspaceMember.entity'; import { WorkspaceMember } from './models/workspaceMember.entity';
import { Workspace } from './models/workspace.entity'; import { Workspace } from './models/workspace.entity';
import { Collaborator } from './models/collaborator.entity'; import { Collaborator } from './models/collaborator.entity';
import { DownloadTask } from './models/downloadTask.entity';
import { Session } from './models/session.entity';
import { LoggerProvider } from './logger/logger.provider'; import { LoggerProvider } from './logger/logger.provider';
import { PluginManagerProvider } from './securityPlugin/pluginManager.provider'; import { PluginManagerProvider } from './securityPlugin/pluginManager.provider';
import { LogRequestMiddleware } from './middlewares/logRequest.middleware'; import { LogRequestMiddleware } from './middlewares/logRequest.middleware';
import { PluginManager } from './securityPlugin/pluginManager'; import { XiaojuSurveyPluginManager } from './securityPlugin/pluginManager';
import { Logger } from './logger'; import { Logger } from './logger';
@Module({ @Module({
imports: [ imports: [
ConfigModule.forRoot({ ConfigModule.forRoot({}),
envFilePath: `.env.${process.env.NODE_ENV}`, // 根据 NODE_ENV 动态加载对应的 .env 文件
isGlobal: true, // 使配置模块在应用的任何地方可用
}),
TypeOrmModule.forRootAsync({ TypeOrmModule.forRootAsync({
imports: [ConfigModule], imports: [ConfigModule],
inject: [ConfigService], inject: [ConfigService],
@ -87,8 +81,6 @@ import { Logger } from './logger';
Workspace, Workspace,
WorkspaceMember, WorkspaceMember,
Collaborator, Collaborator,
DownloadTask,
Session,
], ],
}; };
}, },
@ -108,7 +100,6 @@ import { Logger } from './logger';
MessageModule, MessageModule,
FileModule, FileModule,
WorkspaceModule, WorkspaceModule,
UpgradeModule,
], ],
controllers: [AppController], controllers: [AppController],
providers: [ providers: [
@ -123,7 +114,7 @@ import { Logger } from './logger';
export class AppModule { export class AppModule {
constructor( constructor(
private readonly configService: ConfigService, private readonly configService: ConfigService,
private readonly pluginManager: PluginManager, private readonly pluginManager: XiaojuSurveyPluginManager,
) {} ) {}
configure(consumer: MiddlewareConsumer) { configure(consumer: MiddlewareConsumer) {
consumer.apply(LogRequestMiddleware).forRoutes('*'); consumer.apply(LogRequestMiddleware).forRoutes('*');

View File

@ -1,6 +0,0 @@
export enum DOWNLOAD_TASK_STATUS {
WAITING = 'waiting', // 排队中
COMPUTING = 'computing', // 计算中
SUCCEED = 'succeed', // 导出成功
FAILED = 'failed', // 导出失败
}

View File

@ -6,22 +6,18 @@ export enum EXCEPTION_CODE {
USER_EXISTS = 2001, // 用户已存在 USER_EXISTS = 2001, // 用户已存在
USER_NOT_EXISTS = 2002, // 用户不存在 USER_NOT_EXISTS = 2002, // 用户不存在
USER_PASSWORD_WRONG = 2003, // 用户名或密码错误 USER_PASSWORD_WRONG = 2003, // 用户名或密码错误
PASSWORD_INVALID = 2004, // 密码无效
NO_SURVEY_PERMISSION = 3001, // 没有问卷权限 NO_SURVEY_PERMISSION = 3001, // 没有问卷权限
SURVEY_STATUS_TRANSFORM_ERROR = 3002, // 问卷状态转换报错 SURVEY_STATUS_TRANSFORM_ERROR = 3002, // 问卷状态转换报错
SURVEY_TYPE_ERROR = 3003, // 问卷类型错误 SURVEY_TYPE_ERROR = 3003, // 问卷类型错误
SURVEY_NOT_FOUND = 3004, // 问卷不存在 SURVEY_NOT_FOUND = 3004, // 问卷不存在
SURVEY_CONTENT_NOT_ALLOW = 3005, // 存在禁用内容 SURVEY_CONTENT_NOT_ALLOW = 3005, // 存在禁用内容
SURVEY_SAVE_CONFLICT = 3006, // 问卷冲突
CAPTCHA_INCORRECT = 4001, // 验证码不正确 CAPTCHA_INCORRECT = 4001, // 验证码不正确
WHITELIST_ERROR = 4002, // 白名单校验错误
RESPONSE_SIGN_ERROR = 9001, // 签名不正确 RESPONSE_SIGN_ERROR = 9001, // 签名不正确
RESPONSE_CURRENT_TIME_NOT_ALLOW = 9002, // 当前时间不允许提交 RESPONSE_CURRENT_TIME_NOT_ALLOW = 9002, // 当前时间不允许提交
RESPONSE_OVER_LIMIT = 9003, // 超出限制 RESPONSE_OVER_LIMIT = 9003, // 超出限制
RESPONSE_SCHEMA_REMOVED = 9004, // 问卷已删除 RESPONSE_SCHEMA_REMOVED = 9004, // 问卷已删除
RESPONSE_DATA_DECRYPT_ERROR = 9005, // 问卷已删除 RESPONSE_DATA_DECRYPT_ERROR = 9005, // 问卷已删除
RESPONSE_PAUSING = 9006, // 问卷已暂停
UPLOAD_FILE_ERROR = 5001, // 上传文件错误 UPLOAD_FILE_ERROR = 5001, // 上传文件错误
} }

View File

@ -1,15 +1,11 @@
// 状态类型 // 状态类型
export enum RECORD_STATUS { export enum RECORD_STATUS {
NEW = 'new', // 新建 | 未发布 NEW = 'new', // 新建
PUBLISHED = 'published', // 发布
EDITING = 'editing', // 编辑 EDITING = 'editing', // 编辑
FINISHED = 'finished', // 已结束
REMOVED = 'removed',
}
export const enum RECORD_SUB_STATUS {
DEFAULT = '', // 默认
PAUSING = 'pausing', // 暂停 PAUSING = 'pausing', // 暂停
PUBLISHED = 'published', // 发布
REMOVED = 'removed', // 删除
FORCE_REMOVED = 'forceRemoved', // 从回收站删除
} }
// 历史类型 // 历史类型

View File

@ -1,37 +0,0 @@
/**
* @description
*/
export enum QUESTION_TYPE {
/**
*
*/
TEXT = 'text',
/**
*
*/
TEXTAREA = 'textarea',
/**
*
*/
RADIO = 'radio',
/**
*
*/
CHECKBOX = 'checkbox',
/**
*
*/
BINARY_CHOICE = 'binary-choice',
/**
*
*/
RADIO_STAR = 'radio-star',
/**
* nps评分
*/
RADIO_NPS = 'radio-nps',
/**
*
*/
VOTE = 'vote',
}

View File

@ -1,4 +0,0 @@
export enum SESSION_STATUS {
ACTIVATED = 'activated',
DEACTIVATED = 'deactivated',
}

View File

@ -1,68 +0,0 @@
import { ExecutionContext } from '@nestjs/common';
import { Reflector } from '@nestjs/core';
import { SessionService } from 'src/modules/survey/services/session.service';
import { SessionGuard } from '../session.guard';
import { NoPermissionException } from 'src/exceptions/noPermissionException';
describe('SessionGuard', () => {
let sessionGuard: SessionGuard;
let reflector: Reflector;
let sessionService: SessionService;
beforeEach(() => {
reflector = new Reflector();
sessionService = {
findOne: jest.fn(),
} as unknown as SessionService;
sessionGuard = new SessionGuard(reflector, sessionService);
});
it('should return true when sessionId exists and sessionService returns sessionInfo', async () => {
const mockSessionId = '12345';
const mockSessionInfo = { id: mockSessionId, name: 'test session' };
const context = {
switchToHttp: jest.fn().mockReturnThis(),
getRequest: jest.fn().mockReturnValue({
sessionId: mockSessionId,
}),
getHandler: jest.fn(),
} as unknown as ExecutionContext;
jest.spyOn(reflector, 'get').mockReturnValue('sessionId');
jest
.spyOn(sessionService, 'findOne')
.mockResolvedValue(mockSessionInfo as any);
const result = await sessionGuard.canActivate(context);
const request = context.switchToHttp().getRequest();
expect(result).toBe(true);
expect(reflector.get).toHaveBeenCalledWith(
'sessionId',
context.getHandler(),
);
expect(sessionService.findOne).toHaveBeenCalledWith(mockSessionId);
expect(request.sessionInfo).toEqual(mockSessionInfo);
});
it('should throw NoPermissionException when sessionId is missing', async () => {
const context = {
switchToHttp: jest.fn().mockReturnThis(),
getRequest: jest.fn().mockReturnValue({}),
getHandler: jest.fn(),
} as unknown as ExecutionContext;
jest.spyOn(reflector, 'get').mockReturnValue('sessionId');
await expect(sessionGuard.canActivate(context)).rejects.toThrow(
NoPermissionException,
);
expect(reflector.get).toHaveBeenCalledWith(
'sessionId',
context.getHandler(),
);
});
});

View File

@ -1,30 +0,0 @@
import { Injectable, CanActivate, ExecutionContext } from '@nestjs/common';
import { Reflector } from '@nestjs/core';
import { get } from 'lodash';
import { NoPermissionException } from 'src/exceptions/noPermissionException';
import { SessionService } from 'src/modules/survey/services/session.service';
@Injectable()
export class SessionGuard implements CanActivate {
constructor(
private reflector: Reflector,
private readonly sessionService: SessionService,
) {}
async canActivate(context: ExecutionContext): Promise<boolean> {
const request = context.switchToHttp().getRequest();
const sessionIdKey = this.reflector.get<string>(
'sessionId',
context.getHandler(),
);
const sessionId = get(request, sessionIdKey);
if (!sessionId) {
throw new NoPermissionException('没有权限');
}
const sessionInfo = await this.sessionService.findOne(sessionId);
request.sessionInfo = sessionInfo;
request.surveyId = sessionInfo.surveyId;
return true;
}
}

View File

@ -3,6 +3,7 @@ import { Reflector } from '@nestjs/core';
import { get } from 'lodash'; import { get } from 'lodash';
import { WorkspaceMemberService } from 'src/modules/workspace/services/workspaceMember.service'; import { WorkspaceMemberService } from 'src/modules/workspace/services/workspaceMember.service';
import { CollaboratorService } from 'src/modules/survey/services/collaborator.service'; import { CollaboratorService } from 'src/modules/survey/services/collaborator.service';
import { SurveyMetaService } from 'src/modules/survey/services/surveyMeta.service'; import { SurveyMetaService } from 'src/modules/survey/services/surveyMeta.service';
import { SurveyNotFoundException } from 'src/exceptions/surveyNotFoundException'; import { SurveyNotFoundException } from 'src/exceptions/surveyNotFoundException';

View File

@ -94,58 +94,18 @@ export interface SubmitConf {
msgContent: MsgContent; msgContent: MsgContent;
} }
// 白名单类型
export enum WhitelistType {
ALL = 'ALL',
// 空间成员
MEMBER = 'MEMBER',
// 自定义
CUSTOM = 'CUSTOM',
}
// 白名单用户类型
export enum MemberType {
// 手机号
MOBILE = 'MOBILE',
// 邮箱
EMAIL = 'EMAIL',
}
export interface BaseConf { export interface BaseConf {
beginTime: string; begTime: string;
endTime: string; endTime: string;
answerBegTime: string; answerBegTime: string;
answerEndTime: string; answerEndTime: string;
tLimit: number; tLimit: number;
language: string; language: string;
// 访问密码开关
passwordSwitch?: boolean;
// 密码
password?: string | null;
// 白名单类型
whitelistType?: WhitelistType;
// 白名单用户类型
memberType?: MemberType;
// 白名单列表
whitelist?: string[];
// 提示语
whitelistTip?: string;
} }
export interface SkinConf { export interface SkinConf {
skinColor: string; skinColor: string;
inputBgColor: string; inputBgColor: string;
backgroundConf: {
color: string;
type: string;
image: string;
};
contentConf: {
opacity: number;
};
themeConf: {
color: string;
};
} }
export interface BottomConf { export interface BottomConf {

View File

@ -1,18 +1,15 @@
import * as log4js from 'log4js'; import * as log4js from 'log4js';
import moment from 'moment'; import moment from 'moment';
import { Injectable, Scope, Inject } from '@nestjs/common'; import { Request } from 'express';
import { CONTEXT, RequestContext } from '@nestjs/microservices';
const log4jsLogger = log4js.getLogger(); const log4jsLogger = log4js.getLogger();
@Injectable({ scope: Scope.REQUEST })
export class Logger { export class Logger {
private static inited = false; private static inited = false;
constructor(@Inject(CONTEXT) private readonly ctx: RequestContext) {} constructor() {}
static init(config: { filename: string }) { static init(config: { filename: string }) {
if (Logger.inited) { if (this.inited) {
return; return;
} }
log4js.configure({ log4js.configure({
@ -33,26 +30,25 @@ export class Logger {
default: { appenders: ['app'], level: 'trace' }, default: { appenders: ['app'], level: 'trace' },
}, },
}); });
Logger.inited = true;
} }
_log(message, options: { dltag?: string; level: string }) { _log(message, options: { dltag?: string; level: string; req?: Request }) {
const datetime = moment().format('YYYY-MM-DD HH:mm:ss.SSS'); const datetime = moment().format('YYYY-MM-DD HH:mm:ss.SSS');
const level = options?.level; const level = options?.level;
const dltag = options?.dltag ? `${options.dltag}||` : ''; const dltag = options?.dltag ? `${options.dltag}||` : '';
const traceIdStr = this.ctx?.['traceId'] const traceIdStr = options?.req?.['traceId']
? `traceid=${this.ctx?.['traceId']}||` ? `traceid=${options?.req?.['traceId']}||`
: ''; : '';
return log4jsLogger[level]( return log4jsLogger[level](
`[${datetime}][${level.toUpperCase()}]${dltag}${traceIdStr}${message}`, `[${datetime}][${level.toUpperCase()}]${dltag}${traceIdStr}${message}`,
); );
} }
info(message, options?: { dltag?: string }) { info(message, options?: { dltag?: string; req?: Request }) {
return this._log(message, { ...options, level: 'info' }); return this._log(message, { ...options, level: 'info' });
} }
error(message, options?: { dltag?: string }) { error(message, options: { dltag?: string; req?: Request }) {
return this._log(message, { ...options, level: 'error' }); return this._log(message, { ...options, level: 'error' });
} }
} }

View File

@ -10,9 +10,9 @@ const getCountStr = () => {
export const genTraceId = ({ ip }) => { export const genTraceId = ({ ip }) => {
// ip转16位 + 当前时间戳(毫秒级)+自增序列1000开始自增到9000+ 当前进程id的后5位 // ip转16位 + 当前时间戳(毫秒级)+自增序列1000开始自增到9000+ 当前进程id的后5位
ip = ip.replace('::ffff:', '').replace('::1', ''); ip = ip.replace('::ffff:', '');
let ipArr; let ipArr;
if (ip.indexOf(':') >= 0) { if (ip.indexOf(':') > 0) {
ipArr = ip.split(':').map((segment) => { ipArr = ip.split(':').map((segment) => {
// 将IPv6每个段转为16位并补0到长度为4 // 将IPv6每个段转为16位并补0到长度为4
return parseInt(segment, 16).toString(16).padStart(4, '0'); return parseInt(segment, 16).toString(16).padStart(4, '0');
@ -20,9 +20,7 @@ export const genTraceId = ({ ip }) => {
} else { } else {
ipArr = ip ipArr = ip
.split('.') .split('.')
.map((item) => .map((item) => parseInt(item).toString(16).padStart(2, '0'));
item ? parseInt(item).toString(16).padStart(2, '0') : '',
);
} }
return `${ipArr.join('')}${Date.now().toString()}${getCountStr()}${process.pid.toString().slice(-5)}`; return `${ipArr.join('')}${Date.now().toString()}${getCountStr()}${process.pid.toString().slice(-5)}`;

View File

@ -1,7 +1,6 @@
import { NestFactory } from '@nestjs/core'; import { NestFactory } from '@nestjs/core';
import { AppModule } from './app.module'; import { AppModule } from './app.module';
import { SwaggerModule, DocumentBuilder } from '@nestjs/swagger'; import { SwaggerModule, DocumentBuilder } from '@nestjs/swagger';
import './report';
async function bootstrap() { async function bootstrap() {
const PORT = process.env.PORT || 3000; const PORT = process.env.PORT || 3000;

View File

@ -1,3 +1,4 @@
// logger.middleware.ts
import { Injectable, NestMiddleware } from '@nestjs/common'; import { Injectable, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express'; import { Request, Response, NextFunction } from 'express';
import { Logger } from '../logger/index'; // 替换为你实际的logger路径 import { Logger } from '../logger/index'; // 替换为你实际的logger路径
@ -19,6 +20,7 @@ export class LogRequestMiddleware implements NestMiddleware {
`method=${method}||uri=${originalUrl}||ip=${ip}||ua=${userAgent}||query=${query}||body=${body}`, `method=${method}||uri=${originalUrl}||ip=${ip}||ua=${userAgent}||query=${query}||body=${body}`,
{ {
dltag: 'request_in', dltag: 'request_in',
req,
}, },
); );
@ -28,6 +30,7 @@ export class LogRequestMiddleware implements NestMiddleware {
`status=${res.statusCode.toString()}||duration=${duration}ms`, `status=${res.statusCode.toString()}||duration=${duration}ms`,
{ {
dltag: 'request_out', dltag: 'request_out',
req,
}, },
); );
}); });

View File

@ -0,0 +1,30 @@
import { BaseEntity } from '../base.entity';
import { RECORD_STATUS } from 'src/enums';
describe('BaseEntity', () => {
let baseEntity: BaseEntity;
beforeEach(() => {
baseEntity = new BaseEntity();
});
it('should initialize default info before insert', () => {
const now = Date.now();
baseEntity.initDefaultInfo();
expect(baseEntity.curStatus.status).toBe(RECORD_STATUS.NEW);
expect(baseEntity.curStatus.date).toBeCloseTo(now, -3);
expect(baseEntity.statusList).toHaveLength(1);
expect(baseEntity.statusList[0].status).toBe(RECORD_STATUS.NEW);
expect(baseEntity.statusList[0].date).toBeCloseTo(now, -3);
expect(baseEntity.createDate).toBeCloseTo(now, -3);
expect(baseEntity.updateDate).toBeCloseTo(now, -3);
});
it('should update updateDate before update', () => {
const now = Date.now();
baseEntity.onUpdate();
expect(baseEntity.updateDate).toBeCloseTo(now, -3);
});
});

View File

@ -1,57 +0,0 @@
import { SurveyMeta } from '../surveyMeta.entity';
import { RECORD_STATUS, RECORD_SUB_STATUS } from 'src/enums';
// 模拟日期
const mockDateNow = Date.now();
describe('SurveyMeta Entity', () => {
let surveyMeta: SurveyMeta;
// 在每个测试之前,初始化 SurveyMeta 实例
beforeEach(() => {
surveyMeta = new SurveyMeta();
// 模拟 Date.now() 返回固定的时间
jest.spyOn(Date, 'now').mockReturnValue(mockDateNow);
});
afterEach(() => {
jest.restoreAllMocks(); // 每次测试后还原所有 mock
});
it('should set default curStatus and subStatus on insert when they are not provided', () => {
surveyMeta.initDefaultInfo();
// 验证 curStatus 是否被初始化为默认值
expect(surveyMeta.curStatus).toEqual({
status: RECORD_STATUS.NEW,
date: mockDateNow,
});
// 验证 statusList 是否包含 curStatus
expect(surveyMeta.statusList).toEqual([
{
status: RECORD_STATUS.NEW,
date: mockDateNow,
},
]);
// 验证 subStatus 是否被初始化为默认值
expect(surveyMeta.subStatus).toEqual({
status: RECORD_SUB_STATUS.DEFAULT,
date: mockDateNow,
});
});
it('should initialize statusList if curStatus is provided but statusList is empty', () => {
surveyMeta.curStatus = null;
surveyMeta.initDefaultInfo();
expect(surveyMeta.statusList).toEqual([
{
status: RECORD_STATUS.NEW,
date: expect.any(Number),
},
]);
});
});

View File

@ -1,13 +1,43 @@
import { ObjectIdColumn, CreateDateColumn, UpdateDateColumn } from 'typeorm'; import { Column, ObjectIdColumn, BeforeInsert, BeforeUpdate } from 'typeorm';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
import { RECORD_STATUS } from '../enums';
export class BaseEntity { export class BaseEntity {
@ObjectIdColumn() @ObjectIdColumn()
_id: ObjectId; _id: ObjectId;
@CreateDateColumn({ type: 'timestamp', precision: 3 }) @Column()
createdAt: Date; curStatus: {
status: RECORD_STATUS;
date: number;
};
@UpdateDateColumn({ type: 'timestamp', precision: 3 }) @Column()
updatedAt: Date; statusList: Array<{
status: RECORD_STATUS;
date: number;
}>;
@Column()
createDate: number;
@Column()
updateDate: number;
@BeforeInsert()
initDefaultInfo() {
const now = Date.now();
if (!this.curStatus) {
const curStatus = { status: RECORD_STATUS.NEW, date: now };
this.curStatus = curStatus;
this.statusList = [curStatus];
}
this.createDate = now;
this.updateDate = now;
}
@BeforeUpdate()
onUpdate() {
this.updateDate = Date.now();
}
} }

View File

@ -5,7 +5,8 @@ import { BaseEntity } from './base.entity';
@Entity({ name: 'captcha' }) @Entity({ name: 'captcha' })
export class Captcha extends BaseEntity { export class Captcha extends BaseEntity {
@Index({ @Index({
expireAfterSeconds: 3600, expireAfterSeconds:
new Date(Date.now() + 2 * 60 * 60 * 1000).getTime() / 1000,
}) })
@ObjectIdColumn() @ObjectIdColumn()
_id: ObjectId; _id: ObjectId;

View File

@ -6,7 +6,8 @@ import { BaseEntity } from './base.entity';
@Entity({ name: 'clientEncrypt' }) @Entity({ name: 'clientEncrypt' })
export class ClientEncrypt extends BaseEntity { export class ClientEncrypt extends BaseEntity {
@Index({ @Index({
expireAfterSeconds: 3600, expireAfterSeconds:
new Date(Date.now() + 2 * 60 * 60 * 1000).getTime() / 1000,
}) })
@ObjectIdColumn() @ObjectIdColumn()
_id: ObjectId; _id: ObjectId;

View File

@ -11,16 +11,4 @@ export class Collaborator extends BaseEntity {
@Column('jsonb') @Column('jsonb')
permissions: Array<string>; permissions: Array<string>;
@Column()
creator: string;
@Column()
creatorId: string;
@Column()
operator: string;
@Column()
operatorId: string;
} }

View File

@ -1,48 +0,0 @@
import { Entity, Column } from 'typeorm';
import { BaseEntity } from './base.entity';
import { DOWNLOAD_TASK_STATUS } from 'src/enums/downloadTaskStatus';
@Entity({ name: 'downloadTask' })
export class DownloadTask extends BaseEntity {
@Column()
surveyId: string;
@Column()
surveyPath: string;
// 文件路径
@Column()
url: string;
// 文件key
@Column()
fileKey: string;
// 任务创建人
@Column()
creatorId: string;
// 任务创建人
@Column()
creator: string;
// 文件名
@Column()
filename: string;
// 文件大小
@Column()
fileSize: string;
@Column()
params: string;
@Column()
isDeleted: boolean;
@Column()
deletedAt: Date;
@Column()
status: DOWNLOAD_TASK_STATUS;
}

View File

@ -27,16 +27,4 @@ export class MessagePushingTask extends BaseEntity {
@Column() @Column()
ownerId: string; ownerId: string;
@Column()
isDeleted: boolean;
@Column()
deletedAt: Date;
@Column()
operator: string;
@Column()
operatorId: string;
} }

View File

@ -1,7 +1,6 @@
import { Entity, Column } from 'typeorm'; import { Entity, Column } from 'typeorm';
import { SurveySchemaInterface } from '../interfaces/survey'; import { SurveySchemaInterface } from '../interfaces/survey';
import { BaseEntity } from './base.entity'; import { BaseEntity } from './base.entity';
import { RECORD_STATUS, RECORD_SUB_STATUS } from '../enums';
@Entity({ name: 'surveyPublish' }) @Entity({ name: 'surveyPublish' })
export class ResponseSchema extends BaseEntity { export class ResponseSchema extends BaseEntity {
@ -16,19 +15,4 @@ export class ResponseSchema extends BaseEntity {
@Column() @Column()
pageId: string; pageId: string;
@Column()
curStatus: {
status: RECORD_STATUS;
date: number;
};
@Column()
subStatus: {
status: RECORD_SUB_STATUS;
date: number;
};
@Column()
isDeleted: boolean;
} }

View File

@ -1,22 +0,0 @@
import { Entity, Column, Index, ObjectIdColumn } from 'typeorm';
import { ObjectId } from 'mongodb';
import { BaseEntity } from './base.entity';
import { SESSION_STATUS } from 'src/enums/surveySessionStatus';
@Entity({ name: 'session' })
export class Session extends BaseEntity {
@Index({
expireAfterSeconds: 3600,
})
@ObjectIdColumn()
_id: ObjectId;
@Column()
surveyId: string;
@Column()
userId: string;
@Column()
status: SESSION_STATUS;
}

View File

@ -19,7 +19,4 @@ export class SurveyHistory extends BaseEntity {
username: string; username: string;
_id: string; _id: string;
}; };
@Column('string')
sessionId: string;
} }

View File

@ -1,6 +1,5 @@
import { Entity, Column, BeforeInsert } from 'typeorm'; import { Entity, Column } from 'typeorm';
import { BaseEntity } from './base.entity'; import { BaseEntity } from './base.entity';
import { RECORD_STATUS, RECORD_SUB_STATUS } from '../enums';
@Entity({ name: 'surveyMeta' }) @Entity({ name: 'surveyMeta' })
export class SurveyMeta extends BaseEntity { export class SurveyMeta extends BaseEntity {
@ -19,9 +18,6 @@ export class SurveyMeta extends BaseEntity {
@Column() @Column()
creator: string; creator: string;
@Column()
creatorId: string;
@Column() @Column()
owner: string; owner: string;
@ -36,48 +32,4 @@ export class SurveyMeta extends BaseEntity {
@Column() @Column()
workspaceId: string; workspaceId: string;
@Column()
curStatus: {
status: RECORD_STATUS;
date: number;
};
@Column()
subStatus: {
status: RECORD_SUB_STATUS;
date: number;
};
@Column()
statusList: Array<{
status: RECORD_STATUS | RECORD_SUB_STATUS;
date: number;
}>;
@Column()
operator: string;
@Column()
operatorId: string;
@Column()
isDeleted: boolean;
@Column()
deletedAt: Date;
@BeforeInsert()
initDefaultInfo() {
const now = Date.now();
if (!this.curStatus) {
const curStatus = { status: RECORD_STATUS.NEW, date: now };
this.curStatus = curStatus;
this.statusList = [curStatus];
}
if (!this.subStatus) {
const subStatus = { status: RECORD_SUB_STATUS.DEFAULT, date: now };
this.subStatus = subStatus;
}
}
} }

View File

@ -14,7 +14,7 @@ export class SurveyResponse extends BaseEntity {
data: Record<string, any>; data: Record<string, any>;
@Column() @Column()
diffTime: number; difTime: number;
@Column() @Column()
clientTime: number; clientTime: number;
@ -27,11 +27,11 @@ export class SurveyResponse extends BaseEntity {
@BeforeInsert() @BeforeInsert()
async onDataInsert() { async onDataInsert() {
return await pluginManager.triggerHook('encryptResponseData', this); return await pluginManager.triggerHook('beforeResponseDataCreate', this);
} }
@AfterLoad() @AfterLoad()
async onDataLoaded() { async onDataLoaded() {
return await pluginManager.triggerHook('decryptResponseData', this); return await pluginManager.triggerHook('afterResponseDataReaded', this);
} }
} }

View File

@ -3,33 +3,12 @@ import { BaseEntity } from './base.entity';
@Entity({ name: 'workspace' }) @Entity({ name: 'workspace' })
export class Workspace extends BaseEntity { export class Workspace extends BaseEntity {
@Column()
creatorId: string;
@Column()
creator: string;
@Column() @Column()
ownerId: string; ownerId: string;
@Column()
owner: string;
@Column() @Column()
name: string; name: string;
@Column() @Column()
description: string; description: string;
@Column()
operator: string;
@Column()
operatorId: string;
@Column()
isDeleted: boolean;
@Column()
deletedAt: Date;
} }

View File

@ -11,16 +11,4 @@ export class WorkspaceMember extends BaseEntity {
@Column() @Column()
role: string; role: string;
@Column()
creator: string;
@Column()
creatorId: string;
@Column()
operator: string;
@Column()
operatorId: string;
} }

View File

@ -82,22 +82,6 @@ describe('AuthController', () => {
new HttpException('验证码不正确', EXCEPTION_CODE.CAPTCHA_INCORRECT), new HttpException('验证码不正确', EXCEPTION_CODE.CAPTCHA_INCORRECT),
); );
}); });
it('should throw HttpException with PASSWORD_INVALID code when password is invalid', async () => {
const mockUserInfo = {
username: 'testUser',
password: '无效的密码abc123',
captchaId: 'testCaptchaId',
captcha: 'testCaptcha',
};
await expect(controller.register(mockUserInfo)).rejects.toThrow(
new HttpException(
'密码只能输入数字、字母、特殊字符',
EXCEPTION_CODE.PASSWORD_INVALID,
),
);
});
}); });
describe('login', () => { describe('login', () => {
@ -220,29 +204,4 @@ describe('AuthController', () => {
expect(typeof result.data.img).toBe('string'); expect(typeof result.data.img).toBe('string');
}); });
}); });
describe('password strength', () => {
it('it should return strong', async () => {
await expect(
controller.getPasswordStrength('abcd&1234'),
).resolves.toEqual({
code: 200,
data: 'Strong',
});
});
it('it should return medium', async () => {
await expect(controller.getPasswordStrength('abc123')).resolves.toEqual({
code: 200,
data: 'Medium',
});
});
it('it should return weak', async () => {
await expect(controller.getPasswordStrength('123456')).resolves.toEqual({
code: 200,
data: 'Weak',
});
});
});
}); });

View File

@ -5,6 +5,7 @@ import { UserService } from '../services/user.service';
import { User } from 'src/models/user.entity'; import { User } from 'src/models/user.entity';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { hash256 } from 'src/utils/hash256'; import { hash256 } from 'src/utils/hash256';
import { RECORD_STATUS } from 'src/enums';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
describe('UserService', () => { describe('UserService', () => {
@ -144,6 +145,7 @@ describe('UserService', () => {
expect(userRepository.findOne).toHaveBeenCalledWith({ expect(userRepository.findOne).toHaveBeenCalledWith({
where: { where: {
username: username, username: username,
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
}); });
expect(user).toEqual(userInfo); expect(user).toEqual(userInfo);
@ -161,6 +163,7 @@ describe('UserService', () => {
expect(findOneSpy).toHaveBeenCalledWith({ expect(findOneSpy).toHaveBeenCalledWith({
where: { where: {
username: username, username: username,
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
}); });
expect(user).toBe(null); expect(user).toBe(null);
@ -181,6 +184,7 @@ describe('UserService', () => {
expect(userRepository.findOne).toHaveBeenCalledWith({ expect(userRepository.findOne).toHaveBeenCalledWith({
where: { where: {
_id: new ObjectId(id), _id: new ObjectId(id),
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
}); });
expect(user).toEqual(userInfo); expect(user).toEqual(userInfo);
@ -198,6 +202,7 @@ describe('UserService', () => {
expect(findOneSpy).toHaveBeenCalledWith({ expect(findOneSpy).toHaveBeenCalledWith({
where: { where: {
_id: new ObjectId(id), _id: new ObjectId(id),
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
}); });
expect(user).toBe(null); expect(user).toBe(null);
@ -206,8 +211,8 @@ describe('UserService', () => {
it('should return a list of users by username', async () => { it('should return a list of users by username', async () => {
const username = 'test'; const username = 'test';
const userList = [ const userList = [
{ _id: new ObjectId(), username: 'testUser1', createdAt: new Date() }, { _id: new ObjectId(), username: 'testUser1', createDate: new Date() },
{ _id: new ObjectId(), username: 'testUser2', createdAt: new Date() }, { _id: new ObjectId(), username: 'testUser2', createDate: new Date() },
]; ];
jest jest
@ -223,10 +228,11 @@ describe('UserService', () => {
expect(userRepository.find).toHaveBeenCalledWith({ expect(userRepository.find).toHaveBeenCalledWith({
where: { where: {
username: new RegExp(username), username: new RegExp(username),
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
skip: 0, skip: 0,
take: 10, take: 10,
select: ['_id', 'username', 'createdAt'], select: ['_id', 'username', 'createDate'],
}); });
expect(result).toEqual(userList); expect(result).toEqual(userList);
}); });
@ -237,12 +243,12 @@ describe('UserService', () => {
{ {
_id: new ObjectId(idList[0]), _id: new ObjectId(idList[0]),
username: 'testUser1', username: 'testUser1',
createdAt: new Date(), createDate: new Date(),
}, },
{ {
_id: new ObjectId(idList[1]), _id: new ObjectId(idList[1]),
username: 'testUser2', username: 'testUser2',
createdAt: new Date(), createDate: new Date(),
}, },
]; ];
@ -257,8 +263,9 @@ describe('UserService', () => {
_id: { _id: {
$in: idList.map((id) => new ObjectId(id)), $in: idList.map((id) => new ObjectId(id)),
}, },
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
select: ['_id', 'username', 'createdAt'], select: ['_id', 'username', 'createDate'],
}); });
expect(result).toEqual(userList); expect(result).toEqual(userList);
}); });

View File

@ -1,12 +1,4 @@
import { import { Controller, Post, Body, HttpCode } from '@nestjs/common';
Controller,
Post,
Body,
HttpCode,
Get,
Query,
Request,
} from '@nestjs/common';
import { ConfigService } from '@nestjs/config'; import { ConfigService } from '@nestjs/config';
import { UserService } from '../services/user.service'; import { UserService } from '../services/user.service';
import { CaptchaService } from '../services/captcha.service'; import { CaptchaService } from '../services/captcha.service';
@ -15,9 +7,6 @@ import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { create } from 'svg-captcha'; import { create } from 'svg-captcha';
import { ApiTags } from '@nestjs/swagger'; import { ApiTags } from '@nestjs/swagger';
const passwordReg = /^[a-zA-Z0-9!@#$%^&*()_+\-=[\]{};':"\\|,.<>/?]+$/;
@ApiTags('auth') @ApiTags('auth')
@Controller('/api/auth') @Controller('/api/auth')
export class AuthController { export class AuthController {
@ -39,24 +28,6 @@ export class AuthController {
captcha: string; captcha: string;
}, },
) { ) {
if (!userInfo.password) {
throw new HttpException('密码无效', EXCEPTION_CODE.PASSWORD_INVALID);
}
if (userInfo.password.length < 6 || userInfo.password.length > 16) {
throw new HttpException(
'密码长度在 6 到 16 个字符',
EXCEPTION_CODE.PASSWORD_INVALID,
);
}
if (!passwordReg.test(userInfo.password)) {
throw new HttpException(
'密码只能输入数字、字母、特殊字符',
EXCEPTION_CODE.PASSWORD_INVALID,
);
}
const isCorrect = await this.captchaService.checkCaptchaIsCorrect({ const isCorrect = await this.captchaService.checkCaptchaIsCorrect({
captcha: userInfo.captcha, captcha: userInfo.captcha,
id: userInfo.captchaId, id: userInfo.captchaId,
@ -191,59 +162,4 @@ export class AuthController {
}, },
}; };
} }
/**
*
*/
@Get('/password/strength')
@HttpCode(200)
async getPasswordStrength(@Query('password') password: string) {
const numberReg = /[0-9]/.test(password);
const letterReg = /[a-zA-Z]/.test(password);
const symbolReg = /[!@#$%^&*()_+\-=[\]{};':"\\|,.<>/?]/.test(password);
// 包含三种、且长度大于8
if (numberReg && letterReg && symbolReg && password.length >= 8) {
return {
code: 200,
data: 'Strong',
};
}
// 满足任意两种
if ([numberReg, letterReg, symbolReg].filter(Boolean).length >= 2) {
return {
code: 200,
data: 'Medium',
};
}
return {
code: 200,
data: 'Weak',
};
}
@Get('/verifyToken')
@HttpCode(200)
async verifyToken(@Request() req) {
const token = req.headers.authorization?.split(' ')[1];
if (!token) {
return {
code: 200,
data: false,
};
}
try {
await this.authService.verifyToken(token);
return {
code: 200,
data: true,
};
} catch (error) {
return {
code: 200,
data: false,
};
}
}
} }

View File

@ -1,11 +1,4 @@
import { import { Controller, Get, Query, HttpCode, UseGuards } from '@nestjs/common';
Controller,
Get,
Query,
HttpCode,
UseGuards,
Request,
} from '@nestjs/common';
import { ApiTags, ApiBearerAuth } from '@nestjs/swagger'; import { ApiTags, ApiBearerAuth } from '@nestjs/swagger';
import { Authentication } from 'src/guards/authentication.guard'; import { Authentication } from 'src/guards/authentication.guard';
@ -50,16 +43,4 @@ export class UserController {
}), }),
}; };
} }
@UseGuards(Authentication)
@Get('/getUserInfo')
async getUserInfo(@Request() req) {
return {
code: 200,
data: {
userId: req.user._id.toString(),
username: req.user.username,
},
};
}
} }

View File

@ -5,6 +5,7 @@ import { User } from 'src/models/user.entity';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { hash256 } from 'src/utils/hash256'; import { hash256 } from 'src/utils/hash256';
import { RECORD_STATUS } from 'src/enums';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
@Injectable() @Injectable()
@ -52,6 +53,9 @@ export class UserService {
const user = await this.userRepository.findOne({ const user = await this.userRepository.findOne({
where: { where: {
username: username, username: username,
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
}); });
@ -62,6 +66,9 @@ export class UserService {
const user = await this.userRepository.findOne({ const user = await this.userRepository.findOne({
where: { where: {
_id: new ObjectId(id), _id: new ObjectId(id),
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
}); });
@ -72,10 +79,13 @@ export class UserService {
const list = await this.userRepository.find({ const list = await this.userRepository.find({
where: { where: {
username: new RegExp(username), username: new RegExp(username),
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
skip, skip,
take, take,
select: ['_id', 'username', 'createdAt'], select: ['_id', 'username', 'createDate'],
}); });
return list; return list;
} }
@ -86,8 +96,11 @@ export class UserService {
_id: { _id: {
$in: idList.map((item) => new ObjectId(item)), $in: idList.map((item) => new ObjectId(item)),
}, },
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
select: ['_id', 'username', 'createdAt'], select: ['_id', 'username', 'createDate'],
}); });
return list; return list;
} }

View File

@ -14,18 +14,13 @@ export class FileService {
configKey, configKey,
file, file,
pathPrefix, pathPrefix,
filename,
}: { }: {
configKey: string; configKey: string;
file: Express.Multer.File; file: Express.Multer.File;
pathPrefix: string; pathPrefix: string;
filename?: string;
}) { }) {
const handler = this.getHandler(configKey); const handler = this.getHandler(configKey);
const { key } = await handler.upload(file, { const { key } = await handler.upload(file, { pathPrefix });
pathPrefix,
filename,
});
const url = await handler.getUrl(key); const url = await handler.getUrl(key);
return { return {
key, key,

View File

@ -1,4 +1,4 @@
import { join, dirname, sep } from 'path'; import { join, dirname } from 'path';
import fse from 'fs-extra'; import fse from 'fs-extra';
import { createWriteStream } from 'fs'; import { createWriteStream } from 'fs';
import { FileUploadHandler } from './uploadHandler.interface'; import { FileUploadHandler } from './uploadHandler.interface';
@ -12,20 +12,13 @@ export class LocalHandler implements FileUploadHandler {
async upload( async upload(
file: Express.Multer.File, file: Express.Multer.File,
options?: { pathPrefix?: string; filename?: string }, options?: { pathPrefix?: string },
): Promise<{ key: string }> { ): Promise<{ key: string }> {
let filename; const filename = await generateUniqueFilename(file.originalname);
if (options?.filename) {
filename = file.filename;
} else {
filename = await generateUniqueFilename(file.originalname);
}
const filePath = join( const filePath = join(
options?.pathPrefix ? options?.pathPrefix : '', options?.pathPrefix ? options?.pathPrefix : '',
filename, filename,
) );
.split(sep)
.join('/');
const physicalPath = join(this.physicalRootPath, filePath); const physicalPath = join(this.physicalRootPath, filePath);
await fse.mkdir(dirname(physicalPath), { recursive: true }); await fse.mkdir(dirname(physicalPath), { recursive: true });
const writeStream = createWriteStream(physicalPath); const writeStream = createWriteStream(physicalPath);
@ -42,10 +35,6 @@ export class LocalHandler implements FileUploadHandler {
} }
getUrl(key: string): string { getUrl(key: string): string {
if (process.env.SERVER_ENV === 'local') {
const port = process.env.PORT || 3000;
return `http://localhost:${port}/${key}`;
}
return `/${key}`; return `/${key}`;
} }
} }

View File

@ -8,6 +8,7 @@ import {
MESSAGE_PUSHING_TYPE, MESSAGE_PUSHING_TYPE,
MESSAGE_PUSHING_HOOK, MESSAGE_PUSHING_HOOK,
} from 'src/enums/messagePushing'; } from 'src/enums/messagePushing';
import { RECORD_STATUS } from 'src/enums';
describe('MessagePushingTaskDto', () => { describe('MessagePushingTaskDto', () => {
let dto: MessagePushingTaskDto; let dto: MessagePushingTaskDto;
@ -33,9 +34,9 @@ describe('MessagePushingTaskDto', () => {
}); });
it('should have a type', () => { it('should have a type', () => {
dto.type = MESSAGE_PUSHING_TYPE.HTTP; dto.type = MESSAGE_PUSHING_TYPE.HTTP; // Set your desired type here
expect(dto.type).toBeDefined(); expect(dto.type).toBeDefined();
expect(dto.type).toEqual(MESSAGE_PUSHING_TYPE.HTTP); expect(dto.type).toEqual(MESSAGE_PUSHING_TYPE.HTTP); // Adjust based on your enum
}); });
it('should have a push address', () => { it('should have a push address', () => {
@ -45,13 +46,13 @@ describe('MessagePushingTaskDto', () => {
}); });
it('should have a trigger hook', () => { it('should have a trigger hook', () => {
dto.triggerHook = MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED; dto.triggerHook = MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED; // Set your desired hook here
expect(dto.triggerHook).toBeDefined(); expect(dto.triggerHook).toBeDefined();
expect(dto.triggerHook).toEqual(MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED); expect(dto.triggerHook).toEqual(MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED); // Adjust based on your enum
}); });
it('should have an array of surveys', () => { it('should have an array of surveys', () => {
dto.surveys = ['survey1', 'survey2']; dto.surveys = ['survey1', 'survey2']; // Set your desired surveys here
expect(dto.surveys).toBeDefined(); expect(dto.surveys).toBeDefined();
expect(dto.surveys).toEqual(['survey1', 'survey2']); expect(dto.surveys).toEqual(['survey1', 'survey2']);
}); });
@ -61,6 +62,13 @@ describe('MessagePushingTaskDto', () => {
expect(dto.owner).toBeDefined(); expect(dto.owner).toBeDefined();
expect(dto.owner).toBe('test_owner'); expect(dto.owner).toBe('test_owner');
}); });
it('should have current status', () => {
dto.curStatus = { status: RECORD_STATUS.NEW, date: Date.now() };
expect(dto.curStatus).toBeDefined();
expect(dto.curStatus.status).toEqual(RECORD_STATUS.NEW);
expect(dto.curStatus.date).toBeDefined();
});
}); });
describe('CodeDto', () => { describe('CodeDto', () => {

View File

@ -10,6 +10,7 @@ import { MessagePushingLogService } from '../services/messagePushingLog.service'
import { CreateMessagePushingTaskDto } from '../dto/createMessagePushingTask.dto'; import { CreateMessagePushingTaskDto } from '../dto/createMessagePushingTask.dto';
import { UpdateMessagePushingTaskDto } from '../dto/updateMessagePushingTask.dto'; import { UpdateMessagePushingTaskDto } from '../dto/updateMessagePushingTask.dto';
import { RECORD_STATUS } from 'src/enums';
import { MESSAGE_PUSHING_TYPE } from 'src/enums/messagePushing'; import { MESSAGE_PUSHING_TYPE } from 'src/enums/messagePushing';
import { MESSAGE_PUSHING_HOOK } from 'src/enums/messagePushing'; import { MESSAGE_PUSHING_HOOK } from 'src/enums/messagePushing';
import { MessagePushingTask } from 'src/models/messagePushingTask.entity'; import { MessagePushingTask } from 'src/models/messagePushingTask.entity';
@ -117,12 +118,10 @@ describe('MessagePushingTaskService', () => {
expect(result).toEqual(tasks); expect(result).toEqual(tasks);
expect(repository.find).toHaveBeenCalledWith({ expect(repository.find).toHaveBeenCalledWith({
where: { where: {
isDeleted: {
$ne: true,
},
ownerId: mockOwnerId, ownerId: mockOwnerId,
surveys: { $all: [surveyId] }, surveys: { $all: [surveyId] },
triggerHook: hook, triggerHook: hook,
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
}); });
}); });
@ -147,19 +146,10 @@ describe('MessagePushingTaskService', () => {
where: { where: {
ownerId: mockOwnerId, ownerId: mockOwnerId,
_id: new ObjectId(taskId), _id: new ObjectId(taskId),
isDeleted: { 'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
$ne: true,
},
}, },
}); });
}); });
it('should throw an error when message pushing task is not found', async () => {
const taskId = '65afc62904d5db18534c0f78';
jest.spyOn(repository, 'findOne').mockResolvedValue(null); // 模拟未找到任务
const mockOwnerId = '66028642292c50f8b71a9eee';
await expect(service.findOne({ id: taskId, ownerId: mockOwnerId }));
});
}); });
describe('update', () => { describe('update', () => {
@ -171,6 +161,10 @@ describe('MessagePushingTaskService', () => {
pushAddress: 'http://update.example.com', pushAddress: 'http://update.example.com',
triggerHook: MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED, triggerHook: MESSAGE_PUSHING_HOOK.RESPONSE_INSERTED,
surveys: ['new survey id'], surveys: ['new survey id'],
curStatus: {
status: RECORD_STATUS.EDITING,
date: Date.now(),
},
}; };
const existingTask = new MessagePushingTask(); const existingTask = new MessagePushingTask();
existingTask._id = new ObjectId(taskId); existingTask._id = new ObjectId(taskId);
@ -196,20 +190,6 @@ describe('MessagePushingTaskService', () => {
}); });
expect(repository.save).toHaveBeenCalledWith(updatedTask); expect(repository.save).toHaveBeenCalledWith(updatedTask);
}); });
it('should throw an error if the task to be updated is not found', async () => {
const taskId = '65afc62904d5db18534c0f78';
const updateDto: UpdateMessagePushingTaskDto = { name: 'Updated Task' };
jest.spyOn(repository, 'findOne').mockResolvedValue(null); // 模拟任务未找到
const mockOwnerId = '66028642292c50f8b71a9eee';
await expect(
service.update({
ownerId: mockOwnerId,
id: taskId,
updateData: updateDto,
}),
).rejects.toThrow(`Message pushing task with id ${taskId} not found`);
});
}); });
describe('remove', () => { describe('remove', () => {
@ -217,54 +197,38 @@ describe('MessagePushingTaskService', () => {
const taskId = '65afc62904d5db18534c0f78'; const taskId = '65afc62904d5db18534c0f78';
const updateResult = { modifiedCount: 1 }; const updateResult = { modifiedCount: 1 };
const mockOperatorId = '66028642292c50f8b71a9eee'; const mockOwnerId = '66028642292c50f8b71a9eee';
const mockOperator = 'mockOperator';
jest.spyOn(repository, 'updateOne').mockResolvedValue(updateResult); jest.spyOn(repository, 'updateOne').mockResolvedValue(updateResult);
const result = await service.remove({ const result = await service.remove({
id: taskId, id: taskId,
operatorId: mockOperatorId, ownerId: mockOwnerId,
operator: mockOperator,
}); });
expect(result).toEqual(updateResult); expect(result).toEqual(updateResult);
expect(repository.updateOne).toHaveBeenCalledWith( expect(repository.updateOne).toHaveBeenCalledWith(
{ {
ownerId: mockOwnerId,
_id: new ObjectId(taskId), _id: new ObjectId(taskId),
'curStatus.status': { $ne: RECORD_STATUS.REMOVED },
}, },
{ {
$set: { $set: {
isDeleted: true, curStatus: {
operatorId: mockOperatorId, status: RECORD_STATUS.REMOVED,
operator: mockOperator, date: expect.any(Number),
deletedAt: expect.any(Date), },
},
$push: {
statusList: {
status: RECORD_STATUS.REMOVED,
date: expect.any(Number),
},
}, },
}, },
); );
}); });
it('should throw an error if the task to be removed is not found', async () => {
const taskId = '65afc62904d5db18534c0f78';
jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({ modifiedCount: 0 }); // 模拟删除失败
const mockOperatorId = '66028642292c50f8b71a9eee';
const mockOperator = 'mockOperator';
const result = await service.remove({
id: taskId,
operatorId: mockOperatorId,
operator: mockOperator,
});
expect(result.modifiedCount).toBe(0);
expect(repository.updateOne).toHaveBeenCalledWith(
{
_id: new ObjectId(taskId),
},
expect.any(Object),
);
});
}); });
describe('surveyAuthorizeTask', () => { describe('surveyAuthorizeTask', () => {
@ -294,35 +258,8 @@ describe('MessagePushingTaskService', () => {
$push: { $push: {
surveys: surveyId, surveys: surveyId,
}, },
$set: {
updatedAt: expect.any(Date),
},
}, },
); );
}); });
it('should not add the surveyId if it already exists in the task', async () => {
const taskId = '65afc62904d5db18534c0f78';
const surveyId = '65af380475b64545e5277dd9';
const mockOwnerId = '66028642292c50f8b71a9eee';
jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({ modifiedCount: 0 }); // 模拟重复添加
const result = await service.surveyAuthorizeTask({
taskId,
surveyId,
ownerId: mockOwnerId,
});
expect(result.modifiedCount).toBe(0);
expect(repository.updateOne).toHaveBeenCalledWith(
{
_id: new ObjectId(taskId),
surveys: { $nin: [surveyId] }, // 确保只有不包含时才插入
ownerId: mockOwnerId,
},
expect.any(Object),
);
});
}); });
}); });

View File

@ -37,4 +37,9 @@ describe('UpdateMessagePushingTaskDto', () => {
dto.surveys = null; dto.surveys = null;
expect(dto.surveys).toBeNull(); expect(dto.surveys).toBeNull();
}); });
it('should have a nullable curStatus', () => {
dto.curStatus = null;
expect(dto.curStatus).toBeNull();
});
}); });

View File

@ -150,9 +150,8 @@ export class MessagePushingTaskController {
async remove(@Request() req, @Param('id') id: string) { async remove(@Request() req, @Param('id') id: string) {
const userId = req.user._id; const userId = req.user._id;
const res = await this.messagePushingTaskService.remove({ const res = await this.messagePushingTaskService.remove({
ownerId: userId,
id, id,
operator: req.user.username,
operatorId: userId,
}); });
return { return {
code: 200, code: 200,

View File

@ -4,6 +4,7 @@ import {
MESSAGE_PUSHING_TYPE, MESSAGE_PUSHING_TYPE,
MESSAGE_PUSHING_HOOK, MESSAGE_PUSHING_HOOK,
} from 'src/enums/messagePushing'; } from 'src/enums/messagePushing';
import { RECORD_STATUS } from 'src/enums';
export class MessagePushingTaskDto { export class MessagePushingTaskDto {
@ApiProperty({ description: '任务id' }) @ApiProperty({ description: '任务id' })
@ -26,6 +27,12 @@ export class MessagePushingTaskDto {
@ApiProperty({ description: '所有者' }) @ApiProperty({ description: '所有者' })
owner: string; owner: string;
@ApiProperty({ description: '任务状态', required: false })
curStatus?: {
status: RECORD_STATUS;
date: number;
};
} }
export class CodeDto { export class CodeDto {

View File

@ -1,4 +1,5 @@
import { ApiProperty } from '@nestjs/swagger'; import { ApiProperty } from '@nestjs/swagger';
import { RECORD_STATUS } from 'src/enums';
import { import {
MESSAGE_PUSHING_TYPE, MESSAGE_PUSHING_TYPE,
MESSAGE_PUSHING_HOOK, MESSAGE_PUSHING_HOOK,
@ -19,4 +20,10 @@ export class UpdateMessagePushingTaskDto {
@ApiProperty({ description: '绑定的问卷id', required: false }) @ApiProperty({ description: '绑定的问卷id', required: false })
surveys?: string[]; surveys?: string[];
@ApiProperty({ description: '任务状态', required: false })
curStatus?: {
status: RECORD_STATUS;
date: number;
};
} }

View File

@ -6,6 +6,7 @@ import { MESSAGE_PUSHING_HOOK } from 'src/enums/messagePushing';
import { CreateMessagePushingTaskDto } from '../dto/createMessagePushingTask.dto'; import { CreateMessagePushingTaskDto } from '../dto/createMessagePushingTask.dto';
import { UpdateMessagePushingTaskDto } from '../dto/updateMessagePushingTask.dto'; import { UpdateMessagePushingTaskDto } from '../dto/updateMessagePushingTask.dto';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
import { RECORD_STATUS } from 'src/enums';
import { MESSAGE_PUSHING_TYPE } from 'src/enums/messagePushing'; import { MESSAGE_PUSHING_TYPE } from 'src/enums/messagePushing';
import { MessagePushingLogService } from './messagePushingLog.service'; import { MessagePushingLogService } from './messagePushingLog.service';
import { httpPost } from 'src/utils/request'; import { httpPost } from 'src/utils/request';
@ -43,8 +44,8 @@ export class MessagePushingTaskService {
ownerId?: string; ownerId?: string;
}): Promise<MessagePushingTask[]> { }): Promise<MessagePushingTask[]> {
const where: Record<string, any> = { const where: Record<string, any> = {
isDeleted: { 'curStatus.status': {
$ne: true, $ne: RECORD_STATUS.REMOVED,
}, },
}; };
if (surveyId) { if (surveyId) {
@ -63,19 +64,19 @@ export class MessagePushingTaskService {
}); });
} }
findOne({ async findOne({
id, id,
ownerId, ownerId,
}: { }: {
id: string; id: string;
ownerId: string; ownerId: string;
}): Promise<MessagePushingTask> { }): Promise<MessagePushingTask> {
return this.messagePushingTaskRepository.findOne({ return await this.messagePushingTaskRepository.findOne({
where: { where: {
ownerId, ownerId,
_id: new ObjectId(id), _id: new ObjectId(id),
isDeleted: { 'curStatus.status': {
$ne: true, $ne: RECORD_STATUS.REMOVED,
}, },
}, },
}); });
@ -103,25 +104,25 @@ export class MessagePushingTaskService {
return await this.messagePushingTaskRepository.save(updatedTask); return await this.messagePushingTaskRepository.save(updatedTask);
} }
async remove({ async remove({ id, ownerId }: { id: string; ownerId: string }) {
id, const curStatus = {
operator, status: RECORD_STATUS.REMOVED,
operatorId, date: Date.now(),
}: { };
id: string;
operator: string;
operatorId: string;
}) {
return this.messagePushingTaskRepository.updateOne( return this.messagePushingTaskRepository.updateOne(
{ {
ownerId,
_id: new ObjectId(id), _id: new ObjectId(id),
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
{ {
$set: { $set: {
isDeleted: true, curStatus,
operator, },
operatorId, $push: {
deletedAt: new Date(), statusList: curStatus as never,
}, },
}, },
); );
@ -146,9 +147,6 @@ export class MessagePushingTaskService {
$push: { $push: {
surveys: surveyId as never, surveys: surveyId as never,
}, },
$set: {
updatedAt: new Date(),
},
}, },
); );
} }

View File

@ -1,9 +0,0 @@
// src/redis/redis.module.ts
import { Module } from '@nestjs/common';
import { RedisService } from './redis.service';
@Module({
providers: [RedisService],
exports: [RedisService],
})
export class RedisModule {}

View File

@ -1,32 +0,0 @@
import { Injectable } from '@nestjs/common';
import { Redis } from 'ioredis';
import Redlock, { Lock } from 'redlock';
@Injectable()
export class RedisService {
private readonly redisClient: Redis;
private readonly redlock: Redlock;
constructor() {
this.redisClient = new Redis({
host: process.env.XIAOJU_SURVEY_REDIS_HOST,
port: parseInt(process.env.XIAOJU_SURVEY_REDIS_PORT),
password: process.env.XIAOJU_SURVEY_REDIS_PASSWORD || undefined,
username: process.env.XIAOJU_SURVEY_REDIS_USERNAME || undefined,
db: parseInt(process.env.XIAOJU_SURVEY_REDIS_DB) || 0,
});
this.redlock = new Redlock([this.redisClient], {
retryCount: 10,
retryDelay: 200, // ms
retryJitter: 200, // ms
});
}
async lockResource(resource: string, ttl: number): Promise<Lock> {
return this.redlock.acquire([resource], ttl);
}
async unlockResource(lock: Lock): Promise<void> {
await lock.release();
}
}

View File

@ -191,6 +191,7 @@ describe('CollaboratorController', () => {
describe('getSurveyCollaboratorList', () => { describe('getSurveyCollaboratorList', () => {
it('should return collaborator list', async () => { it('should return collaborator list', async () => {
const query = { surveyId: 'surveyId' }; const query = { surveyId: 'surveyId' };
const req = { user: { _id: 'userId' } };
const result = [ const result = [
{ _id: 'collaboratorId', userId: 'userId', username: '' }, { _id: 'collaboratorId', userId: 'userId', username: '' },
]; ];
@ -201,7 +202,7 @@ describe('CollaboratorController', () => {
jest.spyOn(userService, 'getUserListByIds').mockResolvedValueOnce([]); jest.spyOn(userService, 'getUserListByIds').mockResolvedValueOnce([]);
const response = await controller.getSurveyCollaboratorList(query); const response = await controller.getSurveyCollaboratorList(query, req);
expect(response).toEqual({ expect(response).toEqual({
code: 200, code: 200,
@ -213,10 +214,11 @@ describe('CollaboratorController', () => {
const query: GetSurveyCollaboratorListDto = { const query: GetSurveyCollaboratorListDto = {
surveyId: '', surveyId: '',
}; };
const req = { user: { _id: 'userId' } };
await expect(controller.getSurveyCollaboratorList(query)).rejects.toThrow( await expect(
HttpException, controller.getSurveyCollaboratorList(query, req),
); ).rejects.toThrow(HttpException);
expect(logger.error).toHaveBeenCalledTimes(1); expect(logger.error).toHaveBeenCalledTimes(1);
}); });
}); });
@ -228,13 +230,14 @@ describe('CollaboratorController', () => {
userId: 'userId', userId: 'userId',
permissions: ['read'], permissions: ['read'],
}; };
const req = { user: { _id: 'userId' } };
const result = { _id: 'userId', permissions: ['read'] }; const result = { _id: 'userId', permissions: ['read'] };
jest jest
.spyOn(collaboratorService, 'changeUserPermission') .spyOn(collaboratorService, 'changeUserPermission')
.mockResolvedValue(result); .mockResolvedValue(result);
const response = await controller.changeUserPermission(reqBody); const response = await controller.changeUserPermission(reqBody, req);
expect(response).toEqual({ expect(response).toEqual({
code: 200, code: 200,
@ -248,10 +251,11 @@ describe('CollaboratorController', () => {
userId: '', userId: '',
permissions: ['surveyManage'], permissions: ['surveyManage'],
}; };
const req = { user: { _id: 'userId' } };
await expect(controller.changeUserPermission(reqBody)).rejects.toThrow( await expect(
HttpException, controller.changeUserPermission(reqBody, req),
); ).rejects.toThrow(HttpException);
expect(logger.error).toHaveBeenCalledTimes(1); expect(logger.error).toHaveBeenCalledTimes(1);
}); });
}); });
@ -259,13 +263,14 @@ describe('CollaboratorController', () => {
describe('deleteCollaborator', () => { describe('deleteCollaborator', () => {
it('should delete collaborator successfully', async () => { it('should delete collaborator successfully', async () => {
const query = { surveyId: 'surveyId', userId: 'userId' }; const query = { surveyId: 'surveyId', userId: 'userId' };
const req = { user: { _id: 'userId' } };
const result = { acknowledged: true, deletedCount: 1 }; const result = { acknowledged: true, deletedCount: 1 };
jest jest
.spyOn(collaboratorService, 'deleteCollaborator') .spyOn(collaboratorService, 'deleteCollaborator')
.mockResolvedValue(result); .mockResolvedValue(result);
const response = await controller.deleteCollaborator(query); const response = await controller.deleteCollaborator(query, req);
expect(response).toEqual({ expect(response).toEqual({
code: 200, code: 200,
@ -275,8 +280,9 @@ describe('CollaboratorController', () => {
it('should throw an exception if validation fails', async () => { it('should throw an exception if validation fails', async () => {
const query = { surveyId: '', userId: '' }; const query = { surveyId: '', userId: '' };
const req = { user: { _id: 'userId' } };
await expect(controller.deleteCollaborator(query)).rejects.toThrow( await expect(controller.deleteCollaborator(query, req)).rejects.toThrow(
HttpException, HttpException,
); );
expect(logger.error).toHaveBeenCalledTimes(1); expect(logger.error).toHaveBeenCalledTimes(1);

View File

@ -75,25 +75,6 @@ describe('CollaboratorService', () => {
}); });
}); });
describe('deleteCollaborator', () => {
it('should delete a collaborator by userId and surveyId', async () => {
const deleteOneSpy = jest
.spyOn(repository, 'deleteOne')
.mockResolvedValue({ acknowledged: true, deletedCount: 1 });
const result = await service.deleteCollaborator({
userId: '1',
surveyId: '1',
});
expect(deleteOneSpy).toHaveBeenCalledWith({
userId: '1',
surveyId: '1',
});
expect(result).toEqual({ acknowledged: true, deletedCount: 1 });
});
});
describe('batchCreate', () => { describe('batchCreate', () => {
it('should batch create collaborators', async () => { it('should batch create collaborators', async () => {
const insertManySpy = jest const insertManySpy = jest
@ -105,193 +86,15 @@ describe('CollaboratorService', () => {
const result = await service.batchCreate({ const result = await service.batchCreate({
surveyId: '1', surveyId: '1',
collaboratorList: [{ userId: '1', permissions: [] }], collaboratorList: [{ userId: '1', permissions: [] }],
creator: 'testCreator',
creatorId: 'testCreatorId',
}); });
expect(insertManySpy).toHaveBeenCalledWith([ expect(insertManySpy).toHaveBeenCalledWith([
{ { surveyId: '1', userId: '1', permissions: [] },
userId: '1',
permissions: [],
surveyId: '1',
createdAt: expect.any(Date),
updatedAt: expect.any(Date),
creator: 'testCreator',
creatorId: 'testCreatorId',
},
]); ]);
expect(result).toEqual({ insertedCount: 1 }); expect(result).toEqual({ insertedCount: 1 });
}); });
}); });
describe('changeUserPermission', () => {
it("should update a user's permissions", async () => {
const updateOneSpy = jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({});
const result = await service.changeUserPermission({
userId: '1',
surveyId: '1',
permission: 'read',
operator: 'testOperator',
operatorId: 'testOperatorId',
});
expect(updateOneSpy).toHaveBeenCalledWith(
{
surveyId: '1',
userId: '1',
},
{
$set: {
permission: 'read',
operator: 'testOperator',
operatorId: 'testOperatorId',
updatedAt: expect.any(Date),
},
},
);
expect(result).toEqual({});
});
});
describe('batchDelete', () => {
it('should batch delete collaborators', async () => {
const mockResult = { acknowledged: true, deletedCount: 1 };
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue(mockResult);
const collaboratorId = new ObjectId().toString();
const result = await service.batchDelete({
surveyId: '1',
idList: [collaboratorId],
});
const expectedQuery = {
surveyId: '1',
$or: [
{
_id: {
$in: [new ObjectId(collaboratorId)],
},
},
],
};
expect(logger.info).toHaveBeenCalledWith(JSON.stringify(expectedQuery));
expect(deleteManySpy).toHaveBeenCalledWith(expectedQuery);
expect(result).toEqual(mockResult);
});
it('should batch delete collaborators by idList and userIdList', async () => {
const collaboratorId = new ObjectId().toString();
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue({ acknowledged: true, deletedCount: 2 });
const result = await service.batchDelete({
idList: [collaboratorId],
userIdList: ['user1', 'user2'],
surveyId: '1',
});
const expectedQuery = {
surveyId: '1',
$or: [
{
userId: {
$in: ['user1', 'user2'],
},
},
{
_id: {
$in: [new ObjectId(collaboratorId)],
},
},
],
};
expect(deleteManySpy).toHaveBeenCalledWith(expectedQuery);
expect(result).toEqual({ acknowledged: true, deletedCount: 2 });
});
it('should handle batch delete with neIdList only', async () => {
const neCollaboratorId = new ObjectId().toString();
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue({ acknowledged: true, deletedCount: 1 });
const result = await service.batchDelete({
neIdList: [neCollaboratorId],
surveyId: '1',
});
const expectedQuery = {
surveyId: '1',
$or: [
{
_id: {
$nin: [new ObjectId(neCollaboratorId)],
},
},
],
};
expect(deleteManySpy).toHaveBeenCalledWith(expectedQuery);
expect(result).toEqual({ acknowledged: true, deletedCount: 1 });
});
});
describe('batchDeleteBySurveyId', () => {
it('should batch delete collaborators by survey id', async () => {
const mockResult = { acknowledged: true, deletedCount: 1 };
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue(mockResult);
const surveyId = new ObjectId().toString();
const result = await service.batchDeleteBySurveyId(surveyId);
expect(deleteManySpy).toHaveBeenCalledWith({
surveyId,
});
expect(result).toEqual(mockResult);
});
});
describe('updateById', () => {
it('should update collaborator by id', async () => {
const updateOneSpy = jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({});
const collaboratorId = new ObjectId().toString();
const result = await service.updateById({
collaboratorId,
permissions: [],
operator: 'testOperator',
operatorId: 'testOperatorId',
});
expect(updateOneSpy).toHaveBeenCalledWith(
{
_id: new ObjectId(collaboratorId),
},
{
$set: {
permissions: [],
operator: 'testOperator',
operatorId: 'testOperatorId',
updatedAt: expect.any(Date),
},
},
);
expect(result).toEqual({});
});
});
describe('getSurveyCollaboratorList', () => { describe('getSurveyCollaboratorList', () => {
it('should return a list of collaborators for a survey', async () => { it('should return a list of collaborators for a survey', async () => {
const collaboratorId = new ObjectId().toString(); const collaboratorId = new ObjectId().toString();
@ -318,6 +121,38 @@ describe('CollaboratorService', () => {
}); });
}); });
describe('getCollaboratorListByIds', () => {
it('should return a list of collaborators by ids', async () => {
const collaboratorId = new ObjectId().toString();
const findSpy = jest.spyOn(repository, 'find').mockResolvedValue([
{
_id: new ObjectId(collaboratorId),
surveyId: '1',
userId: '1',
permissions: [],
},
] as Collaborator[]);
const result = await service.getCollaboratorListByIds({
idList: [collaboratorId],
});
expect(findSpy).toHaveBeenCalledWith({
_id: {
$in: [new ObjectId(collaboratorId)],
},
});
expect(result).toEqual([
{
_id: new ObjectId(collaboratorId),
surveyId: '1',
userId: '1',
permissions: [],
},
]);
});
});
describe('getCollaborator', () => { describe('getCollaborator', () => {
it('should return a collaborator', async () => { it('should return a collaborator', async () => {
const collaboratorId = new ObjectId().toString(); const collaboratorId = new ObjectId().toString();
@ -348,6 +183,127 @@ describe('CollaboratorService', () => {
}); });
}); });
describe('changeUserPermission', () => {
it("should update a user's permissions", async () => {
const updateOneSpy = jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({});
const result = await service.changeUserPermission({
userId: '1',
surveyId: '1',
permission: 'read',
});
expect(updateOneSpy).toHaveBeenCalledWith(
{
surveyId: '1',
userId: '1',
},
{
$set: {
permission: 'read',
},
},
);
expect(result).toEqual({});
});
});
describe('deleteCollaborator', () => {
it('should delete a collaborator', async () => {
const mockResult = { acknowledged: true, deletedCount: 1 };
const deleteOneSpy = jest
.spyOn(repository, 'deleteOne')
.mockResolvedValue(mockResult);
const result = await service.deleteCollaborator({
userId: '1',
surveyId: '1',
});
expect(deleteOneSpy).toHaveBeenCalledWith({
userId: '1',
surveyId: '1',
});
expect(result).toEqual(mockResult);
});
});
describe('batchDelete', () => {
it('should batch delete collaborators', async () => {
const mockResult = { acknowledged: true, deletedCount: 1 };
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue(mockResult);
const collaboratorId = new ObjectId().toString();
const result = await service.batchDelete({
surveyId: '1',
idList: [collaboratorId],
});
const expectedQuery = {
surveyId: '1',
$or: [
{
_id: {
$in: [new ObjectId(collaboratorId)],
},
},
],
};
expect(logger.info).toHaveBeenCalledWith(JSON.stringify(expectedQuery));
expect(deleteManySpy).toHaveBeenCalledWith(expectedQuery);
expect(result).toEqual(mockResult);
});
});
describe('batchDeleteBySurveyId', () => {
it('should batch delete collaborators by survey id', async () => {
const mockResult = { acknowledged: true, deletedCount: 1 };
const deleteManySpy = jest
.spyOn(repository, 'deleteMany')
.mockResolvedValue(mockResult);
const surveyId = new ObjectId().toString();
const result = await service.batchDeleteBySurveyId(surveyId);
expect(deleteManySpy).toHaveBeenCalledWith({
surveyId,
});
expect(result).toEqual(mockResult);
});
});
describe('updateById', () => {
it('should update collaborator by id', async () => {
const updateOneSpy = jest
.spyOn(repository, 'updateOne')
.mockResolvedValue({});
const collaboratorId = new ObjectId().toString();
const result = await service.updateById({
collaboratorId,
permissions: [],
});
expect(updateOneSpy).toHaveBeenCalledWith(
{
_id: new ObjectId(collaboratorId),
},
{
$set: {
permissions: [],
},
},
);
expect(result).toEqual({});
});
});
describe('getCollaboratorListByUserId', () => { describe('getCollaboratorListByUserId', () => {
it('should return a list of collaborators by user id', async () => { it('should return a list of collaborators by user id', async () => {
const userId = new ObjectId().toString(); const userId = new ObjectId().toString();
@ -371,48 +327,5 @@ describe('CollaboratorService', () => {
{ _id: '1', surveyId: '1', userId, permissions: [] }, { _id: '1', surveyId: '1', userId, permissions: [] },
]); ]);
}); });
it('should return a list of collaborators by their IDs', async () => {
const collaboratorId1 = new ObjectId().toString();
const collaboratorId2 = new ObjectId().toString();
const findSpy = jest.spyOn(repository, 'find').mockResolvedValue([
{
_id: new ObjectId(collaboratorId1),
surveyId: '1',
userId: 'user1',
permissions: [],
},
{
_id: new ObjectId(collaboratorId2),
surveyId: '2',
userId: 'user2',
permissions: [],
},
] as Collaborator[]);
const result = await service.getCollaboratorListByIds({
idList: [collaboratorId1, collaboratorId2],
});
expect(findSpy).toHaveBeenCalledWith({
_id: {
$in: [new ObjectId(collaboratorId1), new ObjectId(collaboratorId2)],
},
});
expect(result).toEqual([
{
_id: new ObjectId(collaboratorId1),
surveyId: '1',
userId: 'user1',
permissions: [],
},
{
_id: new ObjectId(collaboratorId2),
surveyId: '2',
userId: 'user2',
permissions: [],
},
]);
});
}); });
}); });

View File

@ -8,7 +8,7 @@ import { SurveyMetaService } from '../services/surveyMeta.service';
import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service'; import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service';
import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider'; import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider';
import { PluginManager } from 'src/securityPlugin/pluginManager'; import { XiaojuSurveyPluginManager } from 'src/securityPlugin/pluginManager';
import { Logger } from 'src/logger'; import { Logger } from 'src/logger';
import { UserService } from 'src/modules/auth/services/user.service'; import { UserService } from 'src/modules/auth/services/user.service';
@ -27,7 +27,7 @@ describe('DataStatisticController', () => {
let controller: DataStatisticController; let controller: DataStatisticController;
let dataStatisticService: DataStatisticService; let dataStatisticService: DataStatisticService;
let responseSchemaService: ResponseSchemaService; let responseSchemaService: ResponseSchemaService;
let pluginManager: PluginManager; let pluginManager: XiaojuSurveyPluginManager;
let logger: Logger; let logger: Logger;
beforeEach(async () => { beforeEach(async () => {
@ -70,7 +70,9 @@ describe('DataStatisticController', () => {
responseSchemaService = module.get<ResponseSchemaService>( responseSchemaService = module.get<ResponseSchemaService>(
ResponseSchemaService, ResponseSchemaService,
); );
pluginManager = module.get<PluginManager>(PluginManager); pluginManager = module.get<XiaojuSurveyPluginManager>(
XiaojuSurveyPluginManager,
);
logger = module.get<Logger>(Logger); logger = module.get<Logger>(Logger);
pluginManager.registerPlugin( pluginManager.registerPlugin(
@ -88,7 +90,7 @@ describe('DataStatisticController', () => {
const mockRequest = { const mockRequest = {
query: { query: {
surveyId, surveyId,
isMasked: false, isDesensitive: false,
page: 1, page: 1,
pageSize: 10, pageSize: 10,
}, },
@ -104,13 +106,12 @@ describe('DataStatisticController', () => {
field: 'xxx', field: 'xxx',
title: 'xxx', title: 'xxx',
type: 'xxx', type: 'xxx',
diffTime: 'xxx',
othersCode: 'xxx', othersCode: 'xxx',
}, },
], ],
listBody: [ listBody: [
{ diffTime: '0.5', createdAt: '2024-02-11' }, { difTime: '0.5', createDate: '2024-02-11' },
{ diffTime: '0.5', createdAt: '2024-02-11' }, { difTime: '0.5', createDate: '2024-02-11' },
], ],
}; };
@ -121,7 +122,7 @@ describe('DataStatisticController', () => {
.spyOn(dataStatisticService, 'getDataTable') .spyOn(dataStatisticService, 'getDataTable')
.mockResolvedValueOnce(mockDataTable); .mockResolvedValueOnce(mockDataTable);
const result = await controller.data(mockRequest.query); const result = await controller.data(mockRequest.query, mockRequest);
expect(result).toEqual({ expect(result).toEqual({
code: 200, code: 200,
@ -129,12 +130,12 @@ describe('DataStatisticController', () => {
}); });
}); });
it('should return data table with isMasked', async () => { it('should return data table with isDesensitive', async () => {
const surveyId = new ObjectId().toString(); const surveyId = new ObjectId().toString();
const mockRequest = { const mockRequest = {
query: { query: {
surveyId, surveyId,
isMasked: true, isDesensitive: true,
page: 1, page: 1,
pageSize: 10, pageSize: 10,
}, },
@ -150,13 +151,12 @@ describe('DataStatisticController', () => {
field: 'xxx', field: 'xxx',
title: 'xxx', title: 'xxx',
type: 'xxx', type: 'xxx',
diffTime: 'xxx',
othersCode: 'xxx', othersCode: 'xxx',
}, },
], ],
listBody: [ listBody: [
{ diffTime: '0.5', createdAt: '2024-02-11', data123: '15200000000' }, { difTime: '0.5', createDate: '2024-02-11', data123: '15200000000' },
{ diffTime: '0.5', createdAt: '2024-02-11', data123: '13800000000' }, { difTime: '0.5', createDate: '2024-02-11', data123: '13800000000' },
], ],
}; };
@ -167,7 +167,7 @@ describe('DataStatisticController', () => {
.spyOn(dataStatisticService, 'getDataTable') .spyOn(dataStatisticService, 'getDataTable')
.mockResolvedValueOnce(mockDataTable); .mockResolvedValueOnce(mockDataTable);
const result = await controller.data(mockRequest.query); const result = await controller.data(mockRequest.query, mockRequest);
expect(result).toEqual({ expect(result).toEqual({
code: 200, code: 200,
@ -185,9 +185,9 @@ describe('DataStatisticController', () => {
}, },
}; };
await expect(controller.data(mockRequest.query)).rejects.toThrow( await expect(
HttpException, controller.data(mockRequest.query, mockRequest),
); ).rejects.toThrow(HttpException);
expect(logger.error).toHaveBeenCalledTimes(1); expect(logger.error).toHaveBeenCalledTimes(1);
}); });
}); });
@ -212,8 +212,8 @@ describe('DataStatisticController', () => {
date: 1717158851823, date: 1717158851823,
}, },
], ],
createdAt: 1717158851823, createDate: 1717158851823,
updatedAt: 1717159136025, updateDate: 1717159136025,
title: '问卷调研', title: '问卷调研',
surveyPath: 'ZdGNzTTR', surveyPath: 'ZdGNzTTR',
code: { code: {
@ -233,7 +233,7 @@ describe('DataStatisticController', () => {
}, },
}, },
baseConf: { baseConf: {
beginTime: '2024-05-31 20:31:36', begTime: '2024-05-31 20:31:36',
endTime: '2034-05-31 20:31:36', endTime: '2034-05-31 20:31:36',
language: 'chinese', language: 'chinese',
showVoteProcess: 'allow', showVoteProcess: 'allow',
@ -249,8 +249,6 @@ describe('DataStatisticController', () => {
skinConf: { skinConf: {
backgroundConf: { backgroundConf: {
color: '#fff', color: '#fff',
type: 'color',
image: '',
}, },
themeConf: { themeConf: {
color: '#ffa600', color: '#ffa600',

View File

@ -11,7 +11,7 @@ import { cloneDeep } from 'lodash';
import { getRepositoryToken } from '@nestjs/typeorm'; import { getRepositoryToken } from '@nestjs/typeorm';
import { RECORD_STATUS } from 'src/enums'; import { RECORD_STATUS } from 'src/enums';
import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider'; import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider';
import { PluginManager } from 'src/securityPlugin/pluginManager'; import { XiaojuSurveyPluginManager } from 'src/securityPlugin/pluginManager';
import { ResponseSecurityPlugin } from 'src/securityPlugin/responseSecurityPlugin'; import { ResponseSecurityPlugin } from 'src/securityPlugin/responseSecurityPlugin';
describe('DataStatisticService', () => { describe('DataStatisticService', () => {
@ -34,7 +34,9 @@ describe('DataStatisticService', () => {
surveyResponseRepository = module.get<MongoRepository<SurveyResponse>>( surveyResponseRepository = module.get<MongoRepository<SurveyResponse>>(
getRepositoryToken(SurveyResponse), getRepositoryToken(SurveyResponse),
); );
const manager = module.get<PluginManager>(PluginManager); const manager = module.get<XiaojuSurveyPluginManager>(
XiaojuSurveyPluginManager,
);
manager.registerPlugin( manager.registerPlugin(
new ResponseSecurityPlugin('dataAesEncryptSecretKey'), new ResponseSecurityPlugin('dataAesEncryptSecretKey'),
); );
@ -68,7 +70,7 @@ describe('DataStatisticService', () => {
data413: 3, data413: 3,
data863: '109239', data863: '109239',
}, },
diffTime: 21278, difTime: 21278,
clientTime: 1710340862733.0, clientTime: 1710340862733.0,
secretKeys: [], secretKeys: [],
optionTextAndId: { optionTextAndId: {
@ -151,8 +153,8 @@ describe('DataStatisticService', () => {
date: 1710340863123.0, date: 1710340863123.0,
}, },
], ],
createdAt: 1710340863123.0, createDate: 1710340863123.0,
updatedAt: 1710340863123.0, updateDate: 1710340863123.0,
}, },
] as unknown as Array<SurveyResponse>; ] as unknown as Array<SurveyResponse>;
@ -195,14 +197,14 @@ describe('DataStatisticService', () => {
data413_3: expect.any(String), data413_3: expect.any(String),
data413: expect.any(Number), data413: expect.any(Number),
data863: expect.any(String), data863: expect.any(String),
diffTime: expect.any(String), difTime: expect.any(String),
createdAt: expect.any(String), createDate: expect.any(String),
}), }),
]), ]),
}); });
}); });
it('should return desensitized table data', async () => { it('should return desensitive table data', async () => {
const mockSchema = cloneDeep(mockSensitiveResponseSchema); const mockSchema = cloneDeep(mockSensitiveResponseSchema);
const surveyResponseList: Array<SurveyResponse> = [ const surveyResponseList: Array<SurveyResponse> = [
{ {
@ -218,7 +220,7 @@ describe('DataStatisticService', () => {
'U2FsdGVkX19bRmf3uEmXAJ/6zXd1Znr3cZsD5v4Nocr2v5XG1taXluz8cohFkDyH', 'U2FsdGVkX19bRmf3uEmXAJ/6zXd1Znr3cZsD5v4Nocr2v5XG1taXluz8cohFkDyH',
data770: 'U2FsdGVkX18ldQMhJjFXO8aerjftZLpFnRQ4/FVcCLI=', data770: 'U2FsdGVkX18ldQMhJjFXO8aerjftZLpFnRQ4/FVcCLI=',
}, },
diffTime: 806707, difTime: 806707,
clientTime: 1710400229573.0, clientTime: 1710400229573.0,
secretKeys: ['data458', 'data450', 'data405', 'data770'], secretKeys: ['data458', 'data450', 'data405', 'data770'],
optionTextAndId: { optionTextAndId: {
@ -273,8 +275,8 @@ describe('DataStatisticService', () => {
date: 1710400232161.0, date: 1710400232161.0,
}, },
], ],
createdAt: 1710400232161.0, createDate: 1710400232161.0,
updatedAt: 1710400232161.0, updateDate: 1710400232161.0,
}, },
] as unknown as Array<SurveyResponse>; ] as unknown as Array<SurveyResponse>;
@ -295,13 +297,13 @@ describe('DataStatisticService', () => {
expect(result.listBody).toEqual( expect(result.listBody).toEqual(
expect.arrayContaining([ expect.arrayContaining([
expect.objectContaining({ expect.objectContaining({
createdAt: expect.any(String), createDate: expect.any(String),
data405: expect.any(String), data405: expect.any(String),
data450: expect.any(String), data450: expect.any(String),
data458: expect.any(String), data458: expect.any(String),
data515: expect.any(String), data515: expect.any(String),
data770: expect.any(String), data770: expect.any(String),
diffTime: expect.any(String), difTime: expect.any(String),
}), }),
]), ]),
); );

View File

@ -1,264 +0,0 @@
import { Test, TestingModule } from '@nestjs/testing';
import { ObjectId } from 'mongodb';
import { DownloadTaskController } from '../controllers/downloadTask.controller';
import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service';
import { AuthService } from 'src/modules/auth/services/auth.service';
import { DownloadTaskService } from '../services/downloadTask.service';
import { CollaboratorService } from '../services/collaborator.service';
import { SurveyMetaService } from '../services/surveyMeta.service';
import { WorkspaceMemberService } from 'src/modules/workspace/services/workspaceMember.service';
import { Logger } from 'src/logger';
import { HttpException } from 'src/exceptions/httpException';
import { NoPermissionException } from 'src/exceptions/noPermissionException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { Authentication } from 'src/guards/authentication.guard';
import { SurveyGuard } from 'src/guards/survey.guard';
describe('DownloadTaskController', () => {
let controller: DownloadTaskController;
let responseSchemaService: ResponseSchemaService;
let downloadTaskService: DownloadTaskService;
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
controllers: [DownloadTaskController],
providers: [
{
provide: ResponseSchemaService,
useValue: {
getResponseSchemaByPageId: jest.fn(),
},
},
{
provide: DownloadTaskService,
useValue: {
createDownloadTask: jest.fn(),
processDownloadTask: jest.fn(),
getDownloadTaskList: jest.fn(),
getDownloadTaskById: jest.fn(),
deleteDownloadTask: jest.fn(),
},
},
{
provide: Logger,
useValue: {
error: jest.fn(),
},
},
{
provide: AuthService,
useClass: jest.fn().mockImplementation(() => ({
varifytoken() {
return {};
},
})),
},
{
provide: CollaboratorService,
useValue: {},
},
{
provide: SurveyMetaService,
useValue: {},
},
{
provide: WorkspaceMemberService,
useValue: {},
},
{
provide: Authentication,
useClass: jest.fn().mockImplementation(() => ({
canActivate: () => true,
})),
},
{
provide: SurveyGuard,
useClass: jest.fn().mockImplementation(() => ({
canActivate: () => true,
})),
},
],
}).compile();
controller = module.get<DownloadTaskController>(DownloadTaskController);
responseSchemaService = module.get<ResponseSchemaService>(
ResponseSchemaService,
);
downloadTaskService = module.get<DownloadTaskService>(DownloadTaskService);
});
describe('createTask', () => {
it('should create a download task successfully', async () => {
const mockReqBody = {
surveyId: new ObjectId().toString(),
isMasked: false,
};
const mockReq = { user: { _id: 'mockUserId', username: 'mockUsername' } };
const mockTaskId = 'mockTaskId';
jest
.spyOn(responseSchemaService, 'getResponseSchemaByPageId')
.mockResolvedValue({} as any);
jest
.spyOn(downloadTaskService, 'createDownloadTask')
.mockResolvedValue(mockTaskId);
const result = await controller.createTask(mockReqBody, mockReq);
expect(
responseSchemaService.getResponseSchemaByPageId,
).toHaveBeenCalledWith(mockReqBody.surveyId);
expect(downloadTaskService.createDownloadTask).toHaveBeenCalledWith({
surveyId: mockReqBody.surveyId,
responseSchema: {},
creatorId: mockReq.user._id.toString(),
creator: mockReq.user.username,
params: { isMasked: mockReqBody.isMasked },
});
expect(downloadTaskService.processDownloadTask).toHaveBeenCalledWith({
taskId: mockTaskId,
});
expect(result).toEqual({ code: 200, data: { taskId: mockTaskId } });
});
it('should throw HttpException if validation fails', async () => {
const mockReqBody: any = { isMasked: false };
const mockReq = { user: { _id: 'mockUserId', username: 'mockUsername' } };
await expect(controller.createTask(mockReqBody, mockReq)).rejects.toThrow(
HttpException,
);
});
});
describe('downloadList', () => {
it('should return the download task list', async () => {
const mockQueryInfo = { pageIndex: 1, pageSize: 10 };
const mockReq = { user: { _id: 'mockUserId' } };
const mockTaskList: any = {
total: 1,
list: [
{
_id: 'mockTaskId',
curStatus: 'completed',
filename: 'mockFile.csv',
url: 'http://mock-url.com',
fileSize: 1024,
createdAt: Date.now(),
},
],
};
jest
.spyOn(downloadTaskService, 'getDownloadTaskList')
.mockResolvedValue(mockTaskList);
const result = await controller.downloadList(mockQueryInfo, mockReq);
expect(downloadTaskService.getDownloadTaskList).toHaveBeenCalledWith({
creatorId: mockReq.user._id.toString(),
pageIndex: mockQueryInfo.pageIndex,
pageSize: mockQueryInfo.pageSize,
});
expect(result.data.total).toEqual(mockTaskList.total);
expect(result.data.list[0].taskId).toEqual(
mockTaskList.list[0]._id.toString(),
);
});
it('should throw HttpException if validation fails', async () => {
const mockQueryInfo: any = { pageIndex: 'invalid', pageSize: 10 };
const mockReq = { user: { _id: 'mockUserId' } };
await expect(
controller.downloadList(mockQueryInfo, mockReq),
).rejects.toThrow(HttpException);
});
});
describe('getDownloadTask', () => {
it('should return a download task', async () => {
const mockQuery = { taskId: 'mockTaskId' };
const mockReq = { user: { _id: 'mockUserId' } };
const mockTaskInfo: any = {
_id: 'mockTaskId',
creatorId: 'mockUserId',
curStatus: 'completed',
};
jest
.spyOn(downloadTaskService, 'getDownloadTaskById')
.mockResolvedValue(mockTaskInfo);
const result = await controller.getDownloadTask(mockQuery, mockReq);
expect(downloadTaskService.getDownloadTaskById).toHaveBeenCalledWith({
taskId: mockQuery.taskId,
});
expect(result.data.taskId).toEqual(mockTaskInfo._id.toString());
});
it('should throw NoPermissionException if user has no permission', async () => {
const mockQuery = { taskId: 'mockTaskId' };
const mockReq = { user: { _id: new ObjectId() } };
const mockTaskInfo: any = {
_id: 'mockTaskId',
creatorId: 'mockUserId',
curStatus: 'completed',
};
jest
.spyOn(downloadTaskService, 'getDownloadTaskById')
.mockResolvedValue(mockTaskInfo);
await expect(
controller.getDownloadTask(mockQuery, mockReq),
).rejects.toThrow(new NoPermissionException('没有权限'));
});
});
describe('deleteFileByName', () => {
it('should delete a download task successfully', async () => {
const mockBody = { taskId: 'mockTaskId' };
const mockUserId = new ObjectId();
const mockReq = {
user: { _id: mockUserId, username: 'mockUsername' },
};
const mockTaskInfo: any = {
_id: new ObjectId(),
creatorId: mockUserId.toString(),
};
const mockDelRes = { modifiedCount: 1 };
jest
.spyOn(downloadTaskService, 'getDownloadTaskById')
.mockResolvedValue(mockTaskInfo);
jest
.spyOn(downloadTaskService, 'deleteDownloadTask')
.mockResolvedValue(mockDelRes);
const result = await controller.deleteFileByName(mockBody, mockReq);
expect(downloadTaskService.deleteDownloadTask).toHaveBeenCalledWith({
taskId: mockBody.taskId,
operator: mockReq.user.username,
operatorId: mockReq.user._id.toString(),
});
expect(result).toEqual({ code: 200, data: true });
});
it('should throw HttpException if task does not exist', async () => {
const mockBody = { taskId: 'mockTaskId' };
const mockReq = { user: { _id: 'mockUserId' } };
jest
.spyOn(downloadTaskService, 'getDownloadTaskById')
.mockResolvedValue(null);
await expect(
controller.deleteFileByName(mockBody, mockReq),
).rejects.toThrow(
new HttpException('任务不存在', EXCEPTION_CODE.PARAMETER_ERROR),
);
});
});
});

View File

@ -1,245 +0,0 @@
import { Test, TestingModule } from '@nestjs/testing';
import { DownloadTaskService } from '../services/downloadTask.service';
import { MongoRepository } from 'typeorm';
import { getRepositoryToken } from '@nestjs/typeorm';
import { DownloadTask } from 'src/models/downloadTask.entity';
import { SurveyResponse } from 'src/models/surveyResponse.entity';
import { ResponseSchemaService } from 'src/modules/surveyResponse/services/responseScheme.service';
import { DataStatisticService } from '../services/dataStatistic.service';
import { FileService } from 'src/modules/file/services/file.service';
import { Logger } from 'src/logger';
import { ObjectId } from 'mongodb';
import { DOWNLOAD_TASK_STATUS } from 'src/enums/downloadTaskStatus';
describe('DownloadTaskService', () => {
let service: DownloadTaskService;
let downloadTaskRepository: MongoRepository<DownloadTask>;
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
providers: [
DownloadTaskService,
{
provide: getRepositoryToken(DownloadTask),
useClass: MongoRepository,
},
{
provide: getRepositoryToken(SurveyResponse),
useClass: MongoRepository,
},
{
provide: ResponseSchemaService,
useValue: {
getResponseSchemaByPageId: jest.fn(),
},
},
{
provide: DataStatisticService,
useValue: {
getDataTable: jest.fn(),
},
},
{
provide: FileService,
useValue: {
upload: jest.fn(),
},
},
{
provide: Logger,
useValue: {
info: jest.fn(),
error: jest.fn(),
},
},
],
}).compile();
service = module.get<DownloadTaskService>(DownloadTaskService);
downloadTaskRepository = module.get<MongoRepository<DownloadTask>>(
getRepositoryToken(DownloadTask),
);
});
describe('createDownloadTask', () => {
it('should create and save a download task', async () => {
const mockTaskId = new ObjectId().toString();
const mockDownloadTask = { _id: new ObjectId(mockTaskId) };
const mockParams: any = {
surveyId: 'survey1',
responseSchema: { title: 'test-title', surveyPath: '/path' },
creatorId: 'creator1',
creator: 'creatorName',
params: { isMasked: true },
};
jest
.spyOn(downloadTaskRepository, 'create')
.mockReturnValue(mockDownloadTask as any);
jest
.spyOn(downloadTaskRepository, 'save')
.mockResolvedValue(mockDownloadTask as any);
const result = await service.createDownloadTask(mockParams);
expect(downloadTaskRepository.create).toHaveBeenCalledWith({
surveyId: mockParams.surveyId,
surveyPath: mockParams.responseSchema.surveyPath,
fileSize: '计算中',
creatorId: mockParams.creatorId,
creator: mockParams.creator,
params: {
...mockParams.params,
title: mockParams.responseSchema.title,
},
filename: expect.any(String),
status: DOWNLOAD_TASK_STATUS.WAITING,
});
expect(downloadTaskRepository.save).toHaveBeenCalled();
expect(result).toEqual(mockTaskId);
});
});
describe('getDownloadTaskList', () => {
it('should return task list and total count', async () => {
const mockCreatorId = 'creator1';
const mockTasks = [{ _id: '1' }, { _id: '2' }];
const mockTotal = 2;
jest
.spyOn(downloadTaskRepository, 'findAndCount')
.mockResolvedValue([mockTasks as any, mockTotal]);
const result = await service.getDownloadTaskList({
creatorId: mockCreatorId,
pageIndex: 1,
pageSize: 10,
});
expect(downloadTaskRepository.findAndCount).toHaveBeenCalledWith({
where: {
creatorId: mockCreatorId,
isDeleted: { $ne: true },
},
take: 10,
skip: 0,
order: { createdAt: -1 },
});
expect(result).toEqual({
total: mockTotal,
list: mockTasks,
});
});
});
describe('getDownloadTaskById', () => {
it('should return task by id', async () => {
const mockTaskId = new ObjectId().toString();
const mockTask = { _id: new ObjectId(mockTaskId) };
jest
.spyOn(downloadTaskRepository, 'find')
.mockResolvedValue([mockTask as any]);
const result = await service.getDownloadTaskById({ taskId: mockTaskId });
expect(downloadTaskRepository.find).toHaveBeenCalledWith({
where: { _id: new ObjectId(mockTaskId) },
});
expect(result).toEqual(mockTask);
});
it('should return null if task is not found', async () => {
const mockTaskId = new ObjectId().toString();
jest.spyOn(downloadTaskRepository, 'find').mockResolvedValue([]);
const result = await service.getDownloadTaskById({ taskId: mockTaskId });
expect(result).toBeNull();
});
});
describe('deleteDownloadTask', () => {
it('should mark task as deleted and set deletedAt', async () => {
const mockTaskId = new ObjectId().toString();
const mockOperator = 'operatorName';
const mockOperatorId = 'operatorId1';
const mockUpdateResult = { matchedCount: 1 };
jest
.spyOn(downloadTaskRepository, 'updateOne')
.mockResolvedValue(mockUpdateResult as any);
const result = await service.deleteDownloadTask({
taskId: mockTaskId,
operator: mockOperator,
operatorId: mockOperatorId,
});
expect(downloadTaskRepository.updateOne).toHaveBeenCalledWith(
{ _id: new ObjectId(mockTaskId) },
{
$set: {
isDeleted: true,
operator: mockOperator,
operatorId: mockOperatorId,
deletedAt: expect.any(Date),
},
},
);
expect(result).toEqual(mockUpdateResult);
});
});
describe('processDownloadTask', () => {
it('should push task to queue and execute if not executing', async () => {
const mockTaskId = new ObjectId().toString();
jest.spyOn(service, 'executeTask').mockImplementation(jest.fn());
service.processDownloadTask({ taskId: mockTaskId });
expect(DownloadTaskService.taskList).toContain(mockTaskId);
expect(service.executeTask).toHaveBeenCalled();
});
it('should handle already executing case', async () => {
const mockTaskId = new ObjectId().toString();
DownloadTaskService.isExecuting = true;
jest.spyOn(service, 'executeTask').mockImplementation(jest.fn());
service.processDownloadTask({ taskId: mockTaskId });
expect(DownloadTaskService.taskList).toContain(mockTaskId);
expect(service.executeTask).not.toHaveBeenCalled();
});
});
describe('executeTask', () => {
it('should process and execute tasks in queue', async () => {
const mockTaskId = new ObjectId().toString();
DownloadTaskService.taskList.push(mockTaskId);
jest.spyOn(service, 'getDownloadTaskById').mockResolvedValue({
_id: new ObjectId(mockTaskId),
isDeleted: false,
} as any);
jest.spyOn(service, 'handleDownloadTask').mockResolvedValue(undefined);
await service.executeTask();
expect(service.getDownloadTaskById).toHaveBeenCalledWith({
taskId: mockTaskId,
});
expect(service.handleDownloadTask).toHaveBeenCalled();
});
it('should stop executing when queue is empty', async () => {
DownloadTaskService.taskList = [];
await service.executeTask();
expect(DownloadTaskService.isExecuting).toBe(false);
});
});
});

View File

@ -14,8 +14,8 @@ export const mockSensitiveResponseSchema: ResponseSchema = {
date: 1710399368439, date: 1710399368439,
}, },
], ],
createdAt: 1710399368440, createDate: 1710399368440,
updatedAt: 1710399368440, updateDate: 1710399368440,
title: '加密全流程', title: '加密全流程',
surveyPath: 'EBzdmnSp', surveyPath: 'EBzdmnSp',
code: { code: {
@ -32,7 +32,7 @@ export const mockSensitiveResponseSchema: ResponseSchema = {
}, },
}, },
baseConf: { baseConf: {
beginTime: '2024-03-14 14:54:41', begTime: '2024-03-14 14:54:41',
endTime: '2034-03-14 14:54:41', endTime: '2034-03-14 14:54:41',
language: 'chinese', language: 'chinese',
tLimit: 0, tLimit: 0,
@ -44,17 +44,6 @@ export const mockSensitiveResponseSchema: ResponseSchema = {
logoImageWidth: '60%', logoImageWidth: '60%',
}, },
skinConf: { skinConf: {
backgroundConf: {
color: '#fff',
type: 'color',
image: '',
},
themeConf: {
color: '#ffa600',
},
contentConf: {
opacity: 100,
},
skinColor: '#4a4c5b', skinColor: '#4a4c5b',
inputBgColor: '#ffffff', inputBgColor: '#ffffff',
}, },
@ -295,7 +284,7 @@ export const mockSensitiveResponseSchema: ResponseSchema = {
}, },
}, },
pageId: '65f29f3192862d6a9067ad1c', pageId: '65f29f3192862d6a9067ad1c',
} as unknown as ResponseSchema; } as ResponseSchema;
export const mockResponseSchema: ResponseSchema = { export const mockResponseSchema: ResponseSchema = {
_id: new ObjectId('65b0d46e04d5db18534c0f7c'), _id: new ObjectId('65b0d46e04d5db18534c0f7c'),
@ -326,7 +315,7 @@ export const mockResponseSchema: ResponseSchema = {
}, },
}, },
baseConf: { baseConf: {
beginTime: '2024-01-23 21:59:05', begTime: '2024-01-23 21:59:05',
endTime: '2034-01-23 21:59:05', endTime: '2034-01-23 21:59:05',
language: 'chinese', language: 'chinese',
tLimit: 0, tLimit: 0,
@ -338,17 +327,6 @@ export const mockResponseSchema: ResponseSchema = {
logoImageWidth: '60%', logoImageWidth: '60%',
}, },
skinConf: { skinConf: {
backgroundConf: {
color: '#fff',
type: 'color',
image: '',
},
themeConf: {
color: '#ffa600',
},
contentConf: {
opacity: 100,
},
skinColor: '#4a4c5b', skinColor: '#4a4c5b',
inputBgColor: '#ffffff', inputBgColor: '#ffffff',
}, },
@ -654,6 +632,6 @@ export const mockResponseSchema: ResponseSchema = {
}, },
}, },
pageId: '65afc62904d5db18534c0f78', pageId: '65afc62904d5db18534c0f78',
createdAt: 1710340841289, createDate: 1710340841289,
updatedAt: 1710340841289.0, updateDate: 1710340841289.0,
} as unknown as ResponseSchema; } as ResponseSchema;

View File

@ -1,87 +0,0 @@
import { Test, TestingModule } from '@nestjs/testing';
import { SessionController } from '../controllers/session.controller';
import { SessionService } from '../services/session.service';
import { Logger } from 'src/logger';
import { HttpException } from 'src/exceptions/httpException';
import { Authentication } from 'src/guards/authentication.guard';
import { SurveyGuard } from 'src/guards/survey.guard';
import { SessionGuard } from 'src/guards/session.guard';
describe('SessionController', () => {
let controller: SessionController;
let sessionService: jest.Mocked<SessionService>;
let logger: jest.Mocked<Logger>;
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
controllers: [SessionController],
providers: [
{
provide: SessionService,
useValue: {
create: jest.fn(),
updateSessionToEditing: jest.fn(),
},
},
{
provide: Logger,
useValue: {
error: jest.fn(),
},
},
],
})
.overrideGuard(Authentication)
.useValue({ canActivate: () => true })
.overrideGuard(SurveyGuard)
.useValue({ canActivate: () => true })
.overrideGuard(SessionGuard)
.useValue({ canActivate: () => true })
.compile();
controller = module.get<SessionController>(SessionController);
sessionService = module.get<jest.Mocked<SessionService>>(SessionService);
logger = module.get<jest.Mocked<Logger>>(Logger);
});
it('should create a session', async () => {
const reqBody = { surveyId: '123' };
const req = { user: { _id: 'userId' } };
const session: any = { _id: 'sessionId' };
sessionService.create.mockResolvedValue(session);
const result = await controller.create(reqBody, req);
expect(sessionService.create).toHaveBeenCalledWith({
surveyId: '123',
userId: 'userId',
});
expect(result).toEqual({ code: 200, data: { sessionId: 'sessionId' } });
});
it('should throw an exception if validation fails', async () => {
const reqBody = { surveyId: null };
const req = { user: { _id: 'userId' } };
try {
await controller.create(reqBody, req);
} catch (error) {
expect(error).toBeInstanceOf(HttpException);
expect(logger.error).toHaveBeenCalled();
}
});
it('should seize a session', async () => {
const req = {
sessionInfo: { _id: 'sessionId', surveyId: 'surveyId' },
};
await controller.seize(req);
expect(sessionService.updateSessionToEditing).toHaveBeenCalledWith({
sessionId: 'sessionId',
surveyId: 'surveyId',
});
});
});

View File

@ -1,144 +0,0 @@
import { Test, TestingModule } from '@nestjs/testing';
import { MongoRepository } from 'typeorm';
import { getRepositoryToken } from '@nestjs/typeorm';
import { SessionService } from '../services/session.service';
import { Session } from 'src/models/session.entity';
import { ObjectId } from 'mongodb';
import { SESSION_STATUS } from 'src/enums/surveySessionStatus';
describe('SessionService', () => {
let service: SessionService;
let repository: MongoRepository<Session>;
beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({
providers: [
SessionService,
{
provide: getRepositoryToken(Session),
useClass: MongoRepository,
},
],
}).compile();
service = module.get<SessionService>(SessionService);
repository = module.get<MongoRepository<Session>>(
getRepositoryToken(Session),
);
});
afterEach(() => {
jest.clearAllMocks();
});
describe('create', () => {
it('should create and save a new session', async () => {
const mockSession = {
surveyId: 'survey123',
userId: 'user123',
status: SESSION_STATUS.DEACTIVATED,
};
const createdSession: any = { ...mockSession, _id: new ObjectId() };
jest.spyOn(repository, 'create').mockReturnValue(createdSession);
jest.spyOn(repository, 'save').mockResolvedValue(createdSession);
const result = await service.create({
surveyId: mockSession.surveyId,
userId: mockSession.userId,
});
expect(result).toEqual(createdSession);
expect(repository.create).toHaveBeenCalledWith(mockSession);
expect(repository.save).toHaveBeenCalledWith(createdSession);
});
});
describe('findOne', () => {
it('should find a session by id', async () => {
const sessionId = '65afc62904d5db18534c0f78';
const foundSession = {
_id: new ObjectId(sessionId),
surveyId: 'survey123',
userId: 'user123',
status: SESSION_STATUS.ACTIVATED,
};
jest
.spyOn(repository, 'findOne')
.mockResolvedValue(foundSession as Session);
const result = await service.findOne(sessionId);
expect(result).toEqual(foundSession);
expect(repository.findOne).toHaveBeenCalledWith({
where: { _id: new ObjectId(sessionId) },
});
});
});
describe('findLatestEditingOne', () => {
it('should find the latest editing session for a survey', async () => {
const surveyId = 'survey123';
const latestSession = {
_id: new ObjectId(),
surveyId: surveyId,
userId: 'user123',
status: SESSION_STATUS.ACTIVATED,
};
jest
.spyOn(repository, 'findOne')
.mockResolvedValue(latestSession as Session);
const result = await service.findLatestEditingOne({ surveyId });
expect(result).toEqual(latestSession);
expect(repository.findOne).toHaveBeenCalledWith({
where: {
surveyId,
status: SESSION_STATUS.ACTIVATED,
},
});
});
});
describe('updateSessionToEditing', () => {
it('should update a session to editing and deactivate other sessions', async () => {
const sessionId = '65afc62904d5db18534c0f78';
const surveyId = 'survey123';
const updateResult: any = { affected: 1 };
const updateManyResult = { modifiedCount: 1 };
jest.spyOn(repository, 'update').mockResolvedValue(updateResult);
jest.spyOn(repository, 'updateMany').mockResolvedValue(updateManyResult);
const result = await service.updateSessionToEditing({
sessionId,
surveyId,
});
expect(result).toEqual([updateResult, updateManyResult]);
expect(repository.update).toHaveBeenCalledWith(
{ _id: new ObjectId(sessionId) },
{
status: SESSION_STATUS.ACTIVATED,
updatedAt: expect.any(Date),
},
);
expect(repository.updateMany).toHaveBeenCalledWith(
{
surveyId,
_id: { $ne: new ObjectId(sessionId) },
},
{
$set: {
status: SESSION_STATUS.DEACTIVATED,
updatedAt: expect.any(Date),
},
},
);
});
});
});

View File

@ -5,20 +5,19 @@ import { SurveyConfService } from '../services/surveyConf.service';
import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service'; import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service';
import { ContentSecurityService } from '../services/contentSecurity.service'; import { ContentSecurityService } from '../services/contentSecurity.service';
import { SurveyHistoryService } from '../services/surveyHistory.service'; import { SurveyHistoryService } from '../services/surveyHistory.service';
import { SessionService } from '../services/session.service';
import { UserService } from '../../auth/services/user.service';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
import { Logger } from 'src/logger'; import { SurveyMeta } from 'src/models/surveyMeta.entity';
import { SurveyConf } from 'src/models/surveyConf.entity';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { Authentication } from 'src/guards/authentication.guard'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { LoggerProvider } from 'src/logger/logger.provider';
jest.mock('../services/surveyMeta.service'); jest.mock('../services/surveyMeta.service');
jest.mock('../services/surveyConf.service'); jest.mock('../services/surveyConf.service');
jest.mock('../../surveyResponse/services/responseScheme.service'); jest.mock('../../surveyResponse/services/responseScheme.service');
jest.mock('../services/contentSecurity.service'); jest.mock('../services/contentSecurity.service');
jest.mock('../services/surveyHistory.service'); jest.mock('../services/surveyHistory.service');
jest.mock('../services/session.service');
jest.mock('../../auth/services/user.service');
jest.mock('src/guards/authentication.guard'); jest.mock('src/guards/authentication.guard');
jest.mock('src/guards/survey.guard'); jest.mock('src/guards/survey.guard');
jest.mock('src/guards/workspace.guard'); jest.mock('src/guards/workspace.guard');
@ -28,6 +27,8 @@ describe('SurveyController', () => {
let surveyMetaService: SurveyMetaService; let surveyMetaService: SurveyMetaService;
let surveyConfService: SurveyConfService; let surveyConfService: SurveyConfService;
let responseSchemaService: ResponseSchemaService; let responseSchemaService: ResponseSchemaService;
let contentSecurityService: ContentSecurityService;
let surveyHistoryService: SurveyHistoryService;
beforeEach(async () => { beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({ const module: TestingModule = await Test.createTestingModule({
@ -38,21 +39,7 @@ describe('SurveyController', () => {
ResponseSchemaService, ResponseSchemaService,
ContentSecurityService, ContentSecurityService,
SurveyHistoryService, SurveyHistoryService,
SessionService, LoggerProvider,
UserService,
{
provide: Logger,
useValue: {
error: jest.fn(),
info: jest.fn(),
},
},
{
provide: Authentication,
useClass: jest.fn().mockImplementation(() => ({
canActivate: () => true,
})),
},
], ],
}).compile(); }).compile();
@ -62,11 +49,17 @@ describe('SurveyController', () => {
responseSchemaService = module.get<ResponseSchemaService>( responseSchemaService = module.get<ResponseSchemaService>(
ResponseSchemaService, ResponseSchemaService,
); );
contentSecurityService = module.get<ContentSecurityService>(
ContentSecurityService,
);
surveyHistoryService =
module.get<SurveyHistoryService>(SurveyHistoryService);
}); });
describe('getBannerData', () => { describe('getBannerData', () => {
it('should return banner data', async () => { it('should return banner data', async () => {
const result = await controller.getBannerData(); const result = await controller.getBannerData();
expect(result.code).toBe(200); expect(result.code).toBe(200);
expect(result.data).toBeDefined(); expect(result.data).toBeDefined();
}); });
@ -78,16 +71,33 @@ describe('SurveyController', () => {
surveyType: 'normal', surveyType: 'normal',
remark: '问卷调研', remark: '问卷调研',
title: '问卷调研', title: '问卷调研',
}; } as SurveyMeta;
const newId = new ObjectId(); const newId = new ObjectId();
jest.spyOn(surveyMetaService, 'createSurveyMeta').mockResolvedValue({ jest
_id: newId, .spyOn(surveyMetaService, 'createSurveyMeta')
} as any); .mockImplementation(() => {
const result = {
jest.spyOn(surveyConfService, 'createSurveyConf').mockResolvedValue({ _id: newId,
_id: new ObjectId(), } as SurveyMeta;
} as any); return Promise.resolve(result);
});
jest
.spyOn(surveyConfService, 'createSurveyConf')
.mockImplementation(
(params: {
surveyId: string;
surveyType: string;
createMethod: string;
createFrom: string;
}) => {
const result = {
_id: new ObjectId(),
pageId: params.surveyId,
code: {},
} as SurveyConf;
return Promise.resolve(result);
},
);
const result = await controller.createSurvey(surveyInfo, { const result = await controller.createSurvey(surveyInfo, {
user: { username: 'testUser', _id: new ObjectId() }, user: { username: 'testUser', _id: new ObjectId() },
@ -101,15 +111,13 @@ describe('SurveyController', () => {
}); });
}); });
it('should throw an error if validation fails', async () => {
const surveyInfo = {}; // Invalid data
await expect(
controller.createSurvey(surveyInfo as any, { user: {} }),
).rejects.toThrow(HttpException);
});
it('should create a new survey by copy', async () => { it('should create a new survey by copy', async () => {
const existsSurveyId = new ObjectId(); const existsSurveyId = new ObjectId();
const existsSurveyMeta = {
_id: existsSurveyId,
surveyType: 'exam',
owner: 'testUser',
} as SurveyMeta;
const params = { const params = {
surveyType: 'normal', surveyType: 'normal',
remark: '问卷调研', remark: '问卷调研',
@ -118,15 +126,19 @@ describe('SurveyController', () => {
createFrom: existsSurveyId.toString(), createFrom: existsSurveyId.toString(),
}; };
jest
.spyOn(surveyMetaService, 'createSurveyMeta')
.mockImplementation(() => {
const result = {
_id: new ObjectId(),
} as SurveyMeta;
return Promise.resolve(result);
});
const request = { const request = {
user: { username: 'testUser', _id: new ObjectId() }, user: { username: 'testUser', _id: new ObjectId() },
surveyMeta: { _id: existsSurveyId, surveyType: 'exam' }, surveyMeta: existsSurveyMeta,
}; }; // 模拟请求对象,根据实际情况进行调整
jest.spyOn(surveyMetaService, 'createSurveyMeta').mockResolvedValue({
_id: new ObjectId(),
} as any);
const result = await controller.createSurvey(params, request); const result = await controller.createSurvey(params, request);
expect(result?.data?.id).toBeDefined(); expect(result?.data?.id).toBeDefined();
}); });
@ -135,30 +147,48 @@ describe('SurveyController', () => {
describe('updateConf', () => { describe('updateConf', () => {
it('should update survey configuration', async () => { it('should update survey configuration', async () => {
const surveyId = new ObjectId(); const surveyId = new ObjectId();
const surveyMeta = {
_id: surveyId,
surveyType: 'exam',
owner: 'testUser',
} as SurveyMeta;
jest
.spyOn(surveyConfService, 'saveSurveyConf')
.mockResolvedValue(undefined);
jest
.spyOn(surveyHistoryService, 'addHistory')
.mockResolvedValue(undefined);
const reqBody = { const reqBody = {
surveyId: surveyId.toString(), surveyId: surveyId.toString(),
configData: { configData: {
/* ... your config data here ... */ bannerConf: {
titleConfig: {},
bannerConfig: {},
},
baseConf: {
begTime: '2024-01-23 21:59:05',
endTime: '2034-01-23 21:59:05',
},
bottomConf: { logoImage: '/imgs/Logo.webp', logoImageWidth: '60%' },
skinConf: { skinColor: '#4a4c5b', inputBgColor: '#ffffff' },
submitConf: {},
dataConf: {
dataList: [],
},
}, },
sessionId: 'mock-session-id',
}; };
const result = await controller.updateConf(reqBody, { const result = await controller.updateConf(reqBody, {
user: { username: 'testUser', _id: 'testUserId' }, user: { username: 'testUser', _id: 'testUserId' },
surveyMeta: { _id: surveyId }, surveyMeta,
}); });
expect(result).toEqual({ expect(result).toEqual({
code: 200, code: 200,
}); });
}); });
it('should throw an error if validation fails', async () => {
const reqBody = {}; // Invalid data
await expect(
controller.updateConf(reqBody, { user: {} }),
).rejects.toThrow(HttpException);
});
}); });
describe('deleteSurvey', () => { describe('deleteSurvey', () => {
@ -168,7 +198,7 @@ describe('SurveyController', () => {
_id: surveyId, _id: surveyId,
surveyType: 'exam', surveyType: 'exam',
owner: 'testUser', owner: 'testUser',
}; } as SurveyMeta;
jest jest
.spyOn(surveyMetaService, 'deleteSurveyMeta') .spyOn(surveyMetaService, 'deleteSurveyMeta')
@ -178,10 +208,13 @@ describe('SurveyController', () => {
.mockResolvedValue(undefined); .mockResolvedValue(undefined);
const result = await controller.deleteSurvey({ const result = await controller.deleteSurvey({
user: { username: 'testUser' },
surveyMeta, surveyMeta,
user: { username: 'testUser', _id: new ObjectId() },
}); });
expect(result).toEqual({ code: 200 });
expect(result).toEqual({
code: 200,
});
}); });
}); });
@ -192,102 +225,115 @@ describe('SurveyController', () => {
_id: surveyId, _id: surveyId,
surveyType: 'exam', surveyType: 'exam',
owner: 'testUser', owner: 'testUser',
}; } as SurveyMeta;
jest jest
.spyOn(surveyConfService, 'getSurveyConfBySurveyId') .spyOn(surveyConfService, 'getSurveyConfBySurveyId')
.mockResolvedValue({} as any); .mockResolvedValue(
Promise.resolve({
_id: new ObjectId(),
pageId: surveyId.toString(),
} as SurveyConf),
);
const request = {
user: { username: 'testUser', _id: new ObjectId() },
surveyMeta,
};
const result = await controller.getSurvey( const result = await controller.getSurvey(
{ surveyId: surveyId.toString() }, { surveyId: surveyId.toString() },
{ request,
surveyMeta,
user: { username: 'testUser', _id: new ObjectId() },
},
); );
expect(result?.data?.surveyMetaRes).toBeDefined(); expect(result?.data?.surveyMetaRes).toBeDefined();
expect(result?.data?.surveyConfRes).toBeDefined(); expect(result?.data?.surveyConfRes).toBeDefined();
}); });
}); });
describe('publishSurvey', () => { describe('publishSurvey', () => {
it('should publish a survey successfully', async () => { it('should publish a survey success', async () => {
const surveyId = new ObjectId(); const surveyId = new ObjectId();
const surveyMeta = { const surveyMeta = {
_id: surveyId, _id: surveyId,
surveyType: 'exam', surveyType: 'exam',
owner: 'testUser', owner: 'testUser',
isDeleted: false, } as SurveyMeta;
};
jest jest
.spyOn(surveyConfService, 'getSurveyConfBySurveyId') .spyOn(surveyConfService, 'getSurveyConfBySurveyId')
.mockResolvedValue({ .mockResolvedValue(
code: {}, Promise.resolve({
} as any); _id: new ObjectId(),
pageId: surveyId.toString(),
} as SurveyConf),
);
jest jest
.spyOn(surveyConfService, 'getSurveyContentByCode') .spyOn(surveyConfService, 'getSurveyContentByCode')
.mockResolvedValue({ text: '' }); .mockResolvedValue({
text: '题目1',
});
jest
.spyOn(contentSecurityService, 'isForbiddenContent')
.mockResolvedValue(false);
jest jest
.spyOn(surveyMetaService, 'publishSurveyMeta') .spyOn(surveyMetaService, 'publishSurveyMeta')
.mockResolvedValue(undefined); .mockResolvedValue(undefined);
jest
.spyOn(responseSchemaService, 'publishResponseSchema')
.mockResolvedValue(undefined);
jest
.spyOn(surveyHistoryService, 'addHistory')
.mockResolvedValue(undefined);
const result = await controller.publishSurvey( const result = await controller.publishSurvey(
{ surveyId: surveyId.toString() }, { surveyId: surveyId.toString() },
{ surveyMeta, user: { username: 'testUser', _id: new ObjectId() } }, { user: { username: 'testUser', _id: 'testUserId' }, surveyMeta },
); );
expect(result.code).toBe(200);
expect(result).toEqual({
code: 200,
});
}); });
it('should throw an error if the survey is deleted', async () => { it('should not publish a survey with forbidden content', async () => {
const surveyId = new ObjectId(); const surveyId = new ObjectId();
const surveyMeta = { _id: surveyId, isDeleted: true }; const surveyMeta = {
_id: surveyId,
surveyType: 'normal',
owner: 'testUser',
} as SurveyMeta;
jest
.spyOn(surveyConfService, 'getSurveyConfBySurveyId')
.mockResolvedValue(
Promise.resolve({
_id: new ObjectId(),
pageId: surveyId.toString(),
} as SurveyConf),
);
jest
.spyOn(surveyConfService, 'getSurveyContentByCode')
.mockResolvedValue({
text: '违禁词',
});
jest
.spyOn(contentSecurityService, 'isForbiddenContent')
.mockResolvedValue(true);
await expect( await expect(
controller.publishSurvey( controller.publishSurvey(
{ surveyId: surveyId.toString() }, { surveyId: surveyId.toString() },
{ surveyMeta, user: { username: 'testUser' } }, { user: { username: 'testUser', _id: 'testUserId' }, surveyMeta },
), ),
).rejects.toThrow(HttpException); ).rejects.toThrow(
}); new HttpException(
}); '问卷存在非法关键字,不允许发布',
EXCEPTION_CODE.SURVEY_CONTENT_NOT_ALLOW,
// New tests for additional methods ),
describe('pausingSurvey', () => { );
it('should pause the survey successfully', async () => {
const surveyMeta = { surveyPath: 'some/path' };
jest
.spyOn(surveyMetaService, 'pausingSurveyMeta')
.mockResolvedValue(undefined);
jest
.spyOn(responseSchemaService, 'pausingResponseSchema')
.mockResolvedValue(undefined);
const result = await controller.pausingSurvey({
surveyMeta,
user: { username: 'testUser' },
});
expect(result.code).toBe(200);
});
});
describe('getPreviewSchema', () => {
it('should get the preview schema successfully', async () => {
const surveyId = new ObjectId();
jest
.spyOn(surveyConfService, 'getSurveyConfBySurveyId')
.mockResolvedValue({} as any);
jest.spyOn(surveyMetaService, 'getSurveyById').mockResolvedValue({
title: 'Test Survey',
surveyPath: 'some/path',
} as any);
const result = await controller.getPreviewSchema({
surveyPath: surveyId.toString(),
});
expect(result.code).toBe(200);
}); });
}); });
}); });

View File

@ -66,7 +66,7 @@ describe('SurveyHistoryController', () => {
it('should return history list when query is valid', async () => { it('should return history list when query is valid', async () => {
const queryInfo = { surveyId: 'survey123', historyType: 'published' }; const queryInfo = { surveyId: 'survey123', historyType: 'published' };
await controller.getList(queryInfo); await controller.getList(queryInfo, {});
expect(surveyHistoryService.getHistoryList).toHaveBeenCalledWith({ expect(surveyHistoryService.getHistoryList).toHaveBeenCalledWith({
surveyId: queryInfo.surveyId, surveyId: queryInfo.surveyId,

View File

@ -42,7 +42,7 @@ describe('SurveyHistoryService', () => {
msgContent: undefined, msgContent: undefined,
}, },
baseConf: { baseConf: {
beginTime: '', begTime: '',
endTime: '', endTime: '',
answerBegTime: '', answerBegTime: '',
answerEndTime: '', answerEndTime: '',
@ -78,12 +78,7 @@ describe('SurveyHistoryService', () => {
.spyOn(repository, 'save') .spyOn(repository, 'save')
.mockResolvedValueOnce({} as SurveyHistory); .mockResolvedValueOnce({} as SurveyHistory);
await service.addHistory({ await service.addHistory({ surveyId, schema, type, user });
surveyId,
schema,
type,
user,
});
expect(spyCreate).toHaveBeenCalledWith({ expect(spyCreate).toHaveBeenCalledWith({
pageId: surveyId, pageId: surveyId,
@ -121,9 +116,9 @@ describe('SurveyHistoryService', () => {
}, },
take: 100, take: 100,
order: { order: {
createdAt: -1, createDate: -1,
}, },
select: ['createdAt', 'operator', 'type', '_id'], select: ['createDate', 'operator', 'type', '_id'],
}); });
}); });
}); });

View File

@ -1,7 +1,7 @@
import { Test, TestingModule } from '@nestjs/testing'; import { Test, TestingModule } from '@nestjs/testing';
import { SurveyMetaController } from '../controllers/surveyMeta.controller'; import { SurveyMetaController } from '../controllers/surveyMeta.controller';
import { SurveyMetaService } from '../services/surveyMeta.service'; import { SurveyMetaService } from '../services/surveyMeta.service';
import { Logger } from 'src/logger'; import { LoggerProvider } from 'src/logger/logger.provider';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { CollaboratorService } from '../services/collaborator.service'; import { CollaboratorService } from '../services/collaborator.service';
@ -28,12 +28,7 @@ describe('SurveyMetaController', () => {
.mockResolvedValue({ count: 0, data: [] }), .mockResolvedValue({ count: 0, data: [] }),
}, },
}, },
{ LoggerProvider,
provide: Logger,
useValue: {
error() {},
},
},
{ {
provide: CollaboratorService, provide: CollaboratorService,
useValue: { useValue: {
@ -59,25 +54,18 @@ describe('SurveyMetaController', () => {
remark: '', remark: '',
}; };
const mockUser = {
username: 'test-user',
_id: new ObjectId(),
};
const req = { const req = {
user: mockUser, user: {
username: 'test-user',
},
surveyMeta: survey, surveyMeta: survey,
}; };
const result = await controller.updateMeta(reqBody, req); const result = await controller.updateMeta(reqBody, req);
expect(surveyMetaService.editSurveyMeta).toHaveBeenCalledWith({ expect(surveyMetaService.editSurveyMeta).toHaveBeenCalledWith({
operator: mockUser.username, title: reqBody.title,
operatorId: mockUser._id.toString(), remark: reqBody.remark,
survey: {
title: reqBody.title,
remark: reqBody.remark,
},
}); });
expect(result).toEqual({ code: 200 }); expect(result).toEqual({ code: 200 });
@ -123,15 +111,11 @@ describe('SurveyMetaController', () => {
data: [ data: [
{ {
_id: new ObjectId(), _id: new ObjectId(),
createdAt: date, createDate: date,
updatedAt: date, updateDate: date,
curStatus: { curStatus: {
date: date, date: date,
}, },
subStatus: {
date: date,
},
surveyType: 'normal',
}, },
], ],
}); });
@ -145,7 +129,10 @@ describe('SurveyMetaController', () => {
count: 10, count: 10,
data: expect.arrayContaining([ data: expect.arrayContaining([
expect.objectContaining({ expect.objectContaining({
createdAt: expect.stringMatching( createDate: expect.stringMatching(
/^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/,
),
updateDate: expect.stringMatching(
/^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/, /^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/,
), ),
curStatus: expect.objectContaining({ curStatus: expect.objectContaining({
@ -153,17 +140,10 @@ describe('SurveyMetaController', () => {
/^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/, /^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/,
), ),
}), }),
subStatus: expect.objectContaining({
date: expect.stringMatching(
/^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/,
),
}),
surveyType: 'normal',
}), }),
]), ]),
}, },
}); });
expect(surveyMetaService.getSurveyMetaList).toHaveBeenCalledWith({ expect(surveyMetaService.getSurveyMetaList).toHaveBeenCalledWith({
pageNum: queryInfo.curPage, pageNum: queryInfo.curPage,
pageSize: queryInfo.pageSize, pageSize: queryInfo.pageSize,
@ -190,7 +170,7 @@ describe('SurveyMetaController', () => {
condition: [{ field: 'surveyType', value: 'normal' }], condition: [{ field: 'surveyType', value: 'normal' }],
}, },
]), ]),
order: JSON.stringify([{ field: 'createdAt', value: -1 }]), order: JSON.stringify([{ field: 'createDate', value: -1 }]),
}; };
const userId = new ObjectId().toString(); const userId = new ObjectId().toString();
const req = { const req = {
@ -210,28 +190,8 @@ describe('SurveyMetaController', () => {
surveyIdList: [], surveyIdList: [],
userId, userId,
filter: { surveyType: 'normal', title: { $regex: 'hahah' } }, filter: { surveyType: 'normal', title: { $regex: 'hahah' } },
order: { createdAt: -1 }, order: { createDate: -1 },
workspaceId: undefined, workspaceId: undefined,
}); });
}); });
it('should handle Joi validation in getList', async () => {
const invalidQueryInfo: any = {
curPage: 'invalid',
pageSize: 10,
};
const req = {
user: {
username: 'test-user',
_id: new ObjectId(),
},
};
try {
await controller.getList(invalidQueryInfo, req);
} catch (error) {
expect(error).toBeInstanceOf(HttpException);
expect(error.code).toBe(EXCEPTION_CODE.PARAMETER_ERROR);
}
});
}); });

View File

@ -2,16 +2,18 @@ import { Test, TestingModule } from '@nestjs/testing';
import { SurveyMetaService } from '../services/surveyMeta.service'; import { SurveyMetaService } from '../services/surveyMeta.service';
import { MongoRepository } from 'typeorm'; import { MongoRepository } from 'typeorm';
import { SurveyMeta } from 'src/models/surveyMeta.entity'; import { SurveyMeta } from 'src/models/surveyMeta.entity';
import { PluginManager } from 'src/securityPlugin/pluginManager'; import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider';
import { XiaojuSurveyPluginManager } from 'src/securityPlugin/pluginManager';
import { RECORD_STATUS } from 'src/enums';
import { getRepositoryToken } from '@nestjs/typeorm'; import { getRepositoryToken } from '@nestjs/typeorm';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { RECORD_STATUS, RECORD_SUB_STATUS } from 'src/enums'; import { SurveyUtilPlugin } from 'src/securityPlugin/surveyUtilPlugin';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
describe('SurveyMetaService', () => { describe('SurveyMetaService', () => {
let service: SurveyMetaService; let service: SurveyMetaService;
let surveyRepository: MongoRepository<SurveyMeta>; let surveyRepository: MongoRepository<SurveyMeta>;
let pluginManager: PluginManager; let pluginManager: XiaojuSurveyPluginManager;
beforeEach(async () => { beforeEach(async () => {
const module: TestingModule = await Test.createTestingModule({ const module: TestingModule = await Test.createTestingModule({
@ -24,11 +26,10 @@ describe('SurveyMetaService', () => {
count: jest.fn(), count: jest.fn(),
create: jest.fn(), create: jest.fn(),
save: jest.fn(), save: jest.fn(),
updateOne: jest.fn(),
findAndCount: jest.fn(), findAndCount: jest.fn(),
}, },
}, },
PluginManager, PluginManagerProvider,
], ],
}).compile(); }).compile();
@ -36,19 +37,21 @@ describe('SurveyMetaService', () => {
surveyRepository = module.get<MongoRepository<SurveyMeta>>( surveyRepository = module.get<MongoRepository<SurveyMeta>>(
getRepositoryToken(SurveyMeta), getRepositoryToken(SurveyMeta),
); );
pluginManager = module.get<PluginManager>(PluginManager); pluginManager = module.get<XiaojuSurveyPluginManager>(
XiaojuSurveyPluginManager,
);
pluginManager.registerPlugin(new SurveyUtilPlugin());
}); });
describe('getNewSurveyPath', () => { describe('getNewSurveyPath', () => {
it('should generate a new survey path', async () => { it('should generate a new survey path', async () => {
jest.spyOn(pluginManager, 'triggerHook').mockResolvedValueOnce('path1'); jest.spyOn(surveyRepository, 'count').mockResolvedValueOnce(1);
jest.spyOn(surveyRepository, 'count').mockResolvedValueOnce(0); jest.spyOn(surveyRepository, 'count').mockResolvedValueOnce(0);
const surveyPath = await service.getNewSurveyPath(); const surveyPath = await service.getNewSurveyPath();
expect(surveyPath).toBe('path1'); expect(typeof surveyPath).toBe('string');
expect(pluginManager.triggerHook).toHaveBeenCalledTimes(1); expect(surveyRepository.count).toHaveBeenCalledTimes(2);
expect(surveyRepository.count).toHaveBeenCalledTimes(1);
}); });
}); });
@ -62,11 +65,14 @@ describe('SurveyMetaService', () => {
userId: new ObjectId().toString(), userId: new ObjectId().toString(),
createMethod: '', createMethod: '',
createFrom: '', createFrom: '',
workspaceId: 'workspace1',
}; };
const newSurvey = new SurveyMeta(); const newSurvey = new SurveyMeta();
jest.spyOn(service, 'getNewSurveyPath').mockResolvedValue('path1'); const mockedSurveyPath = 'mockedSurveyPath';
jest
.spyOn(service, 'getNewSurveyPath')
.mockResolvedValue(mockedSurveyPath);
jest jest
.spyOn(surveyRepository, 'create') .spyOn(surveyRepository, 'create')
.mockImplementation(() => newSurvey); .mockImplementation(() => newSurvey);
@ -78,118 +84,86 @@ describe('SurveyMetaService', () => {
title: params.title, title: params.title,
remark: params.remark, remark: params.remark,
surveyType: params.surveyType, surveyType: params.surveyType,
surveyPath: 'path1', surveyPath: mockedSurveyPath,
creator: params.username, creator: params.username,
creatorId: params.userId,
owner: params.username,
ownerId: params.userId, ownerId: params.userId,
owner: params.username,
createMethod: params.createMethod, createMethod: params.createMethod,
createFrom: params.createFrom, createFrom: params.createFrom,
workspaceId: params.workspaceId,
}); });
expect(surveyRepository.save).toHaveBeenCalledWith(newSurvey); expect(surveyRepository.save).toHaveBeenCalledWith(newSurvey);
expect(result).toEqual(newSurvey); expect(result).toEqual(newSurvey);
}); });
}); });
describe('pausingSurveyMeta', () => {
it('should throw an exception if survey is in NEW status', async () => {
const survey = new SurveyMeta();
survey.curStatus = { status: RECORD_STATUS.NEW, date: Date.now() };
await expect(service.pausingSurveyMeta(survey)).rejects.toThrow(
HttpException,
);
});
it('should pause a survey and update subStatus', async () => {
const survey = new SurveyMeta();
survey.curStatus = { status: RECORD_STATUS.PUBLISHED, date: Date.now() };
survey.statusList = [];
jest.spyOn(surveyRepository, 'save').mockResolvedValue(survey);
const result = await service.pausingSurveyMeta(survey);
expect(survey.subStatus.status).toBe(RECORD_SUB_STATUS.PAUSING);
expect(survey.statusList.length).toBe(1);
expect(survey.statusList[0].status).toBe(RECORD_SUB_STATUS.PAUSING);
expect(surveyRepository.save).toHaveBeenCalledWith(survey);
expect(result).toEqual(survey);
});
});
describe('editSurveyMeta', () => { describe('editSurveyMeta', () => {
it('should edit a survey meta and return it', async () => { it('should edit a survey meta and return it if in NEW or EDITING status', async () => {
const survey = new SurveyMeta(); const survey = new SurveyMeta();
survey.curStatus = { status: RECORD_STATUS.PUBLISHED, date: Date.now() }; survey.curStatus = { status: RECORD_STATUS.PUBLISHED, date: Date.now() };
survey.statusList = []; survey.statusList = [];
const operator = 'editor';
const operatorId = 'editorId';
jest.spyOn(surveyRepository, 'save').mockResolvedValue(survey); jest.spyOn(surveyRepository, 'save').mockResolvedValue(survey);
const result = await service.editSurveyMeta({ const result = await service.editSurveyMeta(survey);
survey,
operator,
operatorId,
});
expect(survey.curStatus.status).toBe(RECORD_STATUS.EDITING); expect(survey.curStatus.status).toEqual(RECORD_STATUS.EDITING);
expect(survey.statusList.length).toBe(1); expect(survey.statusList.length).toBe(1);
expect(survey.statusList[0].status).toBe(RECORD_STATUS.EDITING); expect(survey.statusList[0].status).toEqual(RECORD_STATUS.EDITING);
expect(survey.operator).toBe(operator);
expect(survey.operatorId).toBe(operatorId);
expect(surveyRepository.save).toHaveBeenCalledWith(survey); expect(surveyRepository.save).toHaveBeenCalledWith(survey);
expect(result).toEqual(survey); expect(result).toEqual(survey);
}); });
}); });
describe('deleteSurveyMeta', () => { describe('deleteSurveyMeta', () => {
it('should mark a survey as deleted', async () => { it('should delete survey meta and update status', async () => {
const surveyId = new ObjectId().toString(); // 准备假的SurveyMeta对象
const operator = 'deleter'; const survey = new SurveyMeta();
const operatorId = 'deleterId'; survey.curStatus = { status: RECORD_STATUS.NEW, date: Date.now() };
survey.statusList = [];
jest.spyOn(surveyRepository, 'updateOne').mockResolvedValue({ // 模拟save方法
matchedCount: 1, jest.spyOn(surveyRepository, 'save').mockResolvedValue(survey);
modifiedCount: 1,
acknowledged: true,
});
const result = await service.deleteSurveyMeta({ // 调用要测试的方法
surveyId, const result = await service.deleteSurveyMeta(survey);
operator,
operatorId,
});
expect(surveyRepository.updateOne).toHaveBeenCalledWith( // 验证结果
{ _id: new ObjectId(surveyId) }, expect(result).toBe(survey);
{ expect(survey.curStatus.status).toBe(RECORD_STATUS.REMOVED);
$set: { expect(survey.statusList.length).toBe(1);
isDeleted: true, expect(survey.statusList[0].status).toBe(RECORD_STATUS.REMOVED);
operator, expect(surveyRepository.save).toHaveBeenCalledTimes(1);
operatorId, expect(surveyRepository.save).toHaveBeenCalledWith(survey);
deletedAt: expect.any(Date), });
},
}, it('should throw exception when survey is already removed', async () => {
// 准备假的SurveyMeta对象其状态已设置为REMOVED
const survey = new SurveyMeta();
survey.curStatus = { status: RECORD_STATUS.REMOVED, date: Date.now() };
// 调用要测试的方法并期待异常
await expect(service.deleteSurveyMeta(survey)).rejects.toThrow(
HttpException,
); );
expect(result.matchedCount).toBe(1);
// 验证save方法没有被调用
expect(surveyRepository.save).not.toHaveBeenCalled();
}); });
}); });
describe('getSurveyMetaList', () => { describe('getSurveyMetaList', () => {
it('should return a list of survey metadata', async () => { it('should return a list of survey metadata', async () => {
// 准备模拟数据
const mockData = [ const mockData = [
{ _id: 1, title: 'Survey 1' }, { _id: 1, title: 'Survey 1' },
{ _id: 2, title: 'Survey 2' },
] as unknown as Array<SurveyMeta>; ] as unknown as Array<SurveyMeta>;
const mockCount = 1; const mockCount = 2;
jest jest
.spyOn(surveyRepository, 'findAndCount') .spyOn(surveyRepository, 'findAndCount')
.mockResolvedValue([mockData, mockCount]); .mockResolvedValue([mockData, mockCount]);
// 调用方法并检查返回值
const condition = { const condition = {
pageNum: 1, pageNum: 1,
pageSize: 10, pageSize: 10,
@ -198,47 +172,40 @@ describe('SurveyMetaService', () => {
filter: {}, filter: {},
order: {}, order: {},
}; };
const result = await service.getSurveyMetaList(condition); const result = await service.getSurveyMetaList(condition);
// 验证返回值
expect(result).toEqual({ data: mockData, count: mockCount }); expect(result).toEqual({ data: mockData, count: mockCount });
// 验证repository方法被正确调用
expect(surveyRepository.findAndCount).toHaveBeenCalledTimes(1); expect(surveyRepository.findAndCount).toHaveBeenCalledTimes(1);
}); });
}); });
describe('publishSurveyMeta', () => { describe('publishSurveyMeta', () => {
it('should publish a survey and update curStatus', async () => { it('should publish a survey meta and add status to statusList', async () => {
const surveyMeta = new SurveyMeta(); // 准备模拟数据
surveyMeta.statusList = []; const surveyMeta = {
id: 1,
title: 'Test Survey',
statusList: [],
} as unknown as SurveyMeta;
const savedSurveyMeta = {
...surveyMeta,
curStatus: {
status: RECORD_STATUS.PUBLISHED,
date: expect.any(Number),
},
} as unknown as SurveyMeta;
jest.spyOn(surveyRepository, 'save').mockResolvedValue(surveyMeta); jest.spyOn(surveyRepository, 'save').mockResolvedValue(savedSurveyMeta);
// 调用方法并检查返回值
const result = await service.publishSurveyMeta({ surveyMeta }); const result = await service.publishSurveyMeta({ surveyMeta });
expect(surveyMeta.curStatus.status).toBe(RECORD_STATUS.PUBLISHED); // 验证返回值
expect(surveyMeta.statusList.length).toBe(1); expect(result).toEqual(savedSurveyMeta);
expect(surveyMeta.statusList[0].status).toBe(RECORD_STATUS.PUBLISHED); // 验证repository方法被正确调用
expect(surveyRepository.save).toHaveBeenCalledWith(surveyMeta); expect(surveyRepository.save).toHaveBeenCalledWith(savedSurveyMeta);
expect(result).toEqual(surveyMeta);
});
});
describe('countSurveyMetaByWorkspaceId', () => {
it('should return the count of surveys in a workspace', async () => {
const workspaceId = 'workspace1';
const mockCount = 5;
jest.spyOn(surveyRepository, 'count').mockResolvedValue(mockCount);
const result = await service.countSurveyMetaByWorkspaceId({
workspaceId,
});
expect(result).toBe(mockCount);
expect(surveyRepository.count).toHaveBeenCalledWith({
workspaceId,
isDeleted: { $ne: true },
});
}); });
}); });
}); });

View File

@ -69,7 +69,7 @@ export class CollaboratorController {
) { ) {
const { error, value } = CreateCollaboratorDto.validate(reqBody); const { error, value } = CreateCollaboratorDto.validate(reqBody);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException( throw new HttpException(
'系统错误,请联系管理员', '系统错误,请联系管理员',
EXCEPTION_CODE.PARAMETER_ERROR, EXCEPTION_CODE.PARAMETER_ERROR,
@ -124,7 +124,7 @@ export class CollaboratorController {
) { ) {
const { error, value } = BatchSaveCollaboratorDto.validate(reqBody); const { error, value } = BatchSaveCollaboratorDto.validate(reqBody);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException( throw new HttpException(
'系统错误,请联系管理员', '系统错误,请联系管理员',
EXCEPTION_CODE.PARAMETER_ERROR, EXCEPTION_CODE.PARAMETER_ERROR,
@ -184,15 +184,11 @@ export class CollaboratorController {
neIdList: collaboratorIdList, neIdList: collaboratorIdList,
userIdList: newCollaboratorUserIdList, userIdList: newCollaboratorUserIdList,
}); });
this.logger.info('batchDelete:' + JSON.stringify(delRes)); this.logger.info('batchDelete:' + JSON.stringify(delRes), { req });
const username = req.user.username;
const userId = req.user._id.toString();
if (Array.isArray(newCollaborator) && newCollaborator.length > 0) { if (Array.isArray(newCollaborator) && newCollaborator.length > 0) {
const insertRes = await this.collaboratorService.batchCreate({ const insertRes = await this.collaboratorService.batchCreate({
surveyId: value.surveyId, surveyId: value.surveyId,
collaboratorList: newCollaborator, collaboratorList: newCollaborator,
creator: username,
creatorId: userId,
}); });
this.logger.info(`${JSON.stringify(insertRes)}`); this.logger.info(`${JSON.stringify(insertRes)}`);
} }
@ -202,8 +198,6 @@ export class CollaboratorController {
this.collaboratorService.updateById({ this.collaboratorService.updateById({
collaboratorId: item._id, collaboratorId: item._id,
permissions: item.permissions, permissions: item.permissions,
operator: username,
operatorId: userId,
}), }),
), ),
); );
@ -214,7 +208,7 @@ export class CollaboratorController {
const delRes = await this.collaboratorService.batchDeleteBySurveyId( const delRes = await this.collaboratorService.batchDeleteBySurveyId(
value.surveyId, value.surveyId,
); );
this.logger.info(JSON.stringify(delRes)); this.logger.info(JSON.stringify(delRes), { req });
} }
return { return {
@ -231,10 +225,11 @@ export class CollaboratorController {
]) ])
async getSurveyCollaboratorList( async getSurveyCollaboratorList(
@Query() query: GetSurveyCollaboratorListDto, @Query() query: GetSurveyCollaboratorListDto,
@Request() req,
) { ) {
const { error, value } = GetSurveyCollaboratorListDto.validate(query); const { error, value } = GetSurveyCollaboratorListDto.validate(query);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
@ -268,14 +263,17 @@ export class CollaboratorController {
@SetMetadata('surveyPermission', [ @SetMetadata('surveyPermission', [
SURVEY_PERMISSION.SURVEY_COOPERATION_MANAGE, SURVEY_PERMISSION.SURVEY_COOPERATION_MANAGE,
]) ])
async changeUserPermission(@Body() reqBody: ChangeUserPermissionDto) { async changeUserPermission(
@Body() reqBody: ChangeUserPermissionDto,
@Request() req,
) {
const { error, value } = Joi.object({ const { error, value } = Joi.object({
surveyId: Joi.string(), surveyId: Joi.string(),
userId: Joi.string(), userId: Joi.string(),
permissions: Joi.array().items(Joi.string().required()), permissions: Joi.array().items(Joi.string().required()),
}).validate(reqBody); }).validate(reqBody);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
@ -294,13 +292,13 @@ export class CollaboratorController {
@SetMetadata('surveyPermission', [ @SetMetadata('surveyPermission', [
SURVEY_PERMISSION.SURVEY_COOPERATION_MANAGE, SURVEY_PERMISSION.SURVEY_COOPERATION_MANAGE,
]) ])
async deleteCollaborator(@Query() query) { async deleteCollaborator(@Query() query, @Request() req) {
const { error, value } = Joi.object({ const { error, value } = Joi.object({
surveyId: Joi.string(), surveyId: Joi.string(),
userId: Joi.string(), userId: Joi.string(),
}).validate(query); }).validate(query);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
@ -321,7 +319,7 @@ export class CollaboratorController {
const surveyMeta = await this.surveyMetaService.getSurveyById({ surveyId }); const surveyMeta = await this.surveyMetaService.getSurveyById({ surveyId });
if (!surveyMeta) { if (!surveyMeta) {
this.logger.error(`问卷不存在: ${surveyId}`); this.logger.error(`问卷不存在: ${surveyId}`, { req });
throw new HttpException('问卷不存在', EXCEPTION_CODE.SURVEY_NOT_FOUND); throw new HttpException('问卷不存在', EXCEPTION_CODE.SURVEY_NOT_FOUND);
} }

View File

@ -5,6 +5,7 @@ import {
HttpCode, HttpCode,
UseGuards, UseGuards,
SetMetadata, SetMetadata,
Request,
} from '@nestjs/common'; } from '@nestjs/common';
import * as Joi from 'joi'; import * as Joi from 'joi';
import { ApiTags, ApiBearerAuth } from '@nestjs/swagger'; import { ApiTags, ApiBearerAuth } from '@nestjs/swagger';
@ -13,7 +14,7 @@ import { DataStatisticService } from '../services/dataStatistic.service';
import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service'; import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service';
import { Authentication } from 'src/guards/authentication.guard'; import { Authentication } from 'src/guards/authentication.guard';
import { PluginManager } from 'src/securityPlugin/pluginManager'; import { XiaojuSurveyPluginManager } from 'src/securityPlugin/pluginManager';
import { SurveyGuard } from 'src/guards/survey.guard'; import { SurveyGuard } from 'src/guards/survey.guard';
import { SURVEY_PERMISSION } from 'src/enums/surveyPermission'; import { SURVEY_PERMISSION } from 'src/enums/surveyPermission';
import { Logger } from 'src/logger'; import { Logger } from 'src/logger';
@ -21,7 +22,6 @@ import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { AggregationStatisDto } from '../dto/aggregationStatis.dto'; import { AggregationStatisDto } from '../dto/aggregationStatis.dto';
import { handleAggretionData } from '../utils'; import { handleAggretionData } from '../utils';
import { QUESTION_TYPE } from 'src/enums/question';
@ApiTags('survey') @ApiTags('survey')
@ApiBearerAuth() @ApiBearerAuth()
@ -30,7 +30,7 @@ export class DataStatisticController {
constructor( constructor(
private readonly responseSchemaService: ResponseSchemaService, private readonly responseSchemaService: ResponseSchemaService,
private readonly dataStatisticService: DataStatisticService, private readonly dataStatisticService: DataStatisticService,
private readonly pluginManager: PluginManager, private readonly pluginManager: XiaojuSurveyPluginManager,
private readonly logger: Logger, private readonly logger: Logger,
) {} ) {}
@ -43,18 +43,19 @@ export class DataStatisticController {
async data( async data(
@Query() @Query()
queryInfo, queryInfo,
@Request() req,
) { ) {
const { value, error } = await Joi.object({ const { value, error } = await Joi.object({
surveyId: Joi.string().required(), surveyId: Joi.string().required(),
isMasked: Joi.boolean().default(true), // 默认true就是需要脱敏 isDesensitive: Joi.boolean().default(true), // 默认true就是需要脱敏
page: Joi.number().default(1), page: Joi.number().default(1),
pageSize: Joi.number().default(10), pageSize: Joi.number().default(10),
}).validate(queryInfo); }).validate(queryInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const { surveyId, isMasked, page, pageSize } = value; const { surveyId, isDesensitive, page, pageSize } = value;
const responseSchema = const responseSchema =
await this.responseSchemaService.getResponseSchemaByPageId(surveyId); await this.responseSchemaService.getResponseSchemaByPageId(surveyId);
const { total, listHead, listBody } = const { total, listHead, listBody } =
@ -65,10 +66,10 @@ export class DataStatisticController {
pageSize, pageSize,
}); });
if (isMasked) { if (isDesensitive) {
// 脱敏 // 脱敏
listBody.forEach((item) => { listBody.forEach((item) => {
this.pluginManager.triggerHook('maskData', item); this.pluginManager.triggerHook('desensitiveData', item);
}); });
} }
@ -102,15 +103,15 @@ export class DataStatisticController {
}; };
} }
const allowQuestionType = [ const allowQuestionType = [
QUESTION_TYPE.RADIO, 'radio',
QUESTION_TYPE.CHECKBOX, 'checkbox',
QUESTION_TYPE.BINARY_CHOICE, 'binary-choice',
QUESTION_TYPE.RADIO_STAR, 'radio-star',
QUESTION_TYPE.RADIO_NPS, 'radio-nps',
QUESTION_TYPE.VOTE, 'vote',
]; ];
const fieldList = responseSchema.code.dataConf.dataList const fieldList = responseSchema.code.dataConf.dataList
.filter((item) => allowQuestionType.includes(item.type as QUESTION_TYPE)) .filter((item) => allowQuestionType.includes(item.type))
.map((item) => item.field); .map((item) => item.field);
const dataMap = responseSchema.code.dataConf.dataList.reduce((pre, cur) => { const dataMap = responseSchema.code.dataConf.dataList.reduce((pre, cur) => {
pre[cur.field] = cur; pre[cur.field] = cur;

View File

@ -1,189 +0,0 @@
import {
Controller,
Get,
Query,
HttpCode,
UseGuards,
SetMetadata,
Request,
Post,
Body,
} from '@nestjs/common';
import { ApiTags, ApiBearerAuth } from '@nestjs/swagger';
import { ResponseSchemaService } from '../../surveyResponse/services/responseScheme.service';
import { Authentication } from 'src/guards/authentication.guard';
import { SurveyGuard } from 'src/guards/survey.guard';
import { SURVEY_PERMISSION } from 'src/enums/surveyPermission';
import { Logger } from 'src/logger';
import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { DownloadTaskService } from '../services/downloadTask.service';
import {
GetDownloadTaskDto,
CreateDownloadDto,
GetDownloadTaskListDto,
DeleteDownloadTaskDto,
} from '../dto/downloadTask.dto';
import moment from 'moment';
import { NoPermissionException } from 'src/exceptions/noPermissionException';
@ApiTags('downloadTask')
@ApiBearerAuth()
@Controller('/api/downloadTask')
export class DownloadTaskController {
constructor(
private readonly responseSchemaService: ResponseSchemaService,
private readonly downloadTaskService: DownloadTaskService,
private readonly logger: Logger,
) {}
@Post('/createTask')
@HttpCode(200)
@UseGuards(SurveyGuard)
@SetMetadata('surveyId', 'body.surveyId')
@SetMetadata('surveyPermission', [SURVEY_PERMISSION.SURVEY_RESPONSE_MANAGE])
@UseGuards(Authentication)
async createTask(
@Body()
reqBody: CreateDownloadDto,
@Request() req,
) {
const { value, error } = CreateDownloadDto.validate(reqBody);
if (error) {
this.logger.error(error.message);
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
}
const { surveyId, isMasked } = value;
const responseSchema =
await this.responseSchemaService.getResponseSchemaByPageId(surveyId);
const id = await this.downloadTaskService.createDownloadTask({
surveyId,
responseSchema,
creatorId: req.user._id.toString(),
creator: req.user.username,
params: { isMasked },
});
this.downloadTaskService.processDownloadTask({ taskId: id });
return {
code: 200,
data: { taskId: id },
};
}
@Get('/getDownloadTaskList')
@HttpCode(200)
@UseGuards(Authentication)
async downloadList(
@Query()
queryInfo: GetDownloadTaskListDto,
@Request() req,
) {
const { value, error } = GetDownloadTaskListDto.validate(queryInfo);
if (error) {
this.logger.error(error.message);
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
}
const { pageIndex, pageSize } = value;
const { total, list } = await this.downloadTaskService.getDownloadTaskList({
creatorId: req.user._id.toString(),
pageIndex,
pageSize,
});
return {
code: 200,
data: {
total: total,
list: list.map((data) => {
const item: Record<string, any> = {};
item.taskId = data._id.toString();
item.status = data.status;
item.filename = data.filename;
item.url = data.url;
const fmt = 'YYYY-MM-DD HH:mm:ss';
const units = ['Bytes', 'KB', 'MB', 'GB', 'TB'];
let unitIndex = 0;
let size = Number(data.fileSize);
if (isNaN(size)) {
item.fileSize = data.fileSize;
} else {
while (size >= 1024 && unitIndex < units.length - 1) {
size /= 1024;
unitIndex++;
}
item.fileSize = `${size.toFixed()} ${units[unitIndex]}`;
}
item.createdAt = moment(data.createdAt).format(fmt);
return item;
}),
},
};
}
@Get('/getDownloadTask')
@HttpCode(200)
@UseGuards(Authentication)
async getDownloadTask(@Query() query: GetDownloadTaskDto, @Request() req) {
const { value, error } = GetDownloadTaskDto.validate(query);
if (error) {
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
}
const taskInfo = await this.downloadTaskService.getDownloadTaskById({
taskId: value.taskId,
});
if (!taskInfo) {
throw new HttpException('任务不存在', EXCEPTION_CODE.PARAMETER_ERROR);
}
if (taskInfo.creatorId !== req.user._id.toString()) {
throw new NoPermissionException('没有权限');
}
const res: Record<string, any> = {
...taskInfo,
};
res.taskId = taskInfo._id.toString();
delete res._id;
return {
code: 200,
data: res,
};
}
@Post('/deleteDownloadTask')
@HttpCode(200)
@UseGuards(Authentication)
async deleteFileByName(@Body() body: DeleteDownloadTaskDto, @Request() req) {
const { value, error } = DeleteDownloadTaskDto.validate(body);
if (error) {
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
}
const { taskId } = value;
const taskInfo = await this.downloadTaskService.getDownloadTaskById({
taskId,
});
if (!taskInfo) {
throw new HttpException('任务不存在', EXCEPTION_CODE.PARAMETER_ERROR);
}
if (taskInfo.creatorId !== req.user._id.toString()) {
throw new NoPermissionException('没有权限');
}
const delRes = await this.downloadTaskService.deleteDownloadTask({
taskId,
operator: req.user.username,
operatorId: req.user._id.toString(),
});
return {
code: 200,
data: delRes.modifiedCount === 1,
};
}
}

View File

@ -1,90 +0,0 @@
import {
Controller,
Post,
Body,
HttpCode,
UseGuards,
SetMetadata,
Request,
} from '@nestjs/common';
import * as Joi from 'joi';
import { ApiTags } from '@nestjs/swagger';
import { SessionService } from '../services/session.service';
import { Authentication } from 'src/guards/authentication.guard';
import { SurveyGuard } from 'src/guards/survey.guard';
import { SURVEY_PERMISSION } from 'src/enums/surveyPermission';
import { Logger } from 'src/logger';
import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { SessionGuard } from 'src/guards/session.guard';
@ApiTags('survey')
@Controller('/api/session')
export class SessionController {
constructor(
private readonly sessionService: SessionService,
private readonly logger: Logger,
) {}
@Post('/create')
@HttpCode(200)
@UseGuards(SurveyGuard)
@SetMetadata('surveyId', 'body.surveyId')
@SetMetadata('surveyPermission', [SURVEY_PERMISSION.SURVEY_CONF_MANAGE])
@UseGuards(Authentication)
async create(
@Body()
reqBody: {
surveyId: string;
},
@Request()
req,
) {
const { value, error } = Joi.object({
surveyId: Joi.string().required(),
}).validate(reqBody);
if (error) {
this.logger.error(error.message);
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
}
const surveyId = value.surveyId;
const session = await this.sessionService.create({
surveyId,
userId: req.user._id.toString(),
});
return {
code: 200,
data: {
sessionId: session._id.toString(),
},
};
}
@Post('/seize')
@HttpCode(200)
@UseGuards(SessionGuard, SurveyGuard)
@SetMetadata('sessionId', 'body.sessionId')
@SetMetadata('surveyId', 'surveyId')
@SetMetadata('surveyPermission', [SURVEY_PERMISSION.SURVEY_CONF_MANAGE])
@UseGuards(Authentication)
async seize(
@Request()
req,
) {
const sessionInfo = req.sessionInfo;
await this.sessionService.updateSessionToEditing({
sessionId: sessionInfo._id.toString(),
surveyId: sessionInfo.surveyId,
});
return {
code: 200,
};
}
}

View File

@ -31,8 +31,6 @@ import { SURVEY_PERMISSION } from 'src/enums/surveyPermission';
import { WorkspaceGuard } from 'src/guards/workspace.guard'; import { WorkspaceGuard } from 'src/guards/workspace.guard';
import { PERMISSION as WORKSPACE_PERMISSION } from 'src/enums/workspace'; import { PERMISSION as WORKSPACE_PERMISSION } from 'src/enums/workspace';
import { SessionService } from '../services/session.service';
import { UserService } from 'src/modules/auth/services/user.service';
@ApiTags('survey') @ApiTags('survey')
@Controller('/api/survey') @Controller('/api/survey')
@ -44,8 +42,6 @@ export class SurveyController {
private readonly contentSecurityService: ContentSecurityService, private readonly contentSecurityService: ContentSecurityService,
private readonly surveyHistoryService: SurveyHistoryService, private readonly surveyHistoryService: SurveyHistoryService,
private readonly logger: Logger, private readonly logger: Logger,
private readonly sessionService: SessionService,
private readonly userService: UserService,
) {} ) {}
@Get('/getBannerData') @Get('/getBannerData')
@ -74,7 +70,9 @@ export class SurveyController {
) { ) {
const { error, value } = CreateSurveyDto.validate(reqBody); const { error, value } = CreateSurveyDto.validate(reqBody);
if (error) { if (error) {
this.logger.error(`createSurvey_parameter error: ${error.message}`); this.logger.error(`createSurvey_parameter error: ${error.message}`, {
req,
});
throw new HttpException('参数错误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数错误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
@ -130,41 +128,13 @@ export class SurveyController {
const { value, error } = Joi.object({ const { value, error } = Joi.object({
surveyId: Joi.string().required(), surveyId: Joi.string().required(),
configData: Joi.any().required(), configData: Joi.any().required(),
sessionId: Joi.string().required(),
}).validate(surveyInfo); }).validate(surveyInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const sessionId = value.sessionId;
const surveyId = value.surveyId;
const latestEditingOne = await this.sessionService.findLatestEditingOne({
surveyId,
});
if (latestEditingOne && latestEditingOne._id.toString() !== sessionId) {
const curSession = await this.sessionService.findOne(sessionId);
if (curSession.createdAt <= latestEditingOne.updatedAt) {
// 在当前用户打开之后,被其他页面保存过了
const isSameOperator =
latestEditingOne.userId === req.user._id.toString();
let preOperator;
if (!isSameOperator) {
preOperator = await this.userService.getUserById(
latestEditingOne.userId,
);
}
return {
code: EXCEPTION_CODE.SURVEY_SAVE_CONFLICT,
errmsg: isSameOperator
? '当前问卷已在其它页面开启编辑,刷新以获取最新内容'
: `当前问卷已由 ${preOperator.username} 编辑,刷新以获取最新内容`,
};
}
}
await this.sessionService.updateSessionToEditing({ sessionId, surveyId });
const username = req.user.username; const username = req.user.username;
const surveyId = value.surveyId;
const configData = value.configData; const configData = value.configData;
await this.surveyConfService.saveSurveyConf({ await this.surveyConfService.saveSurveyConf({
@ -194,35 +164,8 @@ export class SurveyController {
async deleteSurvey(@Request() req) { async deleteSurvey(@Request() req) {
const surveyMeta = req.surveyMeta; const surveyMeta = req.surveyMeta;
const delMetaRes = await this.surveyMetaService.deleteSurveyMeta({ await this.surveyMetaService.deleteSurveyMeta(surveyMeta);
surveyId: surveyMeta._id.toString(), await this.responseSchemaService.deleteResponseSchema({
operator: req.user.username,
operatorId: req.user._id.toString(),
});
const delResponseRes =
await this.responseSchemaService.deleteResponseSchema({
surveyPath: surveyMeta.surveyPath,
});
this.logger.info(JSON.stringify(delMetaRes));
this.logger.info(JSON.stringify(delResponseRes));
return {
code: 200,
};
}
@HttpCode(200)
@Post('/pausingSurvey')
@UseGuards(SurveyGuard)
@SetMetadata('surveyId', 'body.surveyId')
@SetMetadata('surveyPermission', [SURVEY_PERMISSION.SURVEY_CONF_MANAGE])
@UseGuards(Authentication)
async pausingSurvey(@Request() req) {
const surveyMeta = req.surveyMeta;
await this.surveyMetaService.pausingSurveyMeta(surveyMeta);
await this.responseSchemaService.pausingResponseSchema({
surveyPath: surveyMeta.surveyPath, surveyPath: surveyMeta.surveyPath,
}); });
@ -254,7 +197,7 @@ export class SurveyController {
}).validate(queryInfo); }).validate(queryInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
@ -287,13 +230,15 @@ export class SurveyController {
queryInfo: { queryInfo: {
surveyPath: string; surveyPath: string;
}, },
@Request()
req,
) { ) {
const { value, error } = Joi.object({ const { value, error } = Joi.object({
surveyId: Joi.string().required(), surveyId: Joi.string().required(),
}).validate({ surveyId: queryInfo.surveyPath }); }).validate({ surveyId: queryInfo.surveyPath });
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const surveyId = value.surveyId; const surveyId = value.surveyId;
@ -326,18 +271,12 @@ export class SurveyController {
surveyId: Joi.string().required(), surveyId: Joi.string().required(),
}).validate(surveyInfo); }).validate(surveyInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const username = req.user.username; const username = req.user.username;
const surveyId = value.surveyId; const surveyId = value.surveyId;
const surveyMeta = req.surveyMeta; const surveyMeta = req.surveyMeta;
if (surveyMeta.isDeleted) {
throw new HttpException(
'问卷已删除,无法发布',
EXCEPTION_CODE.SURVEY_NOT_FOUND,
);
}
const surveyConf = const surveyConf =
await this.surveyConfService.getSurveyConfBySurveyId(surveyId); await this.surveyConfService.getSurveyConfBySurveyId(surveyId);
@ -363,8 +302,7 @@ export class SurveyController {
pageId: surveyId, pageId: surveyId,
}); });
// 添加发布历史可以异步添加 await this.surveyHistoryService.addHistory({
this.surveyHistoryService.addHistory({
surveyId, surveyId,
schema: surveyConf.code, schema: surveyConf.code,
type: HISTORY_TYPE.PUBLISH_HIS, type: HISTORY_TYPE.PUBLISH_HIS,

View File

@ -5,6 +5,7 @@ import {
HttpCode, HttpCode,
UseGuards, UseGuards,
SetMetadata, SetMetadata,
Request,
} from '@nestjs/common'; } from '@nestjs/common';
import * as Joi from 'joi'; import * as Joi from 'joi';
import { ApiTags } from '@nestjs/swagger'; import { ApiTags } from '@nestjs/swagger';
@ -17,8 +18,9 @@ import { SURVEY_PERMISSION } from 'src/enums/surveyPermission';
import { Logger } from 'src/logger'; import { Logger } from 'src/logger';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
@ApiTags('survey') @ApiTags('survey')
@Controller('/api/surveyHistory') @Controller('/api/surveyHisotry')
export class SurveyHistoryController { export class SurveyHistoryController {
constructor( constructor(
private readonly surveyHistoryService: SurveyHistoryService, private readonly surveyHistoryService: SurveyHistoryService,
@ -41,6 +43,7 @@ export class SurveyHistoryController {
surveyId: string; surveyId: string;
historyType: string; historyType: string;
}, },
@Request() req,
) { ) {
const { value, error } = Joi.object({ const { value, error } = Joi.object({
surveyId: Joi.string().required(), surveyId: Joi.string().required(),
@ -48,7 +51,7 @@ export class SurveyHistoryController {
}).validate(queryInfo); }).validate(queryInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }

View File

@ -51,18 +51,16 @@ export class SurveyMetaController {
}).validate(reqBody, { allowUnknown: true }); }).validate(reqBody, { allowUnknown: true });
if (error) { if (error) {
this.logger.error(`updateMeta_parameter error: ${error.message}`); this.logger.error(`updateMeta_parameter error: ${error.message}`, {
req,
});
throw new HttpException('参数错误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数错误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const survey = req.surveyMeta; const survey = req.surveyMeta;
survey.title = value.title; survey.title = value.title;
survey.remark = value.remark; survey.remark = value.remark;
await this.surveyMetaService.editSurveyMeta({ await this.surveyMetaService.editSurveyMeta(survey);
survey,
operator: req.user.username,
operatorId: req.user._id.toString(),
});
return { return {
code: 200, code: 200,
@ -83,7 +81,7 @@ export class SurveyMetaController {
) { ) {
const { value, error } = GetSurveyListDto.validate(queryInfo); const { value, error } = GetSurveyListDto.validate(queryInfo);
if (error) { if (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR); throw new HttpException('参数有误', EXCEPTION_CODE.PARAMETER_ERROR);
} }
const { curPage, pageSize, workspaceId } = value; const { curPage, pageSize, workspaceId } = value;
@ -93,14 +91,14 @@ export class SurveyMetaController {
try { try {
filter = getFilter(JSON.parse(decodeURIComponent(value.filter))); filter = getFilter(JSON.parse(decodeURIComponent(value.filter)));
} catch (error) { } catch (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
} }
} }
if (value.order) { if (value.order) {
try { try {
order = order = getOrder(JSON.parse(decodeURIComponent(value.order))); order = order = getOrder(JSON.parse(decodeURIComponent(value.order)));
} catch (error) { } catch (error) {
this.logger.error(error.message); this.logger.error(error.message, { req });
} }
} }
const userId = req.user._id.toString(); const userId = req.user._id.toString();
@ -131,10 +129,9 @@ export class SurveyMetaController {
if (!item.surveyType) { if (!item.surveyType) {
item.surveyType = item.questionType || 'normal'; item.surveyType = item.questionType || 'normal';
} }
item.createdAt = moment(item.createdAt).format(fmt); item.createDate = moment(item.createDate).format(fmt);
item.updateDate = moment(item.updateDate).format(fmt);
item.curStatus.date = moment(item.curStatus.date).format(fmt); item.curStatus.date = moment(item.curStatus.date).format(fmt);
item.subStatus.date = moment(item.subStatus.date).format(fmt);
item.updatedAt = moment(item.updatedAt).format(fmt);
const surveyId = item._id.toString(); const surveyId = item._id.toString();
if (cooperSurveyIdMap[surveyId]) { if (cooperSurveyIdMap[surveyId]) {
item.isCollaborated = true; item.isCollaborated = true;

View File

@ -12,10 +12,10 @@ export class CreateSurveyDto {
surveyType: string; surveyType: string;
@ApiProperty({ description: '创建方法', required: false }) @ApiProperty({ description: '创建方法', required: false })
createMethod?: string; createMethod: string;
@ApiProperty({ description: '创建来源', required: false }) @ApiProperty({ description: '创建来源', required: false })
createFrom?: string; createFrom: string;
@ApiProperty({ description: '问卷创建在哪个空间下', required: false }) @ApiProperty({ description: '问卷创建在哪个空间下', required: false })
workspaceId?: string; workspaceId?: string;

View File

@ -1,51 +0,0 @@
import { ApiProperty } from '@nestjs/swagger';
import Joi from 'joi';
export class CreateDownloadDto {
@ApiProperty({ description: '问卷id', required: true })
surveyId: string;
@ApiProperty({ description: '是否脱敏', required: false })
isMasked: boolean;
static validate(data) {
return Joi.object({
surveyId: Joi.string().required(),
isMasked: Joi.boolean().allow(null).default(false),
}).validate(data);
}
}
export class GetDownloadTaskListDto {
@ApiProperty({ description: '当前页', required: false })
pageIndex: number;
@ApiProperty({ description: '一页大小', required: false })
pageSize: number;
static validate(data) {
return Joi.object({
pageIndex: Joi.number().default(1),
pageSize: Joi.number().default(20),
}).validate(data);
}
}
export class GetDownloadTaskDto {
@ApiProperty({ description: '任务id', required: true })
taskId: string;
static validate(data) {
return Joi.object({
taskId: Joi.string().required(),
}).validate(data);
}
}
export class DeleteDownloadTaskDto {
@ApiProperty({ description: '任务id', required: true })
taskId: string;
static validate(data) {
return Joi.object({
taskId: Joi.string().required(),
}).validate(data);
}
}

View File

@ -22,17 +22,12 @@ export class CollaboratorService {
return this.collaboratorRepository.save(collaborator); return this.collaboratorRepository.save(collaborator);
} }
async batchCreate({ surveyId, collaboratorList, creator, creatorId }) { async batchCreate({ surveyId, collaboratorList }) {
const now = new Date();
const res = await this.collaboratorRepository.insertMany( const res = await this.collaboratorRepository.insertMany(
collaboratorList.map((item) => { collaboratorList.map((item) => {
return { return {
...item, ...item,
surveyId, surveyId,
createdAt: now,
updatedAt: now,
creator,
creatorId,
}; };
}), }),
); );
@ -65,13 +60,7 @@ export class CollaboratorService {
return info; return info;
} }
async changeUserPermission({ async changeUserPermission({ userId, surveyId, permission }) {
userId,
surveyId,
permission,
operator,
operatorId,
}) {
const updateRes = await this.collaboratorRepository.updateOne( const updateRes = await this.collaboratorRepository.updateOne(
{ {
surveyId, surveyId,
@ -80,9 +69,6 @@ export class CollaboratorService {
{ {
$set: { $set: {
permission, permission,
operator,
operatorId,
updatedAt: new Date(),
}, },
}, },
); );
@ -148,7 +134,7 @@ export class CollaboratorService {
return delRes; return delRes;
} }
updateById({ collaboratorId, permissions, operator, operatorId }) { updateById({ collaboratorId, permissions }) {
return this.collaboratorRepository.updateOne( return this.collaboratorRepository.updateOne(
{ {
_id: new ObjectId(collaboratorId), _id: new ObjectId(collaboratorId),
@ -156,9 +142,6 @@ export class CollaboratorService {
{ {
$set: { $set: {
permissions, permissions,
operator,
operatorId,
updatedAt: new Date(),
}, },
}, },
); );

View File

@ -8,10 +8,9 @@ import { keyBy } from 'lodash';
import { DataItem } from 'src/interfaces/survey'; import { DataItem } from 'src/interfaces/survey';
import { ResponseSchema } from 'src/models/responseSchema.entity'; import { ResponseSchema } from 'src/models/responseSchema.entity';
import { getListHeadByDataList, transformAndMergeArrayFields } from '../utils'; import { getListHeadByDataList, transformAndMergeArrayFields } from '../utils';
import { QUESTION_TYPE } from 'src/enums/question';
@Injectable() @Injectable()
export class DataStatisticService { export class DataStatisticService {
private radioType = [QUESTION_TYPE.RADIO_STAR, QUESTION_TYPE.RADIO_NPS]; private radioType = ['radio-star', 'radio-nps'];
constructor( constructor(
@InjectRepository(SurveyResponse) @InjectRepository(SurveyResponse)
@ -34,8 +33,8 @@ export class DataStatisticService {
const dataListMap = keyBy(dataList, 'field'); const dataListMap = keyBy(dataList, 'field');
const where = { const where = {
pageId: surveyId, pageId: surveyId,
isDeleted: { 'curStatus.status': {
$ne: true, $ne: 'removed',
}, },
}; };
const [surveyResponseList, total] = const [surveyResponseList, total] =
@ -44,7 +43,7 @@ export class DataStatisticService {
take: pageSize, take: pageSize,
skip: (pageNum - 1) * pageSize, skip: (pageNum - 1) * pageSize,
order: { order: {
createdAt: -1, createDate: -1,
}, },
}); });
@ -69,7 +68,7 @@ export class DataStatisticService {
} }
// 处理选项的更多输入框 // 处理选项的更多输入框
if ( if (
this.radioType.includes(itemConfig.type as QUESTION_TYPE) && this.radioType.includes(itemConfig.type) &&
!data[`${itemConfigKey}_custom`] !data[`${itemConfigKey}_custom`]
) { ) {
data[`${itemConfigKey}_custom`] = data[`${itemConfigKey}_custom`] =
@ -90,10 +89,10 @@ export class DataStatisticService {
} }
return { return {
...data, ...data,
diffTime: submitedData.diffTime difTime: (submitedData.difTime / 1000).toFixed(2),
? (submitedData.diffTime / 1000).toFixed(2) createDate: moment(submitedData.createDate).format(
: '0', 'YYYY-MM-DD HH:mm:ss',
createdAt: moment(submitedData.createdAt).format('YYYY-MM-DD HH:mm:ss'), ),
}; };
}); });
return { return {
@ -124,8 +123,8 @@ export class DataStatisticService {
{ {
$match: { $match: {
pageId: surveyId, pageId: surveyId,
isDeleted: { 'curStatus.status': {
$ne: true, $ne: 'removed',
}, },
}, },
}, },

View File

@ -1,273 +0,0 @@
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { MongoRepository } from 'typeorm';
import { ResponseSchema } from 'src/models/responseSchema.entity';
import { DownloadTask } from 'src/models/downloadTask.entity';
import { ObjectId } from 'mongodb';
import { ResponseSchemaService } from 'src/modules/surveyResponse/services/responseScheme.service';
import { SurveyResponse } from 'src/models/surveyResponse.entity';
import { DataStatisticService } from './dataStatistic.service';
import xlsx from 'node-xlsx';
import { load } from 'cheerio';
import { get } from 'lodash';
import { FileService } from 'src/modules/file/services/file.service';
import { Logger } from 'src/logger';
import moment from 'moment';
import { DOWNLOAD_TASK_STATUS } from 'src/enums/downloadTaskStatus';
@Injectable()
export class DownloadTaskService {
static taskList: Array<any> = [];
static isExecuting: boolean = false;
constructor(
@InjectRepository(DownloadTask)
private readonly downloadTaskRepository: MongoRepository<DownloadTask>,
private readonly responseSchemaService: ResponseSchemaService,
@InjectRepository(SurveyResponse)
private readonly surveyResponseRepository: MongoRepository<SurveyResponse>,
private readonly dataStatisticService: DataStatisticService,
private readonly fileService: FileService,
private readonly logger: Logger,
) {}
async createDownloadTask({
surveyId,
responseSchema,
creatorId,
creator,
params,
}: {
surveyId: string;
responseSchema: ResponseSchema;
creatorId: string;
creator: string;
params: any;
}) {
const filename = `${responseSchema.title}-${params.isMasked ? '脱敏' : '原'}回收数据-${moment().format('YYYYMMDDHHmmss')}.xlsx`;
const downloadTask = this.downloadTaskRepository.create({
surveyId,
surveyPath: responseSchema.surveyPath,
fileSize: '计算中',
creatorId,
creator,
params: {
...params,
title: responseSchema.title,
},
filename,
status: DOWNLOAD_TASK_STATUS.WAITING,
});
await this.downloadTaskRepository.save(downloadTask);
return downloadTask._id.toString();
}
async getDownloadTaskList({
creatorId,
pageIndex,
pageSize,
}: {
creatorId: string;
pageIndex: number;
pageSize: number;
}) {
const where = {
creatorId,
isDeleted: {
$ne: true,
},
};
const [surveyDownloadList, total] =
await this.downloadTaskRepository.findAndCount({
where,
take: pageSize,
skip: (pageIndex - 1) * pageSize,
order: {
createdAt: -1,
},
});
return {
total,
list: surveyDownloadList,
};
}
async getDownloadTaskById({ taskId }) {
const res = await this.downloadTaskRepository.find({
where: {
_id: new ObjectId(taskId),
},
});
if (Array.isArray(res) && res.length > 0) {
return res[0];
}
return null;
}
async deleteDownloadTask({
taskId,
operator,
operatorId,
}: {
taskId: string;
operator: string;
operatorId: string;
}) {
return this.downloadTaskRepository.updateOne(
{
_id: new ObjectId(taskId),
},
{
$set: {
isDeleted: true,
operator,
operatorId,
deletedAt: new Date(),
},
},
);
}
processDownloadTask({ taskId }) {
DownloadTaskService.taskList.push(taskId);
if (!DownloadTaskService.isExecuting) {
this.executeTask();
DownloadTaskService.isExecuting = true;
}
}
async executeTask() {
try {
while (DownloadTaskService.taskList.length > 0) {
const taskId = DownloadTaskService.taskList.shift();
this.logger.info(`handle taskId: ${taskId}`);
const taskInfo = await this.getDownloadTaskById({ taskId });
if (!taskInfo || taskInfo.isDeleted) {
// 不存在或者已删除的,不处理
continue;
}
await this.handleDownloadTask({ taskInfo });
}
} finally {
DownloadTaskService.isExecuting = false;
}
}
async handleDownloadTask({ taskInfo }) {
try {
// 更新任务状态为计算中
const updateRes = await this.downloadTaskRepository.updateOne(
{
_id: taskInfo._id,
},
{
$set: {
status: DOWNLOAD_TASK_STATUS.COMPUTING,
updatedAt: new Date(),
},
},
);
this.logger.info(JSON.stringify(updateRes));
// 开始计算任务
const surveyId = taskInfo.surveyId;
const responseSchema =
await this.responseSchemaService.getResponseSchemaByPageId(surveyId);
const where = {
pageId: surveyId,
};
const total = await this.surveyResponseRepository.count(where);
const pageSize = 200;
const pageTotal = Math.ceil(total / pageSize);
const xlsxHead = [];
const xlsxBody = [];
for (let pageIndex = 1; pageIndex <= pageTotal; pageIndex++) {
const { listHead, listBody } =
await this.dataStatisticService.getDataTable({
surveyId,
pageNum: pageIndex,
pageSize,
responseSchema,
});
if (xlsxHead.length === 0) {
for (const item of listHead) {
const $ = load(item.title);
const text = $.text();
xlsxHead.push(text);
}
}
for (const bodyItem of listBody) {
const bodyData = [];
for (const headItem of listHead) {
const field = headItem.field;
const val = get(bodyItem, field, '');
if (typeof val === 'string') {
const $ = load(val);
const text = $.text();
bodyData.push(text);
} else {
bodyData.push(val);
}
}
xlsxBody.push(bodyData);
}
}
const xlsxData = [xlsxHead, ...xlsxBody];
const buffer = await xlsx.build([
{ name: 'sheet1', data: xlsxData, options: {} },
]);
const file: Express.Multer.File = {
fieldname: 'file',
originalname: taskInfo.filename,
encoding: '7bit',
mimetype: 'application/octet-stream',
filename: taskInfo.filename,
size: buffer.length,
buffer: buffer,
stream: null,
destination: null,
path: '',
};
const { url, key } = await this.fileService.upload({
configKey: 'SERVER_LOCAL_CONFIG',
file,
pathPrefix: 'exportfile',
filename: taskInfo.filename,
});
// 更新计算结果
const updateFinishRes = await this.downloadTaskRepository.updateOne(
{
_id: taskInfo._id,
},
{
$set: {
status: DOWNLOAD_TASK_STATUS.SUCCEED,
url,
fileKey: key,
fileSize: buffer.length,
updatedAt: new Date(),
},
},
);
this.logger.info(JSON.stringify(updateFinishRes));
} catch (error) {
await this.downloadTaskRepository.updateOne(
{
_id: taskInfo._id,
},
{
$set: {
status: DOWNLOAD_TASK_STATUS.FAILED,
updatedAt: new Date(),
},
},
);
this.logger.error(
`导出文件失败 taskId: ${taskInfo._id.toString()}, surveyId: ${taskInfo.surveyId}, message: ${error.message}`,
);
}
}
}

View File

@ -1,68 +0,0 @@
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { MongoRepository } from 'typeorm';
import { Session } from 'src/models/session.entity';
import { ObjectId } from 'mongodb';
import { SESSION_STATUS } from 'src/enums/surveySessionStatus';
@Injectable()
export class SessionService {
constructor(
@InjectRepository(Session)
private readonly sessionRepository: MongoRepository<Session>,
) {}
create({ surveyId, userId }) {
const session = this.sessionRepository.create({
surveyId,
userId,
status: SESSION_STATUS.DEACTIVATED,
});
return this.sessionRepository.save(session);
}
findOne(sessionId) {
return this.sessionRepository.findOne({
where: {
_id: new ObjectId(sessionId),
},
});
}
findLatestEditingOne({ surveyId }) {
return this.sessionRepository.findOne({
where: {
surveyId,
status: SESSION_STATUS.ACTIVATED,
},
});
}
updateSessionToEditing({ sessionId, surveyId }) {
return Promise.all([
this.sessionRepository.update(
{
_id: new ObjectId(sessionId),
},
{
status: SESSION_STATUS.ACTIVATED,
updatedAt: new Date(),
},
),
this.sessionRepository.updateMany(
{
surveyId,
_id: {
$ne: new ObjectId(sessionId),
},
},
{
$set: {
status: SESSION_STATUS.DEACTIVATED,
updatedAt: new Date(),
},
},
),
]);
}
}

View File

@ -45,9 +45,9 @@ export class SurveyHistoryService {
}, },
take: 100, take: 100,
order: { order: {
createdAt: -1, createDate: -1,
}, },
select: ['createdAt', 'operator', 'type', '_id'], select: ['createDate', 'operator', 'type', '_id'],
}); });
} }
} }

View File

@ -2,18 +2,18 @@ import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm'; import { InjectRepository } from '@nestjs/typeorm';
import { MongoRepository, FindOptionsOrder } from 'typeorm'; import { MongoRepository, FindOptionsOrder } from 'typeorm';
import { SurveyMeta } from 'src/models/surveyMeta.entity'; import { SurveyMeta } from 'src/models/surveyMeta.entity';
import { RECORD_STATUS, RECORD_SUB_STATUS } from 'src/enums'; import { RECORD_STATUS } from 'src/enums';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
import { HttpException } from 'src/exceptions/httpException'; import { HttpException } from 'src/exceptions/httpException';
import { EXCEPTION_CODE } from 'src/enums/exceptionCode'; import { EXCEPTION_CODE } from 'src/enums/exceptionCode';
import { PluginManager } from 'src/securityPlugin/pluginManager'; import { XiaojuSurveyPluginManager } from 'src/securityPlugin/pluginManager';
@Injectable() @Injectable()
export class SurveyMetaService { export class SurveyMetaService {
constructor( constructor(
@InjectRepository(SurveyMeta) @InjectRepository(SurveyMeta)
private readonly surveyRepository: MongoRepository<SurveyMeta>, private readonly surveyRepository: MongoRepository<SurveyMeta>,
private readonly pluginManager: PluginManager, private readonly pluginManager: XiaojuSurveyPluginManager,
) {} ) {}
async getNewSurveyPath(): Promise<string> { async getNewSurveyPath(): Promise<string> {
@ -65,7 +65,6 @@ export class SurveyMetaService {
surveyType: surveyType, surveyType: surveyType,
surveyPath, surveyPath,
creator: username, creator: username,
creatorId: userId,
owner: username, owner: username,
ownerId: userId, ownerId: userId,
createMethod, createMethod,
@ -76,36 +75,11 @@ export class SurveyMetaService {
return await this.surveyRepository.save(newSurvey); return await this.surveyRepository.save(newSurvey);
} }
async pausingSurveyMeta(survey: SurveyMeta) { async editSurveyMeta(survey: SurveyMeta) {
if (survey?.curStatus?.status === RECORD_STATUS.NEW) { if (
throw new HttpException( survey.curStatus.status !== RECORD_STATUS.NEW &&
'问卷不能暂停', survey.curStatus.status !== RECORD_STATUS.EDITING
EXCEPTION_CODE.SURVEY_STATUS_TRANSFORM_ERROR, ) {
);
}
const subCurStatus = {
status: RECORD_SUB_STATUS.PAUSING,
date: Date.now(),
};
survey.subStatus = subCurStatus;
if (Array.isArray(survey.statusList)) {
survey.statusList.push(subCurStatus);
} else {
survey.statusList = [subCurStatus];
}
return this.surveyRepository.save(survey);
}
async editSurveyMeta({
survey,
operator,
operatorId,
}: {
survey: SurveyMeta;
operator: string;
operatorId: string;
}) {
if (survey?.curStatus?.status !== RECORD_STATUS.EDITING) {
const newStatus = { const newStatus = {
status: RECORD_STATUS.EDITING, status: RECORD_STATUS.EDITING,
date: Date.now(), date: Date.now(),
@ -113,26 +87,27 @@ export class SurveyMetaService {
survey.curStatus = newStatus; survey.curStatus = newStatus;
survey.statusList.push(newStatus); survey.statusList.push(newStatus);
} }
survey.updatedAt = new Date();
survey.operator = operator;
survey.operatorId = operatorId;
return this.surveyRepository.save(survey); return this.surveyRepository.save(survey);
} }
async deleteSurveyMeta({ surveyId, operator, operatorId }) { async deleteSurveyMeta(survey: SurveyMeta) {
return this.surveyRepository.updateOne( if (survey.curStatus.status === RECORD_STATUS.REMOVED) {
{ throw new HttpException(
_id: new ObjectId(surveyId), '问卷已删除,不能重复删除',
}, EXCEPTION_CODE.SURVEY_STATUS_TRANSFORM_ERROR,
{ );
$set: { }
isDeleted: true, const newStatusInfo = {
operator, status: RECORD_STATUS.REMOVED,
operatorId, date: Date.now(),
deletedAt: new Date(), };
}, survey.curStatus = newStatusInfo;
}, if (Array.isArray(survey.statusList)) {
); survey.statusList.push(newStatusInfo);
} else {
survey.statusList = [newStatusInfo];
}
return this.surveyRepository.save(survey);
} }
async getSurveyMetaList(condition: { async getSurveyMetaList(condition: {
@ -150,16 +125,14 @@ export class SurveyMetaService {
const skip = (pageNum - 1) * pageSize; const skip = (pageNum - 1) * pageSize;
try { try {
const query: Record<string, any> = Object.assign( const query: Record<string, any> = Object.assign(
{},
{ {
isDeleted: { 'curStatus.status': {
$ne: true, $ne: 'removed',
}, },
}, },
condition.filter, condition.filter,
); );
if (condition.filter['curStatus.status']) {
query['subStatus.status'] = RECORD_SUB_STATUS.DEFAULT;
}
if (workspaceId) { if (workspaceId) {
query.workspaceId = workspaceId; query.workspaceId = workspaceId;
} else { } else {
@ -187,8 +160,9 @@ export class SurveyMetaService {
condition.order && Object.keys(condition.order).length > 0 condition.order && Object.keys(condition.order).length > 0
? (condition.order as FindOptionsOrder<SurveyMeta>) ? (condition.order as FindOptionsOrder<SurveyMeta>)
: ({ : ({
createdAt: -1, createDate: -1,
} as FindOptionsOrder<SurveyMeta>); } as FindOptionsOrder<SurveyMeta>);
const [data, count] = await this.surveyRepository.findAndCount({ const [data, count] = await this.surveyRepository.findAndCount({
where: query, where: query,
skip, skip,
@ -207,10 +181,6 @@ export class SurveyMetaService {
date: Date.now(), date: Date.now(),
}; };
surveyMeta.curStatus = curStatus; surveyMeta.curStatus = curStatus;
surveyMeta.subStatus = {
status: RECORD_SUB_STATUS.DEFAULT,
date: Date.now(),
};
if (Array.isArray(surveyMeta.statusList)) { if (Array.isArray(surveyMeta.statusList)) {
surveyMeta.statusList.push(curStatus); surveyMeta.statusList.push(curStatus);
} else { } else {
@ -222,8 +192,8 @@ export class SurveyMetaService {
async countSurveyMetaByWorkspaceId({ workspaceId }) { async countSurveyMetaByWorkspaceId({ workspaceId }) {
const total = await this.surveyRepository.count({ const total = await this.surveyRepository.count({
workspaceId, workspaceId,
isDeleted: { 'curStatus.status': {
$ne: true, $ne: RECORD_STATUS.REMOVED,
}, },
}); });
return total; return total;

View File

@ -7,7 +7,6 @@ import { LoggerProvider } from 'src/logger/logger.provider';
import { SurveyResponseModule } from '../surveyResponse/surveyResponse.module'; import { SurveyResponseModule } from '../surveyResponse/surveyResponse.module';
import { AuthModule } from '../auth/auth.module'; import { AuthModule } from '../auth/auth.module';
import { WorkspaceModule } from '../workspace/workspace.module'; import { WorkspaceModule } from '../workspace/workspace.module';
import { FileModule } from '../file/file.module';
import { DataStatisticController } from './controllers/dataStatistic.controller'; import { DataStatisticController } from './controllers/dataStatistic.controller';
import { SurveyController } from './controllers/survey.controller'; import { SurveyController } from './controllers/survey.controller';
@ -15,8 +14,6 @@ import { SurveyHistoryController } from './controllers/surveyHistory.controller'
import { SurveyMetaController } from './controllers/surveyMeta.controller'; import { SurveyMetaController } from './controllers/surveyMeta.controller';
import { SurveyUIController } from './controllers/surveyUI.controller'; import { SurveyUIController } from './controllers/surveyUI.controller';
import { CollaboratorController } from './controllers/collaborator.controller'; import { CollaboratorController } from './controllers/collaborator.controller';
import { DownloadTaskController } from './controllers/downloadTask.controller';
import { SessionController } from './controllers/session.controller';
import { SurveyConf } from 'src/models/surveyConf.entity'; import { SurveyConf } from 'src/models/surveyConf.entity';
import { SurveyHistory } from 'src/models/surveyHistory.entity'; import { SurveyHistory } from 'src/models/surveyHistory.entity';
@ -24,21 +21,14 @@ import { SurveyMeta } from 'src/models/surveyMeta.entity';
import { SurveyResponse } from 'src/models/surveyResponse.entity'; import { SurveyResponse } from 'src/models/surveyResponse.entity';
import { Word } from 'src/models/word.entity'; import { Word } from 'src/models/word.entity';
import { Collaborator } from 'src/models/collaborator.entity'; import { Collaborator } from 'src/models/collaborator.entity';
import { DownloadTask } from 'src/models/downloadTask.entity';
import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider'; import { PluginManagerProvider } from 'src/securityPlugin/pluginManager.provider';
import { DataStatisticService } from './services/dataStatistic.service'; import { DataStatisticService } from './services/dataStatistic.service';
import { SurveyConfService } from './services/surveyConf.service'; import { SurveyConfService } from './services/surveyConf.service';
import { SurveyHistoryService } from './services/surveyHistory.service'; import { SurveyHistoryService } from './services/surveyHistory.service';
import { SurveyMetaService } from './services/surveyMeta.service'; import { SurveyMetaService } from './services/surveyMeta.service';
import { ContentSecurityService } from './services/contentSecurity.service'; import { ContentSecurityService } from './services/contentSecurity.service';
import { CollaboratorService } from './services/collaborator.service'; import { CollaboratorService } from './services/collaborator.service';
import { Counter } from 'src/models/counter.entity';
import { CounterService } from '../surveyResponse/services/counter.service';
import { FileService } from '../file/services/file.service';
import { DownloadTaskService } from './services/downloadTask.service';
import { SessionService } from './services/session.service';
import { Session } from 'src/models/session.entity';
@Module({ @Module({
imports: [ imports: [
@ -49,15 +39,11 @@ import { Session } from 'src/models/session.entity';
SurveyResponse, SurveyResponse,
Word, Word,
Collaborator, Collaborator,
Counter,
DownloadTask,
Session,
]), ]),
ConfigModule, ConfigModule,
SurveyResponseModule, SurveyResponseModule,
AuthModule, AuthModule,
WorkspaceModule, WorkspaceModule,
FileModule,
], ],
controllers: [ controllers: [
DataStatisticController, DataStatisticController,
@ -66,8 +52,6 @@ import { Session } from 'src/models/session.entity';
SurveyMetaController, SurveyMetaController,
SurveyUIController, SurveyUIController,
CollaboratorController, CollaboratorController,
DownloadTaskController,
SessionController,
], ],
providers: [ providers: [
DataStatisticService, DataStatisticService,
@ -78,10 +62,6 @@ import { Session } from 'src/models/session.entity';
ContentSecurityService, ContentSecurityService,
CollaboratorService, CollaboratorService,
LoggerProvider, LoggerProvider,
CounterService,
DownloadTaskService,
FileService,
SessionService,
], ],
}) })
export class SurveyModule {} export class SurveyModule {}

View File

@ -43,6 +43,7 @@
"options": [ "options": [
{ {
"text": "选项1", "text": "选项1",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",
@ -51,6 +52,7 @@
}, },
{ {
"text": "选项2", "text": "选项2",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",

View File

@ -39,8 +39,8 @@
"showSpliter": true, "showSpliter": true,
"placeholder": "", "placeholder": "",
"isRequired": true, "isRequired": true,
"starMin": "", "min": "",
"starMax": "", "max": "",
"type": "radio-star", "type": "radio-star",
"title": "标题2" "title": "标题2"
} }

View File

@ -47,6 +47,7 @@
{ {
"text": "课程1", "text": "课程1",
"hash": "115019", "hash": "115019",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",
@ -55,6 +56,7 @@
{ {
"text": "课程2", "text": "课程2",
"hash": "115020", "hash": "115020",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",
@ -63,6 +65,7 @@
{ {
"text": "课程3", "text": "课程3",
"hash": "115021", "hash": "115021",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",
@ -71,6 +74,7 @@
{ {
"text": "课程4", "text": "课程4",
"hash": "115022", "hash": "115022",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",

View File

@ -41,11 +41,12 @@
"innerType": "radio", "innerType": "radio",
"field": "data606", "field": "data606",
"title": "标题2", "title": "标题2",
"minNum": 0, "minNum": "",
"maxNum": 0, "maxNum": "",
"options": [ "options": [
{ {
"text": "选项1", "text": "选项1",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",
@ -54,6 +55,7 @@
}, },
{ {
"text": "选项2", "text": "选项2",
"imageUrl": "",
"others": false, "others": false,
"mustOthers": false, "mustOthers": false,
"othersKey": "", "othersKey": "",

View File

@ -1,69 +1,55 @@
{ {
"bannerConf": { "bannerConf": {
"titleConfig": { "titleConfig": {
"mainTitle": "<h3 style=\"text-align: center\">欢迎填写问卷</h3><p>为了给您提供更好的服务,希望您能抽出几分钟时间,将您的感受和建议告诉我们,<span style=\"color: rgb(204, 0, 0)\">期待您的参与!</span></p>", "mainTitle": "<h3 style=\"text-align: center\">欢迎填写问卷</h3><p>为了给您提供更好的服务,希望您能抽出几分钟时间,将您的感受和建议告诉我们,<span style=\"color: rgb(204, 0, 0)\">期待您的参与!</span></p>",
"subTitle": "<p>为了给您提供更好的服务,希望您能抽出几分钟时间,将您的感受和建议告诉我们,<span style=\"color: rgb(204, 0, 0)\">期待您的参与!</span></p>" "subTitle": ""
},
"bannerConfig": {
"bgImage": "/imgs/skin/17e06b7604a007e1d3e1453b9ddadc3c.webp",
"videoLink": "",
"postImg": ""
}
}, },
"bannerConfig": { "submitConf": {
"bgImage": "/imgs/skin/17e06b7604a007e1d3e1453b9ddadc3c.webp", "submitTitle": "提交",
"bgImageAllowJump": false, "confirmAgain": {
"bgImageJumpLink": "", "is_again": true,
"videoLink": "", "again_text": "确认要提交吗?"
"postImg": "" },
"msgContent": {
"msg_200": "提交成功",
"msg_9001": "您来晚了,感谢支持问卷~",
"msg_9002": "请勿多次提交!",
"msg_9003": "您来晚了,已经满额!",
"msg_9004": "提交失败!"
}
},
"bottomConf": {
"logoImage": "/imgs/Logo.webp",
"logoImageWidth": "60%"
},
"baseConf": {
"begTime": "2024-01-01 00:00:00",
"endTime": "2034-01-01 00:00:00",
"tLimit": 0,
"language": "chinese",
"answerBegTime": "00:00:00",
"answerEndTime": "23:59:59"
},
"skinConf": {
"skinColor": "#4a4c5b",
"inputBgColor": "#ffffff",
"backgroundConf": {
"color": "#fff"
},
"themeConf": {
"color": "#ffa600"
},
"contentConf": {
"opacity": 100
}
},
"logicConf": {
"showLogicConf": []
} }
},
"submitConf": {
"submitTitle": "提交",
"confirmAgain": {
"is_again": true,
"again_text": "确认要提交吗?"
},
"msgContent": {
"msg_200": "提交成功",
"msg_9001": "您来晚了,感谢支持问卷~",
"msg_9002": "请勿多次提交!",
"msg_9003": "您来晚了,已经满额!",
"msg_9004": "提交失败!"
},
"link": ""
},
"bottomConf": {
"logoImage": "/imgs/Logo.webp",
"logoImageWidth": "60%"
},
"baseConf": {
"beginTime": "2024-01-01 00:00:00",
"endTime": "2034-01-01 00:00:00",
"tLimit": 0,
"language": "chinese",
"answerBegTime": "00:00:00",
"answerEndTime": "23:59:59",
"passwordSwitch": false,
"password": "",
"whitelistType": "ALL",
"whitelist": [],
"memberType": "MOBILE",
"fillAnswer": false,
"fillSubmitAnswer": false
},
"skinConf": {
"skinColor": "#4a4c5b",
"inputBgColor": "#ffffff",
"backgroundConf": {
"color": "#b8dbff",
"type": "color",
"image": ""
},
"themeConf": {
"color": "#ffa600"
},
"contentConf": {
"opacity": 100
}
},
"pageConf": [],
"logicConf": {
"showLogicConf": [],
"jumpLogicConf": []
} }
}

View File

@ -5,7 +5,6 @@ import normalCode from '../template/surveyTemplate/survey/normal.json';
import npsCode from '../template/surveyTemplate/survey/nps.json'; import npsCode from '../template/surveyTemplate/survey/nps.json';
import registerCode from '../template/surveyTemplate/survey/register.json'; import registerCode from '../template/surveyTemplate/survey/register.json';
import voteCode from '../template/surveyTemplate/survey/vote.json'; import voteCode from '../template/surveyTemplate/survey/vote.json';
import { QUESTION_TYPE } from 'src/enums/question';
const schemaDataMap = { const schemaDataMap = {
normal: normalCode, normal: normalCode,
@ -22,7 +21,7 @@ export async function getSchemaBySurveyType(surveyType: string) {
} }
const code = Object.assign({}, templateBase, codeData); const code = Object.assign({}, templateBase, codeData);
const nowMoment = moment(); const nowMoment = moment();
code.baseConf.beginTime = nowMoment.format('YYYY-MM-DD HH:mm:ss'); code.baseConf.begTime = nowMoment.format('YYYY-MM-DD HH:mm:ss');
code.baseConf.endTime = nowMoment code.baseConf.endTime = nowMoment
.add(10, 'years') .add(10, 'years')
.format('YYYY-MM-DD HH:mm:ss'); .format('YYYY-MM-DD HH:mm:ss');
@ -32,11 +31,9 @@ export async function getSchemaBySurveyType(surveyType: string) {
export function getListHeadByDataList(dataList) { export function getListHeadByDataList(dataList) {
const listHead = dataList.map((question) => { const listHead = dataList.map((question) => {
let othersCode; let othersCode;
const radioType = [QUESTION_TYPE.RADIO_STAR, QUESTION_TYPE.RADIO_NPS]; const radioType = ['radio-star', 'radio-nps'];
if (radioType.includes(question.type)) { if (radioType.includes(question.type)) {
const rangeConfigKeys = question.rangeConfig const rangeConfigKeys = question.rangeConfig ? Object.keys(question.rangeConfig) : [];
? Object.keys(question.rangeConfig)
: [];
if (rangeConfigKeys.length > 0) { if (rangeConfigKeys.length > 0) {
othersCode = [{ code: `${question.field}_custom`, option: '填写理由' }]; othersCode = [{ code: `${question.field}_custom`, option: '填写理由' }];
} }
@ -58,14 +55,14 @@ export function getListHeadByDataList(dataList) {
}; };
}); });
listHead.push({ listHead.push({
field: 'diffTime', field: 'difTime',
title: '答题耗时(秒)', title: '答题耗时(秒)',
type: QUESTION_TYPE.TEXT, type: 'text',
}); });
listHead.push({ listHead.push({
field: 'createdAt', field: 'createDate',
title: '提交时间', title: '提交时间',
type: QUESTION_TYPE.TEXT, type: 'text',
}); });
return listHead; return listHead;
} }
@ -112,14 +109,7 @@ export function handleAggretionData({ dataMap, item }) {
pre[cur.id] = cur; pre[cur.id] = cur;
return pre; return pre;
}, {}); }, {});
if ( if (['radio', 'checkbox', 'vote', 'binary-choice'].includes(type)) {
[
QUESTION_TYPE.RADIO,
QUESTION_TYPE.CHECKBOX,
QUESTION_TYPE.VOTE,
QUESTION_TYPE.BINARY_CHOICE,
].includes(type)
) {
return { return {
...item, ...item,
title: dataMap[item.field].title, title: dataMap[item.field].title,
@ -135,9 +125,7 @@ export function handleAggretionData({ dataMap, item }) {
}), }),
}, },
}; };
} else if ( } else if (['radio-star', 'radio-nps'].includes(type)) {
[QUESTION_TYPE.RADIO_STAR, QUESTION_TYPE.RADIO_NPS].includes(type)
) {
const summary: Record<string, any> = {}; const summary: Record<string, any> = {};
const average = getAverage({ aggregation: item.data.aggregation }); const average = getAverage({ aggregation: item.data.aggregation });
const median = getMedian({ aggregation: item.data.aggregation }); const median = getMedian({ aggregation: item.data.aggregation });
@ -148,10 +136,10 @@ export function handleAggretionData({ dataMap, item }) {
summary['average'] = average; summary['average'] = average;
summary['median'] = median; summary['median'] = median;
summary['variance'] = variance; summary['variance'] = variance;
if (type === QUESTION_TYPE.RADIO_NPS) { if (type === 'radio-nps') {
summary['nps'] = getNps({ aggregation: item.data.aggregation }); summary['nps'] = getNps({ aggregation: item.data.aggregation });
} }
const range = type === QUESTION_TYPE.RADIO_NPS ? [0, 10] : [1, 5]; const range = type === 'radio-nps' ? [0, 10] : [1, 5];
const arr = []; const arr = [];
for (let i = range[0]; i <= range[1]; i++) { for (let i = range[0]; i <= range[1]; i++) {
arr.push(i); arr.push(i);

View File

@ -3,6 +3,7 @@ import { MongoRepository } from 'typeorm';
import { ClientEncryptService } from '../services/clientEncrypt.service'; import { ClientEncryptService } from '../services/clientEncrypt.service';
import { ClientEncrypt } from 'src/models/clientEncrypt.entity'; import { ClientEncrypt } from 'src/models/clientEncrypt.entity';
import { ENCRYPT_TYPE } from 'src/enums/encrypt'; import { ENCRYPT_TYPE } from 'src/enums/encrypt';
import { RECORD_STATUS } from 'src/enums';
import { ObjectId } from 'mongodb'; import { ObjectId } from 'mongodb';
import { getRepositoryToken } from '@nestjs/typeorm'; import { getRepositoryToken } from '@nestjs/typeorm';
@ -21,7 +22,6 @@ describe('ClientEncryptService', () => {
save: jest.fn(), save: jest.fn(),
findOne: jest.fn(), findOne: jest.fn(),
updateOne: jest.fn(), updateOne: jest.fn(),
deleteOne: jest.fn(),
}, },
}, },
], ],
@ -88,6 +88,9 @@ describe('ClientEncryptService', () => {
expect(repository.findOne).toHaveBeenCalledWith({ expect(repository.findOne).toHaveBeenCalledWith({
where: { where: {
_id: new ObjectId(id), _id: new ObjectId(id),
'curStatus.status': {
$ne: RECORD_STATUS.REMOVED,
},
}, },
}); });
expect(result).toEqual(encryptInfo); expect(result).toEqual(encryptInfo);
@ -106,13 +109,11 @@ describe('ClientEncryptService', () => {
describe('deleteEncryptInfo', () => { describe('deleteEncryptInfo', () => {
it('should delete encrypt info by id', async () => { it('should delete encrypt info by id', async () => {
const id = new ObjectId().toHexString(); const id = new ObjectId().toHexString();
const deleteResult = { matchedCount: 1, modifiedCount: 1 }; const updateResult = { matchedCount: 1, modifiedCount: 1 };
jest jest.spyOn(repository, 'updateOne').mockResolvedValue(updateResult);
.spyOn(repository, 'deleteOne')
.mockResolvedValue(deleteResult as any);
const result = await service.deleteEncryptInfo(id); const result = await service.deleteEncryptInfo(id);
expect(result).toEqual(deleteResult); expect(result).toEqual(updateResult);
}); });
}); });
}); });

View File

@ -50,7 +50,6 @@ describe('CounterService', () => {
surveyPath: 'testPath', surveyPath: 'testPath',
type: 'testType', type: 'testType',
data, data,
updatedAt: expect.any(Date),
}, },
}, },
{ upsert: true }, { upsert: true },

Some files were not shown because too many files have changed in this diff Show More