gosa-si/modules/DBsqlite.pm

   1 package GOSA::DBsqlite;
   2
   3
   4 use strict;
   5 use warnings;
   6 use DBI;
   7 use Data::Dumper;
   8 use GOSA::GosaSupportDaemon;
   9 use threads;
  10 use Time::HiRes qw(usleep);
  11
  12
  13 my $col_names = {};
  14
  15 sub new {
  16     my $class = shift;
  17     my $db_name = shift;
  18
  19     my $lock = $db_name.".si.lock";
  20         # delete existing lock - instance should be running only once
  21         if(stat($lock)) {
  22                 &main::daemon_log("DEBUG: Removed existing lock $lock.", 7);
  23                 unlink($lock);
  24         }
  25     my $self = {dbh=>undef,db_name=>undef,db_lock=>undef,db_lock_handle=>undef};
  26     my $dbh = DBI->connect("dbi:SQLite:dbname=$db_name");
  27     $self->{dbh} = $dbh;
  28     $self->{db_name} = $db_name;
  29     $self->{db_lock} = $lock;
  30     bless($self,$class);
  31
  32     return($self);
  33 }
  34
  35 sub lock_exists : locked {
  36     my $self=shift;
  37     my $funcname=shift;
  38     my $lock = $self->{db_lock};
  39     my $result=(-f $lock);
  40         my $i=0;
  41     if($result) {
  42                 if($i>10) {
  43                 &main::daemon_log("WARNING: (".((defined $funcname)?$funcname:"").") Lock (PID ".$$.") $lock found. Waiting time: ".($i*500)." us", 8);
  44                 }
  45         usleep 500;
  46                 $i++;
  47     }
  48     return $result;
  49 }
  50
  51 sub create_lock : locked {
  52     my $self=shift;
  53     my $funcname=shift;
  54     #&main::daemon_log("(".((defined $funcname)?$funcname:"").") Creating Lock (PID ".$$.") ".($self->{db_lock}),8);
  55
  56     my $lock = $self->{db_lock};
  57     while( -f $lock ) {
  58                 #&main::daemon_log("(".((defined $funcname)?$funcname:"").") Lock (PID ".$$.") $lock found",8);
  59         usleep 100;
  60     }
  61
  62     open($self->{db_lock_handle},'>',$self->{db_lock});
  63 }
  64
  65 sub remove_lock : locked {
  66     my $self=shift;
  67     my $funcname=shift;
  68     #&main::daemon_log("(".((defined $funcname)?$funcname:"").") Removing Lock (PID ".$$.") ".$self->{db_lock}, 8);
  69     close($self->{db_lock_handle});
  70     unlink($self->{db_lock});
  71 }
  72
  73
  74 sub create_table {
  75     my $self = shift;
  76     my $table_name = shift;
  77     my $col_names_ref = shift;
  78     my @col_names;
  79     foreach my $col_name (@$col_names_ref) {
  80         my @t = split(" ", $col_name);
  81         $col_name = $t[0];
  82         push(@col_names, $col_name);
  83     }
  84
  85     $col_names->{ $table_name } = $col_names_ref;
  86     my $col_names_string = join("', '", @col_names);
  87     my $sql_statement = "CREATE TABLE IF NOT EXISTS $table_name ( '$col_names_string' )";
  88     &create_lock($self,'create_table');
  89     $self->{dbh}->do($sql_statement);
  90     &remove_lock($self,'create_table');
  91     return 0;
  92 }
  93
  94
  95 sub add_dbentry {
  96     my $self = shift;
  97     my $arg = shift;
  98
  99     # if dbh not specified, return errorflag 1
 100     my $table = $arg->{table};
 101     if( not defined $table ) {
 102         return 1 ;
 103     }
 104
 105     # specify primary key in table
 106     if (not exists $arg->{primkey}) {
 107         return (2, "a hash key 'primkey' with at least an empty list as value is necessary for add_dbentry");
 108     }
 109     my $primkeys = $arg->{'primkey'};
 110     my $prim_statement="";
 111     if( 0 != @$primkeys ) {
 112         my @prim_list;
 113         foreach my $primkey (@$primkeys) {
 114             if($primkey eq 'id') {
 115                 # if primkey is id, fetch max id from table and give new job id=  max(id)+1
 116                 my $sql_statement = "SELECT MAX(CAST(id AS INTEGER)) FROM $table";
 117                 &create_lock($self,'add_dbentry');
 118                 my $max_id = @{ @{ $self->{dbh}->selectall_arrayref($sql_statement) }[0] }[0];
 119                 &remove_lock($self,'add_dbentry');
 120                 my $id;
 121                 if( defined $max_id) {
 122                     $id = $max_id + 1;
 123                 } else {
 124                     $id = 1;
 125                 }
 126                 $arg->{id} = $id;
 127             }
 128             if( not exists $arg->{$primkey} ) {
 129                 return (3, "primkey '$primkey' has no value for add_dbentry");
 130             }
 131            push(@prim_list, "$primkey='".$arg->{$primkey}."'");
 132         }
 133         $prim_statement = "WHERE ".join(" AND ", @prim_list);
 134     }
 135
 136     # if timestamp is not provided, add timestamp
 137     if( not exists $arg->{timestamp} ) {
 138         $arg->{timestamp} = &get_time;
 139     }
 140
 141     # check wether primkey is unique in table, otherwise return errorflag
 142     my $sql_statement = "SELECT * FROM $table $prim_statement";
 143     &create_lock($self,'add_dbentry');
 144     my $res = @{ $self->{dbh}->selectall_arrayref($sql_statement) };
 145     &remove_lock($self,'add_dbentry');
 146
 147     if ($res == 0) {
 148         # primekey is unique
 149
 150         # fetch column names of table
 151         my $col_names = &get_table_columns($self, $table);
 152
 153         # assign values to column name variables
 154         my @col_list;
 155                 my @val_list;
 156         foreach my $col_name (@{$col_names}) {
 157             # use function parameter for column values
 158             if (exists $arg->{$col_name}) {
 159                 push(@col_list, $col_name);
 160                                 push(@val_list, $arg->{$col_name});
 161             }
 162         }
 163
 164         my $sql_statement = "INSERT INTO $table (".join(", ", @col_list).") VALUES ('".join("', '", @val_list)."')";
 165         &create_lock($self,'add_dbentry');
 166         my $db_res = $self->{dbh}->do($sql_statement);
 167         &remove_lock($self,'add_dbentry');
 168         if( $db_res != 1 ) {
 169             return (4, $sql_statement);
 170         }
 171
 172     } else  {
 173         # entry already exists, so update it
 174         my @update_l;
 175         while( my ($pram, $val) = each %{$arg} ) {
 176             if( $pram eq 'table' ) { next; }
 177             if( $pram eq 'primkey' ) { next; }
 178             push(@update_l, "$pram='$val'");
 179         }
 180         my $update_str= join(", ", @update_l);
 181         $update_str= " SET $update_str";
 182
 183         my $sql_statement= "UPDATE $table $update_str $prim_statement";
 184         my $db_res = &update_dbentry($self, $sql_statement );
 185
 186     }
 187
 188     return 0;
 189 }
 190
 191 sub update_dbentry {
 192     my ($self, $sql)= @_;
 193     my $db_answer= &exec_statement($self, $sql);
 194     return $db_answer;
 195 }
 196
 197
 198 sub del_dbentry {
 199     my ($self, $sql)= @_;;
 200     my $db_res= &exec_statement($self, $sql);
 201     return $db_res;
 202 }
 203
 204
 205 sub get_table_columns {
 206     my $self = shift;
 207     my $table = shift;
 208     my @column_names;
 209
 210     if(exists $col_names->{$table}) {
 211         @column_names = @{$col_names->{$table}};
 212     } else {
 213         &create_lock($self,'get_table_columns');
 214         my @res = @{$self->{dbh}->selectall_arrayref("pragma table_info('$table')")};
 215         &remove_lock($self,'get_table_columns');
 216
 217         foreach my $column (@res) {
 218             push(@column_names, @$column[1]);
 219         }
 220     }
 221     return \@column_names;
 222
 223 }
 224
 225
 226 sub select_dbentry {
 227     my ($self, $sql)= @_;
 228     my $error= 0;
 229     my $answer= {};
 230
 231     my $db_answer= &exec_statement($self, $sql);
 232
 233     # fetch column list of db and create a hash with column_name->column_value of the select query
 234     $sql =~ /FROM ([\S]*?)( |$)/g;
 235     my $table = $1;
 236     my $column_list = &get_table_columns($self, $table);
 237     my $list_len = @{ $column_list } ;
 238     my $hit_counter = 0;
 239     foreach my $hit ( @{ $db_answer }) {
 240         $hit_counter++;
 241         for ( my $i = 0; $i < $list_len; $i++) {
 242             $answer->{ $hit_counter }->{ @{ $column_list }[$i] } = @{ $hit }[$i];
 243         }
 244     }
 245
 246     return $answer;
 247 }
 248
 249
 250 sub show_table {
 251     my $self = shift;
 252     my $table_name = shift;
 253
 254     my $sql_statement= "SELECT * FROM $table_name ORDER BY timestamp";
 255     my $res= &exec_statement($self, $sql_statement);
 256
 257     my @answer;
 258     foreach my $hit (@{$res}) {
 259         push(@answer, "hit: ".join(', ', @{$hit}));
 260     }
 261     return join("\n", @answer);
 262 }
 263
 264
 265 sub exec_statement {
 266     my $self = shift;
 267     my $sql_statement = shift;
 268
 269     &create_lock($self,'exec_statement');
 270     my @db_answer = @{$self->{dbh}->selectall_arrayref($sql_statement)};
 271     &remove_lock($self, 'exec_statement');
 272
 273     return \@db_answer;
 274 }
 275
 276
 277 sub exec_statementlist {
 278     my $self = shift;
 279     my $sql_list = shift;
 280     my @db_answer;
 281
 282     &create_lock($self,'exec_statement');
 283     foreach my $sql (@$sql_list) {
 284         @db_answer = @{$self->{dbh}->selectall_arrayref($sql)};
 285     }
 286     &remove_lock($self, 'exec_statement');
 287
 288     return \@db_answer;
 289 }
 290
 291 sub count_dbentries {
 292     my ($self, $table)= @_;
 293     my $error= 0;
 294     my $answer= -1;
 295
 296     my $sql_statement= "SELECT * FROM $table";
 297     my $db_answer= &select_dbentry($self, $sql_statement);
 298
 299     my $count = keys(%{$db_answer});
 300     return $count;
 301 }
 302
 303
 304
 305 sub move_table {
 306     my ($self, $from, $to) = @_;
 307
 308     my $sql_statement_drop = "DROP TABLE IF EXISTS $to";
 309     my $sql_statement_alter = "ALTER TABLE $from RENAME TO $to";
 310     &create_lock($self,'move_table');
 311     my $db_res = $self->{dbh}->do($sql_statement_drop);
 312     $db_res = $self->{dbh}->do($sql_statement_alter);
 313     &remove_lock($self,'move_table');
 314
 315     return;
 316 }
 317
 318
 319 1;