Method operate_SQL

sql/SQLOperator.py:172–208

(self, sql)

        return result

    def operate_SQL(self, sql):
        # SQL包含表名以及数据库名称
        match = re.search(r'FROM\s+(\S+)\.(\S+)', sql)
        database_name = None
        table_name = None
        if match:
            database_name = match.group(0)
            table_name = match.group(1)
        # SQL不包含数据库名，只包含表名
        match = re.search(r'FROM\s+(\S+)', sql)
        if match:
            table_name = match.group(1)
        else:
            logging.error("Table Name is not found, please check SQL")
        if database_name is None:
            for jdbc in self.jdbcList:
                tables = self.tableList.get(jdbc)
                if tables.__contains__(table_name):
                    database_name = str(jdbc).split('/')[-1]
                    break
        if database_name is None:
            if table_name is None:
                logging.error("Cannot find the database and table")
            else:
                logging.error("Cannot find the database contain table: " + table_name)
        # 执行结果
        if database_name is not None or table_name is not None:
            for jdbc in self.jdbcList:
                if str(jdbc).split('/')[-1] == database_name:
                    engine = self.connMap.get(jdbc)
                    with engine.connect() as conn:
                        start_time = time.time()
                        result = conn.execute(text(sql))
                        end_time = time.time()
                    logging.info("sql: " + sql + " cost time " + str(end_time - start_time))
        result = self.process_result(sql, result)
        return result

    def get_db_structure_and_index(self, sql):
        """