Added rolling log support

+ Log files are limited to 10mb
+ 10 logs are kept at a time
master
Joey Hines 2020-05-22 22:57:11 -05:00
parent f3c977b93b
commit 79d0b1cd09
4 changed files with 68 additions and 20 deletions

1
.gitignore vendored
View File

@ -1,3 +1,4 @@
/target /target
**/*.rs.bk **/*.rs.bk
.env .env
log/*

View File

@ -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/"
``` ```

View File

@ -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(())
} }

View File

@ -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