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 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())