server.js 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. const express = require('express');
  2. const mongoose = require('mongoose');
  3. const dotenv = require('dotenv');
  4. const userRoutes = require('./routes/user');
  5. const triangleRoutes = require('./routes/triangle');
  6. const app = express();
  7. dotenv.config();
  8. // 中间件
  9. app.use(express.json({ limit: '10mb' }));
  10. app.use((req, res, next) => {
  11. res.badRequest = (msg) => {
  12. res.status(400).json({ statusCode: 400, code: -1, message: msg ?? 'Bad Request' });
  13. }
  14. res.unauthorized = (msg) => {
  15. res.status(401).json({ statusCode: 401, code: -1, message: msg ?? 'Unauthorized' });
  16. }
  17. res.serverError = (msg) => {
  18. res.status(500).json({ statusCode: 500, code: -1, message: msg ?? 'Internal Server Error' });
  19. }
  20. res.sendSuccess = (data, msg) => {
  21. const response = { statusCode: 200, code: 0, message: msg ?? 'OK' }
  22. if (data) {
  23. response.data = data;
  24. }
  25. res.status(200).json(response);
  26. }
  27. next();
  28. });
  29. // 路由
  30. app.use('/api/user', userRoutes);
  31. app.use('/api/triangle', triangleRoutes);
  32. // 启动服务
  33. const PORT = process.env.PORT || 9055;
  34. mongoose.connect(process.env.MONGO_URI)
  35. .then(() => {
  36. console.log('MongoDB connected');
  37. app.listen(PORT, () => console.log(`Server running on port ${PORT}`));
  38. })
  39. .catch(err => console.error(err));