瀏覽代碼

[importer_direct_upload_mysql] Supporting mysql dialect for the direct upload

ayush.goyal 4 年之前
父節點
當前提交
c2fda8e1a8
共有 1 個文件被更改,包括 8 次插入2 次删除
  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']
 
-    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 (
 %(columns)s);
       ''' % {
@@ -316,7 +322,7 @@ class SQLIndexer(object):
 
         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;
           '''% {
                   'database': database,