Add Index for CVEs

SpiderLabs · Aug 7, 2020 · 670adc3 · 670adc3
1 parent bfced02
commit 670adc3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/bin/seed b/bin/seed
@@ -40,4 +40,6 @@ Dir.glob(files).sort.each do |infile|
   CVEServer::Cve.bulk_create(@doc.all_cve)
 end
 puts "Reducing the cpe list"
-CVEServer::Cve.reduce_cpes
+CVEServer::Cve.reduce_cpes
+puts 'Creating CVE collection Index'
+CVEServer::Cve.create_index(:cve)
diff --git a/lib/cve_server/db/mongo.rb b/lib/cve_server/db/mongo.rb
@@ -35,6 +35,10 @@ def bulk_create(data)
         db[@collection].bulk_write(inserts, ordered: true)
       end
 
+      def create_index(param)
+        db[@collection].indexes.create_one( { param => 1 } )
+      end
+
       def drop_all
         db[@collection].drop
       end