mongo: Use ordered map for search parameters

Mongo 1.5 requires the use of ordered maps when executing a sort operation or it will panic. changelog: none Signed-off-by: Alf-Rune Siqveland <[email protected]>
mendersoftware · Mar 24, 2021 · da0e78e · da0e78e
1 parent d525aa3
commit da0e78e
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/store/mongo/datastore_mongo.go b/store/mongo/datastore_mongo.go
@@ -210,10 +210,9 @@ func (db *DataStoreMongo) GetDevices(ctx context.Context, q store.ListQuery) ([]
 	if q.Sort != nil {
 		name := fmt.Sprintf("%s-%s", q.Sort.AttrScope, model.GetDeviceAttributeNameReplacer().Replace(q.Sort.AttrName))
 		sortField := fmt.Sprintf("%s.%s.%s", DbDevAttributes, name, DbDevAttributesValue)
-		sortFieldQuery := bson.M{}
-		sortFieldQuery[sortField] = 1
+		sortFieldQuery := bson.D{{Key: sortField, Value: 1}}
 		if !q.Sort.Ascending {
-			sortFieldQuery[sortField] = -1
+			sortFieldQuery[0].Value = -1
 		}
 		findOptions.SetSort(sortFieldQuery)
 	}
@@ -955,13 +954,13 @@ func (db *DataStoreMongo) SearchDevices(ctx context.Context, searchParams model.
 	}
 
 	if len(searchParams.Sort) > 0 {
-		sortField := bson.M{}
-		for _, sortQ := range searchParams.Sort {
+		sortField := make(bson.D, len(searchParams.Sort))
+		for i, sortQ := range searchParams.Sort {
 			name := fmt.Sprintf("%s-%s", sortQ.Scope, model.GetDeviceAttributeNameReplacer().Replace(sortQ.Attribute))
 			field := fmt.Sprintf("%s.%s.%s", DbDevAttributes, name, DbDevAttributesValue)
-			sortField[field] = 1
+			sortField[i] = bson.E{Key: field, Value: 1}
 			if sortQ.Order == "desc" {
-				sortField[field] = -1
+				sortField[i].Value = -1
 			}
 		}
 		findOptions.SetSort(sortField)