Jelajahi Sumber

[importer_direct_upload_mysql] Supporting mysql dialect for the direct upload

ayush.goyal 4 tahun lalu
induk
melakukan
c2fda8e1a8
1 mengubah file dengan 8 tambahan dan 2 penghapusan
  1. 8 2
      desktop/libs/indexer/src/indexer/indexers/sql.py

+ 8 - 2
desktop/libs/indexer/src/indexer/indexers/sql.py

@@ -295,7 +295,13 @@ class SQLIndexer(object):
 
 
     columns = destination['columns']
     columns = destination['columns']
 
 
-    if editor_type == 'hive':
+    if editor_type in ('hive', 'mysql'):
+
+      if editor_type == 'mysql':
+        for col in columns:
+          if col['type'] == 'string':
+            col['type'] = 'VARCHAR(255)'
+
       sql = '''CREATE TABLE IF NOT EXISTS %(database)s.%(table_name)s (
       sql = '''CREATE TABLE IF NOT EXISTS %(database)s.%(table_name)s (
 %(columns)s);
 %(columns)s);
       ''' % {
       ''' % {
@@ -316,7 +322,7 @@ class SQLIndexer(object):
 
 
         csv_rows = str(list_of_tuples)[1:-1]
         csv_rows = str(list_of_tuples)[1:-1]
 
 
-        if editor_type == 'hive':
+        if editor_type in ('hive', 'mysql'):
           sql += '''\nINSERT INTO %(database)s.%(table_name)s VALUES %(csv_rows)s;
           sql += '''\nINSERT INTO %(database)s.%(table_name)s VALUES %(csv_rows)s;
           '''% {
           '''% {
                   'database': database,
                   'database': database,