async_hooks: use instanceof to identify resource types

Signed-off-by: Darshan Sen <[email protected]>

lib/async_hooks.js

@@ -252,6 +252,30 @@ class AsyncResource {
   }
 }
 
+let _tls_wrap;
+
+function lazyLoadTlsWrap() {
+  _tls_wrap ??= require('_tls_wrap');
+}
+
+let dgram;
+
+function lazyLoadDgram() {
+  dgram ??= require('dgram');
+}
+
+let internal_dgram;
+
+function lazyLoadInternalDgram() {
+  internal_dgram ??= require('internal/dgram');
+}
+
+let net;
+
+function lazyLoadNet() {
+  net ??= require('net');
+}
+
 function lookupResourceWithStorage() {
   // When a TCP/UDP object is created, its owner_symbol is uninitialized, so
   // calling executionAsyncResource() would return the TCP/UDP object itself.
@@ -263,11 +287,23 @@ function lookupResourceWithStorage() {
   // object, which does not contain the storage that was saved initially. Hence,
   // in the case of a Socket/TLSSocket object, we must use the underlying
   // TCP/UDP object, if available, for storage.
+
+  lazyLoadDgram();
+  lazyLoadInternalDgram();
+  lazyLoadNet();
+  lazyLoadTlsWrap();
+
   let resource = executionAsyncResource();
-  if ((resource.constructor.name === 'Socket' ||
-       resource.constructor.name === 'TLSSocket') &&
-      resource._handle != null)
+
+  if ((resource instanceof net.Socket ||
+       resource instanceof _tls_wrap.TLSSocket) &&
+      resource._handle != null) {
     resource = resource._handle;
+  } else if (resource instanceof dgram.Socket &&
+             resource[internal_dgram.kStateSymbol].handle != null) {
+    resource = resource[internal_dgram.kStateSymbol].handle;
+  }
+
   return resource;
 }