fix slow response: remove batchSize options

elmasy-com · Jul 19, 2023 · 1effae8 · 1effae8
1 parent 04fe093
commit 1effae8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/db/lookup.go b/db/lookup.go
@@ -10,7 +10,6 @@ import (
 	"github.com/elmasy-com/slices"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
-	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 // Lookup query the DB and returns a list subdomains only.
@@ -52,7 +51,7 @@ func Lookup(d string, days int) ([]string, error) {
 	}
 
 	// Use Find() to find every shard of the domain
-	cursor, err := Domains.Find(context.TODO(), doc, options.Find().SetBatchSize(100))
+	cursor, err := Domains.Find(context.TODO(), doc)
 	if err != nil {
 		return nil, fmt.Errorf("failed to find: %s", err)
 	}
@@ -118,7 +117,7 @@ func LookupFull(d string, days int) ([]string, error) {
 	}
 
 	// Use Find() to find every shard of the domain
-	cursor, err := Domains.Find(context.TODO(), doc, options.Find().SetBatchSize(100))
+	cursor, err := Domains.Find(context.TODO(), doc)
 	if err != nil {
 		return nil, fmt.Errorf("failed to find: %s", err)
 	}

diff --git a/db/records.go b/db/records.go
@@ -405,7 +405,7 @@ func TopListUpdater(wg *sync.WaitGroup) {
 
 		start := time.Now()
 
-		cursor, err := TopList.Find(context.TODO(), bson.M{}, options.Find().SetBatchSize(10).SetSort(bson.M{"count": -1}))
+		cursor, err := TopList.Find(context.TODO(), bson.M{}, options.Find().SetSort(bson.M{"count": -1}))
 		if err != nil {
 			fmt.Fprintf(os.Stderr, "TopListUpdater() failed to find toplist: %s\n", err)
 			continue