Guard against undefined.

open-telemetry · Dec 5, 2024 · bd89bcc · bd89bcc
1 parent d3bf605
commit bd89bcc
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/plugins/node/opentelemetry-instrumentation-pg/src/instrumentation.ts b/plugins/node/opentelemetry-instrumentation-pg/src/instrumentation.ts
@@ -152,11 +152,15 @@ export class PgInstrumentation extends InstrumentationBase<PgInstrumentationConf
       'pg',
       SUPPORTED_PG_VERSIONS,
       (module: any) => {
-        this._patchPgClient(module.Client);
+        const moduleExports = extractModuleExports(module);
+
+        this._patchPgClient(moduleExports.Client);
         return module;
       },
       (module: any) => {
-        this._unpatchPgClient(module.Client);
+        const moduleExports = extractModuleExports(module);
+
+        this._unpatchPgClient(moduleExports.Client);
         return module;
       },
       [modulePgClient, modulePgNativeClient]
@@ -187,6 +191,10 @@ export class PgInstrumentation extends InstrumentationBase<PgInstrumentationConf
   }
 
   private _patchPgClient(module: any) {
+    if (!module) {
+      return;
+    }
+
     const moduleExports = extractModuleExports(module);
 
     if (isWrapped(moduleExports.prototype.query)) {