From a56596a1d5f2146268653b2dce40ecdf5b8711db Mon Sep 17 00:00:00 2001
From: Dave Mueller <dmueller@liquidweb.com>
Date: Wed, 6 May 2015 15:03:19 -0400
Subject: [PATCH] Adding a 'select_for_update' wrapper around 'select' method

---
 lib/SQL/Abstract.pm | 17 +++++++++++++++++
 t/01generate.t      |  7 +++++++
 2 files changed, 24 insertions(+)

diff --git a/lib/SQL/Abstract.pm b/lib/SQL/Abstract.pm
index 7e12187b..2e2c7b7e 100644
--- a/lib/SQL/Abstract.pm
+++ b/lib/SQL/Abstract.pm
@@ -454,6 +454,17 @@ sub select {
   return wantarray ? ($sql, @bind) : $sql;
 }
 
+#======================================================================
+# SELECT FOR UPDATE
+#======================================================================
+
+sub select_for_update {
+  my $self = shift;
+  my($sql, @bind) = $self->select(@_);
+  $sql .= $self->_sqlcase(' for update');
+  return wantarray ? ($sql, @bind) : $sql;
+}
+
 #======================================================================
 # DELETE
 #======================================================================
@@ -2145,6 +2156,12 @@ for details.
 
 =back
 
+=head2 select_for_update($source, $fields, $where, $order)
+
+This is a wrapper around L<select> which will append the
+"FOR UPDATE" statement to the end of a select statement.
+
+=back
 
 =head2 delete($table, \%where)
 
diff --git a/t/01generate.t b/t/01generate.t
index 948317ff..231469df 100644
--- a/t/01generate.t
+++ b/t/01generate.t
@@ -74,6 +74,13 @@ my @tests = (
               stmt_q => 'SELECT * FROM `test` WHERE ( `a` != ? )',
               bind   => ['boom']
       },
+      {
+              func   => 'select_for_update',
+              args   => ['test', '*'],
+              stmt   => 'SELECT * FROM test FOR UPDATE',
+              stmt_q => 'SELECT * FROM `test` FOR UPDATE',
+              bind   => []
+      },
       {
               func   => 'update',
               args   => ['test', {a => 'boom'}, {a => undef}],