Skip to content

Commit

Permalink
[Hack] Replace RLock by FakeLock
Browse files Browse the repository at this point in the history
Signed-off-by: Kent Huang <[email protected]>
  • Loading branch information
kentwelcome committed Sep 6, 2023
1 parent 162cbbb commit 27dc8bc
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 4 deletions.
25 changes: 22 additions & 3 deletions dbt/adapters/piperider/connections.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
from contextlib import contextmanager
from dataclasses import dataclass
from typing import Tuple, List
from typing import Tuple, List, Dict, Hashable, Optional

import agate
import dbt.exceptions # noqa
from dbt.adapters.base import Credentials
from dbt.adapters.base.query_headers import MacroQueryStringSetter
from dbt.adapters.sql import SQLConnectionManager
from dbt.contracts.connection import AdapterResponse
from dbt.contracts.connection import AdapterResponse, AdapterRequiredConfig, Connection


# from dbt.logger import GLOBAL_LOGGER as logger
Expand Down Expand Up @@ -50,11 +51,29 @@ def _connection_keys(self):
return ()


class FakeLock:
def __enter__(self):
"""In __enter__()"""
pass

def __exit__(self, exception_type, exception_value, exception_traceback):
"""In __exit__()"""
pass


class PipeRiderAdapterConnectionManager(SQLConnectionManager):
def __init__(self, profile: AdapterRequiredConfig):
self.profile = profile
self.thread_connections: Dict[Hashable, Connection] = {}
self.lock: FakeLock = FakeLock() # type: ignore
self.query_header: Optional[MacroQueryStringSetter] = None

def cancel_open(self) -> List[str]:
return super().cancel_open()

def execute(self, sql: str, auto_begin: bool = False, fetch: bool = False) -> Tuple[AdapterResponse, agate.Table]:
def execute(
self, sql: str, auto_begin: bool = False, fetch: bool = False, limit: Optional[int] = None
) -> Tuple[AdapterResponse, agate.Table]:
return super().execute(sql, auto_begin, fetch)

def begin(self):
Expand Down
2 changes: 1 addition & 1 deletion dbt/adapters/piperider/impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ class PipeRiderAdapter(MockAdapter):
Controls actual implmentation of adapter, and ability to override certain methods.
"""

ConnectionManager = PipeRiderAdapterConnectionManager
ConnectionManager = PipeRiderAdapterConnectionManager # type: ignore

@classmethod
def date_function(cls):
Expand Down

0 comments on commit 27dc8bc

Please sign in to comment.