From 0cef53257f66b437890c8aed9c7a0fab23e9d980 Mon Sep 17 00:00:00 2001
From: Jarek Jarcec Cecho <jarcec@apache.org>
Date: Tue, 7 Aug 2012 06:12:50 +0000
Subject: [PATCH] SQOOP-539. getPrimaryKeyQuery is very slow in mysql 5.0.77.

(Jhovanny via Jarek Jarcec Cecho)


git-svn-id: https://svn.apache.org/repos/asf/sqoop/trunk@1370124 13f79535-47bb-0310-9956-ffa450edef68
---
 src/java/org/apache/sqoop/manager/MySQLManager.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/java/org/apache/sqoop/manager/MySQLManager.java b/src/java/org/apache/sqoop/manager/MySQLManager.java
index d7abb44f..a817aa41 100644
--- a/src/java/org/apache/sqoop/manager/MySQLManager.java
+++ b/src/java/org/apache/sqoop/manager/MySQLManager.java
@@ -69,6 +69,14 @@ protected void initOptionDefaults() {
     }
   }
 
+  @Override
+  protected String getPrimaryKeyQuery(String tableName) {
+    return "SELECT column_name FROM INFORMATION_SCHEMA.COLUMNS "
+         + "WHERE TABLE_SCHEMA = (" + getSchemaQuery() + ") "
+         + "AND TABLE_NAME = '"+tableName+"' "
+         + "AND COLUMN_KEY = 'PRI'";
+  }
+
   @Override
   protected String getColNamesQuery(String tableName) {
     // Use mysql-specific hints and LIMIT to return fast