ler vários arquivos usando multiprocessamento

Eu preciso ler alguns arquivos de texto muito grandes (100 + Mb), processar todas as linhas com regex e armazenar os dados em uma estrutura. Minha estrutura herda de defaultdict, tem um método de leitura (self) que lê o arquivo self.file_name.

Veja este exemplo muito simples (mas não real), não estou usando regex, mas estou dividindo linhas:


import multiprocessing
from collections import defaultdict

def SingleContainer():
    return list()

class Container(defaultdict):
    """
    this class store odd line in self["odd"] and even line in self["even"].
    It is stupid, but it's only an example. In the real case the class
    has additional methods that do computation on readen data.
    """
    def __init__(self,file_name):
        if type(file_name) != str:
            raise AttributeError, "%s is not a string" % file_name
        defaultdict.__init__(self,SingleContainer)
        self.file_name = file_name
        self.readen_lines = 0
    def read(self):
        f = open(self.file_name)
        print "start reading file %s" % self.file_name
        for line in f:
            self.readen_lines += 1
            values = line.split()
            key = {0: "even", 1: "odd"}[self.readen_lines %2]
            self[key].append(values)
        print "readen %d lines from file %s" % (self.readen_lines, self.file_name)

def do(file_name):
    container = Container(file_name)
    container.read()
    return container.items()

if __name__ == "__main__":
    file_names = ["r1_200909.log", "r1_200910.log"]
    pool = multiprocessing.Pool(len(file_names))
    result = pool.map(do,file_names)
    pool.close()
    pool.join()
    print "Finish"      

No final, preciso juntar todos os resultados em um único contêiner. É importante que a ordem das linhas seja preservada. Minha abordagem é muito lenta ao retornar valores. Melhor solução? Estou usando o python 2.6 no Linux

questionAnswers(3)

yourAnswerToTheQuestion