def get_db(): thread_local = local() if not hasattr(thread_local, "db"): thread_local.db = sqlite3.connect("reptile.db") return thread_local.db 如何在线程里开关

在多线程环境下，为了确保每个线程都有自己的数据库连接，可以使用 `Thread-local Storage` 技术来实现。你可以使用 Python 标准库中的 `threading.local()` 函数来创建一个本地存储对象，然后在每个线程中存储该线程的数据库连接。以下是一个示例代码：

import sqlite3
import threading

# 创建线程本地存储对象
local_data = threading.local()

def get_db():
    # 在当前线程中查找数据库连接
    if not hasattr(local_data, "db"):
        # 如果当前线程没有连接，则创建一个新连接
        local_data.db = sqlite3.connect("reptile.db")
    return local_data.db

def close_db():
    # 关闭当前线程的数据库连接
    if hasattr(local_data, "db"):
        local_data.db.close()
        del local_data.db

这个示例代码中，我们首先使用 `threading.local()` 函数创建了一个本地存储对象 `local_data`，然后在 `get_db` 函数中使用该对象来存储当前线程的数据库连接。在每个线程中，`get_db` 函数会检查当前线程是否已经有数据库连接，如果没有，则创建一个新连接并存储到本地存储对象中。在程序结束时，可以调用 `close_db` 函数来关闭当前线程的数据库连接。

在使用本地存储对象时，需要注意以下几点：

- 本地存储对象是线程安全的，可以在多个线程中同时使用。
- 每个线程都有自己的本地存储对象，存储在其中的数据只能被当前线程访问。
- 在每个线程中，本地存储对象的属性都是独立的，不会和其他线程中的属性发生冲突。
- 在使用本地存储对象时，需要注意及时清理已经不需要的属性，以避免内存泄漏。