← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/user_locks/get_lock.cc

  • Committer: Patrick Crews
  • Date: 2010-09-14 20:21:03 UTC
  • mto: (1771.1.1 pcrews)
  • mto: This revision was merged to the branch mainline in revision 1772.
  • Revision ID: gleebix@gmail.com-20100914202103-1db2n0bshzafep19
Moved transaction_log tests into updated non-publisher-based tree

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/user_locks/get_lock.cc
1
 
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2010 Brian Aker
5
 
 *
6
 
 *  This program is free software; you can redistribute it and/or modify
7
 
 *  it under the terms of the GNU General Public License as published by
8
 
 *  the Free Software Foundation; either version 2 of the License, or
9
 
 *  (at your option) any later version.
10
 
 *
11
 
 *  This program is distributed in the hope that it will be useful,
12
 
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 
 *  GNU General Public License for more details.
15
 
 *
16
 
 *  You should have received a copy of the GNU General Public License
17
 
 *  along with this program; if not, write to the Free Software
18
 
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
 
 */
20
 
 
21
 
#include "config.h"
22
 
#include "plugin/user_locks/module.h"
23
 
#include "plugin/user_locks/lock_storage.h"
24
 
 
25
 
#include <string>
26
 
 
27
 
namespace user_locks {
28
 
 
29
 
int64_t GetLock::val_int()
30
 
{
31
 
  drizzled::String *res= args[0]->val_str(&value);
32
 
  int64_t wait_time= 0;
33
 
 
34
 
  if (arg_count == 2)
35
 
  {
36
 
    wait_time= args[1]->val_int();
37
 
  }
38
 
 
39
 
  if (not res || not res->length())
40
 
  {
41
 
    my_error(drizzled::ER_USER_LOCKS_INVALID_NAME_LOCK, MYF(0));
42
 
    return 0;
43
 
  }
44
 
  null_value= false;
45
 
 
46
 
  user_locks::Storable *list= static_cast<user_locks::Storable *>(getSession().getProperty("user_locks"));
47
 
  if (list) // To be compatible with MySQL, we will now release all other locks we might have.
48
 
    list->erase_all();
49
 
 
50
 
  bool result;
51
 
  drizzled::identifier::User::const_shared_ptr user_identifier(getSession().user());
52
 
  {
53
 
    boost::this_thread::restore_interruption dl(getSession().getThreadInterupt());
54
 
 
55
 
    try {
56
 
      result= user_locks::Locks::getInstance().lock(getSession().getSessionId(), Key(*user_identifier, res->c_str()), wait_time);
57
 
    }
58
 
    catch(boost::thread_interrupted const& error)
59
 
    {
60
 
      my_error(drizzled::ER_QUERY_INTERRUPTED, MYF(0));
61
 
      null_value= true;
62
 
 
63
 
      return 0;
64
 
    }
65
 
  }
66
 
 
67
 
  if (result)
68
 
  {
69
 
    if (not list)
70
 
    {
71
 
      list= new user_locks::Storable(getSession().getSessionId());
72
 
      getSession().setProperty("user_locks", list);
73
 
    }
74
 
 
75
 
    list->insert(Key(*user_identifier, res->c_str()));
76
 
 
77
 
    return 1;
78
 
  }
79
 
 
80
 
  return 0;
81
 
}
82
 
 
83
 
} /* namespace user_locks */