Merge pull request #20 from seunggabi/feature/#19_singleton

(#19) feat: singleton
jkklee · Sep 5, 2023 · b7f1de5 · b7f1de5
2 parents cfa69f8 + 38fa566
commit b7f1de5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -9,3 +9,5 @@ eggs
 .idea
 test.py
 *.swp
+
+venv
diff --git a/pymysqlpool.py b/pymysqlpool.py
@@ -327,3 +327,17 @@ def wrapper(*args, **kwargs):
 for name, fn in inspect.getmembers(Connection, inspect.isfunction):
     if not name.startswith('_'):
         setattr(Connection, name, already_returned_conn(fn))
+
+
+class ConnectionPoolSingleton:
+    _instance = None
+
+    def __new__(cls, *args, **kwargs):
+        if cls._instance is None:
+            cls._instance = cls._create_pool(*args, **kwargs)
+
+        return cls._instance
+
+    @staticmethod
+    def _create_pool(*args, **kwargs):
+        return ConnectionPool(*args, **kwargs)