Added rolling log support
+ Log files are limited to 10mb + 10 logs are kept at a timemaster
parent
f3c977b93b
commit
79d0b1cd09
|
@ -1,3 +1,4 @@
|
||||||
/target
|
/target
|
||||||
**/*.rs.bk
|
**/*.rs.bk
|
||||||
.env
|
.env
|
||||||
|
log/*
|
||||||
|
|
|
@ -37,6 +37,6 @@ event_channel = 0
|
||||||
event_roles = [0]
|
event_roles = [0]
|
||||||
# Timezone to display events, supported timezones can be found at https://docs.rs/chrono-tz/0.5.1/chrono_tz/#modules
|
# Timezone to display events, supported timezones can be found at https://docs.rs/chrono-tz/0.5.1/chrono_tz/#modules
|
||||||
event_timezone = "America/New_York"
|
event_timezone = "America/New_York"
|
||||||
# Path to logger
|
# Path to place logs in
|
||||||
log_path = "hype_bot.log"
|
log_path = "log/"
|
||||||
```
|
```
|
|
@ -16,6 +16,8 @@ use url::Url;
|
||||||
#[command]
|
#[command]
|
||||||
/// Posts a previewed event
|
/// Posts a previewed event
|
||||||
///
|
///
|
||||||
|
/// `~confirm`
|
||||||
|
///
|
||||||
/// **Note**
|
/// **Note**
|
||||||
/// You can only post events you have created. Only one preview event can exist at a time.
|
/// You can only post events you have created. Only one preview event can exist at a time.
|
||||||
fn confirm(ctx: &mut Context, msg: &Message, _args: Args) -> CommandResult {
|
fn confirm(ctx: &mut Context, msg: &Message, _args: Args) -> CommandResult {
|
||||||
|
@ -162,8 +164,8 @@ fn create(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
||||||
#[command]
|
#[command]
|
||||||
/// Cancels an already scheduled event
|
/// Cancels an already scheduled event
|
||||||
///
|
///
|
||||||
/// `~create_event "event name"`
|
/// `~cancel "event name"`
|
||||||
fn cancel(ctx: &mut Context, _msg: &Message, mut args: Args) -> CommandResult {
|
fn cancel(ctx: &mut Context, msg: &Message, mut args: Args) -> CommandResult {
|
||||||
let config = get_config(&ctx.data)?;
|
let config = get_config(&ctx.data)?;
|
||||||
|
|
||||||
// Parse args
|
// Parse args
|
||||||
|
@ -187,5 +189,7 @@ fn cancel(ctx: &mut Context, _msg: &Message, mut args: Args) -> CommandResult {
|
||||||
|
|
||||||
message.delete(&ctx)?;
|
message.delete(&ctx)?;
|
||||||
|
|
||||||
|
msg.reply(&ctx, "**{}** has been canceled")?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
75
src/main.rs
75
src/main.rs
|
@ -10,9 +10,15 @@ extern crate log4rs;
|
||||||
use chrono::{DateTime, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
use clap::{App, Arg};
|
use clap::{App, Arg};
|
||||||
use log::LevelFilter;
|
use log::LevelFilter;
|
||||||
use log4rs::append::file::FileAppender;
|
use log4rs::append::rolling_file::{RollingFileAppender};
|
||||||
use log4rs::config::{Appender, Config, Root};
|
use log4rs::config::{Appender, Config, Root};
|
||||||
use log4rs::encode::pattern::PatternEncoder;
|
use log4rs::encode::pattern::PatternEncoder;
|
||||||
|
use log4rs::append::console::ConsoleAppender;
|
||||||
|
use log4rs::append::rolling_file::policy::compound::roll::fixed_window::FixedWindowRoller;
|
||||||
|
use log4rs::append::rolling_file::policy::compound::trigger::size::SizeTrigger;
|
||||||
|
use log4rs::append::rolling_file::policy::compound::CompoundPolicy;
|
||||||
|
use log4rs::filter::threshold::ThresholdFilter;
|
||||||
|
use log4rs::init_config;
|
||||||
use serenity::client::Client;
|
use serenity::client::Client;
|
||||||
use serenity::framework::standard::macros::{group, help};
|
use serenity::framework::standard::macros::{group, help};
|
||||||
use serenity::framework::standard::{help_commands, Args, CommandGroup, HelpOptions};
|
use serenity::framework::standard::{help_commands, Args, CommandGroup, HelpOptions};
|
||||||
|
@ -25,18 +31,18 @@ use std::collections::HashSet;
|
||||||
use std::process::exit;
|
use std::process::exit;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use white_rabbit::{DateResult, Scheduler};
|
use white_rabbit::{DateResult, Scheduler};
|
||||||
|
use std::path::Path;
|
||||||
|
|
||||||
mod database;
|
mod database;
|
||||||
mod discord;
|
mod discord;
|
||||||
mod hypebot_config;
|
mod hypebot_config;
|
||||||
|
|
||||||
use crate::discord::schedule_event;
|
|
||||||
use database::models::NewEvent;
|
use database::models::NewEvent;
|
||||||
use database::*;
|
use database::*;
|
||||||
use discord::events::{CANCEL_COMMAND, CONFIRM_COMMAND, CREATE_COMMAND};
|
use discord::events::{CANCEL_COMMAND, CONFIRM_COMMAND, CREATE_COMMAND};
|
||||||
use discord::{
|
use discord::{
|
||||||
delete_event, get_config, get_scheduler, log_error, permission_check,
|
delete_event, get_config, get_scheduler, log_error, permission_check,
|
||||||
send_message_to_reaction_users, DraftEvent, SchedulerKey,
|
send_message_to_reaction_users, schedule_event, DraftEvent, SchedulerKey,
|
||||||
};
|
};
|
||||||
use hypebot_config::HypeBotConfig;
|
use hypebot_config::HypeBotConfig;
|
||||||
|
|
||||||
|
@ -80,7 +86,6 @@ impl EventHandler for Handler {
|
||||||
|
|
||||||
/// On bot ready
|
/// On bot ready
|
||||||
fn ready(&self, _: Context, ready: Ready) {
|
fn ready(&self, _: Context, ready: Ready) {
|
||||||
println!("Connected to Discord as {}", ready.user.name);
|
|
||||||
info!("Connected to Discord as {}", ready.user.name);
|
info!("Connected to Discord as {}", ready.user.name);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -101,18 +106,57 @@ fn bot_help(
|
||||||
|
|
||||||
/// Does the setup for logging
|
/// Does the setup for logging
|
||||||
fn setup_logging(config: &HypeBotConfig) -> HypeBotResult<()> {
|
fn setup_logging(config: &HypeBotConfig) -> HypeBotResult<()> {
|
||||||
// Setup logging
|
// Build log file path
|
||||||
let log_encode = Box::new(PatternEncoder::new("{d}:{l}-{m}\n"));
|
let log_file_path = Path::new(&config.log_path);
|
||||||
|
let log_file_path = log_file_path.join("hype_bot.log");
|
||||||
|
|
||||||
let logfile = FileAppender::builder()
|
// Number of logs to keep
|
||||||
.encoder(log_encode)
|
let window_size = 10;
|
||||||
.build(&config.log_path)?;
|
|
||||||
|
|
||||||
let log_config = Config::builder()
|
// 10MB file size limit
|
||||||
.appender(Appender::builder().build("logfile", Box::new(logfile)))
|
let size_limit = 1 * 1024 * 1024;
|
||||||
.build(Root::builder().appender("logfile").build(LevelFilter::Info))?;
|
let size_trigger = SizeTrigger::new(size_limit);
|
||||||
|
|
||||||
log4rs::init_config(log_config)?;
|
let fixed_window_roller = FixedWindowRoller::builder()
|
||||||
|
.build("hype_bot.{}.log", window_size)
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let compound_policy =
|
||||||
|
CompoundPolicy::new(Box::new(size_trigger), Box::new(fixed_window_roller));
|
||||||
|
|
||||||
|
let config = Config::builder()
|
||||||
|
.appender(
|
||||||
|
Appender::builder()
|
||||||
|
.filter(Box::new(ThresholdFilter::new(LevelFilter::Info)))
|
||||||
|
.build(
|
||||||
|
"logfile",
|
||||||
|
Box::new(
|
||||||
|
RollingFileAppender::builder()
|
||||||
|
.encoder(Box::new(PatternEncoder::new("{d} {l}::{m}{n}")))
|
||||||
|
.build(log_file_path, Box::new(compound_policy))?,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.appender(
|
||||||
|
Appender::builder()
|
||||||
|
.filter(Box::new(ThresholdFilter::new(LevelFilter::Info)))
|
||||||
|
.build(
|
||||||
|
"stdout",
|
||||||
|
Box::new(
|
||||||
|
ConsoleAppender::builder()
|
||||||
|
.encoder(Box::new(PatternEncoder::new("{l}::{m}{n}")))
|
||||||
|
.build(),
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.build(
|
||||||
|
Root::builder()
|
||||||
|
.appender("logfile")
|
||||||
|
.appender("stdout")
|
||||||
|
.build(LevelFilter::Info),
|
||||||
|
)?;
|
||||||
|
|
||||||
|
init_config(config)?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
@ -195,7 +239,7 @@ fn main() -> HypeBotResult<()> {
|
||||||
data.insert::<SchedulerKey>(scheduler);
|
data.insert::<SchedulerKey>(scheduler);
|
||||||
}
|
}
|
||||||
|
|
||||||
// schedule
|
// Schedule current events
|
||||||
let config = get_config(&client.data).expect("Unable to find get config");
|
let config = get_config(&client.data).expect("Unable to find get config");
|
||||||
let duration = chrono::Duration::minutes(60);
|
let duration = chrono::Duration::minutes(60);
|
||||||
for event in get_all_events(config.db_url.clone()).unwrap() {
|
for event in get_all_events(config.db_url.clone()).unwrap() {
|
||||||
|
@ -222,9 +266,8 @@ fn main() -> HypeBotResult<()> {
|
||||||
|
|
||||||
// Start bot
|
// Start bot
|
||||||
info!("Starting HypeBot!");
|
info!("Starting HypeBot!");
|
||||||
println!("Starting HypeBot!");
|
|
||||||
if let Err(why) = client.start() {
|
if let Err(why) = client.start() {
|
||||||
println!("An error occurred while running the client: {:?}", why);
|
error!("An error occurred while running the client: {:?}", why);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// Print help
|
// Print help
|
||||||
|
|
Loading…
Reference in New Issue