← Back to branch summary

~thomir-deactivatedaccount/drizzle/drizzle-fix-bug653747

« back to all changes in this revision

Viewing changes to plugin/user_locks/locks.cc

  • Committer: Brian Aker
  • Date: 2010-10-10 02:07:52 UTC
  • mfrom: (1827.2.3 staging)
  • Revision ID: brian@tangent.org-20101010020752-ktv73isay5dxtvp3
Merge in switch on table_share_instance inheritance.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/user_locks/locks.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
 
 
24
 
#include <boost/thread/locks.hpp>
25
 
 
26
 
#include <string>
27
 
 
28
 
namespace user_locks {
29
 
 
30
 
bool Locks::lock(drizzled::session_id_t id_arg, const user_locks::Key &arg, int64_t wait_for)
31
 
{
32
 
  boost::system_time timeout= boost::get_system_time() + boost::posix_time::seconds(wait_for);
33
 
  boost::unique_lock<boost::mutex> scope(mutex);
34
 
  
35
 
  LockMap::iterator iter;
36
 
  while ((iter= lock_map.find(arg)) != lock_map.end())
37
 
  {
38
 
    if (id_arg == (*iter).second->id)
39
 
    {
40
 
      // We own the lock, so we just exit.
41
 
      return true;
42
 
    }
43
 
    bool success= cond.timed_wait(scope, timeout);
44
 
 
45
 
    if (not success)
46
 
      return false;
47
 
  }
48
 
 
49
 
  if (iter == lock_map.end())
50
 
  {
51
 
    std::pair<LockMap::iterator, bool> is_locked;
52
 
 
53
 
    is_locked= lock_map.insert(std::make_pair(arg, new lock_st(id_arg)));
54
 
    return is_locked.second;
55
 
  }
56
 
 
57
 
  return false;
58
 
}
59
 
 
60
 
bool Locks::lock(drizzled::session_id_t id_arg, const user_locks::Key &arg)
61
 
{
62
 
  std::pair<LockMap::iterator, bool> is_locked;
63
 
  boost::unique_lock<boost::mutex> scope(mutex);
64
 
  is_locked= lock_map.insert(std::make_pair(arg, new lock_st(id_arg)));
65
 
 
66
 
  return is_locked.second;
67
 
}
68
 
 
69
 
bool Locks::lock(drizzled::session_id_t id_arg, const user_locks::Keys &arg)
70
 
{
71
 
  boost::unique_lock<boost::mutex> scope(mutex);
72
 
 
73
 
  for (user_locks::Keys::const_iterator iter= arg.begin(); iter != arg.end(); iter++)
74
 
  {
75
 
    LockMap::iterator record= lock_map.find(*iter);
76
 
 
77
 
    if (record != lock_map.end())
78
 
    {
79
 
      if (id_arg != (*record).second->id)
80
 
        return false;
81
 
    }
82
 
  }
83
 
 
84
 
  for (Keys::iterator iter= arg.begin(); iter != arg.end(); iter++)
85
 
  {
86
 
    std::pair<LockMap::iterator, bool> is_locked;
87
 
    //is_locked can fail in cases where we already own the lock.
88
 
    is_locked= lock_map.insert(std::make_pair(*iter, new lock_st(id_arg)));
89
 
  }
90
 
 
91
 
  return true;
92
 
}
93
 
 
94
 
bool Locks::isUsed(const user_locks::Key &arg, drizzled::session_id_t &id_arg)
95
 
{
96
 
  boost::unique_lock<boost::mutex> scope(mutex);
97
 
  
98
 
  LockMap::iterator iter= lock_map.find(arg);
99
 
  
100
 
  if ( iter == lock_map.end())
101
 
    return false;
102
 
 
103
 
  id_arg= (*iter).second->id;
104
 
 
105
 
  return true;
106
 
}
107
 
 
108
 
bool Locks::isFree(const user_locks::Key &arg)
109
 
{
110
 
  boost::unique_lock<boost::mutex> scope(mutex);
111
 
 
112
 
  LockMap::iterator iter= lock_map.find(arg);
113
 
  
114
 
  return iter != lock_map.end();
115
 
}
116
 
 
117
 
void Locks::Copy(LockMap &lock_map_arg)
118
 
{
119
 
  lock_map_arg= lock_map;
120
 
}
121
 
 
122
 
boost::tribool Locks::release(const user_locks::Key &arg, drizzled::session_id_t &id_arg)
123
 
{
124
 
  size_t elements= 0;
125
 
  boost::unique_lock<boost::mutex> scope(mutex);
126
 
  LockMap::iterator iter= lock_map.find(arg);
127
 
 
128
 
  // Nothing is found
129
 
  if ( iter == lock_map.end())
130
 
    return boost::indeterminate;
131
 
 
132
 
  if ((*iter).second->id == id_arg)
133
 
    elements= lock_map.erase(arg);
134
 
 
135
 
  if (elements)
136
 
  {
137
 
    cond.notify_one();
138
 
    return true;
139
 
  }
140
 
 
141
 
  return false;
142
 
}
143
 
 
144
 
} /* namespace user_locks */