如何在main等待服务器关闭时定期运行函数

import threading
import time

# 定期运行的函数
def periodic_function():
    while not exit_flag:  # 根据退出标志决定是否继续运行
        # 执行需要的操作
        print("Running periodic function...")
        time.sleep(1)  # 暂停一秒钟

# 主函数
def main():
    global exit_flag
    exit_flag = False  # 退出标志

    # 创建定期运行的线程
    thread = threading.Thread(target=periodic_function)
    thread.start()

    # 在这里执行服务器的初始化操作
    print("Server started.")

    # 等待服务器关闭
    while not exit_flag:
        # 假设这里是服务器的主循环
        time.sleep(1)

    # 等待定期运行的线程执行完毕
    thread.join()

    # 在这里执行服务器的清理操作
    print("Server stopped.")

if __name__ == "__main__":
    main()