perl/DXSql.pm

   1 #
   2 # The master SQL module
   3 #
   4 #
   5 #
   6 # Copyright (c) 2006 Dirk Koopman G1TLH
   7 #
   8
   9 package DXSql;
  10
  11 use strict;
  12
  13 use DXSql::SQLite;
  14 use DXSql::mysql;
  15 use DXDebug;
  16
  17 use vars qw($active);
  18 $active = 0;
  19
  20 sub init
  21 {
  22         my $dsn = shift;
  23         return unless $dsn;
  24         return $active if $active;
  25
  26         eval {
  27                 require DBI;
  28         };
  29         unless ($@) {
  30                 import DBI;
  31                 $active++;
  32         }
  33         undef $@;
  34         return $active;
  35 }
  36
  37 sub new
  38 {
  39         my $class = shift;
  40         my $dsn = shift;
  41         my $self;
  42
  43         return undef unless $active;
  44         my $dbh;
  45         my ($style) = $dsn =~ /^dbi:(\w+):/;
  46         my $newclass = "DXSql::$style";
  47         eval "require $newclass";
  48         if ($@) {
  49                 $active = 0;
  50                 return undef;
  51         }
  52         return bless {}, $newclass;
  53 }
  54
  55 sub connect
  56 {
  57         my $self = shift;
  58         my $dsn = shift;
  59         my $user = shift;
  60         my $passwd = shift;
  61
  62         my $dbh;
  63         eval {
  64                 no strict 'refs';
  65                 $dbh = DBI->connect($dsn, $user, $passwd);
  66         };
  67         unless ($dbh) {
  68                 $active = 0;
  69                 return undef;
  70         }
  71         $self->{dbh} = $dbh;
  72         return $self;
  73 }
  74
  75 sub finish
  76 {
  77         my $self = shift;
  78         $self->{dbh}->disconnect;
  79 }
  80
  81 sub do
  82 {
  83         my $self = shift;
  84         my $s = shift;
  85
  86         eval { $self->{dbh}->do($s); };
  87 }
  88
  89 sub begin_work
  90 {
  91         $_[0]->{dbh}->begin_work;
  92 }
  93
  94 sub commit
  95 {
  96         $_[0]->{dbh}->commit;
  97 }
  98
  99 sub rollback
 100 {
 101         $_[0]->{dbh}->rollback;
 102 }
 103
 104 sub quote
 105 {
 106         return $_[0]->{dbh}->quote($_[1]);
 107 }
 108
 109 sub prepare
 110 {
 111         return $_[0]->{dbh}->prepare($_[1]);
 112 }
 113
 114 sub spot_insert_prepare
 115 {
 116         my $self = shift;
 117         return $self->prepare('insert into spot values(?' . ',?' x 15 . ')');
 118 }
 119
 120 sub spot_insert
 121 {
 122         my $self = shift;
 123         my $spot = shift;
 124         my $sth = shift;
 125
 126         if ($sth) {
 127                 push @$spot, undef while  @$spot < 15;
 128                 pop @$spot while @$spot > 15;
 129                 eval {$sth->execute(undef, @$spot)};
 130         } else {
 131                 my $s = "insert into spot values(NULL,";
 132                 $s .= sprintf("%.1f,", $spot->[0]);
 133                 $s .= $self->quote($spot->[1]) . "," ;
 134                 $s .= $spot->[2] . ',';
 135                 $s .= (length $spot->[3] ? $self->quote($spot->[3]) : 'NULL') . ',';
 136                 $s .= $self->quote($spot->[4]) . ',';
 137                 $s .= $spot->[5] . ',';
 138                 $s .= $spot->[6] . ',';
 139                 $s .= (length $spot->[7] ? $self->quote($spot->[7]) : 'NULL') . ',';
 140                 $s .= $spot->[8] . ',';
 141                 $s .= $spot->[9] . ',';
 142                 $s .= $spot->[10] . ',';
 143                 $s .= $spot->[11] . ',';
 144                 $s .= (length $spot->[12] ? $self->quote($spot->[12]) : 'NULL') . ',';
 145                 $s .= (length $spot->[13] ? $self->quote($spot->[13]) : 'NULL') . ',';
 146                 $s .= (length $spot->[14] ? $self->quote($spot->[14]) : 'NULL') . ')';
 147                 eval {$self->do($s)};
 148         }
 149 }
 150
 151 sub spot_search
 152 {
 153         my $self = shift;
 154         my $expr = shift;
 155         my $dayfrom = shift;
 156         my $dayto = shift;
 157         my $n = shift;
 158         my $dxchan = shift;
 159
 160         dbg("expr: $expr") if isdbg('search');
 161         if ($expr =~ /\$f/) {
 162                 $expr =~ s/(?:==|eq)/ = /g;
 163                 $expr =~ s/\$f10/spotteritu/g;
 164                 $expr =~ s/\$f11/spottercq/g;
 165                 $expr =~ s/\$f12/spotstate/g;
 166                 $expr =~ s/\$f13/spotterstate/g;
 167                 $expr =~ s/\$f0/freq/g;
 168                 $expr =~ s/\$f1/spotcall/g;
 169                 $expr =~ s/\$f2/time/g;
 170                 $expr =~ s/\$f3/comment/g;
 171                 $expr =~ s/\$f4/spotter/g;
 172                 $expr =~ s/\$f5/spotdxcc/g;
 173                 $expr =~ s/\$f6/spotterdxcc/g;
 174                 $expr =~ s/\$f7/origin/g;
 175                 $expr =~ s/\$f8/spotitu/g;
 176                 $expr =~ s/\$f9/spotcq/g;
 177                 $expr =~ s/\|\|/ or /g;
 178                 $expr =~ s/\&\&/ and /g;
 179                 $expr =~ s/=~\s+m\{\^([%\w]+)[^\}]*\}/ like '$1'/g;
 180         } else {
 181                 $expr = '';
 182         }
 183         my $fdays = $dayfrom ? "time <= " . ($main::systime - ($dayfrom * 86400)) : "";
 184         my $days = "time >= " . ($main::systime - ($dayto * 86400));
 185         my $trange = $fdays ? "($fdays and $days)" : $days;
 186         $expr .= $expr ? " and $trange" : $trange;
 187     my $s = qq{select freq,spotcall,time,comment,spotter,spotdxcc,spotterdxcc,
 188 origin,spotitu,spotcq,spotteritu,spottercq,spotstate,spotterstate from spot
 189 where $expr order by time desc limit $n};
 190     dbg("sql expr: $s") if isdbg('search');
 191         my $ref = $self->{dbh}->selectall_arrayref($s);
 192         return @$ref;
 193 }
 194
 195 1;
 196