From c3f8d65a4fb1f5674557ee67cf7f74369df86ad1 Mon Sep 17 00:00:00 2001
From: Darkelarious <darkelarious@333networks.com>
Date: Sat, 19 Nov 2016 20:56:04 +0100
Subject: Massive improvements on efficiency, robustness, security, reliability
 and more

---
 util/tools/r_database.pl | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100755 util/tools/r_database.pl

(limited to 'util/tools/r_database.pl')

diff --git a/util/tools/r_database.pl b/util/tools/r_database.pl
new file mode 100755
index 0000000..601b1ae
--- /dev/null
+++ b/util/tools/r_database.pl
@@ -0,0 +1,40 @@
+#!/usr/bin/perl
+use strict;
+use warnings;
+use DBI;
+
+our $dbh = open_database();
+
+sub open_database {
+  my $dbh = DBI->connect('dbi:Pg:dbname=masterserver', 'user', 'password')
+    or die "Cannot connect: $DBI::errstr\n";
+  
+  # don't forget at end!
+  #$dbh->disconnect;
+}
+
+################################################################################
+## Subroutine dbAddServer
+## Add a server to the database (address + port only)
+## If the address already exists in the database, it will be ignored.
+################################################################################
+sub db_add_server {
+  my %o = (@_);
+
+  # Try to get the address out of the database.
+  my $exists = $dbh->selectall_arrayref(
+    "SELECT * FROM serverlist WHERE ip = ? AND port = ?", 
+    undef, $o{ip}, $o{port});
+  return if (defined $exists->[0]);
+  
+  $exists = $dbh->selectall_arrayref(
+    "SELECT * FROM pending WHERE ip = ? AND heartbeat = ?", 
+    undef, $o{ip}, $o{port});
+  return if (defined $exists->[0]);
+  
+  #print "Add $o{ip}\t $o{port}\n";
+  $exists = $dbh->do("INSERT INTO pending (ip, heartbeat) VALUES(?, ?)",
+     undef, $o{ip}, $o{port});
+}
+
+1;
-- 
cgit v1.2.3