fix: one tag match only

ShugetsuSoft · Nov 30, 2023 · e977700 · e977700
1 parent c3635a3
commit e977700
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/common/database/operations/illust.go b/common/database/operations/illust.go
@@ -376,11 +376,17 @@ func (ops *DatabaseOperations) SearchIllust(ctx context.Context, keyword string,
 func (ops *DatabaseOperations) QueryIllustsByTags(ctx context.Context, tags []string, page int64, limit int64, sortpopularity bool, sortdate bool, resultbanned bool) ([]models.Illust, error) {
 	var results []models.Illust
 
+	var filter interface{}
+
+	if len(tags) > 1 {
+		filter = bson.M{"$in": tags}
+	} else {
+		filter = tags[0]
+	}
+
 	query := bson.M{
-		"tags.name": bson.M{
-			"$in": tags,
-		},
-		"banned": false,
+		"tags.name": filter,
+		"banned":    false,
 	}
 	if resultbanned {
 		query["banned"] = true