~bzr-pqm/bzr/bzr.dev

# $Id: http_manager.py 270 2004-10-09 10:38:54Z fredrik $

# effnews http

#

# manage a set of http clients

#

# Copyright (c) 2001-2004 by Fredrik Lundh.  All rights reserved.

#

import asyncore, time

import http_client

class http_manager:

    max_connections = 8

    max_size = 1000000

    max_time = 60

    def __init__(self):

        self.queue = []

    def request(self, uri, consumer, extra_headers=None):

        self.queue.append((uri, consumer, extra_headers))

    def priority_request(self, uri, consumer, extra_headers=None):

        self.queue.insert(0, (uri, consumer, extra_headers))

    def purge(self):

        for channel in asyncore.socket_map.values():

            channel.close()

        del self.queue[:]

    def prioritize(self, priority_uri):

        i = 0

        for uri, consumer, extra_headers in self.queue:

            if uri == priority_uri:

                del self.queue[i]

                self.priority_request(uri, consumer, extra_headers)

                return

            i = i + 1

    def poll(self, timeout=0.1):

        # sanity checks

        now = time.time()

        for channel in asyncore.socket_map.values():

            if channel.bytes_in > self.max_size:

                channel.close() # too much data

                try:

                    channel.consumer.http(

                        0, channel, ("HTTPManager", "too much data", None)

                        )

                except:

                    pass

            if channel.timestamp and now - channel.timestamp > self.max_time:

                channel.close() # too slow

                try:

                    channel.consumer.http(

                        0, channel, ("HTTPManager", "timeout", None)

                        )

                except:

                    pass

        # activate up to max_connections channels

        while self.queue and len(asyncore.socket_map) < self.max_connections:

            http_client.do_request(*self.queue.pop(0))

        # keep the network running

        asyncore.poll(timeout)

        # return non-zero if we should keep on polling

        return len(self.queue) or len(asyncore.socket_map)