AimRT/_deps/libunifex-src/examples/stream_cancellation.cpp

54 lines
1.6 KiB
C++
Raw Normal View History

2025-01-12 20:42:42 +08:00
/*
* Copyright (c) Facebook, Inc. and its affiliates.
*
* Licensed under the Apache License Version 2.0 with LLVM Exceptions
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* https://llvm.org/LICENSE.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <unifex/for_each.hpp>
#include <unifex/inplace_stop_token.hpp>
#include <unifex/on.hpp>
#include <unifex/on_stream.hpp>
#include <unifex/range_stream.hpp>
#include <unifex/scheduler_concepts.hpp>
#include <unifex/timed_single_thread_context.hpp>
#include <unifex/sync_wait.hpp>
#include <unifex/stop_when.hpp>
#include <chrono>
#include <cstdio>
#include <thread>
using namespace unifex;
using namespace std::literals::chrono_literals;
int main() {
timed_single_thread_context context;
using namespace std::chrono;
auto startTime = steady_clock::now();
auto op = on_stream(current_scheduler, range_stream{0, 20})
| for_each([](int value) {
// Simulate some work
std::printf("processing %i\n", value);
std::this_thread::sleep_for(10ms);
})
| stop_when(schedule_after(100ms));
sync_wait(on(context.get_scheduler(), std::move(op)));
auto endTime = steady_clock::now();
std::printf(
"took %i ms\n", (int)duration_cast<milliseconds>(endTime - startTime).count());
}