67 lines
1.7 KiB
C++
67 lines
1.7 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#ifndef OCEANBASE_TRANSACTION_TEST_OB_ASYNC_UTILS_
|
|
#define OCEANBASE_TRANSACTION_TEST_OB_ASYNC_UTILS_
|
|
#include "lib/thread/ob_simple_thread_pool.h"
|
|
namespace oceanbase {
|
|
namespace test {
|
|
class Async {
|
|
public:
|
|
class AsyncRunner : public common::ObSimpleThreadPool {
|
|
public:
|
|
AsyncRunner() {
|
|
ObSimpleThreadPool::init(2, 256, "AsyncRunner-for-Testing");
|
|
}
|
|
void handle(void *task) {
|
|
Async *async = (Async*)task;
|
|
async->eval();
|
|
}
|
|
};
|
|
|
|
Async(std::function<int(void)> &f) {
|
|
static AsyncRunner runner;
|
|
started_ = false;
|
|
evaluated_ = false;
|
|
f_ = f;
|
|
runner.push(this);
|
|
}
|
|
void wait_started() {
|
|
while(!ATOMIC_LOAD(&started_)) {
|
|
usleep(1000);
|
|
}
|
|
}
|
|
void eval() {
|
|
ATOMIC_STORE(&started_, true);
|
|
ret_ = f_();
|
|
evaluated_ = true;
|
|
}
|
|
bool is_evaled() const { return evaluated_; }
|
|
int get() const {
|
|
while(!evaluated_) usleep(100);
|
|
return ret_;
|
|
}
|
|
private:
|
|
bool started_ = false;
|
|
bool evaluated_ = false;
|
|
std::function<int(void)> f_;
|
|
int ret_;
|
|
};
|
|
template<typename F>
|
|
inline Async static make_async(F &f) {
|
|
std::function<int(void)> ff = f;
|
|
return Async(ff);
|
|
}
|
|
} // test
|
|
} // oceanbase
|
|
#endif
|