include/ylt/coro_rpc/impl/errno.h

/*
 * Copyright (c) 2023, Alibaba Group Holding Limited;
 *
 * 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 the License.
 */
#include <cstdint>
#include <ylt/struct_pack.hpp>
#pragma once
namespace coro_rpc {
enum class errc : uint16_t {
  ok,
  io_error,
  not_connected,
  timed_out,
  invalid_rpc_arguments,
  address_in_used,
  bad_address,
  open_error,
  listen_error,
  operation_canceled,
  rpc_throw_exception,
  function_not_registered,
  protocol_error,
  unknown_protocol_version,
  message_too_large,
  server_has_ran,
  invalid_rpc_result,
  serial_number_conflict,
};
inline constexpr std::string_view make_error_message(errc ec) noexcept {
  switch (ec) {
    case errc::ok:
      return "ok";
    case errc::io_error:
      return "io error";
    case errc::not_connected:
      return "not connected";
    case errc::timed_out:
      return "time out";
    case errc::invalid_rpc_arguments:
      return "invalid rpc arg";
    case errc::address_in_used:
      return "address in used";
    case errc::bad_address:
      return "bad_address";
    case errc::open_error:
      return "open_error";
    case errc::listen_error:
      return "listen_error";
    case errc::operation_canceled:
      return "operation canceled";
    case errc::rpc_throw_exception:
      return "rpc throw exception";
    case errc::function_not_registered:
      return "function not registered";
    case errc::protocol_error:
      return "protocol error";
    case errc::message_too_large:
      return "message too large";
    case errc::server_has_ran:
      return "server has ran";
    case errc::invalid_rpc_result:
      return "invalid rpc result";
    case errc::serial_number_conflict:
      return "serial number conflict";
    default:
      return "unknown user-defined error";
  }
}
struct err_code {
 public:
  errc ec;
  constexpr err_code() noexcept : ec(errc::ok) {}
  explicit constexpr err_code(uint16_t ec) noexcept : ec{ec} {};
  constexpr err_code(errc ec) noexcept : ec(ec){};
  constexpr err_code& operator=(errc ec) noexcept {
    this->ec = ec;
    return *this;
  }
  constexpr err_code(const err_code& err_code) noexcept = default;
  constexpr err_code& operator=(const err_code& o) noexcept = default;
  constexpr operator errc() const noexcept { return ec; }
  constexpr operator bool() const noexcept { return static_cast<uint16_t>(ec); }
  constexpr explicit operator uint16_t() const noexcept {
    return static_cast<uint16_t>(ec);
  }
  constexpr uint16_t val() const noexcept { return static_cast<uint16_t>(ec); }
  constexpr std::string_view message() const noexcept {
    return make_error_message(ec);
  }
};

inline bool operator!(err_code ec) noexcept { return ec == errc::ok; }
inline bool operator!(errc ec) noexcept { return ec == errc::ok; }

struct rpc_error {
  coro_rpc::err_code code;  //!< error code
  std::string msg;          //!< error message
  rpc_error() {}
  rpc_error(coro_rpc::err_code code, std::string_view msg)
      : code(code), msg(std::string{msg}) {}
  rpc_error(coro_rpc::err_code code)
      : code(code), msg(std::string{make_error_message(code)}) {}
  uint16_t& val() { return *(uint16_t*)&(code.ec); }
  const uint16_t& val() const { return *(uint16_t*)&(code.ec); }
  constexpr operator bool() const noexcept { return code; }
};
YLT_REFL(rpc_error, val(), msg);

};  // namespace coro_rpc