def test_remove(self): result_map = es_query.execute_sql(None, """ SELECT * FROM abc; SAVE RESULT AS result1; REMOVE RESULT result1; """) self.assertEqual([], result_map.keys())
def test_save_as(self): result_map = es_query.execute_sql( None, """ SELECT * FROM abc; SAVE RESULT AS result1; """) self.assertEqual(['result1'], result_map.keys())
def test_complex_python_code(self): result_map = es_query.execute_sql(None, """ result_map['result1'] = [] for i in range(100): result_map['result1'].append(i) """) self.assertEqual(['result1'], result_map.keys())
def test_complex_python_code(self): result_map = es_query.execute_sql( None, """ result_map['result1'] = [] for i in range(100): result_map['result1'].append(i) """) self.assertEqual(['result1'], result_map.keys())
def test_remove(self): result_map = es_query.execute_sql( None, """ SELECT * FROM abc; SAVE RESULT AS result1; REMOVE RESULT result1; """) self.assertEqual([], result_map.keys())
def search(): if flask.request.method == 'GET': sql = flask.request.args.get('q') else: sql = flask.request.get_data(parse_form_data=False) es_hosts = flask.request.args.get('by') try: resp = {'error': None, 'data': es_query.execute_sql(es_hosts, sql)} return json.dumps(resp, indent=2) except: etype, value, tb = sys.exc_info() resp = { 'traceback': traceback.format_exception(etype, value, tb), 'error': str(value), 'data': None } return json.dumps(resp, indent=2)
def query_datapoints(config): lines = config.splitlines() es_hosts = lines[0] sql = ''.join(lines[1:]) datapoints = [] ts = int(time.time()) result_map = es_query.execute_sql(es_hosts, sql) for metric_name, rows in result_map.iteritems(): for row in rows or []: datapoint = {'value': row.pop('value', 0)} if row: tags = {} for k, v in row.iteritems(): tags[to_str(k)] = to_str(v) datapoint['tags'] = tags datapoint['name'] = metric_name datapoint['timestamp'] = ts datapoints.append(datapoint) return datapoints
def search(): if flask.request.method == 'GET': sql = flask.request.args.get('q') else: sql = flask.request.get_data(parse_form_data=False) es_hosts = flask.request.args.get('by') try: resp = { 'error': None, 'data': es_query.execute_sql(es_hosts, sql) } return json.dumps(resp, indent=2) except: etype, value, tb = sys.exc_info() resp = { 'traceback': traceback.format_exception(etype, value, tb), 'error': str(value), 'data': None } return json.dumps(resp, indent=2)
def test_python_code(self): result_map = es_query.execute_sql( None, """ result_map['result1'] = [] """) self.assertEqual(['result1'], result_map.keys())
def test_no_save_as(self): result_map = es_query.execute_sql( None, """ SELECT * FROM abc """) self.assertEqual(['result'], result_map.keys())
def test_python_code(self): result_map = es_query.execute_sql(None, """ result_map['result1'] = [] """) self.assertEqual(['result1'], result_map.keys())
def test_save_as(self): result_map = es_query.execute_sql(None, """ SELECT * FROM abc; SAVE RESULT AS result1; """) self.assertEqual(['result1'], result_map.keys())
def test_no_save_as(self): result_map = es_query.execute_sql(None, """ SELECT * FROM abc """) self.assertEqual(['result'], result_map.keys())