package Plack::TempBuffer::Auto;
use strict;
use parent 'Plack::TempBuffer';
sub new {
my($class, undef, $max_memory_size) = @_;
bless {
_buffer => Plack::TempBuffer->create('PerlIO'),
_max => $max_memory_size,
}, $class;
}
sub print {
my $self = shift;
$self->{_buffer}->print(@_);
if ($self->{_max} && $self->{_buffer}->size > $self->{_max}) {
my $buf = $self->{_buffer}->{buffer};
$self->{_buffer} = Plack::TempBuffer->create('File'),
$self->{_buffer}->print($buf);
delete $self->{_max};
}
}
sub size {
my $self = shift;
$self->{_buffer}->size;
}
sub rewind {
my $self = shift;
$self->{_buffer}->rewind;
}
1;