def _gen_data_simple_table(self, number_of_rows=1000): helper = RQGQueryHelper() map = self._get_pkey_map_for_tables_wit_primary_key_column() for table_name in map.keys(): for x in range(0, number_of_rows): statement = helper._generate_insert_statement(table_name, map[table_name], "\"" + str(x + 1) + "\"") self._insert_execute_query(statement)
def _convert_delete_template_query_info_with_merge(self, source_table="copy_simple_table", target_table="simple_table" ,n1ql_queries=[], table_map={}): helper = RQGQueryHelper() query_input_list = [] for n1ql_query in n1ql_queries: query_input_list.append(helper._delete_sql_template_to_values_with_merge(source_table=source_table, target_table=target_table, sql=n1ql_query, table_map=table_map)) return query_input_list
def _gen_queries_from_template(self, query_path="./queries.txt", table_name=None): helper = RQGQueryHelper() map = self._get_values_with_type_for_fields_in_table() table_map = map[table_name] with open(query_path) as f: content = f.readlines() for query in content: helper._convert_sql_template_to_value(sql=query, table_map=table_map, table_name=table_name)
def _convert_delete_template_query_info(self, n1ql_queries=[], table_map={}): helper = RQGQueryHelper() query_input_list = [] for n1ql_query in n1ql_queries: query_input_list.append( helper._delete_sql_template_to_values(sql=n1ql_query, table_map=table_map)) return query_input_list
def _convert_template_query_info_with_gsi(self, file_path, gsi_index_file_path=None, table_map={}, table_name="simple_table", define_gsi_index=True, gen_expected_result=False): helper = RQGQueryHelper() f = open(gsi_index_file_path, 'w') n1ql_queries = self._read_from_file(file_path) for n1ql_query in n1ql_queries: check = True if not helper._check_deeper_query_condition(n1ql_query): if "SUBQUERY" in n1ql_query: map = helper._convert_sql_template_to_value_with_subqueries(n1ql_query, table_map=table_map, define_gsi_index=define_gsi_index) else: map=helper._convert_sql_template_to_value_for_secondary_indexes(n1ql_query, table_map=table_map, table_name=table_name, define_gsi_index=define_gsi_index) else: map=helper._convert_sql_template_to_value_for_secondary_indexes_sub_queries(n1ql_query, table_map=table_map, table_name=table_name, define_gsi_index=define_gsi_index) if gen_expected_result: query = map["sql"] try: sql_result = self._query_and_convert_to_json(query) map["expected_result"] = sql_result except Exception, ex: print ex check = False if check: f.write(json.dumps(map)+"\n")
def _convert_template_query_info(self, n1ql_queries=[], table_map={}, define_gsi_index=True, gen_expected_result=False, ansi_joins=False, aggregate_pushdown=False, partitioned_indexes=False, with_let=False): helper = RQGQueryHelper() query_input_list = [] for n1ql_query in n1ql_queries: # check if ["UNION ALL", "INTERSECT ALL", "EXCEPT ALL", "UNION", "INTERSECT", "EXCEPT"] not in query if not helper._check_deeper_query_condition(n1ql_query): if "SUBTABLE" in n1ql_query: sql_n1ql_index_map = helper._convert_sql_template_to_value_with_subqueryenhancements( n1ql_query, table_map=table_map, define_gsi_index=define_gsi_index) elif "SUBQUERY" in n1ql_query: sql_n1ql_index_map = helper._convert_sql_template_to_value_with_subqueries( n1ql_query, table_map=table_map, define_gsi_index=define_gsi_index) else: # takes in sql and n1ql queries and create indexes for them sql_n1ql_index_map = helper._convert_sql_template_to_value_for_secondary_indexes( n1ql_query, table_map=table_map, define_gsi_index=define_gsi_index, ansi_joins=ansi_joins, aggregate_pushdown=aggregate_pushdown, partitioned_indexes=partitioned_indexes, with_let=with_let) else: sql_n1ql_index_map = helper._convert_sql_template_to_value_for_secondary_indexes_sub_queries( n1ql_query, table_map=table_map, define_gsi_index=define_gsi_index, partitioned_indexes=partitioned_indexes) if gen_expected_result: sql_query = sql_n1ql_index_map["sql"] try: sql_result = self._query_and_convert_to_json(sql_query) sql_n1ql_index_map["expected_result"] = sql_result except Exception as ex: print(ex) sql_n1ql_index_map = self._translate_function_names( sql_n1ql_index_map) query_input_list.append(sql_n1ql_index_map) return query_input_list