RetryPrint.h

Go to the documentation of this file.

#pragma once
#include "Print.h"
 
namespace tiny_dlna {
 
class RetryPrint : public Print {
 public:
  RetryPrint(Print& out, int maxRetries = 3)
      : out(out), maxRetries(maxRetries) {}
 
  size_t write(uint8_t ch) override { return write(&ch, 1); }
 
  size_t write(const uint8_t* buffer, size_t size) override {
    size_t totalWritten = 0;
    int retries = 0;
    while (totalWritten < size && retries < maxRetries) {
      size_t written = out.write(buffer + totalWritten, size - totalWritten);
      if (written == 0) {
        ++retries;
        delay(10);  // Small delay before retry
      } else {
        totalWritten += written;
        retries = 0;  // Reset retries on progress
      }
    }
    return totalWritten;
  }
 
 protected:
  Print& out;
  int maxRetries;
};
 
}  // namespace tiny_dlna