import torrus 1.0.9

[freeside.git] / torrus / bin / monitor.in

diff --git a/torrus/bin/monitor.in b/torrus/bin/monitor.in
new file mode 100644 (file)
index 0000000..c857999
--- /dev/null
+++ b/torrus/bin/monitor.in
@@ -0,0 +1,176 @@
+#!@PERL@
+#  Copyright (C) 2002  Stanislav Sinyagin
+#
+#  This program is free software; you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation; either version 2 of the License, or
+#  (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software
+#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
+
+# $Id: monitor.in,v 1.1 2010-12-27 00:04:03 ivan Exp $
+# Stanislav Sinyagin <ssinyagin@yahoo.com>
+
+BEGIN { require '@torrus_config_pl@'; }
+
+use strict;
+use Proc::Daemon;
+use Getopt::Long;
+
+use Torrus::Log;
+use Torrus::Monitor;
+use Torrus::SiteConfig;
+
+exit(1) if not Torrus::SiteConfig::verify();
+
+our $tree;
+our $nodaemon;
+our $runonce;
+our $delay = 0;
+our $debug;
+our $verbose;
+our $help_needed;
+
+# Derive the process name from the command line
+our $process_name = $0;
+$process_name =~ s/^.*\/([^\/]+)$/$1/;
+$process_name .= ' ' . join(' ', @ARGV);
+
+
+my $ok = GetOptions ('tree=s'   => \$tree,
+                     'nodaemon' => \$nodaemon,
+                     'runonce'  => \$runonce,
+                     'delay=i'  => \$delay,
+                     'debug'    => \$debug,
+                     'verbose'  => \$verbose,
+                     'help'     => \$help_needed);
+
+if( not $ok or not $tree or $help_needed or scalar(@ARGV) > 0 )
+{
+    print STDERR "Usage: $0 --tree=NAME [options...]\n",
+    "Options:\n",
+    "  --tree=NAME     tree name\n",
+    "  --nodaemon      do not fork daemon and log to STDERR\n",
+    "  --runonce       run one time and exit. Implies --nodaemon\n",
+    "  --delay         delay the start of the first cycle, minutes\n",
+    "  --debug         set the log level to debug\n",
+    "  --verbose       set the log level to info\n",
+    "  --help          this help message\n";
+    exit 1;
+}
+
+if( not Torrus::SiteConfig::mayRunMonitor( $tree ) )
+{
+    Error('Tree ' . $tree . ' is not configured to run monitor');
+    exit 1;
+}
+
+
+if( $debug )
+{
+    Torrus::Log::setLevel('debug');
+}
+elsif( $verbose )
+{
+    Torrus::Log::setLevel('verbose');
+}
+
+my $logfile = $Torrus::Global::logDir . '/monitor.' . $tree . '.log';
+my $pidfile;
+
+my $rotateLogs = sub
+{
+    Info('Caught SIGHUP. Reopening log file');
+    close( STDERR );
+    open( STDERR, ">>$logfile" );
+};
+
+if( not $nodaemon and not $runonce )
+{
+    my $pidfilename =
+        $Torrus::Global::pidDir . '/monitor.' . $tree . '.pid';
+    
+    if( -r $pidfilename )
+    {
+        Error("Another monitor daemon is running, pid=",
+              `cat $pidfilename`);
+        exit 1;
+    }
+
+    &Proc::Daemon::Init();
+    umask 0017; # Proc::Daemon::Init sets the mask to all-writable
+
+    $SIG{'HUP'} = $rotateLogs;
+
+    # At this point, we cannot tell anyone if "open" fails
+    open(STDERR, ">>$logfile");
+
+    $pidfile = $pidfilename;
+    
+    if( open( PID, ">$pidfile" ) )
+    {
+        printf PID ( "%d", $$ );
+        close PID;
+    }
+    else
+    {
+        Error("Cannot open $pidfile for writing: $!");
+    }
+}
+
+Info(sprintf("Torrus version %s", '@VERSION@'));
+Info(sprintf("%s started for tree %s", $0, $tree));
+Debug(sprintf("Process ID %d", $$));
+
+if( $delay > 0 )
+{
+    Info(sprintf('Delaying for %d minutes', $delay));
+    sleep($delay * 60);
+}
+
+&Torrus::DB::setSafeSignalHandlers();
+
+my %options =
+    (
+     '-ProcessName' => $process_name,
+     '-Tree'      => $tree,
+     '-Delay'     => $delay
+     );
+if( $runonce )
+{
+    $options{'-RunOnce'} = 1;
+}
+
+my $scheduler = new Torrus::MonitorScheduler( %options );
+$scheduler->run();
+
+if( not $options{'-RunOnce'} )
+{
+    Error("Monitor process exited: nothing to collect");
+    unlink $pidfile;
+}
+
+exit;
+
+
+END
+{
+    if( defined($pidfile) and -r $pidfile )
+    {
+        unlink $pidfile;
+    }
+}
+
+
+# Local Variables:
+# mode: perl
+# indent-tabs-mode: nil
+# perl-indent-level: 4
+# End: