← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/global_buffer.h

Merge Stewart's dead code removal

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/global_buffer.h
1
 
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2010 Andrew Hutchings
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; version 2 of the License.
9
 
 *
10
 
 *  This program is distributed in the hope that it will be useful,
11
 
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
 *  GNU General Public License for more details.
14
 
 *
15
 
 *  You should have received a copy of the GNU General Public License
16
 
 *  along with this program; if not, write to the Free Software
17
 
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
 
 */
19
 
 
20
 
#ifndef DRIZZLED_GLOBAL_BUFFER_H
21
 
#define DRIZZLED_GLOBAL_BUFFER_H
22
 
 
23
 
#include <drizzled/atomics.h>
24
 
 
25
 
namespace drizzled
26
 
{
27
 
template <class T>
28
 
class global_buffer_constraint
29
 
{
30
 
public:
31
 
  global_buffer_constraint(T max)
32
 
  {
33
 
    setMaxSize(max); 
34
 
  }
35
 
 
36
 
  T getMaxSize() const { return max_size; }
37
 
  void setMaxSize(T new_size) 
38
 
  {
39
 
    if (new_size == 0) new_size = std::numeric_limits<T>::max(); 
40
 
    max_size= new_size;
41
 
  }
42
 
 
43
 
  bool add(T addition)
44
 
  {
45
 
    if (current_size.add_and_fetch(addition) > max_size)
46
 
    {
47
 
      current_size.add_and_fetch(T(0) - addition);
48
 
      return false;
49
 
    }
50
 
    return true;
51
 
  }
52
 
 
53
 
  bool sub(T subtract)
54
 
  {
55
 
    if (current_size < subtract)
56
 
      return false;
57
 
    else
58
 
      current_size.add_and_fetch(T(0) - subtract);
59
 
 
60
 
    return true;
61
 
  }
62
 
 
63
 
private:
64
 
  atomic<T> current_size;
65
 
  T max_size;
66
 
};
67
 
 
68
 
}
69
 
#endif /* DRIZZLED_GLOBAL_BUFFER_H */