scripts/check-mongo-connection.js

   1 #!/usr/bin/env node
   2
   3 'use strict';
   4
   5 require('gitter-web-persistence');
   6 var mongoose = require('mongoose');
   7
   8 function die(stage, error) {
   9   console.error('DB connection failed during ' + stage);
  10   console.error(error);
  11   console.error(error.stack);
  12   process.exit(1);
  13 }
  14
  15 try {
  16   console.log('Connecting to database');
  17
  18   var db = mongoose.connection.db;
  19   var adminDb = db.admin();
  20
  21   mongoose.connection.on('open', function() {
  22     adminDb.ping(function(err, pingResult) {
  23       if (err) return die('ping', err);
  24       if (
  25         !pingResult ||
  26         !pingResult.documents ||
  27         !pingResult.documents.length ||
  28         !pingResult.documents[0] ||
  29         !pingResult.documents[0].ok
  30       )
  31         return die('ping', 'ping fail');
  32
  33       adminDb.replSetGetStatus(function(err, info) {
  34         if (err) return die('repl', err);
  35         if (!info || info.myState !== 1) return die('repl', 'replica set fail');
  36
  37         var pingtestCollection = db.collection('pingtest');
  38         pingtestCollection.insert({ ping: Date.now() }, function(err) {
  39           if (err) return die('insert', err);
  40
  41           pingtestCollection.remove({}, function(err) {
  42             if (err) return die('remove', err);
  43
  44             process.exit(0);
  45           });
  46         });
  47       });
  48     });
  49   });
  50 } catch (e) {
  51   die('ping', e);
  52 }