dialup_admin/bin/tot_stats

   1 #!/usr/bin/perl
   2 use POSIX;
   3
   4 # Log in the totacct table aggregated daily accounting information for
   5 # each user.
   6 # We keep a row per user for each day.
   7 # Works with mysql and postgresql
   8 #
   9
  10 $conf=shift||'/usr/local/dialup_admin/conf/admin.conf';
  11
  12
  13 open CONF, "<$conf"
  14         or die "Could not open configuration file\n";
  15 while(<CONF>){
  16         chomp;
  17         ($key,$val)=(split /:\s*/,$_);
  18         $sql_type = $val if ($key eq 'sql_type');
  19         $sql_server = $val if ($key eq 'sql_server');
  20         $sql_username = $val if ($key eq 'sql_username');
  21         $sql_password = $val if ($key eq 'sql_password');
  22         $sql_database = $val if ($key eq 'sql_database');
  23         $sql_accounting_table = $val if ($key eq 'sql_accounting_table');
  24         $sqlcmd = $val if ($key eq 'sql_command');
  25 }
  26 close CONF;
  27
  28 die "sql_command directive is not set in admin.conf\n" if ($sqlcmd eq '');
  29 die "sql command '$sqlcmd' not found or does not seem to be executable\n" if (! -x $sqlcmd);
  30
  31 if ($sql_type eq 'mysql'){
  32         $sql_password = ($sql_password eq '') ? '' : "-p$sql_password";
  33 }
  34 $sql_password =~ s/(\W)/\\$1/g;
  35
  36 ($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime;
  37 $date_start = POSIX::strftime("%Y-%m-%d %T",0,0,0,($mday - 1),$mon,$year,$wday,$yday,$isdst);
  38 $date_small_start = POSIX::strftime("%Y-%m-%d",0,0,0,($mday - 1),$mon,$year,$wday,$yday,$isdst);
  39 $date_end = POSIX::strftime("%Y-%m-%d %T",0,0,0,$mday,$mon,$year,$wday,$yday,$isdst);
  40
  41 $query1 = "DELETE FROM totacct WHERE AcctDate = '$date_start';";
  42 $query2 = "INSERT INTO totacct (UserName,AcctDate,ConnNum,ConnTotDuration,
  43         ConnMaxDuration,ConnMinDuration,InputOctets,OutputOctets,NASIPAddress)
  44         SELECT UserName,'$date_small_start',COUNT(*),SUM(AcctSessionTime),
  45         MAX(AcctSessionTime),MIN(AcctSessionTime),SUM(AcctInputOctets),
  46         SUM(AcctOutputOctets),NASIPAddress FROM radacct
  47         WHERE AcctStopTime >= '$date_start' AND
  48         AcctStopTime < '$date_end' GROUP BY UserName,NASIPAddress;";
  49 print "$query1\n";
  50 print "$query2\n";
  51 open TMP, ">/tmp/tot_stats.query"
  52         or die "Could not open tmp file\n";
  53 print TMP "ALTER SESSION SET NLS_TIMESTAMP_TZ_FORMAT='YYYY-MM-DD HH24:MI:SS.FF TZH:TZM';\n" if ($sql_type eq 'oracle');
  54 print TMP $query1;
  55 print TMP $query2;
  56 close TMP;
  57 $command = "$sqlcmd -h $sql_server -u $sql_username $sql_password $sql_database </tmp/tot_stats.query" if ($sql_type eq 'mysql');
  58 $command = "$sqlcmd  -U $sql_username -f /tmp/tot_stats.query $sql_database" if ($sql_type eq 'pg');
  59 $command = "$sqlcmd  $sql_username/$pass" . "@" . "$sql_database <$tmpfile.$server" if ($sql_type eq 'oracle');
  60 $command = "$sqlcmd '$sql_server' '$sql_port' '' '$sql_username' '$sql_password' </tmp/tot_stats.query" if ($sql_type eq 'sqlrelay');
  61 `$command`;