From d12ec6db92b0ca63f2a2697b2895f5c6d6988d96 Mon Sep 17 00:00:00 2001
From: Andrea Zagli <azagli@libero.it>
Date: Wed, 1 Jun 2011 14:29:43 +0200
Subject: [PATCH] Identification of the primary key fields.

---
 src/table.c | 41 ++++++++++++++++++++++++++++++++++++++---
 1 file changed, 38 insertions(+), 3 deletions(-)

diff --git a/src/table.c b/src/table.c
index e21f0b1..f427c60 100644
--- a/src/table.c
+++ b/src/table.c
@@ -426,6 +426,8 @@ table_save (Table *table)
 
 	guint new_id;
 
+	gboolean is_key;
+
 	gchar *table_name;
 	guint id_field;
 	gchar *field_name;
@@ -489,6 +491,7 @@ table_save (Table *table)
 			                       priv->id);
 			stmt = gda_sql_parser_parse_string (priv->commons->gdaparser, sql, NULL, NULL);
 		}
+	g_free (sql);
 
 	error = NULL;
 	if (gda_connection_statement_execute_non_select (priv->commons->gdacon, stmt, NULL, NULL, &error) == 1)
@@ -517,6 +520,30 @@ table_save (Table *table)
 								{
 									if (id_field == 0)
 										{
+											/* find if field is the primary key */
+											sql = g_strdup_printf ("SELECT column_name"
+											                       " FROM _key_column_usage AS c"
+											                       " INNER JOIN _table_constraints AS t"
+											                       " ON c.table_catalog = t.table_catalog"
+											                       " AND c.table_schema = t.table_schema"
+											                       " AND c.table_name = t.table_name"
+											                       " WHERE t.table_name = '%s'"
+											                       " AND t.constraint_type = 'PRIMARY KEY'"
+											                       " AND c.column_name = '%s'",
+											                       table_name,
+											                       field_name);
+											error = NULL;
+											dm = gda_meta_store_extract (gda_connection_get_meta_store (priv->gdacon),
+											                             sql,
+											                             &error,
+											                             NULL);
+											if (dm == NULL || error != NULL)
+												{
+													g_warning ("Unable to get the primary key: %s.",
+													           error != NULL && error->message != NULL ? error->message : "no details");
+												}
+											is_key = (gda_data_model_get_n_rows (dm) > 0);
+
 											error = NULL;
 											sql = g_strdup_printf ("SELECT COALESCE (MAX (id), 0) FROM %sfields",
 											                       priv->commons->prefix);
@@ -532,16 +559,20 @@ table_save (Table *table)
 													           (error != NULL && error->message != NULL ? error->message : "No details."));
 												}
 											id_field++;
+
+											g_free (sql);
 											g_object_unref (stmt);
+											g_object_unref (dm);
 
 											error = NULL;
 											sql = g_strdup_printf ("INSERT INTO %sfields"
-											                       " (id, id_tables, name)"
-											                      " VALUES (%d, %d, '%s')",
+											                       " (id, id_tables, name, is_key)"
+											                      " VALUES (%d, %d, '%s', %s)",
 											                       priv->commons->prefix,
 											                       id_field,
 											                       priv->id,
-											                       field_name);
+											                       field_name,
+											                       is_key ? "TRUE" : "FALSE");
 											stmt = gda_sql_parser_parse_string (priv->commons->gdaparser, sql, NULL, NULL);
 											gda_connection_statement_execute_non_select (priv->commons->gdacon, stmt, NULL, NULL, &error);
 											if (error != NULL)
@@ -549,7 +580,10 @@ table_save (Table *table)
 													g_warning ("Error saving new field: %s",
 													           (error->message != NULL ? error->message : "No details."));
 												}
+
+											g_free (sql);
 											g_object_unref (stmt);
+											g_object_unref (dm);
 										}
 									id_saved = g_strconcat (id_saved, g_strdup_printf ("%d,", id_field), NULL);
 								}
@@ -573,6 +607,7 @@ table_save (Table *table)
 					           (error->message != NULL ? error->message : "No details."));
 				}
 			g_object_unref (stmt);
+			g_free (sql);
 
 			g_signal_emit (table, klass->updated_signal_id, 0);
 
-- 
2.49.0