~drizzle-trunk/drizzle/development

934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 *
1999.6.1 by kalebral at gmail
update Copyright strings to a more common format to help with creating the master debian copyright file
4
 *  Copyright (C) 2008 Sun Microsystems, Inc.
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
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
1241.9.1 by Monty Taylor
Removed global.h. Fixed all the headers.
20
#include "config.h"
21
22
#include <string>
23
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
24
#include "drizzled/foreign_key.h"
25
#include "drizzled/error.h"
1237.9.4 by Padraig O'Sullivan
Removed the inclusion of drizzled/field.h in the server_includes header file.
26
#include "drizzled/create_field.h"
1241.9.64 by Monty Taylor
Moved remaining non-public portions of mysys and mystrings to drizzled/internal.
27
#include "drizzled/internal/my_sys.h"
1638.10.78 by Stewart Smith
separate out adding the foreign key to the table protobuf message from mysql_prepare_create_table and call a function to do it directly from the parser where we set up the foreign key structures
28
#include "drizzled/table_ident.h"
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
29
1280.1.10 by Monty Taylor
Put everything in drizzled into drizzled namespace.
30
namespace drizzled
31
{
1253.1.3 by Monty Taylor
MEM_ROOT == memory::Root
32
1241.9.1 by Monty Taylor
Removed global.h. Fixed all the headers.
33
extern const CHARSET_INFO *system_charset_info;
34
1638.10.78 by Stewart Smith
separate out adding the foreign key to the table protobuf message from mysql_prepare_create_table and call a function to do it directly from the parser where we set up the foreign key structures
35
void add_foreign_key_to_table_message(
36
    message::Table *table_message,
37
    const char* fkey_name,
38
    List<Key_part_spec> &cols,
39
    Table_ident *table,
40
    List<Key_part_spec> &ref_cols,
41
    message::Table::ForeignKeyConstraint::ForeignKeyOption delete_opt_arg,
42
    message::Table::ForeignKeyConstraint::ForeignKeyOption update_opt_arg,
43
    message::Table::ForeignKeyConstraint::ForeignKeyMatchOption match_opt_arg)
44
{
45
  message::Table::ForeignKeyConstraint *pfkey= table_message->add_fk_constraint();
46
  if (fkey_name)
47
    pfkey->set_name(fkey_name);
1638.10.80 by Stewart Smith
fix storing and manipulating foreign keys in the proto around ALTER TABLE, CREATE TABLE and ALTER TABLE ADD/DROP FOREIGN KEY. We also (mostly) emulate the naming of innodb foreign keys in the upper layer.
48
  else if (table_message->has_name())
49
  {
50
    std::string name(table_message->name());
51
    char number[20];
52
53
    name.append("_ibfk_");
54
    snprintf(number, sizeof(number), "%d", table_message->fk_constraint_size());
55
    name.append(number);
56
57
    pfkey->set_name(name);
58
  }
1638.10.78 by Stewart Smith
separate out adding the foreign key to the table protobuf message from mysql_prepare_create_table and call a function to do it directly from the parser where we set up the foreign key structures
59
60
  pfkey->set_match(match_opt_arg);
61
  pfkey->set_update_option(update_opt_arg);
62
  pfkey->set_delete_option(delete_opt_arg);
63
64
  pfkey->set_references_table_name(table->table.str);
65
66
  Key_part_spec *keypart;
67
  List_iterator<Key_part_spec> col_it(cols);
68
  while ((keypart= col_it++))
69
  {
70
    pfkey->add_column_names(keypart->field_name.str);
71
  }
72
73
  List_iterator<Key_part_spec> ref_it(ref_cols);
74
  while ((keypart= ref_it++))
75
  {
76
    pfkey->add_references_columns(keypart->field_name.str);
77
  }
78
79
}
80
1253.1.3 by Monty Taylor
MEM_ROOT == memory::Root
81
Foreign_key::Foreign_key(const Foreign_key &rhs, memory::Root *mem_root)
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
82
  :Key(rhs),
83
  ref_table(rhs.ref_table),
84
  ref_columns(rhs.ref_columns),
85
  delete_opt(rhs.delete_opt),
86
  update_opt(rhs.update_opt),
87
  match_opt(rhs.match_opt)
88
{
89
  list_copy_and_replace_each_value(ref_columns, mem_root);
90
}
91
92
/*
93
  Test if a foreign key (= generated key) is a prefix of the given key
94
  (ignoring key name, key type and order of columns)
95
96
  NOTES:
97
    This is only used to test if an index for a FOREIGN KEY exists
98
99
  IMPLEMENTATION
100
    We only compare field names
101
102
  RETURN
103
    0	Generated key is a prefix of other key
104
    1	Not equal
105
*/
106
bool foreign_key_prefix(Key *a, Key *b)
107
{
108
  /* Ensure that 'a' is the generated key */
109
  if (a->generated)
110
  {
111
    if (b->generated && a->columns.elements > b->columns.elements)
112
      std::swap(a, b);                       // Put shorter key in 'a'
113
  }
114
  else
115
  {
116
    if (!b->generated)
117
      return true;                              // No foreign key
118
    std::swap(a, b);                       // Put generated key in 'a'
119
  }
120
121
  /* Test if 'a' is a prefix of 'b' */
122
  if (a->columns.elements > b->columns.elements)
123
    return true;                                // Can't be prefix
124
125
  List_iterator<Key_part_spec> col_it1(a->columns);
126
  List_iterator<Key_part_spec> col_it2(b->columns);
127
  const Key_part_spec *col1, *col2;
128
129
#ifdef ENABLE_WHEN_INNODB_CAN_HANDLE_SWAPED_FOREIGN_KEY_COLUMNS
130
  while ((col1= col_it1++))
131
  {
132
    bool found= 0;
133
    col_it2.rewind();
134
    while ((col2= col_it2++))
135
    {
136
      if (*col1 == *col2)
137
      {
138
        found= true;
139
	break;
140
      }
141
    }
142
    if (!found)
143
      return true;                              // Error
144
  }
145
  return false;                                 // Is prefix
146
#else
147
  while ((col1= col_it1++))
148
  {
149
    col2= col_it2++;
150
    if (!(*col1 == *col2))
151
      return true;
152
  }
153
  return false;                                 // Is prefix
154
#endif
155
}
156
157
/*
158
  Check if the foreign key options are compatible with columns
159
  on which the FK is created.
160
161
  RETURN
162
    0   Key valid
163
    1   Key invalid
164
*/
1052.2.3 by Nathan Williams
No actual code changes. Changed Create_field to CreateField to be consistent with coding standards.
165
bool Foreign_key::validate(List<CreateField> &table_fields)
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
166
{
1052.2.3 by Nathan Williams
No actual code changes. Changed Create_field to CreateField to be consistent with coding standards.
167
  CreateField  *sql_field;
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
168
  Key_part_spec *column;
169
  List_iterator<Key_part_spec> cols(columns);
1052.2.3 by Nathan Williams
No actual code changes. Changed Create_field to CreateField to be consistent with coding standards.
170
  List_iterator<CreateField> it(table_fields);
934.2.20 by Jay Pipes
Move Foreign_key implementation into its own implementation file and out of session.cc
171
  while ((column= cols++))
172
  {
173
    it.rewind();
174
    while ((sql_field= it++) &&
175
           my_strcasecmp(system_charset_info,
176
                         column->field_name.str,
177
                         sql_field->field_name)) {}
178
    if (!sql_field)
179
    {
180
      my_error(ER_KEY_COLUMN_DOES_NOT_EXITS, MYF(0), column->field_name.str);
181
      return true;
182
    }
183
  }
184
  return false;
185
}
186
1280.1.10 by Monty Taylor
Put everything in drizzled into drizzled namespace.
187
} /* namespace drizzled */