~drizzle-trunk/drizzle/development

1861.4.8 by Brian Aker
Adds in a portion of the ref constraints table.
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
2234 by Brian Aker
Mass removal of ifdef/endif in favor of pragma once.
21
#pragma once
1861.4.8 by Brian Aker
Adds in a portion of the ref constraints table.
22
23
namespace drizzled {
24
namespace generator {
25
26
typedef std::pair<const drizzled::message::Table *, int32_t> ForeignKeyConstraintPair;
27
28
class AllForeignKeys
29
{
30
  Session &session;
31
  message::Table table_message;
1938.4.2 by Brian Aker
Fix style issue around table for message (though this is imperfect,...)
32
  drizzled::message::table::shared_ptr table_ptr;
1861.4.8 by Brian Aker
Adds in a portion of the ref constraints table.
33
  int32_t foreign_keys_iterator;
34
35
  drizzled::generator::AllTables all_tables_generator;
36
37
  bool table_setup();
38
39
public:
40
41
  AllForeignKeys(Session &arg);
42
43
  void reset();
44
45
  operator const drizzled::message::Table::ForeignKeyConstraint*()
46
  {
47
    if (table_ptr)
48
    {
49
      do {
50
        if (foreign_keys_iterator != table_message.fk_constraint_size())
51
        {
2385.2.5 by Olaf van der Spek
Refactor
52
          return &table_message.fk_constraint(foreign_keys_iterator++);
1861.4.8 by Brian Aker
Adds in a portion of the ref constraints table.
53
        }
54
55
      } while ((table_ptr= all_tables_generator) && table_setup());
56
    }
57
58
    return NULL;
59
  }
60
61
  operator const ForeignKeyConstraintPair()
62
  {
63
    if (table_ptr)
64
    {
65
      do {
66
        if (foreign_keys_iterator != table_message.fk_constraint_size())
67
        {
68
          return std::make_pair(&table_message, foreign_keys_iterator++);
69
        }
70
      } while ((table_ptr= all_tables_generator) && table_setup());
71
    }
72
2385.2.5 by Olaf van der Spek
Refactor
73
    return ForeignKeyConstraintPair();
1861.4.8 by Brian Aker
Adds in a portion of the ref constraints table.
74
  }
75
};
76
77
} /* namespace generator */
78
} /* namespace drizzled */
79