~drizzle-trunk/drizzle/development

1823.6.1 by Brian Aker
First pass through adding back user_locks.
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"
1823.6.2 by Brian Aker
Solves the "dangling lock" problem.
23
#include "plugin/user_locks/lock_storage.h"
1823.6.1 by Brian Aker
First pass through adding back user_locks.
24
25
#include <string>
26
27
namespace user_locks {
28
29
int64_t ReleaseLock::val_int()
30
{
31
  drizzled::String *res= args[0]->val_str(&value);
32
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
33
  if (not res || not res->length())
1823.6.1 by Brian Aker
First pass through adding back user_locks.
34
  {
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
35
    my_error(drizzled::ER_USER_LOCKS_INVALID_NAME_LOCK, MYF(0));
1823.6.1 by Brian Aker
First pass through adding back user_locks.
36
    return 0;
37
  }
38
  null_value= false;
39
1823.6.2 by Brian Aker
Solves the "dangling lock" problem.
40
  drizzled::session_id_t id= getSession().getSessionId();
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
41
  locks::return_t result;
42
  {
43
    boost::this_thread::restore_interruption dl(getSession().getThreadInterupt());
44
    try {
45
      result= user_locks::Locks::getInstance().release(Key(getSession().getSecurityContext(), res->c_str()), id);
46
    }
47
    catch(boost::thread_interrupted const& error)
48
    {
49
      my_error(drizzled::ER_QUERY_INTERRUPTED, MYF(0));
50
      return 0;
51
    }
52
  }
53
54
  switch (result)
55
  {
56
  default:
57
  case locks::SUCCESS:
58
    {
59
      user_locks::Storable *list= static_cast<user_locks::Storable *>(getSession().getProperty("user_locks"));
60
      assert(list);
61
      if (list) // Just in case we ever blow the assert
62
        list->erase(Key(getSession().getSecurityContext(), res->c_str()));
63
64
      return 1;
65
    }
66
  case locks::NOT_FOUND:
67
    null_value= true;
68
    return 0;
69
  case locks::NOT_OWNED_BY:
70
    my_error(drizzled::ER_USER_LOCKS_NOT_OWNER_OF_LOCK, MYF(0));
71
    return 0;
72
  }
1823.6.1 by Brian Aker
First pass through adding back user_locks.
73
}
74
75
} /* namespace user_locks */