fix context.TODO()

micromdm · Oct 21, 2022 · 96e93e5 · 96e93e5
1 parent 41b47ef
commit 96e93e5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/storage/mongodb/push.go b/storage/mongodb/push.go
@@ -16,13 +16,13 @@ func (m MongoDBStorage) RetrievePushInfo(ctx context.Context, ids []string) (map
 			"$in": ids,
 		},
 	}
-	cursor, err := m.CheckinCollection.Find(context.TODO(), filter)
+	cursor, err := m.CheckinCollection.Find(ctx, filter)
 	if err != nil {
 		return nil, err
 	}
 
 	records := []DeviceCheckinRecord{}
-	err = cursor.All(context.TODO(), &records)
+	err = cursor.All(ctx, &records)
 	if err != nil {
 		return nil, err
 	}

diff --git a/storage/mongodb/pushcert.go b/storage/mongodb/pushcert.go
@@ -27,7 +27,7 @@ func (m MongoDBStorage) IsPushCertStale(ctx context.Context, topic string, stale
 		"topic": topic,
 	}
 	res := PushCertRecord{}
-	err := m.PushCertCollection.FindOne(context.TODO(), filter, options.FindOne().SetSort(latestSort)).Decode(&res)
+	err := m.PushCertCollection.FindOne(ctx, filter, options.FindOne().SetSort(latestSort)).Decode(&res)
 	if err != nil {
 		return false, err
 	}
@@ -39,7 +39,7 @@ func (m MongoDBStorage) RetrievePushCert(ctx context.Context, topic string) (cer
 		"topic": topic,
 	}
 	res := PushCertRecord{}
-	err = m.PushCertCollection.FindOne(context.TODO(), filter, options.FindOne().SetSort(latestSort)).Decode(&res)
+	err = m.PushCertCollection.FindOne(ctx, filter, options.FindOne().SetSort(latestSort)).Decode(&res)
 	if err != nil {
 		return nil, "", err
 	}
@@ -58,7 +58,7 @@ func (m MongoDBStorage) StorePushCert(ctx context.Context, pemCert, pemKey []byt
 		return err
 	}
 
-	_, err = m.PushCertCollection.InsertOne(context.TODO(), PushCertRecord{
+	_, err = m.PushCertCollection.InsertOne(ctx, PushCertRecord{
 		Timestamp:   strconv.FormatInt(time.Now().UnixNano(), 10),
 		Certificate: string(pemCert),
 		PrivateKey:  string(pemKey),