|
@@ -102,6 +102,9 @@ class MysqlDS(DataSourceBase):
|
|
|
return flat_map(lambda x: x, res[0])
|
|
|
|
|
|
def get_table_schema(self, table_name):
|
|
|
+ def handle_col(x):
|
|
|
+ line = list(map(lambda s: s.decode('utf-8') if type(s) == type(b'bytes') else str(s), x))
|
|
|
+ return [':'.join(line[:3])]
|
|
|
sql = f'describe `{self.database_name}`.{table_name}'
|
|
|
logger.info(sql)
|
|
|
res = self._execute_sql([sql])
|
|
@@ -109,7 +112,7 @@ class MysqlDS(DataSourceBase):
|
|
|
res = [[str(i) , *x]for i, x in enumerate(res[0])]
|
|
|
logger.info(res)
|
|
|
|
|
|
- return flat_map(lambda x: [':'.join(x[:3])], res)
|
|
|
+ return flat_map(lambda x: handle_col(x), res)
|
|
|
else:
|
|
|
raise Exception('table not found')
|
|
|
|