app.js

const winston = require('./config/winston');

var cluster = require('cluster');
var os = require('os');
var uuid = require('uuid');

//키생성 - 서버 확인용
var instance_id = uuid.v4();

// 워커 생성

// var cpuCount = os.cpus().length; //CPU 수
var cpuCount = 1;
var workerCount = cpuCount / 2; //2개의 컨테이너에 돌릴 예정 CPU수 / 2

//마스터일 경우
if (cluster.isMaster) {
  winston.log('info', '서버 ID : ' + instance_id);
  winston.log('info', '서버 CPU 수 : ' + cpuCount + ', 생성할 워커 수 : ' + workerCount);

  //워커 메시지 리스너
  var workerMsgListener = function (msg) {
    var worker_id = msg.worker_id;

    //마스터 아이디 요청
    if (msg.cmd === 'MASTER_ID') {
      cluster.workers[worker_id].send({ cmd: 'MASTER_ID', master_id: instance_id });
    }
  }

  //CPU 수 만큼 워커 생성
  for (var i = 0; i < workerCount; i++) {
    winston.log('info', "워커 생성 [" + (i + 1) + "/" + workerCount + "]");
    var worker = cluster.fork();

    //워커의 요청메시지 리스너
    worker.on('message', workerMsgListener);
  }

  //워커가 online상태가 되었을때
  cluster.on('online', function (worker) {
    winston.log('info', '워커 온라인 - 워커 ID : [' + worker.process.pid + ']');
  });

  //워커가 죽었을 경우 다시 살림
  cluster.on('exit', function (worker) {
    winston.log('info', '워커 사망 - 사망한 워커 ID : [' + worker.process.pid + ']');
    winston.log('info', '다른 워커를 생성합니다.');

    var worker = cluster.fork();
    //워커의 요청메시지 리스너
    worker.on('message', workerMsgListener);
  });

  //워커일 경우
} else if (cluster.isWorker) {
  var createError = require('http-errors');
  var express = require('express');
  var path = require('path');
  var cookieParser = require('cookie-parser');
  var morgan = require('morgan');

  // 마스터, 워커 ID 설정
  var worker_id = cluster.worker.id;
  var master_id;

   //마스터에게 master_id 요청
   process.send({worker_id: worker_id, cmd:'MASTER_ID'});
   process.on('message', function (msg){
       if (msg.cmd === 'MASTER_ID') {
           master_id = msg.master_id;
       }
   }).on('unhandledRejection', (reason, p) => {
    winston.log('error', reason, ' [Unhandled Rejection at Promise] ', p);
  }).on('uncaughtException', (err) => {
    try{
      // 에러가 발생하면 3초 이내에 process를 죽인다.
      const killtimer = setTimeout(function() {
        process.exit(1);
      }, 3000);

      // setTimeout을 현재 process와 독립적으로 동작하도록 레퍼런스 제거
      killtimer.unref();

      // 워커가 죽은 것을 마스터에게 알린다.
      cluster.worker.disconnect();

      // 에러 로그 남김
      winston.log('error', '[Uncaught Exception thrown] ' + err.stack);
    }catch (e) {
      winston.log('error', '프로세스 종료 과정에서 에러 발생 ' + e.stack);
    }
  });

  const session = require('express-session');
  const flash = require('connect-flash');
  const passport = require('passport');
  const cors = require('cors');
  const hpp = require('hpp');
  const helmet = require('helmet');
  require('dotenv').config(); //.env 설정

  //router 객체 선언
  const authRouter = require('./routes/auth');
  const feedbackRouter = require('./routes/feedback');
  const categoryRouter = require('./routes/category');
  const boardRouter = require('./routes/board/all');
  const boardTextRouter = require('./routes/board/text');
  const boardPictureRouter = require('./routes/board/picture');
  const boardVideoRouter = require('./routes/board/video');
  const boardRecordRouter = require('./routes/board/record');
  const mypageRouter = require('./routes/mypage');
  const friendRouter = require('./routes/friend');
  const commentRouter = require('./routes/comment');

  const prod = process.env.NODE_ENV === 'production';

  const { sequelize } = require('./models');
  const passportConfig = require('./passport');
  const app = express();
  sequelize.sync();
  passportConfig(passport);

  app.set('port', process.env.PORT || 3000); //포트 설정

  if (prod) {
    app.use(hpp());
    app.use(helmet());
    app.use(morgan('combined'));
    app.use(cors({
      origin: true,
      credentials: true,
    }));
  } else {
    app.use(morgan('dev'));
    app.use(cors({
      origin: true,
      credentials: true,
    }));
  }

  app.use('/', express.static('public'));
  app.use('/mypage', express.static('public/uploads'));

  app.use(express.json());
  app.use(express.urlencoded({ extended: false }));
  app.use(cookieParser(process.env.COOKIE_SECRET));
  app.use(session({
    resave: false,
    saveUninitialized: false,
    secret: process.env.COOKIE_SECRET,
    cookie: {
      httpOnly: true,
      secure: false,
      domain: '.remindfeedback.com',
    },
  }));
  // app.use(express.static(path.join(__dirname, 'public')));
  app.use(flash()); //1회성 메세지
  app.use(passport.initialize()); //설정초기화 (미들웨어 연결)
  app.use(passport.session()); //로그인시 로컬로 로그인했을때 세션에 저장하는 역할
  app.get('/favicon.ico', (req, res) => {
    res.status(204);
  })

  app.get('/', (req, res) => { 
    res.send('remindfeedback 백엔드 정상 동작!');
  });
  
  // // Worker 테스트 주소
  // app.get('/worker', function (req, res) {
  //   res.send('안녕하세요 저는<br>['+master_id+']서버의<br>워커 ['+ cluster.worker.id+'] 입니다.');
  // });
  // Worker Killer 호출
  // app.get("/workerKiller", function (req, res) {
  //   cluster.worker.kill();
  //   res.send('워커킬러 호출됨');
  // });
  
  app.use('/auth', authRouter);
  app.use('/feedbacks', feedbackRouter);
  app.use('/categories', categoryRouter);
  app.use('/board/cards', boardRouter);
  app.use('/board/cards/text', boardTextRouter);
  app.use('/board/cards/picture', boardPictureRouter);
  app.use('/board/cards/video', boardVideoRouter);
  app.use('/board/cards/record', boardRecordRouter);
  app.use('/mypage', mypageRouter);
  app.use('/friends', friendRouter);
  app.use('/comments', commentRouter);

  // catch 404 and forward to error handler
  app.use(function (req, res, next) {
    next(createError(404));
  });

  // error handler
  app.use(function (err, req, res, next) {
    // set locals, only providing error in development
    res.locals.message = err.message;
    res.locals.error = req.app.get('env') === 'development' ? err : {};

    // 에러 로그 기록
    winston.log('error', err.stack);

    // render the error page
    res.status(err.status || 500);
    res.render('error');
  });

  // require("greenlock-express")
  //   .init({
  //     packageRoot: __dirname,
  //     configDir: process.env.HTTPS_CONFIGDIR,
  //     maintainerEmail: process.env.DOMAIN_EMAIL,
  //     cluster: false
  //   })
  //   .serve(httpsServer);

  // function httpsServer(glx) {
  //   glx.serveApp(app);
  // }

  app.listen(prod ? app.get('port') : 3000, () => {
    winston.log('info', `${app.get('port')}번 포트에서 서버 실행중입니다.`);
  });

  module.exports = app;
}