# -*- coding: utf-8 -*-

# @Author  : ydf

# @Time    : 2019/7/2 14:11

import atexit

import time

import warnings

from collections import Callable

import gevent

from gevent import pool as gevent_pool

from gevent import monkey

from gevent.queue import JoinableQueue

from app.utils_ydf import LoggerMixin, nb_print, LogManager

def check_gevent_monkey_patch(raise_exc=True):

    if not monkey.is_module_patched('socket'):

        if raise_exc:

            warnings.warn(f'检测到 你还没有打gevent包的猴子补丁，请在所运行的脚本第一行写上  【import gevent.monkey;gevent.monkey.patch_all()】  这句话。')

            raise Exception(f'检测到 你还没有打gevent包的猴子补丁，请在所运行的脚本第一行写上  【import gevent.monkey;gevent.monkey.patch_all()】  这句话。')

    else:

        return 1

logger_gevent_timeout_deco = LogManager('logger_gevent_timeout_deco').get_logger_and_add_handlers()

def gevent_timeout_deco(timeout_t):

    def _gevent_timeout_deco(f):

        def __gevent_timeout_deceo(*args, **kwargs):

            timeout = gevent.Timeout(timeout_t, )

            timeout.start()

            try:

                f(*args, **kwargs)

            except gevent.Timeout as t:

                logger_gevent_timeout_deco.error(f'函数 {f} 运行超过了 {timeout_t} 秒')

                if t is not timeout:

                    nb_print(t)

                    # raise  # not my timeout

            finally:

                timeout.close()

        return __gevent_timeout_deceo

    return _gevent_timeout_deco

class GeventPoolExecutor(gevent_pool.Pool):

    def __init__(self, size=None, ):

        check_gevent_monkey_patch()

        super().__init__(size, )

    def submit(self, *args, **kwargs):

        self.spawn(*args, **kwargs)

    def shutdown(self):

        self.join()

if __name__ == '__main__':

    monkey.patch_all()

    def f2(x):

        time.sleep(1)

        nb_print(x)

    pool = GeventPoolExecutor(4)

    for i in range(15):

        nb_print(f'放入{i}')

        pool.submit(gevent_timeout_deco(8)(f2), i)

    nb_print(66666666)