Merge pull request #65 from KevinCarterDev/proper_exit_code

Return a proper exit code if an error occurs

Author: mla

pg_sample

@@ -211,6 +211,8 @@ $SIG{TERM} = $SIG{INT} = $SIG{QUIT} = $SIG{HUP} = sub {
   die "Received signal '$signal'; cleaning up and terminating.\n";
 };
 
+$SIG{__DIE__} = sub { our @reason = @_ };
+
 BEGIN {
   # Encapsulate a fully-qualified Postgresql table name
   package Table;
@@ -943,12 +945,18 @@ foreach my $table (@tables) {
 print "\n";
 
 END {
+  our @reason;
+
   # remove sample tables unless requested not to
   if ($created_schema && !$opt{keep}) {
     notice "Dropping sample schema $opt{sample_schema}\n";
     $dbh->do("DROP SCHEMA $opt{sample_schema} CASCADE");
   }
 
   notice "Done.\n";
-  exit 0;
+  if (@reason) {
+    exit 1;
+  } else {
+    exit 0;
+  }
 }