ext/cron/handler_impl.rs - main - githubmirror/deno

ext/cron/handler_impl.rs

64 строки2 KB

Bartek Iwańczuk

refactor(ext): drop generics from deno_kv and deno_cron extensions (#33966)

12 май 2026, 01:09

12 май 2026, 01:0986a5745

// Copyright 2018-2026 the Deno authors. MIT license.

use std::rc::Rc;

use crate::CronError;
use crate::CronHandle;
use crate::CronHandler;
use crate::CronSpec;
use crate::local::LocalCronHandler;
use crate::socket::SocketCronHandler;

pub enum CronHandlerImpl {
  Local(LocalCronHandler),
  Socket(SocketCronHandler),
}

impl CronHandlerImpl {
  pub fn create_from_env() -> Self {
    match std::env::var("DENO_UNSTABLE_CRON_SOCK") {
      Ok(socket_addr) => Self::Socket(SocketCronHandler::new(socket_addr)),
      Err(_) => Self::Local(LocalCronHandler::new()),
    }
  }
}

impl CronHandler for CronHandlerImpl {
  fn create(&self, spec: CronSpec) -> Result<Rc<dyn CronHandle>, CronError> {
    match self {
      Self::Local(h) => h.create(spec),
      Self::Socket(h) => h.create(spec),
    }
  }

  /// Reload happens when:
  /// - Local → Socket (upgrade)
  /// - Socket(addr1) → Socket(addr2) where addr1 != addr2
  ///
  /// Never downgrades from Socket → Local.
  fn maybe_reload(&self) -> Option<Box<dyn CronHandler>> {
    let current_sock = std::env::var("DENO_UNSTABLE_CRON_SOCK").ok();

    match (self, current_sock) {
      // Local → Socket: upgrade
      (Self::Local(_), Some(new_addr)) => {
        Some(Box::new(Self::Socket(SocketCronHandler::new(new_addr))))
      }

      // Socket → Socket with different address: reload
      (Self::Socket(handler), Some(new_addr)) => {
        if handler.socket_addr() != new_addr {
          Some(Box::new(Self::Socket(SocketCronHandler::new(new_addr))))
        } else {
          None
        }
      }

      // Socket → Local: never downgrade, keep socket
      (Self::Socket(_), None) => None,

      // Local → Local: no change
      (Self::Local(_), None) => None,
    }
  }
}