From e18d4374c16e3d9e8164eec8af3c4cd6c52c4406 Mon Sep 17 00:00:00 2001
From: "mkanat%bugzilla.org" <>
Date: Tue, 11 Dec 2007 11:36:02 +0000
Subject: Bug 310717: [Oracle] Bugzilla::DB::Oracle module Bug 310718: [Oracle]
 Bugzilla::DB::Schema::Oracle module Patch By Xiaoou Wu <xiaoou.wu@oracle.com>
 r=mkanat, a=mkanat

---
 Bugzilla/DB.pm | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

(limited to 'Bugzilla/DB.pm')

diff --git a/Bugzilla/DB.pm b/Bugzilla/DB.pm
index 4aad803c6..49692eec0 100644
--- a/Bugzilla/DB.pm
+++ b/Bugzilla/DB.pm
@@ -301,6 +301,62 @@ sub import {
     $Exporter::ExportLevel-- if $is_exporter;
 }
 
+sub bz_lock_tables {
+    my ($self, @tables) = @_;
+
+    my $list = join(', ', @tables);
+    # Check first if there was no lock before
+    if ($self->{private_bz_tables_locked}) {
+        ThrowCodeError("already_locked", { current => $self->{private_bz_tables_locked},
+                                           new => $list });
+    } else {
+        my %read_tables;
+        my %write_tables;
+        foreach my $table (@tables) {
+            $table =~ /^([\d\w]+)([\s]+AS[\s]+[\d\w]+)?[\s]+(WRITE|READ)$/i;
+            my $table_name = $1;
+            if ($3 =~ /READ/i) {
+                if (!exists $read_tables{$table_name}) {
+                    $read_tables{$table_name} = undef;
+                }
+            }
+            else {
+                if (!exists $write_tables{$table_name}) {
+                    $write_tables{$table_name} = undef;
+                }
+            }
+        }
+
+        # Begin Transaction
+        $self->bz_start_transaction();
+
+        Bugzilla->dbh->do('LOCK TABLE ' . join(', ', keys %read_tables) .
+                          ' IN ROW SHARE MODE') if keys %read_tables;
+        Bugzilla->dbh->do('LOCK TABLE ' . join(', ', keys %write_tables) .
+                          ' IN ROW EXCLUSIVE MODE') if keys %write_tables;
+        $self->{private_bz_tables_locked} = $list;
+    }
+}
+
+sub bz_unlock_tables {
+    my ($self, $abort) = @_;
+
+    # Check first if there was previous matching lock
+    if (!$self->{private_bz_tables_locked}) {
+        # Abort is allowed even without previous lock for error handling
+        return if $abort;
+        ThrowCodeError("no_matching_lock");
+    } else {
+        $self->{private_bz_tables_locked} = "";
+        # End transaction, tables will be unlocked automatically
+        if ($abort) {
+            $self->bz_rollback_transaction();
+        } else {
+            $self->bz_commit_transaction();
+        }
+    }
+}
+
 sub sql_istrcmp {
     my ($self, $left, $right, $op) = @_;
     $op ||= "=";
-- 
cgit v1.2.3-24-g4f1b