Bug 1443537 - Add utility functions for switching database handle to writeable or read-only mode

1 files changed, 26 insertions, 0 deletions

diff --git a/Bugzilla/Util.pm b/Bugzilla/Util.pm
index 91f06e650..7d85a4dfd 100644
--- a/Bugzilla/Util.pm
+++ b/Bugzilla/Util.pm
@@ -14,6 +14,7 @@ use warnings;
 use base qw(Exporter);
 @Bugzilla::Util::EXPORT = qw(trick_taint detaint_natural
                              detaint_signed
+                             with_writable_database with_readonly_database
                              html_quote url_quote xml_quote
                              css_class_quote html_light_quote
                              i_am_cgi i_am_webservice correct_urlbase remote_ip
@@ -44,6 +45,31 @@ use Encode qw(encode decode resolve_alias);
 use Encode::Guess;
 use POSIX qw(floor ceil);
 use Taint::Util qw(untaint);
+use Try::Tiny;
+
+sub with_writable_database(&) {
+    my ($code) = @_;
+    my $dbh = Bugzilla->dbh_main;
+    local Bugzilla->request_cache->{dbh} = $dbh;
+    local Bugzilla->request_cache->{error_mode} = ERROR_MODE_DIE;
+    try {
+        $dbh->bz_start_transaction;
+        $code->();
+        $dbh->bz_commit_transaction;
+    } catch {
+        $dbh->bz_rollback_transaction;
+        # re-throw
+        die $_;
+    };
+}
+
+sub with_readonly_database(&) {
+    my ($code) = @_;
+    local Bugzilla->request_cache->{dbh} = undef;
+    local Bugzilla->request_cache->{error_mode} = ERROR_MODE_DIE;
+    Bugzilla->switch_to_shadow_db();
+    $code->();
+}
 
 sub trick_taint {
     untaint($_[0]);