->wrap can now be called on instance too

ray1729 · Oct 9, 2009 · 4a1e2a1 · 4a1e2a1
1 parent 9a81edc
commit 4a1e2a1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/Plack/Middleware.pm b/lib/Plack/Middleware.pm
@@ -14,8 +14,13 @@ sub import {
 }
 
 sub wrap {
-    my($class, $app, @args) = @_;
-    $class->new({ app => $app, @args })->to_app;
+    my($self, $app, @args) = @_;
+    if (ref $self) {
+        $self->{app} = $app;
+    } else {
+        $self = $self->new({ app => $app, @args });
+    }
+    return $self->to_app;
 }
 
 sub to_app {