← 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: Eric Day
  • Date: 2009-11-10 21:50:22 UTC
  • mto: This revision was merged to the branch mainline in revision 1218.
  • Revision ID: eday@oddments.org-20091110215022-0b2nqmurv7b2l6wo
Duplicated oldlibdrizzle module, one for Drizzle protocol and one for MySQL, per Brian's request from merge proposal. Port options are now --drizzle-protocol-port and --mysql-protocol-port.

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
 
  std::cerr << "Trying to create " << res->c_str() << "\n";
47
 
 
48
 
  user_locks::Storable *list= static_cast<user_locks::Storable *>(getSession().getProperty("user_locks"));
49
 
  if (list) // To be compatible with MySQL, we will now release all other locks we might have.
50
 
    list->erase_all();
51
 
 
52
 
  bool result;
53
 
  {
54
 
    boost::this_thread::restore_interruption dl(getSession().getThreadInterupt());
55
 
 
56
 
    try {
57
 
      result= user_locks::Locks::getInstance().lock(getSession().getSessionId(), Key(getSession().getSecurityContext(), res->c_str()), wait_time);
58
 
    }
59
 
    catch(boost::thread_interrupted const& error)
60
 
    {
61
 
      my_error(drizzled::ER_QUERY_INTERRUPTED, MYF(0));
62
 
      null_value= true;
63
 
 
64
 
      return 0;
65
 
    }
66
 
  }
67
 
 
68
 
  if (boost::indeterminate(result))
69
 
    null_value= true;
70
 
 
71
 
  if (result)
72
 
  {
73
 
    if (not list)
74
 
    {
75
 
      list= new user_locks::Storable(getSession().getSessionId());
76
 
      getSession().setProperty("user_locks", list);
77
 
    }
78
 
 
79
 
    list->insert(Key(getSession().getSecurityContext(), res->c_str()));
80
 
 
81
 
    return 1;
82
 
  }
83
 
 
84
 
  return 0;
85
 
}
86
 
 
87
 
} /* namespace user_locks */