~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
2173.2.1 by Monty Taylor
Fixes incorrect usage of include
21
#include <config.h>
22
#include <plugin/user_locks/module.h>
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
2252.1.8 by Olaf van der Spek
Common fwd
40
  drizzled::identifier::user::ptr user_identifier(getSession().user());
1823.6.2 by Brian Aker
Solves the "dangling lock" problem.
41
  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.
42
  locks::return_t result;
43
  {
44
    boost::this_thread::restore_interruption dl(getSession().getThreadInterupt());
45
    try {
2008.1.1 by Brian Aker
Adding user identifier that makes use of a shared ptr to handle concurrency
46
      result= user_locks::Locks::getInstance().release(Key(*user_identifier, res->c_str()), id);
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
47
    }
48
    catch(boost::thread_interrupted const& error)
49
    {
50
      my_error(drizzled::ER_QUERY_INTERRUPTED, MYF(0));
51
      return 0;
52
    }
53
  }
54
55
  switch (result)
56
  {
57
  default:
58
  case locks::SUCCESS:
59
    {
2240.5.1 by Olaf van der Spek
Session::getProperty
60
      user_locks::Storable *list= getSession().getProperty<user_locks::Storable>("user_locks");
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
61
      assert(list);
62
      if (list) // Just in case we ever blow the assert
2008.1.1 by Brian Aker
Adding user identifier that makes use of a shared ptr to handle concurrency
63
        list->erase(Key(*user_identifier, res->c_str()));
1955.1.2 by Brian Aker
Update such that we can now do a lock test with a wait.
64
65
      return 1;
66
    }
67
  case locks::NOT_FOUND:
68
    null_value= true;
69
    return 0;
70
  case locks::NOT_OWNED_BY:
71
    my_error(drizzled::ER_USER_LOCKS_NOT_OWNER_OF_LOCK, MYF(0));
72
    return 0;
73
  }
1823.6.1 by Brian Aker
First pass through adding back user_locks.
74
}
75
76
} /* namespace user_locks */