ref(socket): Use new scopes API (#2915)

Author: sentrivana

sentry_sdk/integrations/socket.py

@@ -1,5 +1,6 @@
 import socket
-from sentry_sdk import Hub
+
+import sentry_sdk
 from sentry_sdk._types import MYPY
 from sentry_sdk.consts import OP
 from sentry_sdk.integrations import Integration
@@ -47,13 +48,11 @@ def create_connection(
         source_address=None,
     ):
         # type: (Tuple[Optional[str], int], Optional[float], Optional[Tuple[Union[bytearray, bytes, str], int]])-> socket.socket
-        hub = Hub.current
-        if hub.get_integration(SocketIntegration) is None:
-            return real_create_connection(
-                address=address, timeout=timeout, source_address=source_address
-            )
+        integration = sentry_sdk.get_client().get_integration(SocketIntegration)
+        if integration is None:
+            return real_create_connection(address, timeout, source_address)
 
-        with hub.start_span(
+        with sentry_sdk.start_span(
             op=OP.SOCKET_CONNECTION,
             description=_get_span_description(address[0], address[1]),
         ) as span:
@@ -74,11 +73,11 @@ def _patch_getaddrinfo():
 
     def getaddrinfo(host, port, family=0, type=0, proto=0, flags=0):
         # type: (Union[bytes, str, None], Union[str, int, None], int, int, int, int) -> List[Tuple[AddressFamily, SocketKind, int, str, Union[Tuple[str, int], Tuple[str, int, int, int]]]]
-        hub = Hub.current
-        if hub.get_integration(SocketIntegration) is None:
+        integration = sentry_sdk.get_client().get_integration(SocketIntegration)
+        if integration is None:
             return real_getaddrinfo(host, port, family, type, proto, flags)
 
-        with hub.start_span(
+        with sentry_sdk.start_span(
             op=OP.SOCKET_DNS, description=_get_span_description(host, port)
         ) as span:
             span.set_data("host", host)