← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/rand_function/rand_function.cc

This patch completes the first step in the splitting of
the XA resource manager API from the storage engine API,
as outlined in the specification here:

http://drizzle.org/wiki/XaStorageEngine

* Splits plugin::StorageEngine into a base StorageEngine
  class and two derived classes, TransactionalStorageEngine
  and XaStorageEngine.  XaStorageEngine derives from
  TransactionalStorageEngine and creates the XA Resource
  Manager API for storage engines.

  - The methods moved from StorageEngine to TransactionalStorageEngine
    include releaseTemporaryLatches(), startConsistentSnapshot(), 
    commit(), rollback(), setSavepoint(), releaseSavepoint(),
    rollbackToSavepoint() and hasTwoPhaseCommit()
  - The methods moved from StorageEngine to XaStorageEngine
    include recover(), commitXid(), rollbackXid(), and prepare()

* Places all static "EngineVector"s into their proper
  namespaces (typedefs belong in header files, not implementation files)
  and places all static methods corresponding
  to either only transactional engines or only XA engines
  into their respective files in /drizzled/plugin/

* Modifies the InnoDB "handler" files to extend plugin::XaStorageEngine
  and not plugin::StorageEngine

The next step, as outlined in the wiki spec page above, is to isolate
the XA Resource Manager API into its own plugin class and modify
plugin::XaStorageEngine to implement plugin::XaResourceManager via
composition.  This is necessary to enable building plugins which can
participate in an XA transaction *without having to have that plugin
implement the entire storage engine API*

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/rand_function/rand_function.cc
1
 
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems, Inc.
5
 
 *  Copyright (C) 2010 Stewart Smith
6
 
 *
7
 
 *  This program is free software; you can redistribute it and/or modify
8
 
 *  it under the terms of the GNU General Public License as published by
9
 
 *  the Free Software Foundation; version 2 of the License.
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 <drizzled/plugin/function.h>
23
 
#include <drizzled/item/func.h>
24
 
#include <drizzled/function/math/real.h>
25
 
#include <drizzled/session.h>
26
 
 
27
 
using namespace std;
28
 
using namespace drizzled;
29
 
 
30
 
class RandFunction :public Item_real_func
31
 
{
32
 
  uint64_t seed1;
33
 
  uint64_t seed2;
34
 
  uint64_t max_value;
35
 
  double max_value_dbl;
36
 
  void _seed_random_int(uint64_t new_seed1, uint64_t new_seed2);
37
 
 
38
 
public:
39
 
  RandFunction()        :Item_real_func() {}
40
 
  double val_real();
41
 
  const char *func_name() const { return "rand"; }
42
 
  bool const_item() const { return 0; }
43
 
  void update_used_tables();
44
 
  bool fix_fields(Session *session, Item **ref);
45
 
 
46
 
  bool check_argument_count(int n)
47
 
  {
48
 
    return (n == 0 || n == 1);
49
 
  }
50
 
 
51
 
private:
52
 
  void seed_random (Item * val);
53
 
};
54
 
 
55
 
static uint32_t sql_rnd()
56
 
{
57
 
  return (uint32_t) (rand() * 0xffffffff); /* make all bits random */
58
 
}
59
 
 
60
 
 
61
 
void RandFunction::seed_random(Item *arg)
62
 
{
63
 
  /*
64
 
    TODO: do not do reinit 'rand' for every execute of PS/SP if
65
 
    args[0] is a constant.
66
 
  */
67
 
  uint64_t tmp= (uint64_t) arg->val_int();
68
 
  _seed_random_int(tmp * 0x10001L + 55555555L, tmp * 0x10000001L);
69
 
}
70
 
 
71
 
void RandFunction::_seed_random_int(uint64_t new_seed1, uint64_t new_seed2)
72
 
{
73
 
  max_value= 0x3FFFFFFFL;
74
 
  max_value_dbl=(double) max_value;
75
 
  seed1= new_seed1 % max_value;
76
 
  seed2= new_seed2 % max_value;
77
 
}
78
 
 
79
 
bool RandFunction::fix_fields(Session *session,Item **ref)
80
 
{
81
 
  if (Item_real_func::fix_fields(session, ref))
82
 
    return true;
83
 
 
84
 
  used_tables_cache|= RAND_TABLE_BIT;
85
 
  if (arg_count)
86
 
  {                                     // Only use argument once in query
87
 
    /*
88
 
      No need to send a Rand log event if seed was given eg: RAND(seed),
89
 
      as it will be replicated in the query as such.
90
 
    */
91
 
    if (args[0]->const_item())
92
 
      seed_random(args[0]);
93
 
  }
94
 
  else
95
 
  {
96
 
    uint64_t tmp= sql_rnd();
97
 
    _seed_random_int(tmp + (uint64_t) ref, tmp + (uint64_t) session->thread_id);
98
 
  }
99
 
 
100
 
  return false;
101
 
}
102
 
 
103
 
void RandFunction::update_used_tables()
104
 
{
105
 
  Item_real_func::update_used_tables();
106
 
  used_tables_cache|= RAND_TABLE_BIT;
107
 
}
108
 
 
109
 
double RandFunction::val_real()
110
 
{
111
 
  assert(fixed == 1);
112
 
  if (arg_count && !args[0]->const_item())
113
 
    seed_random(args[0]);
114
 
 
115
 
  seed1= (seed1 * 3 + seed2) % max_value;
116
 
  seed2= (seed1 + seed2 + 33) % max_value;
117
 
  return (((double) seed1) / max_value_dbl);
118
 
}
119
 
 
120
 
plugin::Create_function<RandFunction> *rand_function= NULL;
121
 
 
122
 
static int initialize(module::Context &registry)
123
 
{
124
 
  rand_function= new plugin::Create_function<RandFunction>("rand");
125
 
  registry.add(rand_function);
126
 
  return 0;
127
 
}
128
 
 
129
 
DRIZZLE_PLUGIN(initialize, NULL, NULL);