mirror of
https://github.com/Lumither/nocturne.git
synced 2026-10-17 10:26:21 +00:00
fix(scheduler) fn type
This commit is contained in:
11 files changed
+102
-9
No files matched your search
@@ -20,3 +20,4 @@ uuid = { version = "1.10.0", features = ["v4", "fast-rng", "serde"] }
|
|||||||
futures = { version = "0.3.30", features = ["executor"] }
|
futures = { version = "0.3.30", features = ["executor"] }
|
||||||
sha2 = "0.10.8"
|
sha2 = "0.10.8"
|
||||||
cron = "0.12.1"
|
cron = "0.12.1"
|
||||||
|
git2 = "0.19.0"
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
use dirs::home_dir;
|
||||||
|
use std::env;
|
||||||
|
|
||||||
|
pub fn task() {
|
||||||
|
// get dir
|
||||||
|
let git_work_dir_path = &env::var("GIT_WORK_DIR").unwrap_or_else(|e| {
|
||||||
|
if let Ok(default_log_path) = env::var("WORK_DIR") {
|
||||||
|
if default_log_path.starts_with('~') {
|
||||||
|
let home_path = home_dir().unwrap_or_else(|| panic!("Cannot find home directory"));
|
||||||
|
home_path
|
||||||
|
.join(default_log_path.strip_prefix("~/").unwrap())
|
||||||
|
.into_os_string()
|
||||||
|
.into_string()
|
||||||
|
.expect(
|
||||||
|
"[fatal] failed to parse env `WORK_DIR`, try to use absolute path instead",
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
default_log_path
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
panic!("[fatal] failed to parse env `GIT_WORK_DIR`: {}", e);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
pub mod check_update;
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
mod api;
|
mod api;
|
||||||
|
pub mod cron;
|
||||||
|
|
||||||
use crate::blog::api::{
|
use crate::blog::api::{
|
||||||
get_page_count::get_page_count, get_post::get_post, get_post_list::get_post_list,
|
get_page_count::get_page_count, get_post::get_post, get_post_list::get_post_list,
|
||||||
|
|||||||
@@ -1,11 +1,17 @@
|
|||||||
use std::env;
|
|
||||||
|
|
||||||
use crate::logger;
|
use crate::logger;
|
||||||
|
use std::{
|
||||||
|
backtrace::{Backtrace, BacktraceStatus},
|
||||||
|
env, panic,
|
||||||
|
panic::PanicHookInfo,
|
||||||
|
};
|
||||||
|
use tracing::{error, event, span, Level};
|
||||||
|
|
||||||
use tracing_appender::non_blocking::WorkerGuard;
|
use tracing_appender::non_blocking::WorkerGuard;
|
||||||
|
|
||||||
pub fn logger_init() -> (WorkerGuard, WorkerGuard) {
|
pub fn logger_init() -> (WorkerGuard, WorkerGuard) {
|
||||||
logger::init()
|
let ret = logger::init();
|
||||||
|
panic::set_hook(Box::new(panic_hook));
|
||||||
|
ret
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn load_env() {
|
pub fn load_env() {
|
||||||
@@ -22,3 +28,29 @@ pub fn load_env() {
|
|||||||
println!("[info] no .env file referred, starting up")
|
println!("[info] no .env file referred, starting up")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn panic_hook(panic_info: &PanicHookInfo) {
|
||||||
|
let payload = panic_info.payload();
|
||||||
|
// panic_info.
|
||||||
|
|
||||||
|
let payload = if let Some(s) = payload.downcast_ref::<&str>() {
|
||||||
|
Some(&**s)
|
||||||
|
} else {
|
||||||
|
payload.downcast_ref::<String>().map(|s| s.as_str())
|
||||||
|
};
|
||||||
|
|
||||||
|
// let location = panic_info.location().map(|l| l.to_string());
|
||||||
|
// let backtrace = Backtrace::capture();
|
||||||
|
// let note = (backtrace.status() == BacktraceStatus::Disabled)
|
||||||
|
// .then_some("run with RUST_BACKTRACE=1 environment variable to display a backtrace");
|
||||||
|
// (Some(backtrace), note)
|
||||||
|
// };
|
||||||
|
//
|
||||||
|
// panic_info.
|
||||||
|
|
||||||
|
if let Some(location) = panic_info.location() {
|
||||||
|
eprintln!("err filename: {}", location.file())
|
||||||
|
}
|
||||||
|
|
||||||
|
event!(target: "test", Level::ERROR, "panic hook: {:?}", panic_info);
|
||||||
|
}
|
||||||
@@ -9,5 +9,12 @@ async fn main() {
|
|||||||
init::load_env();
|
init::load_env();
|
||||||
let _guards = init::logger_init();
|
let _guards = init::logger_init();
|
||||||
|
|
||||||
server::start().await;
|
// panic!("test");
|
||||||
|
|
||||||
|
match server::start().await {
|
||||||
|
Ok(_) => {}
|
||||||
|
Err(e) => {
|
||||||
|
panic!("server start failed: {}", e);
|
||||||
|
}
|
||||||
|
};
|
||||||
}
|
}
|
||||||
@@ -156,6 +156,10 @@ impl Scheduler {
|
|||||||
self.tasks.write()?.insert(id, Arc::new(task));
|
self.tasks.write()?.insert(id, Arc::new(task));
|
||||||
Ok(id)
|
Ok(id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self::default()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
|
|||||||
@@ -11,12 +11,19 @@ pub struct BasicTask {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl BasicTask {
|
impl BasicTask {
|
||||||
pub fn new(func: fn(), cron_expr: &str) -> Result<Self, Box<dyn Error>> {
|
pub fn new<F>(func: F, cron_expr: &str) -> Result<Self, Box<dyn Error>>
|
||||||
|
where
|
||||||
|
F: TaskFunc,
|
||||||
|
{
|
||||||
Ok(BasicTask {
|
Ok(BasicTask {
|
||||||
schedule: Schedule::from_str(cron_expr)?,
|
schedule: Schedule::from_str(cron_expr)?,
|
||||||
func: Box::new(func),
|
func: Box::new(func),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn to_task(self) -> Box<dyn Task> {
|
||||||
|
Box::new(self)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Task for BasicTask {
|
impl Task for BasicTask {
|
||||||
|
|||||||
@@ -1,9 +1,12 @@
|
|||||||
use std::env;
|
use std::env;
|
||||||
|
use std::error::Error;
|
||||||
use std::num::ParseIntError;
|
use std::num::ParseIntError;
|
||||||
use std::process::exit;
|
use std::process::exit;
|
||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use crate::blog;
|
use crate::blog;
|
||||||
|
use crate::scheduler::tasks::basic::BasicTask;
|
||||||
|
use crate::scheduler::Scheduler;
|
||||||
use axum::Router;
|
use axum::Router;
|
||||||
use sqlx::{
|
use sqlx::{
|
||||||
postgres::{PgConnectOptions, PgPoolOptions},
|
postgres::{PgConnectOptions, PgPoolOptions},
|
||||||
@@ -11,7 +14,7 @@ use sqlx::{
|
|||||||
};
|
};
|
||||||
use tracing::{error, info, warn};
|
use tracing::{error, info, warn};
|
||||||
|
|
||||||
pub async fn start() {
|
pub async fn start() -> Result<(), Box<dyn Error>> {
|
||||||
let port: u32 = match env::var("BACKEND_PORT") {
|
let port: u32 = match env::var("BACKEND_PORT") {
|
||||||
Ok(value) => value.parse().unwrap_or_else(|e: ParseIntError| {
|
Ok(value) => value.parse().unwrap_or_else(|e: ParseIntError| {
|
||||||
warn!("failed to parse BACKEND_PORT: {}", e.to_string());
|
warn!("failed to parse BACKEND_PORT: {}", e.to_string());
|
||||||
@@ -35,6 +38,18 @@ pub async fn start() {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// cron task
|
||||||
|
let scheduler = Scheduler::new();
|
||||||
|
let test_str = "hello world".to_string();
|
||||||
|
// scheduler.insert(
|
||||||
|
// BasicTask::new(
|
||||||
|
// move || blog::cron::check_update::task(test_str.clone()),
|
||||||
|
// "* * * * * *",
|
||||||
|
// )?
|
||||||
|
// .to_task(),
|
||||||
|
// )?;
|
||||||
|
|
||||||
|
// http server
|
||||||
let app = Router::new()
|
let app = Router::new()
|
||||||
.nest("blog", blog::get_router())
|
.nest("blog", blog::get_router())
|
||||||
.with_state(db_pool);
|
.with_state(db_pool);
|
||||||
@@ -50,6 +65,7 @@ pub async fn start() {
|
|||||||
error!("failed to start axum: {}", e.to_string());
|
error!("failed to start axum: {}", e.to_string());
|
||||||
exit(1);
|
exit(1);
|
||||||
});
|
});
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_db_uri() -> String {
|
fn parse_db_uri() -> String {
|
||||||
|
|||||||
@@ -9,6 +9,6 @@ chrono = { version = "0.4.38", features = ["serde", "alloc"] }
|
|||||||
serde = { version = "1.0.209", features = ["derive"] }
|
serde = { version = "1.0.209", features = ["derive"] }
|
||||||
serde_json = "1.0.128"
|
serde_json = "1.0.128"
|
||||||
regex = "1.10.4"
|
regex = "1.10.4"
|
||||||
yaml-rust2 = "0.8.1"
|
yaml-rust2 = "0.9.0"
|
||||||
sha2 = "0.10.8"
|
sha2 = "0.10.8"
|
||||||
base64 = "0.22.1"
|
base64 = "0.22.1"
|
||||||
@@ -60,7 +60,7 @@ tags: [ "test1", "test2" ]
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_front_matter_parser() {
|
fn test_front_matter_parser() {
|
||||||
parse_front_matter(
|
dbg!(parse_front_matter(
|
||||||
r##"---
|
r##"---
|
||||||
title: test_title
|
title: test_title
|
||||||
date: 0000-00-00+0000
|
date: 0000-00-00+0000
|
||||||
@@ -71,7 +71,7 @@ summary: summary
|
|||||||
tags: [ "test1", "test2" ]
|
tags: [ "test1", "test2" ]
|
||||||
---
|
---
|
||||||
"##,
|
"##,
|
||||||
)
|
))
|
||||||
.expect("Failed to parse front matter");
|
.expect("Failed to parse front matter");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Reference in new issue
Block a user