StimBuff: Add skeleton common impl with rate limiting

This commit is contained in:
2025-10-31 13:43:23 -04:00
parent 3bf8146ca3
commit ebbb2b1345
3 changed files with 112 additions and 116 deletions
+110 -2
View File
@@ -6,6 +6,13 @@
#include <cstdint>
#include <atomic>
#include <mutex>
#include <functional>
#include <iostream>
#include <config.h>
#include <boost/asio/io_service.hpp>
#include <boost/asio/deadline_timer.hpp>
#include <spinLock.h>
#include <asynchronousBridge.h>
#include "stimFrame.h"
#include "deviceAttachmentSpec.h"
@@ -40,8 +47,11 @@ public:
public:
explicit StimulusBuffer(
const device::DeviceAttachmentSpec& deviceAttachmentSpec)
: deviceAttachmentSpec(deviceAttachmentSpec)
const device::DeviceAttachmentSpec& deviceAttachmentSpec,
boost::asio::io_service& ioService)
: deviceAttachmentSpec(deviceAttachmentSpec),
ioService(ioService),
shouldContinue(false), timer(ioService)
{}
~StimulusBuffer() = default;
@@ -52,9 +62,107 @@ public:
StimulusBuffer(StimulusBuffer&&) = default;
StimulusBuffer& operator=(StimulusBuffer&&) = default;
// Control methods
void start()
{
shouldContinue.store(true);
scheduleNextTimeout();
}
void stop()
{
shouldContinue.store(false);
// Set up a timeout bridge using the io_service
boost::asio::deadline_timer delayTimer(ioService);
AsynchronousBridge bridge(ioService);
// Set up the delay for CONFIG_STIMBUFF_FRAME_PERIOD_MS to let in-flight
// operation finish
delayTimer.expires_from_now(
boost::posix_time::milliseconds(CONFIG_STIMBUFF_FRAME_PERIOD_MS));
delayTimer.async_wait(
[&bridge](const boost::system::error_code& error)
{
(void)error;
// Always signal complete, whether timeout expired or was cancelled
bridge.setAsyncOperationComplete();
});
bridge.waitForAsyncOperationCompleteOrIoServiceStopped();
// After delay, cancel timer and perform cleanup
timer.cancel();
}
public:
device::DeviceAttachmentSpec deviceAttachmentSpec;
std::vector<StimFrame> frames_;
protected:
SpinLock frameAssemblyRateLimiter;
private:
boost::asio::io_service& ioService;
std::atomic<bool> shouldContinue;
boost::asio::deadline_timer timer;
void scheduleNextTimeout(int delayMs = CONFIG_STIMBUFF_FRAME_PERIOD_MS)
{
if (!shouldContinue.load())
{ return; }
// Schedule the next timeout using the provided delay
timer.expires_from_now(
boost::posix_time::milliseconds(delayMs));
timer.async_wait(
std::bind(
&StimulusBuffer::onTimeout, this, std::placeholders::_1));
}
void onTimeout(const boost::system::error_code& error)
{
// Timer was cancelled, which is expected when stopping
if (error == boost::asio::error::operation_aborted) {
return;
}
if (error)
{
std::cerr << "StimulusBuffer: Timer error: " << error.message()
<< std::endl;
return;
}
if (!shouldContinue.load())
{ return; }
/** EXPLANATION:
* We need to ensure that there's only ever one stimframe being produced
* during any CONFIG_STIMBUFF_FRAME_PERIOD_MS period. To guarantee this, we
* use a spinlock.
*
* When a new frame is to be produced, the async producer will first acquire
* the frameAssemblyLimiter spinlock. This way, when the next timeout is
* fired it can check whether its predecessor stimframe has finished being
* produced. If the preceding stimframe is still being produced, then we'll
* sleep for CONFIG_STIMBUFF_FRAME_RETRY_DELAY_MS ms before trying again.
*/
int nextWakeupDelayMs;
if (frameAssemblyRateLimiter.tryAcquire())
{ nextWakeupDelayMs = CONFIG_STIMBUFF_FRAME_PERIOD_MS; }
else
{ nextWakeupDelayMs = CONFIG_STIMBUFF_FRAME_RETRY_DELAY_MS; }
// Placeholder handler (empty for now)
// Note: The lock should be released when frame production completes
// Schedule next timeout with the pre-determined duration
scheduleNextTimeout(nextWakeupDelayMs);
}
};
} // namespace stim_buff