diff --git a/src/app.ts b/src/app.ts index bb98bbc..b629bca 100644 --- a/src/app.ts +++ b/src/app.ts @@ -6,6 +6,7 @@ import imageController from './modules/image/file.controller'; import config from './utils/core/config'; import './services/ws/wsServer'; import compression from 'compression'; +import testController from './modules/test/test.controller'; const apps = { async createApp() { @@ -23,6 +24,7 @@ const apps = { const modules = [ { path: '/api/sample', name: '测试模块', controller: sampleController }, { path: '/public', name: '文件模块', controller: imageController }, + { path: '/test', name: '测试', controller: testController }, ]; modules.forEach((module) => { diff --git a/src/modules/test/test.controller.ts b/src/modules/test/test.controller.ts new file mode 100644 index 0000000..7ad9fc0 --- /dev/null +++ b/src/modules/test/test.controller.ts @@ -0,0 +1,31 @@ +import express from 'express'; +import TestService from './test.service'; +import response from '../../utils/core/response'; +import logger from '../../utils/core/logger'; + +class TestController { + private readonly router: express.Router; + + constructor() { + this.router = express.Router(); + this.initRouter(); + } + + public getRouter(): express.Router { + return this.router; + } + + public initRouter(): void { + this.router.get('/test', this.test); + } + private test = async (req: express.Request, res: express.Response): Promise => { + try { + const result = await TestService.test(); + await response.success(res, result); + } catch (err) { + logger.error(err); + } + }; +} + +export default new TestController(); diff --git a/src/modules/test/test.service.ts b/src/modules/test/test.service.ts new file mode 100644 index 0000000..3dead06 --- /dev/null +++ b/src/modules/test/test.service.ts @@ -0,0 +1,17 @@ +import wsServer from '../../services/ws/wsServer'; +import wsClientManager from '../../services/ws/wsClientManager'; +import logger from '../../utils/core/logger'; + +class TestService { + public async test() { + try { + const testData = { type: 'test', data: '114514' }; + await wsClientManager.send('test', JSON.stringify(testData)); + return { message: 'ok' }; + } catch (err) { + logger.error(err); + } + } +} + +export default new TestService();