Fix MemoryUsage extension for OS X. Fix GH-529.

This commit is contained in:
Mikhail Korobov 2014-01-15 03:24:48 +06:00
parent 62a7b6fe73
commit a83c6f545c
1 changed files with 7 additions and 3 deletions

View File

@ -3,7 +3,7 @@ MemoryUsage extension
See documentation in docs/topics/extensions.rst
"""
import sys
import socket
from pprint import pformat
from importlib import import_module
@ -16,7 +16,7 @@ from scrapy.mail import MailSender
from scrapy.utils.engine import get_engine_status
class MemoryUsage(object):
def __init__(self, crawler):
if not crawler.settings.getbool('MEMUSAGE_ENABLED'):
raise NotConfigured
@ -41,7 +41,11 @@ class MemoryUsage(object):
return cls(crawler)
def get_virtual_size(self):
return self.resource.getrusage(self.resource.RUSAGE_SELF).ru_maxrss * 1024
size = self.resource.getrusage(self.resource.RUSAGE_SELF).ru_maxrss
if sys.platform != 'darwin':
# on Mac OS X ru_maxrss is in bytes, on Linux it is in KB
size *= 1024
return size
def engine_started(self):
self.crawler.stats.set_value('memusage/startup', self.get_virtual_size())