python - 在 Python 中循环 : Do stuff before first iteration

Question

我要优化。

简单的解决方案

connection = get_db_connection()
for item in my_iterator:
    push_item_to_db(item, connection)

缺点:

get_db_connection() 很慢。如果 my_iterator 为空，那么我想避免调用它。

“if None”解决方案

connection = None
for item in my_iterator:
    if connection is None:
        connection = get_db_connection()
    push_item_to_db(item, connection)

缺点:

如果 my_iterator 中有 100k 个项目，则 if connection is None 将被调用 100k 次(尽管只需要一次)。我想避免这种情况。

完美解决...

1. 如果迭代器为空，不要调用 get_db_connection()
2. 不要调用 if connection is None: 每次迭代都无用。

有什么想法吗？

Answer 1

你可以这样做:

connection = None
for item in my_iterator:
    if connection is None:
        connection = get_db_connection()
    push_item_to_db(item, connection)

简单的解决方案。没必要想太多。即使有 100k 次操作，x is None 也只是采用一个 Python 操作码的引用比较。与每次插入时发生的完整 tcp 往返 + 磁盘写入相比，您真的不需要优化它。