Make sure we hold the same conn for listen forever

Stephen Paul Weber created 2 years ago

hold by itself will never cut it, because we use EM timers inside to
break the connection even to the promise, so we can't sync on that.
Since we really do want to essentially leak this connection for the life
of the process, then let's do that directly.

Change summary

lib/postgres.rb |  7 +++++++
sgx_jmp.rb      | 21 ++++++++++-----------
2 files changed, 17 insertions(+), 11 deletions(-)

Detailed changes

lib/postgres.rb

@@ -10,6 +10,13 @@ class Postgres < SimpleDelegator
 		})
 	end
 
+	# WARNING: this uses up a connection from the pool which never releases
+	# Use it only if you need the same connection for the life of the process
+	def acquire
+		conn = __getobj__.send(:acquire, Fiber.current) until conn
+		conn
+	end
+
 	def query_one(sql, *args, field_names_as: :symbol, default: nil)
 		query_defer(sql, args).then do |rows|
 			rows.field_names_as(field_names_as)&.first || default

sgx_jmp.rb

@@ -223,17 +223,16 @@ when_ready do
 	DB = Postgres.connect(dbname: "jmp", size: 5)
 	TEL_SELECTIONS = TelSelections.new
 
-	EMPromise.resolve(nil).then do
-		DB.hold do |conn|
-			conn.query("LISTEN low_balance")
-			conn.query("LISTEN possible_renewal")
-			catchup_notify_low_balance(conn)
-			catchup_notify_possible_renewal(conn)
-
-			repo = CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
-			poll_for_notify(conn, repo).sync
-		end
-	end
+	EMPromise.resolve(nil).then {
+		conn = DB.acquire
+		conn.query("LISTEN low_balance")
+		conn.query("LISTEN possible_renewal")
+		catchup_notify_low_balance(conn)
+		catchup_notify_possible_renewal(conn)
+
+		repo = CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
+		poll_for_notify(conn, repo)
+	}.catch(&method(:panic))
 
 	load_plans_to_db!