Merge pull request #19 from fujiwara/fix-multi-use

Fix for multiple use.
onishi · May 9, 2019 · 6abdcf2 · 6abdcf2
2 parents c81bd0c + eed0819
commit 6abdcf2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/lib/Devel/KYTProf.pm b/lib/Devel/KYTProf.pm
@@ -4,6 +4,8 @@ use warnings;
 
 our $VERSION = '0.9991';
 
+my $Applied = {};
+
 use Class::Data::Lite (
     rw => {
         namespace_regex         => undef,
@@ -50,6 +52,7 @@ sub apply_prof {
     unless ($prof_pkg->can('apply')) {
         die qq{"$prof_pkg" has no `apply` method. A profiler package should implement it.\n};
     }
+    return if ++$Applied->{$prof_pkg} > 1; # skip if already applied
     $prof_pkg->apply(@args);
 }
 

diff --git a/t/dbi.t b/t/dbi.t
@@ -2,6 +2,7 @@ use strict;
 use warnings;
 use Test::More;
 use Devel::KYTProf;
+use Devel::KYTProf; # test for multiple applied problem
 
 use Test::Requires 'DBI', 'DBD::SQLite';