Працюючи над зовсім іншим сценарієм, я дізнався, що за допомогою 29 мільйонів рядків тексту використання seek()
та оперування даними побіжно часто буває швидше, ніж за рядковим принципом. Ця ж ідея застосовується і в нижченаведеному сценарії: ми відкриваємо файл, і замість того, щоб прокручувати через відкриття та закриття файлу (що може додавати накладні витрати, навіть якщо вони не значні), ми тримаємо файл відкритим і прагнемо повернутися до початку.
#!/usr/bin/env python3
from __future__ import print_function
import sys,os
def error_out(string):
sys.stderr.write(string+"\n")
sys.exit(1)
def read_bytewise(fp):
data = fp.read(1024)
print(data.decode(),end="",flush=True)
while data:
data = fp.read(1024)
print(data.decode(),end="",flush=True)
#fp.seek(0,1)
def main():
howmany = int(sys.argv[1]) + 1
if not os.path.isfile(sys.argv[2]):
error_out("Needs a valid file")
fp = open(sys.argv[2],'rb')
for i in range(1,howmany):
#print(i)
fp.seek(0)
read_bytewise(fp)
fp.close()
if __name__ == '__main__': main()
Сам сценарій у використанні досить простий:
./repeat_text.py <INT> <TEXT.txt>
Для текстового файлу з 3 рядками та 1000 ітерацій він іде повністю добре, приблизно за 0,1 секунди:
$ /usr/bin/time ./repeat_text.py 1000 input.txt > /dev/null
0.10user 0.00system 0:00.23elapsed 45%CPU (0avgtext+0avgdata 9172maxresident)k
0inputs+0outputs (0major+1033minor)pagefaults 0swaps
Сам сценарій не найелегантніший, напевно, його можна скоротити, але це робить роботу. Звичайно, я додав туди-сюди кілька зайвих бітів, як error_out()
функція, яка не потрібна - це лише невеликий зручний для користувачів штрих.