stream_producer.h

Go to the documentation of this file.

#ifndef SENSESP_SYSTEM_STREAM_PRODUCER_H
#define SENSESP_SYSTEM_STREAM_PRODUCER_H
 
#include "sensesp.h"
 
#include <memory>
 
#include "ReactESP.h"
#include "sensesp/system/valueproducer.h"
#include "sensesp_base_app.h"
 
namespace sensesp {

class StreamCharProducer : public ValueProducer<char> {
 public:
  StreamCharProducer(Stream* stream) : stream_{stream} {
    read_event_ = event_loop()->onAvailable(*stream_, [this]() {
      while (stream_->available()) {
        char c = stream_->read();
        this->emit(c);
      }
    });
  }
 
 protected:
  Stream* stream_;
  reactesp::StreamEvent* read_event_;
};

class StreamLineProducer : public ValueProducer<String> {
 public:
  StreamLineProducer(
      Stream* stream,
      std::shared_ptr<reactesp::EventLoop> event_loop_param = event_loop(),
      int max_line_length = 256)
      : stream_{stream},
        max_line_length_{max_line_length},
        buf_(new char[max_line_length_ + 1]) {
    read_event_ =
        event_loop_param->onAvailable(*stream_, [this]() { this->receive_line(); });
  }
 
 protected:
  const int max_line_length_;
  int buf_pos = 0;
  std::unique_ptr<char[]> buf_;
  Stream* stream_;
  reactesp::StreamEvent* read_event_;
 
  void receive_line() {
    while (stream_->available()) {
      char c = stream_->read();
      if (c == '\n') {
        // Include the newline character in the output
        buf_[buf_pos++] = c;
        buf_[buf_pos] = '\0';
        this->emit(buf_);
        buf_pos = 0;
      } else {
        buf_[buf_pos++] = c;
        if (buf_pos >= max_line_length_ - 1) {
          buf_pos = 0;
        }
      }
    }
  }
};
 
}  // namespace sensesp
 
#endif  // SENSESP_SRC_SENSESP_SYSTEM_STREAM_PRODUCER_H_