util.py 文件源码

def get_stats(url, spids):
    stats = {}
    conf = parse_redis_url(url)
    r = redis.Redis(host=conf.host,
                    port=conf.port,
                    db=conf.database)
    for spid in spids:
        n = None
        try:
            n = r.get(spid)
            r.delete(spid)
        except redis.exceptions.ConnectionError:
            logger.error('Error in get_stats failed to connect redis server')
        n = 0 if n is None else int(n)
        stats[spid] = n
    return stats