add more options for executor (#487)

Signed-off-by: liulanzheng <lanzheng.liulz@alibaba-inc.com>
alibaba · May 22, 2024 · e46ae97 · e46ae97
1 parent 24e35e7
commit e46ae97
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 7 deletions.
diff --git a/common/executor/executor.cpp b/common/executor/executor.cpp
@@ -23,9 +23,11 @@ class Executor::ExecutorImpl {
  CBList queue;
  photon::ThreadPoolBase *pool;
 
- ExecutorImpl(int init_ev, int init_io) {
+ ExecutorImpl(int init_ev, int init_io, const PhotonOptions& options,
+ const ExecutorQueueOption& queue_options)
+ : queue(queue_options.max_yield_turn, queue_options.max_yield_usec) {
  th.reset(
- new std::thread(&ExecutorImpl::launch, this, init_ev, init_io));
+ new std::thread(&ExecutorImpl::launch, this, init_ev, init_io, options));
  }
 
  ExecutorImpl() {}
@@ -75,15 +77,16 @@ class Executor::ExecutorImpl {
  pool = nullptr;
  }
 
- void launch(int init_ev, int init_io) {
- photon::init(init_ev, init_io);
+ void launch(int init_ev, int init_io, const PhotonOptions& options) {
+ photon::init(init_ev, init_io, options);
  DEFER(photon::fini());
  do_loop();
  }
 };
 
-Executor::Executor(int init_ev, int init_io)
- : e(new Executor::ExecutorImpl(init_ev, init_io)) {}
+Executor::Executor(int init_ev, int init_io, const PhotonOptions& options,
+ const ExecutorQueueOption& queue_options)
+ : e(new Executor::ExecutorImpl(init_ev, init_io, options, queue_options)) {}
 
 Executor::Executor(create_on_current_vcpu) : e(new Executor::ExecutorImpl()) {}
 

diff --git a/common/executor/executor.h b/common/executor/executor.h
@@ -22,13 +22,21 @@ limitations under the License.
 #include <photon/thread/thread.h>
 
 namespace photon {
+
+struct ExecutorQueueOption {
+ uint64_t max_yield_turn;
+ uint64_t max_yield_usec;
+};
+
 class Executor {
 public:
  class ExecutorImpl;
 
  ExecutorImpl *e;
  Executor(int init_ev = photon::INIT_EVENT_DEFAULT,
- int init_io = photon::INIT_IO_DEFAULT);
+ int init_io = photon::INIT_IO_DEFAULT,
+ const PhotonOptions& options = {},
+ const ExecutorQueueOption& queue_options = {-1UL, 1024});
  ~Executor();
 
  template <