fixup! refactor: tighten up cloudflare detection

brianc · Mar 13, 2024 · a6099dd · a6099dd
1 parent c5946f3
commit a6099dd
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/packages/pg/lib/stream.js b/packages/pg/lib/stream.js
@@ -3,12 +3,13 @@
  * @returns {Duplex}
  */
 module.exports.getStream = function getStream(ssl) {
-  try {
-    const { CloudflareSocket } = require('pg-cloudflare')
-    return new CloudflareSocket(ssl)
-  } catch {
+  console.log(process?.release?.name)
+  if (process?.release?.name === 'node') {
     const net = require('net')
     return new net.Socket()
+  } else {
+    const { CloudflareSocket } = require('pg-cloudflare')
+    return new CloudflareSocket(ssl)
   }
 }
 
@@ -18,11 +19,12 @@ module.exports.getStream = function getStream(ssl) {
  * @returns {Duplex}
  */
 module.exports.getSecureStream = function getSecureStream(options) {
-  try {
-    options.socket.startTls(options)
-    return options.socket
-  } catch {
+  if (process?.release?.name === 'node') {
     var tls = require('tls')
     return tls.connect(options)
   }
+  else {
+    options.socket.startTls(options)
+    return options.socket
+  }
 }