src/lib/Torrus/Collector/ExtDBI.pm

   1 #  Copyright (C) 2005  Stanislav Sinyagin
   2 #
   3 #  This program is free software; you can redistribute it and/or modify
   4 #  it under the terms of the GNU General Public License as published by
   5 #  the Free Software Foundation; either version 2 of the License, or
   6 #  (at your option) any later version.
   7 #
   8 #  This program is distributed in the hope that it will be useful,
   9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 #  GNU General Public License for more details.
  12 #
  13 #  You should have received a copy of the GNU General Public License
  14 #  along with this program; if not, write to the Free Software
  15 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
  16
  17 # $Id$
  18 # Stanislav Sinyagin <ssinyagin@yahoo.com>
  19
  20 ## Pluggable backend module for ExternalStorage
  21 ## Stores data in a generic SQL database
  22
  23 # We use some internals of Torrus::SQL::SrvExport, but
  24 # handle the SQL by ourselves, for better efficiency.
  25
  26 package Torrus::Collector::ExtDBI;
  27
  28 use strict;
  29 use warnings;
  30
  31 use DBI;
  32 use Date::Format;
  33
  34 use Torrus::SQL::SrvExport;
  35 use Torrus::Log;
  36
  37 our $VERSION = 1.0;
  38
  39 $Torrus::Collector::ExternalStorage::backendInit =
  40     \&Torrus::Collector::ExtDBI::backendInit;
  41
  42 $Torrus::Collector::ExternalStorage::backendOpenSession =
  43     \&Torrus::Collector::ExtDBI::backendOpenSession;
  44
  45 $Torrus::Collector::ExternalStorage::backendStoreData =
  46     \&Torrus::Collector::ExtDBI::backendStoreData;
  47
  48 $Torrus::Collector::ExternalStorage::backendCloseSession =
  49     \&Torrus::Collector::ExtDBI::backendCloseSession;
  50
  51
  52 # Optional SQL connection subtype, configurable from torrus-siteconfig.pl
  53 our $subtype;
  54
  55 my $dbh;
  56 my $sth;
  57
  58 sub backendInit
  59 {
  60     my $collector = shift;
  61     my $token = shift;
  62 }
  63
  64 sub backendOpenSession
  65 {
  66     $dbh = Torrus::SQL::SrvExport->dbh( $subtype );
  67
  68     if( defined( $dbh ) )
  69     {
  70         $sth = $dbh->prepare( Torrus::SQL::SrvExport->sqlInsertStatement() );
  71         if( not defined( $sth ) )
  72         {
  73             Error('Error preparing the SQL statement: ' . $dbh->errstr);
  74         }
  75     }
  76 }
  77
  78
  79 sub backendStoreData
  80 {
  81     my $timestamp = shift;
  82     my $serviceid = shift;
  83     my $value = shift;
  84     my $interval = shift;
  85
  86     if( defined( $dbh ) and defined( $sth ) )
  87     {
  88         my $datestr = time2str('%Y-%m-%d', $timestamp);
  89         my $timestr = time2str('%H:%M:%S', $timestamp);
  90         if( isDebug() )
  91         {
  92             Debug('Updating SQL database: ' .
  93                   join(', ', $datestr, $timestr,
  94                        $serviceid, $value, $interval ));
  95         }
  96
  97         if( $sth->execute( $datestr, $timestr,
  98                            $serviceid, $value, $interval ) )
  99         {
 100             return 1;
 101         }
 102         else
 103         {
 104             Error('Error executing SQL: ' . $dbh->errstr);
 105         }
 106     }
 107
 108     return;
 109 }
 110
 111
 112 sub backendCloseSession
 113 {
 114     undef $sth;
 115     if( defined( $dbh ) )
 116     {
 117         $dbh->commit();
 118         $dbh->disconnect();
 119         undef $dbh;
 120     }
 121 }
 122
 123
 124
 125 1;
 126
 127
 128 # Local Variables:
 129 # mode: perl
 130 # indent-tabs-mode: nil
 131 # perl-indent-level: 4
 132 # End: