op.cc

/*
 * Copyright (C) 2022 SKTT1Ryze. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 License.
 * */

#include "kuro.h"
#include "slab.h"

#define SLAB_CAPACITY 100

thread_local slab<std::coroutine_handle<>> CO_HANDLES =
    slab<std::coroutine_handle<>>(SLAB_CAPACITY);
thread_local std::map<unsigned long, __s32*> URING_RESULTS;

template <typename T>
bool Op<T>::await_ready() {
  return false;
}

template <typename T>
void Op<T>::await_suspend(std::coroutine_handle<> h) {
  struct io_uring* handle = uring_handle.get();
  struct io_uring_sqe* sqe = io_uring_get_sqe(handle);

  cb(sqe);

  token = CO_HANDLES.insert(std::move(h));
  URING_RESULTS.insert(std::pair{token, &res});

  io_uring_sqe_set_data(sqe, (void*)token);
}

template <typename T>
__s32 Op<T>::await_resume() {
  CO_HANDLES.remove(token);
  URING_RESULTS.erase(token);

  return res;
}

template <typename T>
Op<T>::Op(const T val, std::shared_ptr<io_uring>& uring, Callback f)
    : cb(f), value(val) {
  uring_handle = uring;
}

// todo: remove it
template class Op<int>;