Changeset 1042 for trunk/packages/sipb-xen-dhcp/files/usr

Timestamp:

Oct 4, 2008, 11:23:12 PM (16 years ago)

Author:

broder

Message:

Remove the hack to reconnect to the database if it goes away

File:

• trunk/packages/sipb-xen-dhcp/files/usr/sbin/sipb-xen-dhcpserver (modified) (3 diffs)

trunk/packages/sipb-xen-dhcp/files/usr/sbin/sipb-xen-dhcpserver

@@ -12 +12 @@
 import syslog as s
 
-import psycopg2
 import time
 from invirt import database
@@ -27 +26 @@
     def findNIC(self, mac):
         database.clear_cache()
-        for i in range(3):
-            try:
-                value = database.NIC.query().filter_by(mac_addr=mac).one()
-            except psycopg2.OperationalError:
-                time.sleep(0.5)
-                if i == 2:  #Try twice to reconnect.
-                    raise
-                #Sigh.  SQLAlchemy should do this itself.
-                database.connect()
-            else:
-                break
-        return value
+        return database.NIC.query().filter_by(mac_addr=mac).one()
     def find_interface(self, packet):
         chaddr = hwmac(packet.GetHardwareAddress())
         nic = self.findNIC(str(chaddr))
         if nic is None or nic.ip is None:
-            return
+            return None
         ipstr = ''.join(reversed(['%02X' % i for i in ipv4(nic.ip).list()]))
         for line in open('/proc/net/route'):
@@ -50 +38 @@
                 s.syslog(s.LOG_DEBUG, "find_interface found "+str(nic.ip)+" on "+parts[0])
                 return parts[0]
-        return
+        return None
                             
     def getParameters(self, **extra):