BitLogger/src-async/utils/async_models.mbt

///|
pub(all) enum AsyncOverflowPolicy {
  Blocking
  DropOldest
  DropNewest
}

///|
pub(all) enum AsyncFlushPolicy {
  Never
  Batch
  Shutdown
}

///|
pub enum AsyncRuntimeMode {
  NativeWorker
  Compatibility
}

///|
pub fn async_runtime_mode_label(mode : AsyncRuntimeMode) -> String {
  match mode {
    AsyncRuntimeMode::NativeWorker => "native_worker"
    AsyncRuntimeMode::Compatibility => "compatibility"
  }
}

///|
pub fn native_worker_async_runtime_mode() -> AsyncRuntimeMode {
  AsyncRuntimeMode::NativeWorker
}

///|
pub fn compatibility_async_runtime_mode() -> AsyncRuntimeMode {
  AsyncRuntimeMode::Compatibility
}

///|
pub struct AsyncRuntimeState {
  mode : AsyncRuntimeMode
  background_worker : Bool
}

///|
pub fn AsyncRuntimeState::new(
  mode : AsyncRuntimeMode,
  background_worker : Bool,
) -> AsyncRuntimeState {
  { mode, background_worker }
}

///|
pub struct AsyncLoggerState {
  runtime : AsyncRuntimeState
  pending_count : Int
  dropped_count : Int
  is_closed : Bool
  is_running : Bool
  has_failed : Bool
  last_error : String
  flush_policy : AsyncFlushPolicy
}

///|
pub fn AsyncLoggerState::new(
  runtime : AsyncRuntimeState,
  pending_count : Int,
  dropped_count : Int,
  is_closed : Bool,
  is_running : Bool,
  has_failed : Bool,
  last_error : String,
  flush_policy : AsyncFlushPolicy,
) -> AsyncLoggerState {
  {
    runtime,
    pending_count,
    dropped_count,
    is_closed,
    is_running,
    has_failed,
    last_error,
    flush_policy,
  }
}

///|
pub fn async_runtime_state_to_json(
  state : AsyncRuntimeState,
) -> @json_parser.JsonValue {
  @json_parser.JsonValue::Object({
    "mode": @json_parser.JsonValue::String(async_runtime_mode_label(state.mode)),
    "background_worker": @json_parser.JsonValue::Bool(state.background_worker),
  })
}

///|
pub fn stringify_async_runtime_state(
  state : AsyncRuntimeState,
  pretty? : Bool = false,
) -> String {
  let value = async_runtime_state_to_json(state)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}

///|
fn async_flush_policy_label(policy : AsyncFlushPolicy) -> String {
  match policy {
    AsyncFlushPolicy::Never => "Never"
    AsyncFlushPolicy::Batch => "Batch"
    AsyncFlushPolicy::Shutdown => "Shutdown"
  }
}

///|
fn async_logger_state_to_json_value(
  state : AsyncLoggerState,
) -> @json_parser.JsonValue {
  @json_parser.JsonValue::Object({
    "runtime": async_runtime_state_to_json(state.runtime),
    "pending_count": @json_parser.JsonValue::Number(
      state.pending_count.to_double(),
    ),
    "dropped_count": @json_parser.JsonValue::Number(
      state.dropped_count.to_double(),
    ),
    "is_closed": @json_parser.JsonValue::Bool(state.is_closed),
    "is_running": @json_parser.JsonValue::Bool(state.is_running),
    "has_failed": @json_parser.JsonValue::Bool(state.has_failed),
    "last_error": @json_parser.JsonValue::String(state.last_error),
    "flush_policy": @json_parser.JsonValue::String(
      async_flush_policy_label(state.flush_policy),
    ),
  })
}

///|
pub fn async_logger_state_to_json(
  state : AsyncLoggerState,
) -> @json_parser.JsonValue {
  async_logger_state_to_json_value(state)
}

///|
pub fn stringify_async_logger_state(
  state : AsyncLoggerState,
  pretty? : Bool = false,
) -> String {
  let value = async_logger_state_to_json_value(state)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}

///|
pub struct AsyncLoggerConfig {
  max_pending : Int
  overflow : AsyncOverflowPolicy
  max_batch : Int
  linger_ms : Int
  flush : AsyncFlushPolicy
}

///|
pub fn AsyncLoggerConfig::new(
  max_pending? : Int = 0,
  overflow? : AsyncOverflowPolicy = AsyncOverflowPolicy::Blocking,
  max_batch? : Int = 1,
  linger_ms? : Int = 0,
  flush? : AsyncFlushPolicy = AsyncFlushPolicy::Never,
) -> AsyncLoggerConfig {
  {
    max_pending,
    overflow,
    max_batch: if max_batch <= 1 {
      1
    } else {
      max_batch
    },
    linger_ms: if linger_ms < 0 {
      0
    } else {
      linger_ms
    },
    flush,
  }
}

///|
fn parse_async_overflow(name : String) -> AsyncOverflowPolicy raise {
  match name.to_upper() {
    "BLOCKING" => AsyncOverflowPolicy::Blocking
    "DROPOLDEST" => AsyncOverflowPolicy::DropOldest
    "DROPLATEST" => AsyncOverflowPolicy::DropNewest
    "DROPNEWEST" => AsyncOverflowPolicy::DropNewest
    _ => raise Failure::Failure("Unsupported async overflow policy: " + name)
  }
}

///|
fn parse_async_flush(name : String) -> AsyncFlushPolicy raise {
  match name.to_upper() {
    "NEVER" => AsyncFlushPolicy::Never
    "NONE" => AsyncFlushPolicy::Never
    "BATCH" => AsyncFlushPolicy::Batch
    "SHUTDOWN" => AsyncFlushPolicy::Shutdown
    _ => raise Failure::Failure("Unsupported async flush policy: " + name)
  }
}

///|
pub fn parse_async_logger_config_text(
  input : String,
) -> AsyncLoggerConfig raise {
  let root = @json_parser.parse(input)
  let obj = match root.as_object() {
    Some(obj) => obj
    None => raise Failure::Failure("Expected object for async logger config")
  }
  let max_pending = match obj.get("max_pending") {
    Some(value) =>
      match value.as_number() {
        Some(number) => number.to_int()
        None =>
          raise Failure::Failure("Expected number at async_config.max_pending")
      }
    None => 0
  }
  let overflow = match obj.get("overflow") {
    Some(value) =>
      match value.as_string() {
        Some(text) => parse_async_overflow(text)
        None =>
          raise Failure::Failure("Expected string at async_config.overflow")
      }
    None => AsyncOverflowPolicy::Blocking
  }
  let max_batch = match obj.get("max_batch") {
    Some(value) =>
      match value.as_number() {
        Some(number) => number.to_int()
        None =>
          raise Failure::Failure("Expected number at async_config.max_batch")
      }
    None => 1
  }
  let linger_ms = match obj.get("linger_ms") {
    Some(value) =>
      match value.as_number() {
        Some(number) => number.to_int()
        None =>
          raise Failure::Failure("Expected number at async_config.linger_ms")
      }
    None => 0
  }
  let flush = match obj.get("flush") {
    Some(value) =>
      match value.as_string() {
        Some(text) => parse_async_flush(text)
        None => raise Failure::Failure("Expected string at async_config.flush")
      }
    None => AsyncFlushPolicy::Never
  }
  AsyncLoggerConfig::new(
    max_pending~,
    overflow~,
    max_batch~,
    linger_ms~,
    flush~,
  )
}

///|
pub fn async_logger_config_to_json(
  config : AsyncLoggerConfig,
) -> @json_parser.JsonValue {
  @json_parser.JsonValue::Object({
    "max_pending": @json_parser.JsonValue::Number(
      config.max_pending.to_double(),
    ),
    "max_batch": @json_parser.JsonValue::Number(config.max_batch.to_double()),
    "linger_ms": @json_parser.JsonValue::Number(config.linger_ms.to_double()),
    "overflow": @json_parser.JsonValue::String(
      match config.overflow {
        AsyncOverflowPolicy::Blocking => "Blocking"
        AsyncOverflowPolicy::DropOldest => "DropOldest"
        AsyncOverflowPolicy::DropNewest => "DropNewest"
      },
    ),
    "flush": @json_parser.JsonValue::String(
      match config.flush {
        AsyncFlushPolicy::Never => "Never"
        AsyncFlushPolicy::Batch => "Batch"
        AsyncFlushPolicy::Shutdown => "Shutdown"
      },
    ),
  })
}

///|
pub fn stringify_async_logger_config(
  config : AsyncLoggerConfig,
  pretty? : Bool = false,
) -> String {
  let value = async_logger_config_to_json(config)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}

///|
pub struct AsyncLoggerBuildConfig {
  logger : @bitlogger.LoggerConfig
  async_config : AsyncLoggerConfig
}

///|
pub fn AsyncLoggerBuildConfig::new(
  logger? : @bitlogger.LoggerConfig = @bitlogger.default_logger_config(),
  async_config? : AsyncLoggerConfig = AsyncLoggerConfig::new(),
) -> AsyncLoggerBuildConfig {
  { logger, async_config }
}

///|
pub fn parse_async_logger_build_config_text(
  input : String,
) -> AsyncLoggerBuildConfig raise {
  let root = @json_parser.parse(input)
  let obj = match root.as_object() {
    Some(obj) => obj
    None =>
      raise Failure::Failure(
        "Expected object at async logger build config root",
      )
  }
  let logger = match obj.get("logger") {
    Some(value) =>
      @bitlogger.parse_logger_config_text(@json_parser.stringify(value))
    None => @bitlogger.default_logger_config()
  }
  let async_config = match obj.get("async_config") {
    Some(value) => parse_async_logger_config_text(@json_parser.stringify(value))
    None => AsyncLoggerConfig::new()
  }
  AsyncLoggerBuildConfig::new(logger~, async_config~)
}

///|
pub fn async_logger_build_config_to_json(
  config : AsyncLoggerBuildConfig,
) -> @json_parser.JsonValue {
  @json_parser.JsonValue::Object({
    "logger": @bitlogger.logger_config_to_json(config.logger),
    "async_config": async_logger_config_to_json(config.async_config),
  })
}

///|
pub fn stringify_async_logger_build_config(
  config : AsyncLoggerBuildConfig,
  pretty? : Bool = false,
) -> String {
  let value = async_logger_build_config_to_json(config)
  if pretty {
    @json_parser.stringify_pretty(value, 2)
  } else {
    @json_parser.stringify(value)
  }
}