1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
package MasterServer::Database::SQLite::dbCore;
use strict;
use warnings;
use POSIX qw/strftime/;
use Exporter 'import';
our @EXPORT = qw| database_login dump_database |;
################################################################################
## login to the database with credentials provided in the config file.
## returns dbh object or quits application on error.
##
## Recommended database types: Postgresql, MySQL or SQLite. Warranty void if
## other database types are used. Use at your own risk.
################################################################################
sub database_login {
my $self = shift;
# get db info
my @db_type = split(':', $self->{dblogin}->[0]);
# inform what db we try to load
$self->log("info","Database: $db_type[1], $db_type[2]");
# check if database file exists
my $db_file = [split(':', $self->{dblogin}->[0])]->[2];
$db_file =~ s/dbname=//i;
unless (-e $db_file) {
# fatal error
$self->log("fatal", "Database file $db_file does not exist!");
# end program
$self->halt();
}
# create the dbi object
my $dbh = DBI->connect(@{$self->{dblogin}}, {PrintError => 1});
# verify that the database connected
if (defined $dbh) {
# log the event
$self->log("info","Connected to the $db_type[1] database.");
# turn on error printing
$dbh->{printerror} = 1;
# synchronous read/writing to the SQLite file OFF. That means: when the script
# shuts down unexpectedly, i.e. because of power failure or a crash, changes
# to the database are NOT SAVED. However, if this setting is not turned OFF,
# it takes too long to write to the database, which means that new beacons,
# requests and servers cannot be processed. You don't have a choice, really..
$dbh->do("PRAGMA synchronous = OFF");
# allow the use of foreign keys (referencing)
$dbh->do("PRAGMA foreign_keys = ON");
# return the dbi object for further use
return $dbh;
}
else {
# fatal error
$self->log("fatal", "$DBI::errstr!");
# end program
$self->halt();
}
# in case of any other error, return undef.
return undef;
}
################################################################################
## Dump the database in the data/dump folder.
## useful for backups, historical data
################################################################################
sub dump_database {
my $self = shift;
# filename / time
my $time = strftime('%Y-%m-%d-%H-%M',localtime);
# read db credentials from db login
my @db_type = split ':', $self->{dblogin}->[0];
$db_type[2] =~ s/dbname=//;
# split db path
my @db_path = split '/', $db_type[2];
# use pg_dump to dump Postgresql databases
system("cp $db_type[2] $self->{root}/data/dumps/SQLite-$time-$db_path[-1]");
# log
$self->log("dump", "Dumping database to /data/dumps/SQLite-$time-$db_path[-1]");
}
1;
|
