void no_read(A4Input &in) {
shared<InputStream> stream = in.get_stream();
if (!stream) return;
assert(stream->good());
if (stream->error()) {
ERROR("stream problem in no_read thread ", boost::this_thread::get_id());
}
}
void my_read(A4Input &in) {
shared<InputStream> stream = in.get_stream();
if (!stream) return;
int cnt = 0;
while (shared<A4Message> msg = stream->next()) {
if (const TestEvent* te = msg->as<TestEvent>()) {
assert((cnt++%N) == te->event_number());
}
}
if (stream->error())
ERROR("stream problem in thread ", boost::this_thread::get_id());
assert(cnt == 5*N);
}
int main(int argc, char ** argv) {
const int N = 1000;
{
A4Output a4o("test_io.a4", "Event");
shared<OutputStream> stream = a4o.get_stream();
stream->set_compression(OutputStream::ZLIB, 1);
Event e;
for(int i = 0; i < N; i++) {
e.set_event_number(i);
stream->write(e);
}
EventStreamInfo m;
m.set_total_events(N);
stream->metadata(m);
}
{
A4Input in;
in.add_file("test_io.a4");
Event e;
const Event * te;
shared<A4Message> msg;
while (shared<InputStream> stream = in.get_stream()) {
int cnt = 0;
while (true) {
if (stream->try_read(e)) {
te = &e;
} else {
msg = stream->next();
if (not msg) break;
te = msg->as<Event>();
}
auto * me = stream->current_metadata()->as<EventStreamInfo>();
assert(cnt++ == te->event_number());
assert(me->total_events() == N);
}
if (stream->error()) throw "AJS";
}
}
}