From 3755e1047db0b9ffdd7b8e0ea54d5e2ef3683fb7 Mon Sep 17 00:00:00 2001
From: Jordan <me@jordan.im>
Date: Sun, 10 Apr 2022 23:06:42 -0700
Subject: asn: performance improvements; re-use connection between threads

---
 asn.py            | 13 ++++++-------
 location-database |  2 +-
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/asn.py b/asn.py
index 6bc2c98..f79ee6c 100755
--- a/asn.py
+++ b/asn.py
@@ -18,7 +18,8 @@ log = logging.getLogger('asn')
 log.setLevel(logging.DEBUG)
 
 class Listener:
-    def __init__(self, host, port):
+    def __init__(self, db, host, port):
+        self.db = db
         self._listen(host, port)
 
     def _listen(self, host, port):
@@ -54,7 +55,6 @@ class Listener:
             conn.close()
 
     def _get_announcements(self, recv):
-        db = DB()
         hosts = set()
         try:
             ip = ipaddress.ip_address(recv)
@@ -72,7 +72,7 @@ class Listener:
 
                 n = self._get_netblock(host)
                 if n:
-                    announcements.extend(db.query(n))
+                    announcements.extend(self.db.query(n))
 
             return announcements
 
@@ -128,7 +128,7 @@ class DB:
     def __init__(self):
         self.repo_path = os.path.dirname(os.path.abspath(__file__))
         self.db_path = os.path.join(self.repo_path, 'cache.db')
-        self.con = sqlite3.connect(self.db_path)
+        self.con = sqlite3.connect(self.db_path, check_same_thread=False)
 
         loc = os.path.join(self.repo_path, 'location-database')
         self.dataset = os.path.join(loc, 'database.txt')
@@ -266,10 +266,10 @@ if __name__ == '__main__':
                         required=False)
     args = parser.parse_args()
 
+    db = DB()
     if args.host and args.port:
-        listen = Listener(args.host, args.port)
+        listen = Listener(db, args.host, args.port)
     elif args.update:
-        db = DB()
         log.info('checking remote repository for new dataset...')
         if db.update():
             log.info('dataset updated, creating/populating cache...')
@@ -277,7 +277,6 @@ if __name__ == '__main__':
         else:
             log.info('no changes since last update')
     elif args.populate:
-        db = DB()
         log.info('creating/populating cache...')
         db.populate_db()
     else:
diff --git a/location-database b/location-database
index 7ad5449..80cd5c7 160000
--- a/location-database
+++ b/location-database
@@ -1 +1 @@
-Subproject commit 7ad544905562799b98ae99b3971d6db770ba0b47
+Subproject commit 80cd5c76ee0c04960da618c18f2feaad2c1f5f0f
-- 
cgit v1.2.3-54-g00ecf