MongoDB
 sql >> Database >  >> NoSQL >> MongoDB

Il tasso di richiesta di CosmosDb è elevato con insertMany

Il driver Mongo ti dice quali record hanno ricevuto errori e quali non sono stati elaborati affatto. Se tutti gli errori (di solito uno) hanno il codice 16500, il problema è la limitazione e riprovare sugli errori e i record rimanenti sono al sicuro. Altrimenti i tuoi errori sono causati da qualcos'altro e dovresti fare un'analisi e decidere se continuare con i tentativi.

Il driver Mongo non restituisce l'intestazione HTTP in cui Cosmos DB suggerisce un ritardo prima di riprovare, ma non è un grosso problema. Il ritardo non garantisce comunque il successo, perché altre richieste che colpiscono lo stesso database potrebbero esaurire le RU. È meglio sperimentare e determinare le proprie regole per i tentativi. Di seguito è riportata una semplice soluzione ricorsiva che continua a riprovare finché tutto non va bene o viene raggiunto il limite di tentativi.

    private async Task InsertManyWithRetry(IMongoCollection<BsonDocument> collection, 
        IEnumerable<BsonDocument> batch, int retries = 10, int delay = 300)
    {
        var batchArray = batch.ToArray();

        try
        {
            await collection.InsertManyAsync(batchArray);
        }
        catch (MongoBulkWriteException<BsonDocument> e)
        {
            if (retries <= 0)
                throw;

            //Check if there were any errors other than throttling.
            var realErrors = e.WriteErrors.Where(we => we.Code != 16500).ToArray();
            //Re-throw original exception for now.
            //TODO: We can make it more sophisticated by continuing with unprocessed records and collecting all errors from all retries.
            if (realErrors.Any())
                throw;

            //Take all records that had errors.
            var errors = e.WriteErrors.Select(we => batchArray[we.Index]);
            //Take all unprocessed records.
            var unprocessed = e.UnprocessedRequests
                .Where(ur => ur.ModelType == WriteModelType.InsertOne)
                .OfType<InsertOneModel<BsonDocument>>() 
                .Select(ur => ur.Document);

            var retryBatchArray = errors.Union(unprocessed).ToArray();

            _logger($"Retry {retryBatchArray.Length} records after {delay} ms");

            await Task.Delay(delay);

            await InsertManyWithRetry(collection, retryBatchArray, retries - 1, delay);
        }
    }