Cách xử lý các kết nối cơ sở dữ liệu trong mô-đun thư viện Python

Tôi đã tạo một thư viện trong Python có chứa các hàm để truy cập cơ sở dữ liệu. Đây là một thư viện trình bao quanh cơ sở dữ liệu ứng dụng của bên thứ ba, được viết do thực tế là ứng dụng của bên thứ ba không cung cấp API tốt. Bây giờ tôi ban đầu cho phép mỗi hàm mở một kết nối cơ sở dữ liệu trong suốt thời gian của lệnh gọi là ổn, cho đến khi logic chương trình của tôi sử dụng các cuộc gọi lồng nhau đến các hàm mà sau đó tôi sẽ gọi một hàm cụ thể vài nghìn lần. Đây không phải là rất hiệu quả. Cấu hình này cho thấy chi phí hoạt động trong thiết lập kết nối cơ sở dữ liệu - một lần cho mỗi lần gọi hàm. Vì vậy, tôi đã chuyển kết nối mở từ bên trong (các) chức năng sang chính mô-đun, để kết nối cơ sở dữ liệu sẽ được mở khi mô-đun thư viện được nhập. Điều này đã cho tôi một hiệu suất chấp nhận được.

Bây giờ tôi có hai câu hỏi liên quan đến điều này. Đầu tiên, tôi có cần phải lo lắng rằng tôi không còn đóng kết nối cơ sở dữ liệu một cách rõ ràng không và làm cách nào tôi có thể làm điều đó một cách rõ ràng với thiết lập này? Thứ hai, những gì tôi đã làm rơi vào bất cứ nơi nào gần với lĩnh vực thực hành tốt và làm thế nào tôi có thể tiếp cận điều này?

Nó thực sự phụ thuộc vào thư viện bạn đang sử dụng. Một số trong số chúng có thể tự đóng kết nối (Lưu ý: Tôi đã kiểm tra thư viện sqlite3 dựng sẵn, nhưng không được). Python sẽ gọi một hàm hủy khi một đối tượng đi ra khỏi phạm vi và các thư viện này có thể thực hiện một hàm hủy đóng các kết nối một cách duyên dáng.

Tuy nhiên, đó có thể không phải là trường hợp! Tôi muốn giới thiệu, như những người khác có trong các ý kiến, để bọc nó trong một đối tượng.

class MyDB(object):

    def __init__(self):
        self._db_connection = db_module.connect('host', 'user', 'password', 'db')
        self._db_cur = self._db_connection.cursor()

    def query(self, query, params):
        return self._db_cur.execute(query, params)

    def __del__(self):
        self._db_connection.close()

Điều này sẽ khởi tạo kết nối cơ sở dữ liệu của bạn khi bắt đầu và đóng nó khi vị trí đối tượng của bạn được khởi tạo nằm ngoài phạm vi. Lưu ý: Nếu bạn khởi tạo đối tượng này ở cấp mô-đun, nó sẽ tồn tại cho toàn bộ ứng dụng của bạn. Trừ khi điều này được dự định, tôi sẽ đề nghị tách các chức năng cơ sở dữ liệu của bạn khỏi các chức năng không phải là cơ sở dữ liệu.

May mắn thay, python đã tiêu chuẩn hóa API cơ sở dữ liệu , vì vậy, nó sẽ hoạt động với tất cả các DB tuân thủ cho bạn :)

Trong khi xử lý các kết nối cơ sở dữ liệu, có hai điều cần quan tâm:

1. ngăn chặn nhiều kết nối tức thời, cho phép mỗi chức năng mở một kết nối cơ sở dữ liệu được coi là thông lệ xấu, với số lượng phiên cơ sở dữ liệu hạn chế, bạn sẽ hết phiên; ít nhất là giải pháp của bạn sẽ không mở rộng ra, thay vào đó, sử dụng mẫu singleton, lớp của bạn sẽ chỉ được khởi tạo một lần, để biết thêm thông tin về mẫu này, hãy xem liên kết

2. đóng kết nối khi thoát ứng dụng, giả sử bạn đã không làm và bạn có ít nhất hàng chục ứng dụng đang hoạt động giống nhau, lúc đầu mọi thứ sẽ ổn, nhưng bạn sẽ hết các phiên cơ sở dữ liệu và cách khắc phục duy nhất sẽ khởi động lại máy chủ cơ sở dữ liệu, đây không phải là một điều tốt cho một ứng dụng trực tiếp do đó sử dụng cùng một kết nối bất cứ khi nào có thể.

để củng cố tất cả các khái niệm này, xem ví dụ sau đây kết thúc psycopg2

import psycopg2


class Postgres(object):
"""docstring for Postgres"""
    _instance = None

    def __new__(cls):
        if cls._instance is None:
            cls._instance = object.__new__(cls)
            # normally the db_credenials would be fetched from a config file or the enviroment
            # meaning shouldn't be hardcoded as follow
            db_config = {'dbname': 'demo', 'host': 'localhost',
                     'password': 'postgres', 'port': 5432, 'user': 'postgres'}
            try:
                print('connecting to PostgreSQL database...')
                connection = Postgres._instance.connection = psycopg2.connect(**db_config)
                cursor = Postgres._instance.cursor = connection.cursor()
                cursor.execute('SELECT VERSION()')
                db_version = cursor.fetchone()

            except Exception as error:
                print('Error: connection not established {}'.format(error))
                Postgres._instance = None

            else:
                print('connection established\n{}'.format(db_version[0]))

        return cls._instance

    def __init__(self):
        self.connection = self._instance.connection
        self.cursor = self._instance.cursor

    def query(self, query):
        try:
            result = self.cursor.execute(query)
        except Exception as error:
            print('error execting query "{}", error: {}'.format(query, error))
            return None
        else:
            return result

    def __del__(self):
        self.connection.close()
        self.cursor.close()

Thật thú vị khi cung cấp khả năng quản lý bối cảnh cho các đối tượng của bạn. Điều này có nghĩa là bạn có thể viết một mã như thế này:

class MyClass:
    def __init__(self):
       # connect to DB
    def __enter__(self):
       return self
    def __exit__(self):
       # close the connection

Điều này sẽ cung cấp cho bạn một cách thuận tiện để tự động đóng kết nối với cơ sở dữ liệu bằng cách gọi lớp bằng câu lệnh with:

with MyClass() as my_class:
   # do what you need
# at this point, the connection is safely closed.

Một thời gian dài để suy nghĩ về điều này. Đến hôm nay, tôi đã tìm được đường. tôi không biết đó là cách tốt nhất bạn tạo một tệp có tên: Conn.py và lưu nó trong thư mục /usr/local/lib/python3.5/site-packages/conn/. Tôi sử dụng freebsd và đây là đường dẫn của thư mục gói trang web của tôi. trong phần kết nối của tôi: Conn = "dbname = omnivore user = postgres password = 12345678"

`` `` `` `` `` `` `` `` `` `` `` `` `` `` `` `` `` và trong kịch bản tôi muốn gọi kết nối, tôi viết:

nhập psycopg2 nhập psycopg2.extras nhập psycopg2.extensions

từ nhập khẩu liên kết thử: Conn = psycopg2.connect (Conn.conn) ngoại trừ: page = "Không thể truy cập cơ sở dữ liệu"

cur = Conn.cthon ()

và blah blah ....

tôi hy vọng điều này hữu ích