mirror of https://github.com/LemmyNet/lemmy.git
Move connection creation into scheduler. (#3120)
* Move connection creation into scheduler. - #3076 * Fix clippy.pull/3215/head
parent
56de68c77a
commit
981c111567
|
@ -31,35 +31,34 @@ pub fn setup(db_url: String, user_agent: String) -> Result<(), LemmyError> {
|
||||||
// Setup the connections
|
// Setup the connections
|
||||||
let mut scheduler = Scheduler::new();
|
let mut scheduler = Scheduler::new();
|
||||||
|
|
||||||
let mut conn_1 = PgConnection::establish(&db_url).expect("could not establish connection");
|
startup_jobs(&db_url);
|
||||||
let mut conn_2 = PgConnection::establish(&db_url).expect("could not establish connection");
|
|
||||||
let mut conn_3 = PgConnection::establish(&db_url).expect("could not establish connection");
|
|
||||||
let mut conn_4 = PgConnection::establish(&db_url).expect("could not establish connection");
|
|
||||||
|
|
||||||
// Run on startup
|
|
||||||
active_counts(&mut conn_1);
|
|
||||||
update_hot_ranks(&mut conn_1, false);
|
|
||||||
update_banned_when_expired(&mut conn_1);
|
|
||||||
clear_old_activities(&mut conn_1);
|
|
||||||
|
|
||||||
// Update active counts every hour
|
// Update active counts every hour
|
||||||
|
let url = db_url.clone();
|
||||||
scheduler.every(CTimeUnits::hour(1)).run(move || {
|
scheduler.every(CTimeUnits::hour(1)).run(move || {
|
||||||
active_counts(&mut conn_1);
|
let mut conn = PgConnection::establish(&url).expect("could not establish connection");
|
||||||
update_banned_when_expired(&mut conn_1);
|
active_counts(&mut conn);
|
||||||
|
update_banned_when_expired(&mut conn);
|
||||||
});
|
});
|
||||||
|
|
||||||
// Update hot ranks every 5 minutes
|
// Update hot ranks every 5 minutes
|
||||||
|
let url = db_url.clone();
|
||||||
scheduler.every(CTimeUnits::minutes(5)).run(move || {
|
scheduler.every(CTimeUnits::minutes(5)).run(move || {
|
||||||
update_hot_ranks(&mut conn_2, true);
|
let mut conn = PgConnection::establish(&url).expect("could not establish connection");
|
||||||
|
update_hot_ranks(&mut conn, true);
|
||||||
});
|
});
|
||||||
|
|
||||||
// Clear old activities every week
|
// Clear old activities every week
|
||||||
|
let url = db_url.clone();
|
||||||
scheduler.every(CTimeUnits::weeks(1)).run(move || {
|
scheduler.every(CTimeUnits::weeks(1)).run(move || {
|
||||||
clear_old_activities(&mut conn_3);
|
let mut conn = PgConnection::establish(&url).expect("could not establish connection");
|
||||||
|
clear_old_activities(&mut conn);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Update the Instance Software
|
||||||
scheduler.every(CTimeUnits::days(1)).run(move || {
|
scheduler.every(CTimeUnits::days(1)).run(move || {
|
||||||
update_instance_software(&mut conn_4, &user_agent);
|
let mut conn = PgConnection::establish(&db_url).expect("could not establish connection");
|
||||||
|
update_instance_software(&mut conn, &user_agent);
|
||||||
});
|
});
|
||||||
|
|
||||||
// Manually run the scheduler in an event loop
|
// Manually run the scheduler in an event loop
|
||||||
|
@ -69,6 +68,15 @@ pub fn setup(db_url: String, user_agent: String) -> Result<(), LemmyError> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Run these on server startup
|
||||||
|
fn startup_jobs(db_url: &str) {
|
||||||
|
let mut conn = PgConnection::establish(db_url).expect("could not establish connection");
|
||||||
|
active_counts(&mut conn);
|
||||||
|
update_hot_ranks(&mut conn, false);
|
||||||
|
update_banned_when_expired(&mut conn);
|
||||||
|
clear_old_activities(&mut conn);
|
||||||
|
}
|
||||||
|
|
||||||
/// Update the hot_rank columns for the aggregates tables
|
/// Update the hot_rank columns for the aggregates tables
|
||||||
fn update_hot_ranks(conn: &mut PgConnection, last_week_only: bool) {
|
fn update_hot_ranks(conn: &mut PgConnection, last_week_only: bool) {
|
||||||
let mut post_update = diesel::update(post_aggregates::table).into_boxed();
|
let mut post_update = diesel::update(post_aggregates::table).into_boxed();
|
||||||
|
|
Loading…
Reference in New Issue