1 files changed, 8 insertions, 15 deletions
diff --git a/lib/MasterServer/Database/SQLite/dbMaintenance.pm b/lib/MasterServer/Database/SQLite/dbMaintenance.pm
index 019f31f..ec9b5cc 100755
--- a/lib/MasterServer/Database/SQLite/dbMaintenance.pm
+++ b/lib/MasterServer/Database/SQLite/dbMaintenance.pm
@@ -1,9 +1,8 @@
-package MasterServer::Database::SQLite::dbMaintenance;
+package MasterServer::Database::Pg::dbMaintenance;
 
 use strict;
 use warnings;
 use Exporter 'import';
-
 our @EXPORT = qw| delete_old_pending 
                       remove_pending |;
 
@@ -12,14 +11,10 @@ our @EXPORT = qw| delete_old_pending
 ## where the server is unresponsive for more than 1 hour
 ################################################################################
 sub delete_old_pending {
-  my ($self) = shift;
-
-  # remove servers
+  my $self = shift;
   my $u = $self->{dbh}->do(
      "DELETE FROM pending 
-      WHERE added < datetime(?, 'unixepoch')", undef, time-3600);
-  
-  # notify 
+      WHERE added < datetime(?, \'unixepoch\')", undef, time-3600);
   $self->log("delete", "Removed $u entries from pending.") if ($u > 0);
 }
 
@@ -28,13 +23,11 @@ sub delete_old_pending {
 ## of error(s).
 ################################################################################
 sub remove_pending {
-  my ($self, $id) = @_;
-  
-  # if address is in list, update the timestamp
-  my $u = $self->{dbh}->do("DELETE FROM pending WHERE id = ?", undef, $id);
-  
-  # notify 
-  $self->log("delete", "removed pending id $id from pending") if ($u > 0);
+  my $self = shift;
+  my %o = ( @_); 
+  my $u = $self->{dbh}->do("DELETE FROM pending WHERE ip = ? AND heartbeat = ?", 
+    undef, $o{ip}, $o{port});
+  $self->log("delete", "removed $o{ip}, $o{port} from pending (".$u."x)") if ($u > 0);
 }
 
 1;