Sha256: 5a486828f859f55a4aa475b73837bfeb6ae3f9b7d638e6d35ce0bad2d9b44f5a
Contents?: true
Size: 1.41 KB
Versions: 9
Compression:
Stored size: 1.41 KB
Contents
require 'unicorn-cuba-base/root_logger'
class MemoryLimit
include ClassLogging
class MemoryLimitedExceededError < RuntimeError
def initialize
super "memory limit exceeded"
end
end
module IO
def root_limit(ml)
@root_limit = ml
end
def read(bytes = nil)
data = @root_limit.get("#{self.class.name} IO read") do |max_read_bytes|
if not bytes or bytes > max_read_bytes
data = super max_read_bytes
raise MemoryLimitedExceededError unless eof?
data or '' # read() always returns '' on EOF
else
super bytes or ''
end
end
end
end
def initialize(bytes = 256 * 1024 ** 2)
log.debug "setting up new memory limit of #{bytes} bytes" if bytes
@limit = bytes
end
attr_reader :limit
def get(reason = nil)
yield(@limit).tap do |data|
borrow(data.bytesize, reason) if data
end
end
def borrow(bytes, reason = nil)
if reason
log.debug "borrowing #{bytes} from #{@limit} bytes of limit for #{reason}"
else
log.debug "borrowing #{bytes} from #{@limit} bytes of limit"
end
bytes > @limit and raise MemoryLimitedExceededError
@limit -= bytes
bytes
end
def return(bytes, reason = nil)
if reason
log.debug "returning #{bytes} to #{@limit} bytes of limit used for #{reason}"
else
log.debug "returning #{bytes} to #{@limit} bytes of limit"
end
@limit += bytes
bytes
end
def io(io)
io.extend MemoryLimit::IO
io.root_limit self
io
end
end
Version data entries
9 entries across 9 versions & 1 rubygems