refactor: add config for autoIndex

AlmogVC · Dec 21, 2019 · d239a92 · d239a92
1 parent 5d7fbef
commit d239a92
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/src/aliveSignal/aliveSignal.model.ts b/src/aliveSignal/aliveSignal.model.ts
@@ -28,7 +28,7 @@ const aliveSignalSchema: mongoose.Schema = new mongoose.Schema(
     },
     {
         versionKey: false,
-        autoIndex: false,
+        autoIndex: config.db.autoIndex,
         id: true,
         toJSON: {
             virtuals: true,

diff --git a/src/config.ts b/src/config.ts
@@ -6,6 +6,7 @@ const config = {
     db: {
         connectionString: `mongodb://${process.env.DB_SERVERS || 'localhost:27017'}/${process.env.KEEP_ALIVE_DB_NAME ||
             'drop-box-keep-alive'}${process.env.DB_REPLICA_NAME ? `?replicaSet=${process.env.DB_REPLICA_NAME}` : ''}`,
+        autoIndex: process.env.MONGO_AUTO_INDEX === '1' || false,
     },
     cors: {
         allowedOrigins: process.env.ALLOWED_ORIGINS ? process.env.ALLOWED_ORIGINS.split(',') : ['*'],

diff --git a/src/service/service.model.ts b/src/service/service.model.ts
@@ -1,12 +1,14 @@
 import * as mongoose from 'mongoose';
 import IService from './service.interface';
+import config from '../config';
 
 const serviceSchema: mongoose.Schema = new mongoose.Schema(
     {
         name: {
             type: String,
             required: true,
             unique: true,
+            index: true,
             immutable: true,
         },
         lastContactDate: {
@@ -31,7 +33,7 @@ const serviceSchema: mongoose.Schema = new mongoose.Schema(
     },
     {
         versionKey: false,
-        autoIndex: false,
+        autoIndex: config.db.autoIndex,
         timestamps: { createdAt: true, updatedAt: true },
         id: true,
         toJSON: {
@@ -43,7 +45,6 @@ const serviceSchema: mongoose.Schema = new mongoose.Schema(
     },
 );
 
-serviceSchema.index({ name: 1 });
 const ServiceModel = mongoose.model<IService & mongoose.Document>('Service', serviceSchema);
 
 export default ServiceModel;
diff --git a/src/serviceHost/serviceHost.model.ts b/src/serviceHost/serviceHost.model.ts
@@ -24,7 +24,7 @@ const serviceHostSchema: mongoose.Schema = new mongoose.Schema(
     },
     {
         versionKey: false,
-        autoIndex: false,
+        autoIndex: config.db.autoIndex,
         timestamps: { createdAt: true },
         id: true,
         toJSON: {